dataset

type CreateDatasetFile struct {
	Name   string
	Reader io.Reader
}

type CreateDatasetRequest struct {
	Name        string              `json:"name"`
	Description string              `json:"description"`
	Type        db_dataset.Type     `json:"type"`
	Data        []string            `json:"data"`  // for list type
	Files       []CreateDatasetFile `json:"files"` // for csv type
}

type DatasetAPIRequest struct {
	Name        string                  `form:"name"`
	Description string                  `form:"description"`
	Type        string                  `form:"type"`
	Data        []string                `form:"data"`
	Files       []*multipart.FileHeader `form:"files"`
}

type DatasetInfo struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
	Type        string `json:"type"`
	ColumnCount int    `json:"column_count"`
	ValueCount  int    `json:"value_count"`
	// for list type, data is the available options
	// for files type, data is the files in the dataset dir
	Data    []string `json:"data"`
	Columns []string `json:"columns"`
}

type DatasetRows struct {
	Rows []map[string]any `json:"rows"`
	Data []string         `json:"data"`
	Type db_dataset.Type  `json:"type"`
}

type DatasetService interface {
	Get(ctx context.Context, source string) (*DatasetInfo, error)
	List(ctx context.Context) ([]*DatasetInfo, error)
	Create(ctx context.Context, req *CreateDatasetRequest) (string, error)
	Update(ctx context.Context, dataset string, req *UpdateDatasetRequest) error
	Delete(ctx context.Context, dataset string) error
	Preview(ctx context.Context, source string) (*DatasetRows, error)
}

type DatasetServiceImpl struct {
	// contains filtered or unexported fields
}

type DatasetServiceMock struct {
	// CreateFunc mocks the Create method.
	CreateFunc func(ctx context.Context, req *CreateDatasetRequest) (string, error)

	// DeleteFunc mocks the Delete method.
	DeleteFunc func(ctx context.Context, dataset string) error

	// GetFunc mocks the Get method.
	GetFunc func(ctx context.Context, source string) (*DatasetInfo, error)

	// ListFunc mocks the List method.
	ListFunc func(ctx context.Context) ([]*DatasetInfo, error)

	// PreviewFunc mocks the Preview method.
	PreviewFunc func(ctx context.Context, source string) (*DatasetRows, error)

	// UpdateFunc mocks the Update method.
	UpdateFunc func(ctx context.Context, dataset string, req *UpdateDatasetRequest) error
	// contains filtered or unexported fields
}

type UpdateDatasetRequest struct {
	CreateDatasetRequest
	Fields []string `json:"fields"`
}