datastore

type UploadFile struct {
	Name                string             `bson:"name" json:"name"`
	Description         string             `bson:"description" json:"description"`
	Filename            string             `bson:"filename" json:"filename"`
	ID                  primitive.ObjectID `bson:"_id" json:"id"`
	OpenAIFileID        string             `bson:"openai_file_id" json:"openai_file_id"` // https://platform.openai.com/docs/assistants/tools/supported-files
	CreatedAt           time.Time          `bson:"created_at,omitempty" json:"created_at,omitempty"`
	CreatedByUserName   string             `bson:"created_by_user_name" json:"created_by_user_name"`
	CreatedByUserID     primitive.ObjectID `bson:"created_by_user_id" json:"created_by_user_id"`
	ModifiedAt          time.Time          `bson:"modified_at,omitempty" json:"modified_at,omitempty"`
	ModifiedByUserName  string             `bson:"modified_by_user_name" json:"modified_by_user_name"`
	ModifiedByUserID    primitive.ObjectID `bson:"modified_by_user_id" json:"modified_by_user_id"`
	ObjectKey           string             `bson:"object_key" json:"object_key"`
	ObjectURL           string             `bson:"object_url" json:"object_url"`
	Status              int8               `bson:"status" json:"status"`
	ContentType         int8               `bson:"content_type" json:"content_type"`
	TenantID            primitive.ObjectID `bson:"tenant_id,omitempty" json:"tenant_id,omitempty"`
	TenantName          string             `bson:"tenant_name" json:"tenant_name"`
	UploadDirectoryID   primitive.ObjectID `bson:"upload_directory_id,omitempty" json:"upload_directory_id,omitempty"`
	UploadDirectoryName string             `bson:"upload_directory_name" json:"upload_directory_name"`
	UserID              primitive.ObjectID `bson:"user_id" json:"user_id"`
	UserName            string             `bson:"user_name" json:"user_name"`
	UserLexicalName     string             `bson:"user_lexical_name" json:"user_lexical_name"`
}

type UploadFileAsSelectOption struct {
	Value primitive.ObjectID `bson:"_id" json:"value"` // Extract from the database `_id` field and output through API as `value`.
	Label string             `bson:"name" json:"label"`
}

type UploadFilePaginationListFilter struct {
	// Pagination related.
	Cursor    string
	PageSize  int64
	SortField string
	SortOrder int8 // 1=ascending | -1=descending

	// Filter related.
	TenantID          primitive.ObjectID
	UploadDirectoryID primitive.ObjectID
	Status            int8
	UUIDs             []string
	SearchText        string

	Name        string
	Description string

	CreatedAtGT  time.Time
	CreatedAtGTE time.Time
	CreatedAtLT  time.Time
	CreatedAtLTE time.Time
}

type UploadFilePaginationListResult struct {
	Results     []*UploadFile `json:"results"`
	NextCursor  string        `json:"next_cursor"`
	HasNextPage bool          `json:"has_next_page"`
}

type UploadFileStorer interface {
	Create(ctx context.Context, m *UploadFile) error
	GetByID(ctx context.Context, id primitive.ObjectID) (*UploadFile, error)
	UpdateByID(ctx context.Context, m *UploadFile) error
	ListByFilter(ctx context.Context, m *UploadFilePaginationListFilter) (*UploadFilePaginationListResult, error)
	ListAsSelectOptionByFilter(ctx context.Context, f *UploadFilePaginationListFilter) ([]*UploadFileAsSelectOption, error)
	DeleteByID(ctx context.Context, id primitive.ObjectID) error
	GetOpenAIFileIDsInUploadDirectoryIDs(ctx context.Context, uploadDirectoryIDs []primitive.ObjectID) ([]string, error)
	ListByUploadDirectoryID(ctx context.Context, uploadDirectoryID primitive.ObjectID) (*UploadFilePaginationListResult, error)
	ListByUploadDirectoryIDs(ctx context.Context, uploadDirectoryIDs []primitive.ObjectID) (*UploadFilePaginationListResult, error)
}

type UploadFileStorerImpl struct {
	Logger     *slog.Logger
	DbClient   *mongo.Client
	Collection *mongo.Collection
}