storage

type CScan struct {
	DsName    string `json:"dsName"`
	DbName    string `json:"dbName"`
	TableName string `json:"tableName"`
	Columns   string `json:"columns"`
	Classes   string `json:"classes"`
	Tags      string `json:"tags"`
}

type Class struct {
	Id          int    `csv:"id"`
	Description string `csv:"description"`
	Rule        string `csv:"rule"`
	Class       string `csv:"class"`
}

type DpDataSource struct {
	ID           int    `json:"id"`
	Datadomain   string `json:"Datadomain"`
	Dsname       string `json:"Dsname"`
	Dsdecription string `json:"Dsdecription"`
	Dstype       string `json:"Dstype"`
	DsKey        string `json:"DsKey"`
	Dsversion    string `json:"Dsversion"`
	Host         string `json:"Host"`
	Port         string `json:"Port"`
	User         string `json:"User"`
	Password     string `json:"Password"`
	CreatedAt    string `json:"CreatedAt"`
}

type DpDbColumn struct {
	ColumnName    string `json:"column_name"`
	ColumnType    string `json:"ColumnType"`
	ColumnComment string `json:"Column_Comment"`
	Tags          string `json:"Tags"`
	Classes       string `json:"Classes"`
}

type DpDbDatabase struct {
	DsKey      string      `json:"DbKey"`
	Name       string      `json:"Name"`
	Type       string      `json:"Type"`
	DpDbTables []DpDbTable `json:"DpDbTable"`
}

type DpDbTable struct {
	Name        string       `json:"Name"`
	Tags        string       `json:"Tags"`
	DpDbColumns []DpDbColumn `json:"DpDbColumns"`
}

type DpSensitiveDataElementTag struct {
	ID                     int    `json:"id" gorm:"primary_key"`
	SensitiveDataElementID int    `json:"sensitive_data_element_id"`
	WorkspaceID            int    `json:"workspace_id"`
	Tag                    string `json:"tag"`
}

type InternalStorage struct {
	SqliteConnc *sql.DB
	Path        string
}

type RecommendedPolicy struct {
	PolicyId   int32
	PolicyName string
}

type ScanDto struct {
	DatabaseId int `json:"db_id"`
}

type SensitiveDataElement struct {
	ID                      int                         `json:"id" gorm:"primary_key"`
	WorkspaceID             int                         `json:"workspace_id"`
	DpDbColumn              string                      `json:"dp_db_column"`
	ElementID               int                         `json:"element_id"`
	DeletedAt               time.Time                   `json:"deleted_at"`
	SensitivityClass        string                      `json:"sensitivity_class"`
	SensitiveDataElementTag []DpSensitiveDataElementTag `gorm:"foreignKey:SensitiveDataElementID;references:ID" json:"sensitive_data_element_tag"`
}

type SensitiveElementTag struct {
	SensitiveElementTags []SensitiveElementTags `json:"sensitive_data_element"`
}

type SensitiveElementTags struct {
	SensitivityClass string `json:"sensitivity_class"`
	Tags             string `json:"tags"`
}

type Storage interface {
	GetClasses() ([]Class, error)
	AddClass(string, string, string) error
	DeleteClasses(ids []int64) (bool, error)

	GetTags() ([]Tag, error)
	AddTag(string, string, []string) error
	DeleteTags(ids []int64) (bool, error)

	GetAssociatedTags(string) ([]Tag, error)
	GetAssociatedClasses(string) ([]Class, error)
	SetSchemaData(DpDbDatabase) error

	AddDataSource(DpDataSource) (int64, error)
	GetDataSources() ([]DpDataSource, error)
	DeleteDataSources(ids []int64) (bool, error)
	Scan(dsname string) error
	GetStatus(dsname string) (string, error)
	GetRecommendedPolicy(dsname string) ([]RecommendedPolicy, error)
	ApplyPolicy(dsname string, ids []int64) error
	GetDataSource(dsname string) (DpDataSource, error)

	GetScanLog(ds string, db string, table string, columns []string) ([]CScan, error)

	UpdateDSStatus(dsid int64, statusid int64) error
}

type StorageConfig struct {
	Type string
	Path string
}

type Tag struct {
	Id          int    `csv:"id"`
	TagName     string `csv:"tag_name"`
	Rule        string `csv:"rule"`
	Description string `csv:"description"`
}