db

package
v0.3.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jul 11, 2022 License: MIT Imports: 12 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type AWSRepoInterface 

type AWSRepoInterface interface {
	ListAWS(ctx context.Context, projectID, awsID uint32, awsAccountID string) (*[]model.AWS, error)
	GetAWSByAccountID(ctx context.Context, projectID uint32, awsAccountID string) (*model.AWS, error)
	UpsertAWS(ctx context.Context, data *model.AWS) (*model.AWS, error)
	DeleteAWS(ctx context.Context, projectID, awsID uint32) error
	ListAWSDataSource(ctx context.Context, projectID, awsID uint32, ds string) (*[]DataSource, error)
	ListDataSourceByAWSDataSourceID(ctx context.Context, awsDataSourceID uint32) (*[]DataSource, error)
	ListAWSRelDataSource(ctx context.Context, projectID, awsID uint32) (*[]model.AWSRelDataSource, error)
	UpsertAWSRelDataSource(ctx context.Context, data *aws.DataSourceForAttach) (*model.AWSRelDataSource, error)
	GetAWSRelDataSourceByID(ctx context.Context, awsID, awsDataSourceID, projectID uint32) (*model.AWSRelDataSource, error)
	DeleteAWSRelDataSource(ctx context.Context, projectID, awsID, awsDataSourceID uint32) error
	GetAWSDataSourceForMessage(ctx context.Context, awsID, awsDataSourceID, projectID uint32) (*message.AWSQueueMessage, error)
}

type Client 

type Client struct {
	MasterDB *gorm.DB
	SlaveDB  *gorm.DB
	// contains filtered or unexported fields
}

func NewClient 

func NewClient(conf *Config, l logging.Logger) *Client

func (*Client) DeleteAWS 

func (c *Client) DeleteAWS(ctx context.Context, projectID, awsID uint32) error

func (*Client) DeleteAWSRelDataSource 

func (c *Client) DeleteAWSRelDataSource(ctx context.Context, projectID, awsID, awsDataSourceID uint32) error

func (*Client) DeleteApplicationScan 

func (c *Client) DeleteApplicationScan(ctx context.Context, projectID uint32, applicationScanID uint32) error

func (*Client) DeleteApplicationScanBasicSetting 

func (c *Client) DeleteApplicationScanBasicSetting(ctx context.Context, projectID uint32, applicationScanBasicSettingID uint32) error

func (*Client) DeleteDiagnosisDataSource 

func (c *Client) DeleteDiagnosisDataSource(ctx context.Context, projectID uint32, diagnosisDataSourceID uint32) error

func (*Client) DeleteEnterpriseOrg 

func (c *Client) DeleteEnterpriseOrg(ctx context.Context, projectID, gitleaksID uint32, login string) error

func (*Client) DeleteGCP 

func (c *Client) DeleteGCP(ctx context.Context, projectID, gcpID uint32) error

func (*Client) DeleteGCPDataSource 

func (c *Client) DeleteGCPDataSource(ctx context.Context, projectID, gcpID, googleDataSourceID uint32) error

func (*Client) DeleteGitleaks 

func (c *Client) DeleteGitleaks(ctx context.Context, projectID, gitleaksID uint32) error

func (*Client) DeleteOsint 

func (c *Client) DeleteOsint(ctx context.Context, projectID uint32, osintID uint32) error

func (*Client) DeleteOsintDataSource 

func (c *Client) DeleteOsintDataSource(ctx context.Context, projectID uint32, osintDataSourceID uint32) error

func (*Client) DeleteOsintDetectWord 

func (c *Client) DeleteOsintDetectWord(ctx context.Context, projectID uint32, osintDetectWordID uint32) error

func (*Client) DeletePortscanSetting 

func (c *Client) DeletePortscanSetting(ctx context.Context, projectID uint32, portscanSettingID uint32) error

func (*Client) DeletePortscanTarget 

func (c *Client) DeletePortscanTarget(ctx context.Context, projectID uint32, portscanTargetID uint32) error

func (*Client) DeletePortscanTargetByPortscanSettingID 

func (c *Client) DeletePortscanTargetByPortscanSettingID(ctx context.Context, projectID uint32, portscanSettingID uint32) error

func (*Client) DeleteRelOsintDataSource 

func (c *Client) DeleteRelOsintDataSource(ctx context.Context, projectID uint32, relOsintDataSourceID uint32) error

func (*Client) DeleteWpscanSetting 

func (c *Client) DeleteWpscanSetting(ctx context.Context, projectID uint32, wpscanSettingID uint32) error

func (*Client) GetAWSByAccountID 

func (c *Client) GetAWSByAccountID(ctx context.Context, projectID uint32, awsAccountID string) (*model.AWS, error)

func (*Client) GetAWSDataSourceForMessage 

func (c *Client) GetAWSDataSourceForMessage(ctx context.Context, awsID, awsDataSourceID, projectID uint32) (*message.AWSQueueMessage, error)

func (*Client) GetAWSRelDataSourceByID 

func (c *Client) GetAWSRelDataSourceByID(ctx context.Context, awsID, awsDataSourceID, projectID uint32) (*model.AWSRelDataSource, error)

func (*Client) GetApplicationScan 

func (c *Client) GetApplicationScan(ctx context.Context, projectID uint32, applicationScanID uint32) (*model.ApplicationScan, error)

func (*Client) GetApplicationScanBasicSetting 

func (c *Client) GetApplicationScanBasicSetting(ctx context.Context, projectID uint32, applicationScanID uint32) (*model.ApplicationScanBasicSetting, error)

func (*Client) GetDiagnosisDataSource 

func (c *Client) GetDiagnosisDataSource(ctx context.Context, projectID uint32, diagnosisDataSourceID uint32) (*model.DiagnosisDataSource, error)

func (*Client) GetGCP 

func (c *Client) GetGCP(ctx context.Context, projectID, gcpID uint32) (*model.GCP, error)

func (*Client) GetGCPByUniqueIndex 

func (c *Client) GetGCPByUniqueIndex(ctx context.Context, projectID uint32, gcpProjectID string) (*model.GCP, error)

func (*Client) GetGCPDataSource 

func (c *Client) GetGCPDataSource(ctx context.Context, projectID, gcpID, googleDataSourceID uint32) (*GCPDataSource, error)

func (*Client) GetGitleaks 

func (c *Client) GetGitleaks(ctx context.Context, projectID, gitleaksID uint32) (*model.CodeGitleaks, error)

func (*Client) GetGitleaksByUniqueIndex 

func (c *Client) GetGitleaksByUniqueIndex(ctx context.Context, projectID, codeDataSourceID uint32, name string) (*model.CodeGitleaks, error)

func (*Client) GetGoogleDataSource 

func (c *Client) GetGoogleDataSource(ctx context.Context, googleDataSourceID uint32) (*model.GCP, error)

func (*Client) GetOsint 

func (c *Client) GetOsint(ctx context.Context, projectID uint32, osintID uint32) (*model.Osint, error)

func (*Client) GetOsintDataSource 

func (c *Client) GetOsintDataSource(ctx context.Context, projectID uint32, osintDataSourceID uint32) (*model.OsintDataSource, error)

func (*Client) GetOsintDetectWord 

func (c *Client) GetOsintDetectWord(ctx context.Context, projectID uint32, osintDetectWordID uint32) (*model.OsintDetectWord, error)

func (*Client) GetPortscanSetting 

func (c *Client) GetPortscanSetting(ctx context.Context, projectID uint32, portscanSettingID uint32) (*model.PortscanSetting, error)

func (*Client) GetPortscanTarget 

func (c *Client) GetPortscanTarget(ctx context.Context, projectID uint32, portscanTargetID uint32) (*model.PortscanTarget, error)

func (*Client) GetRelOsintDataSource 

func (c *Client) GetRelOsintDataSource(ctx context.Context, projectID uint32, relOsintDataSourceID uint32) (*model.RelOsintDataSource, error)

func (*Client) GetWpscanSetting 

func (c *Client) GetWpscanSetting(ctx context.Context, projectID uint32, wpscanSettingID uint32) (*model.WpscanSetting, error)

func (*Client) ListAWS 

func (c *Client) ListAWS(ctx context.Context, projectID, awsID uint32, awsAccountID string) (*[]model.AWS, error)

func (*Client) ListAWSDataSource 

func (c *Client) ListAWSDataSource(ctx context.Context, projectID, awsID uint32, ds string) (*[]DataSource, error)

func (*Client) ListAWSRelDataSource 

func (c *Client) ListAWSRelDataSource(ctx context.Context, projectID, awsID uint32) (*[]model.AWSRelDataSource, error)

func (*Client) ListAllApplicationScan 

func (c *Client) ListAllApplicationScan(ctx context.Context) (*[]model.ApplicationScan, error)

func (*Client) ListAllPortscanSetting 

func (c *Client) ListAllPortscanSetting(ctx context.Context) (*[]model.PortscanSetting, error)

func (*Client) ListAllRelOsintDataSource 

func (c *Client) ListAllRelOsintDataSource(ctx context.Context, osintDataSourceID uint32) (*[]model.RelOsintDataSource, error)

func (*Client) ListAllWpscanSetting 

func (c *Client) ListAllWpscanSetting(ctx context.Context) (*[]model.WpscanSetting, error)

func (*Client) ListApplicationScan 

func (c *Client) ListApplicationScan(ctx context.Context, projectID, diagnoosisDataSourceID uint32) (*[]model.ApplicationScan, error)

func (*Client) ListApplicationScanBasicSetting 

func (c *Client) ListApplicationScanBasicSetting(ctx context.Context, projectID, applicationScanID uint32) (*[]model.ApplicationScanBasicSetting, error)

func (*Client) ListCodeDataSource 

func (c *Client) ListCodeDataSource(ctx context.Context, codeDataSourceID uint32, name string) (*[]model.CodeDataSource, error)

func (*Client) ListDataSourceByAWSDataSourceID 

func (c *Client) ListDataSourceByAWSDataSourceID(ctx context.Context, dataSourceID uint32) (*[]DataSource, error)

func (*Client) ListDiagnosisDataSource 

func (c *Client) ListDiagnosisDataSource(ctx context.Context, projectID uint32, name string) (*[]model.DiagnosisDataSource, error)

func (*Client) ListEnterpriseOrg 

func (c *Client) ListEnterpriseOrg(ctx context.Context, projectID, gitleaksID uint32) (*[]model.CodeEnterpriseOrg, error)

func (*Client) ListGCP 

func (c *Client) ListGCP(ctx context.Context, projectID, gcpID uint32, gcpProjectID string) (*[]model.GCP, error)

func (*Client) ListGCPDataSource 

func (c *Client) ListGCPDataSource(ctx context.Context, projectID, gcpID uint32) (*[]GCPDataSource, error)

func (*Client) ListGCPDataSourceByDataSourceID 

func (c *Client) ListGCPDataSourceByDataSourceID(ctx context.Context, googleDataSourceID uint32) (*[]GCPDataSource, error)

func (*Client) ListGitleaks 

func (c *Client) ListGitleaks(ctx context.Context, projectID, codeDataSourceID, gitleaksID uint32) (*[]model.CodeGitleaks, error)

func (*Client) ListGoogleDataSource 

func (c *Client) ListGoogleDataSource(ctx context.Context, googleDataSourceID uint32, name string) (*[]model.GoogleDataSource, error)

func (*Client) ListOsint 

func (c *Client) ListOsint(ctx context.Context, projectID uint32) (*[]model.Osint, error)

func (*Client) ListOsintDataSource 

func (c *Client) ListOsintDataSource(ctx context.Context, projectID uint32, name string) (*[]model.OsintDataSource, error)

func (*Client) ListOsintDetectWord 

func (c *Client) ListOsintDetectWord(ctx context.Context, projectID, relOsintDataSourceID uint32) (*[]model.OsintDetectWord, error)

func (*Client) ListPortscanSetting 

func (c *Client) ListPortscanSetting(ctx context.Context, projectID, diagnoosisDataSourceID uint32) (*[]model.PortscanSetting, error)

func (*Client) ListPortscanTarget 

func (c *Client) ListPortscanTarget(ctx context.Context, projectID, portscanSettingID uint32) (*[]model.PortscanTarget, error)

func (*Client) ListRelOsintDataSource 

func (c *Client) ListRelOsintDataSource(ctx context.Context, projectID, osintID, osintDataSourceID uint32) (*[]model.RelOsintDataSource, error)

func (*Client) ListWpscanSetting 

func (c *Client) ListWpscanSetting(ctx context.Context, projectID, diagnoosisDataSourceID uint32) (*[]model.WpscanSetting, error)

func (*Client) UpsertAWS 

func (c *Client) UpsertAWS(ctx context.Context, data *model.AWS) (*model.AWS, error)

func (*Client) UpsertAWSRelDataSource 

func (c *Client) UpsertAWSRelDataSource(ctx context.Context, data *aws.DataSourceForAttach) (*model.AWSRelDataSource, error)

func (*Client) UpsertApplicationScan 

func (c *Client) UpsertApplicationScan(ctx context.Context, data *model.ApplicationScan) (*model.ApplicationScan, error)

func (*Client) UpsertApplicationScanBasicSetting 

func (c *Client) UpsertApplicationScanBasicSetting(ctx context.Context, data *model.ApplicationScanBasicSetting) (*model.ApplicationScanBasicSetting, error)

func (*Client) UpsertDiagnosisDataSource 

func (c *Client) UpsertDiagnosisDataSource(ctx context.Context, input *model.DiagnosisDataSource) (*model.DiagnosisDataSource, error)

func (*Client) UpsertEnterpriseOrg 

func (c *Client) UpsertEnterpriseOrg(ctx context.Context, data *code.EnterpriseOrgForUpsert) (*model.CodeEnterpriseOrg, error)

func (*Client) UpsertGCP 

func (c *Client) UpsertGCP(ctx context.Context, gcp *google.GCPForUpsert) (*model.GCP, error)

func (*Client) UpsertGCPDataSource 

func (c *Client) UpsertGCPDataSource(ctx context.Context, gcpDataSource *google.GCPDataSourceForUpsert) (*GCPDataSource, error)

func (*Client) UpsertGitleaks 

func (c *Client) UpsertGitleaks(ctx context.Context, data *code.GitleaksForUpsert) (*model.CodeGitleaks, error)

func (*Client) UpsertGitleaksWithToken 

func (c *Client) UpsertGitleaksWithToken(ctx context.Context, data *code.GitleaksForUpsert) (*model.CodeGitleaks, error)

func (*Client) UpsertGitleaksWithoutToken 

func (c *Client) UpsertGitleaksWithoutToken(ctx context.Context, data *code.GitleaksForUpsert) (*model.CodeGitleaks, error)

func (*Client) UpsertOsint 

func (c *Client) UpsertOsint(ctx context.Context, data *model.Osint) (*model.Osint, error)

func (*Client) UpsertOsintDataSource 

func (c *Client) UpsertOsintDataSource(ctx context.Context, input *model.OsintDataSource) (*model.OsintDataSource, error)

func (*Client) UpsertOsintDetectWord 

func (c *Client) UpsertOsintDetectWord(ctx context.Context, data *model.OsintDetectWord) (*model.OsintDetectWord, error)

func (*Client) UpsertPortscanSetting 

func (c *Client) UpsertPortscanSetting(ctx context.Context, data *model.PortscanSetting) (*model.PortscanSetting, error)

func (*Client) UpsertPortscanTarget 

func (c *Client) UpsertPortscanTarget(ctx context.Context, data *model.PortscanTarget) (*model.PortscanTarget, error)

func (*Client) UpsertRelOsintDataSource 

func (c *Client) UpsertRelOsintDataSource(ctx context.Context, data *model.RelOsintDataSource) (*model.RelOsintDataSource, error)

func (*Client) UpsertWpscanSetting 

func (c *Client) UpsertWpscanSetting(ctx context.Context, data *model.WpscanSetting) (*model.WpscanSetting, error)

type CodeRepoInterface 

type CodeRepoInterface interface {
	// code_data_source
	ListCodeDataSource(ctx context.Context, codeDataSourceID uint32, name string) (*[]model.CodeDataSource, error)

	// code_gitleaks
	ListGitleaks(ctx context.Context, projectID, codeDataSourceID, gitleaksID uint32) (*[]model.CodeGitleaks, error)
	UpsertGitleaks(ctx context.Context, data *code.GitleaksForUpsert) (*model.CodeGitleaks, error)
	DeleteGitleaks(ctx context.Context, projectID uint32, gitleaksID uint32) error
	GetGitleaks(ctx context.Context, projectID, gitleaksID uint32) (*model.CodeGitleaks, error)

	// code_enterprise_org
	ListEnterpriseOrg(ctx context.Context, projectID, gitleaksID uint32) (*[]model.CodeEnterpriseOrg, error)
	UpsertEnterpriseOrg(ctx context.Context, data *code.EnterpriseOrgForUpsert) (*model.CodeEnterpriseOrg, error)
	DeleteEnterpriseOrg(ctx context.Context, projectID, gitleaksID uint32, login string) error
}

type Config 

type Config struct {
	MasterHost     string
	MasterUser     string
	MasterPassword string
	SlaveHost      string
	SlaveUser      string
	SlavePassword  string

	Schema        string
	Port          int
	LogMode       bool
	MaxConnection int
}

type DataSource 

type DataSource struct {
	AWSDataSourceID uint32
	DataSource      string
	MaxScore        float32
	AWSID           uint32 `gorm:"column:aws_id"`
	ProjectID       uint32
	AssumeRoleArn   string
	ExternalID      string
	Status          string
	StatusDetail    string
	ScanAt          time.Time
}

type DiagnosisRepoInterface 

type DiagnosisRepoInterface interface {
	ListDiagnosisDataSource(context.Context, uint32, string) (*[]model.DiagnosisDataSource, error)
	GetDiagnosisDataSource(context.Context, uint32, uint32) (*model.DiagnosisDataSource, error)
	UpsertDiagnosisDataSource(context.Context, *model.DiagnosisDataSource) (*model.DiagnosisDataSource, error)
	DeleteDiagnosisDataSource(context.Context, uint32, uint32) error
	ListWpscanSetting(context.Context, uint32, uint32) (*[]model.WpscanSetting, error)
	GetWpscanSetting(context.Context, uint32, uint32) (*model.WpscanSetting, error)
	UpsertWpscanSetting(context.Context, *model.WpscanSetting) (*model.WpscanSetting, error)
	DeleteWpscanSetting(context.Context, uint32, uint32) error
	ListPortscanSetting(context.Context, uint32, uint32) (*[]model.PortscanSetting, error)
	GetPortscanSetting(context.Context, uint32, uint32) (*model.PortscanSetting, error)
	UpsertPortscanSetting(context.Context, *model.PortscanSetting) (*model.PortscanSetting, error)
	DeletePortscanSetting(context.Context, uint32, uint32) error
	ListPortscanTarget(context.Context, uint32, uint32) (*[]model.PortscanTarget, error)
	GetPortscanTarget(context.Context, uint32, uint32) (*model.PortscanTarget, error)
	UpsertPortscanTarget(context.Context, *model.PortscanTarget) (*model.PortscanTarget, error)
	DeletePortscanTarget(context.Context, uint32, uint32) error
	DeletePortscanTargetByPortscanSettingID(context.Context, uint32, uint32) error
	ListApplicationScan(context.Context, uint32, uint32) (*[]model.ApplicationScan, error)
	GetApplicationScan(context.Context, uint32, uint32) (*model.ApplicationScan, error)
	UpsertApplicationScan(context.Context, *model.ApplicationScan) (*model.ApplicationScan, error)
	DeleteApplicationScan(context.Context, uint32, uint32) error
	ListApplicationScanBasicSetting(context.Context, uint32, uint32) (*[]model.ApplicationScanBasicSetting, error)
	GetApplicationScanBasicSetting(context.Context, uint32, uint32) (*model.ApplicationScanBasicSetting, error)
	UpsertApplicationScanBasicSetting(context.Context, *model.ApplicationScanBasicSetting) (*model.ApplicationScanBasicSetting, error)
	DeleteApplicationScanBasicSetting(context.Context, uint32, uint32) error

	//for InvokeScan
	ListAllWpscanSetting(context.Context) (*[]model.WpscanSetting, error)
}

type GCPDataSource 

type GCPDataSource struct {
	GCPID              uint32 `gorm:"primary_key column:gcp_id"`
	GoogleDataSourceID uint32 `gorm:"primary_key"`
	ProjectID          uint32
	Status             string
	StatusDetail       string
	ScanAt             time.Time
	CreatedAt          time.Time
	UpdatedAt          time.Time
	Name               string  // google_data_source.name
	Description        string  // google_data_source.description
	MaxScore           float32 // google_data_source.max_score
	GCPOrganizationID  string  // gcp.gcp_organization_id
	GCPProjectID       string  // gcp.gcp_project_id
}

type GoogleRepoInterface 

type GoogleRepoInterface interface {
	// google_data_source
	ListGoogleDataSource(ctx context.Context, googleDataSourceID uint32, name string) (*[]model.GoogleDataSource, error)

	// gcp
	ListGCP(ctx context.Context, projectID, gcpID uint32, gcpProjectID string) (*[]model.GCP, error)
	GetGCP(ctx context.Context, projectID, gcpID uint32) (*model.GCP, error)
	UpsertGCP(ctx context.Context, gcp *google.GCPForUpsert) (*model.GCP, error)
	DeleteGCP(ctx context.Context, projectID uint32, gcpID uint32) error

	// gcp_data_source
	ListGCPDataSource(ctx context.Context, projectID, gcpID uint32) (*[]GCPDataSource, error)
	GetGCPDataSource(ctx context.Context, projectID, gcpID, googleDataSourceID uint32) (*GCPDataSource, error)
	UpsertGCPDataSource(ctx context.Context, gcpDataSource *google.GCPDataSourceForUpsert) (*GCPDataSource, error)
	DeleteGCPDataSource(ctx context.Context, projectID, gcpID, googleDataSourceID uint32) error
	ListGCPDataSourceByDataSourceID(ctx context.Context, googleDataSourceID uint32) (*[]GCPDataSource, error)
}

type OSINTRepoInterface 

type OSINTRepoInterface interface {
	ListOsint(context.Context, uint32) (*[]model.Osint, error)
	GetOsint(context.Context, uint32, uint32) (*model.Osint, error)
	UpsertOsint(context.Context, *model.Osint) (*model.Osint, error)
	DeleteOsint(context.Context, uint32, uint32) error
	ListOsintDataSource(context.Context, uint32, string) (*[]model.OsintDataSource, error)
	GetOsintDataSource(context.Context, uint32, uint32) (*model.OsintDataSource, error)
	UpsertOsintDataSource(context.Context, *model.OsintDataSource) (*model.OsintDataSource, error)
	DeleteOsintDataSource(context.Context, uint32, uint32) error
	ListRelOsintDataSource(context.Context, uint32, uint32, uint32) (*[]model.RelOsintDataSource, error)
	GetRelOsintDataSource(context.Context, uint32, uint32) (*model.RelOsintDataSource, error)
	UpsertRelOsintDataSource(context.Context, *model.RelOsintDataSource) (*model.RelOsintDataSource, error)
	DeleteRelOsintDataSource(context.Context, uint32, uint32) error
	ListOsintDetectWord(context.Context, uint32, uint32) (*[]model.OsintDetectWord, error)
	GetOsintDetectWord(context.Context, uint32, uint32) (*model.OsintDetectWord, error)
	UpsertOsintDetectWord(context.Context, *model.OsintDetectWord) (*model.OsintDetectWord, error)
	DeleteOsintDetectWord(context.Context, uint32, uint32) error
	// For Invoke
	ListAllRelOsintDataSource(context.Context, uint32) (*[]model.RelOsintDataSource, error)
}

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.