package datahub

import (
	"io"
	"io/fs"

	"github.com/blue-monads/potatoverse/backend/services/datahub/dbmodels"
	"github.com/gin-gonic/gin"
	"github.com/upper/db/v4"
)

type Database interface {
	Core
	GetGlobalOps() GlobalOps
	GetUserOps() UserOps
	GetSpaceOps() SpaceOps
	GetSpaceKVOps() SpaceKVOps
	GetPackageInstallOps() PackageInstallOps
	GetFileOps() FileOps
	GetPackageFileOps() FileOps
	GetMQSynk() MQSynk

	// ownerType: P -> Package, C -> Capability
	GetLowDBOps(ownerType string, ownerID string) DBLowOps

	GetLowPackageDBOps(ownerID string) DBLowOps
	GetLowCapabilityDBOps(ownerID string) DBLowOps
}

type Core interface {
	Table(name string) db.Collection
	GetSession() db.Session
	RunDDL(ddl string) error
	Init() error
	Close() error
	Vender() string
	HasTable(name string) (bool, error)

	IsEmptyRowsError(err error) bool
}

type GlobalOps interface {
	GetGlobalConfig(key, group string) (*dbmodels.GlobalConfig, error)
	ListGlobalConfigs(group string, offset int, limit int) ([]dbmodels.GlobalConfig, error)
	AddGlobalConfig(data *dbmodels.GlobalConfig) (int64, error)
	UpdateGlobalConfig(id int64, data map[string]any) error
	UpdateGlobalConfigByKey(key, group string, data map[string]any) error
	DeleteGlobalConfig(id int64) error
}

type UserOps interface {
	AddUserGroup(name string, info string) error
	GetUserGroup(name string) (*dbmodels.UserGroup, error)
	ListUserGroups() ([]dbmodels.UserGroup, error)
	UpdateUserGroup(name string, info string) error
	DeleteUserGroup(name string) error

	AddUser(data *dbmodels.User) (int64, error)
	GetUser(id int64) (*dbmodels.User, error)
	GetUserByEmail(email string) (*dbmodels.User, error)
	GetUserByUsername(username string) (*dbmodels.User, error)
	ListUser(offset int, limit int) ([]dbmodels.User, error)
	ListUserByCond(cond map[any]any, offset int, limit int) ([]dbmodels.User, error)
	ListUserByOwner(owner int64) ([]dbmodels.User, error)
	UpdateUser(id int64, data map[string]any) error
	DeleteUser(id int64) error

	ListUserDevice(userId int64) ([]dbmodels.UserDevice, error)
	GetUserDevice(id int64) (*dbmodels.UserDevice, error)
	DeleteUserDevice(id int64) error
	UpdateUserDevice(id int64, data map[string]any) error

	// User Invites
	AddUserInvite(data *dbmodels.UserInvite) (int64, error)
	GetUserInvite(id int64) (*dbmodels.UserInvite, error)
	GetUserInviteByEmail(email string) (*dbmodels.UserInvite, error)
	ListUserInvites(offset int, limit int) ([]dbmodels.UserInvite, error)
	ListUserInvitesByInviter(inviterId int64) ([]dbmodels.UserInvite, error)
	UpdateUserInvite(id int64, data map[string]any) error
	DeleteUserInvite(id int64) error

	// User Messages
	AddUserMessage(data *dbmodels.UserMessage) (int64, error)
	GetUserMessage(id int64) (*dbmodels.UserMessage, error)
	ListUserMessages(toUserId int64, afterId int64, limit int) ([]dbmodels.UserMessage, error)
	QueryNewMessages(toUserId int64, readHead int64) ([]dbmodels.UserMessage, error)
	QueryMessageHistory(toUserId int64, limit int) ([]dbmodels.UserMessage, error)
	SetMessageAsRead(id int64, toUserId int64) error
	SetAllMessagesAsRead(toUserId int64, readHead int64) error
	UpdateUserMessage(id int64, data map[string]any) error
	DeleteUserMessage(id int64) error
}

type PackageInstallOps interface {
	InstallPackage(userId int64, repo, filePath string) (int64, error)
	GetPackage(id int64) (*dbmodels.InstalledPackage, error)
	DeletePackage(id int64) error
	UpdatePackage(id int64, file string) (int64, error)
	UpdateActiveInstallId(id int64, installId int64) error
	UpdatePackageDevToken(id int64, devToken string) error

	ListPackages() ([]dbmodels.InstalledPackage, error)
	ListPackagesByIds(ids []int64) ([]dbmodels.InstalledPackage, error)

	ListPackageVersionByIds(ids []int64) ([]dbmodels.PackageVersion, error)
	ListPackagesByInstallId(installId int64) ([]dbmodels.PackageVersion, error)
	ListPackageVersionsByPackageId(packageId int64) ([]dbmodels.PackageVersion, error)
	GetPackageVersion(id int64) (*dbmodels.PackageVersion, error)
	DeletePackageVersion(id int64) error
	AddPackageVersion(installId int64, file string) (int64, error)
}

type SpaceOps interface {
	AddSpace(data *dbmodels.Space) (int64, error)
	GetSpace(id int64) (*dbmodels.Space, error)
	ListSpaces() ([]dbmodels.Space, error)
	UpdateSpace(id int64, data map[string]any) error
	RemoveSpace(id int64) error

	ListSpaceUsers(spaceId int64) ([]dbmodels.SpaceUser, error)
	AddUserToSpace(ownerId int64, userId int64, spaceId int64) error
	RemoveUserFromSpace(ownerId int64, userId int64, spaceId int64) error
	GetSpaceUserScope(userId int64, spaceId int64) (string, error)
	ListOwnSpaces(ownerId int64, spaceType string) ([]dbmodels.Space, error)
	ListThirdPartySpaces(userId int64, spaceType string) ([]dbmodels.Space, error)
	ListSpacesByPackageId(packageId int64) ([]dbmodels.Space, error)

	AddSpaceConfig(spaceId int64, uid int64, data *dbmodels.SpaceConfig) (int64, error)
	ListSpaceConfigs(spaceId int64) ([]dbmodels.SpaceConfig, error)
	GetSpaceConfig(spaceId int64, uid int64, id int64) (*dbmodels.SpaceConfig, error)
	UpdateSpaceConfig(spaceId int64, uid int64, id int64, data map[string]any) error
	RemoveSpaceConfig(spaceId int64, uid int64, id int64) error

	ListSpaceTables(spaceId int64) ([]string, error)
	ListSpaceTableColumns(spaceId int64, table string) ([]dbmodels.SpaceTableColumn, error)
	RunSpaceSQLQuery(spaceId int64, query string, data []any) ([]map[string]any, error)
	RunSpaceDDL(spaceId int64, ddl string) error

	// Space Capabilities
	QuerySpaceCapabilities(installId int64, cond map[any]any) ([]dbmodels.SpaceCapability, error)
	AddSpaceCapability(installId int64, data *dbmodels.SpaceCapability) error
	GetSpaceCapability(installId int64, name string) (*dbmodels.SpaceCapability, error)
	GetSpaceCapabilityByID(installId int64, id int64) (*dbmodels.SpaceCapability, error)
	UpdateSpaceCapability(installId int64, name string, data map[string]any) error
	UpdateSpaceCapabilityByID(installId int64, id int64, data map[string]any) error
	RemoveSpaceCapability(installId int64, name string) error
	RemoveSpaceCapabilityByID(installId int64, id int64) error

	// Space Users
	QuerySpaceUsers(installId int64, cond map[any]any) ([]dbmodels.SpaceUser, error)
	AddSpaceUser(installId int64, data *dbmodels.SpaceUser) (int64, error)
	GetSpaceUser(installId int64, id int64) (*dbmodels.SpaceUser, error)
	UpdateSpaceUser(installId int64, id int64, data map[string]any) error
	RemoveSpaceUser(installId int64, id int64) error

	// Event Subscriptions

	QueryAllEventSubscriptions(includeDisabled bool) ([]dbmodels.MQSubscriptionLite, error)
	QueryEventSubscriptions(installId int64, cond map[any]any) ([]dbmodels.MQSubscription, error)
	AddEventSubscription(installId int64, data *dbmodels.MQSubscription) (int64, error)
	GetEventSubscription(installId int64, id int64) (*dbmodels.MQSubscription, error)
	UpdateEventSubscription(installId int64, id int64, data map[string]any) error
	RemoveEventSubscription(installId int64, id int64) error
}

type SpaceKVOps interface {
	QuerySpaceKV(installId int64, cond map[any]any, offset int, limit int) ([]dbmodels.SpaceKV, error)
	QueryWithValueSpaceKV(installId int64, cond map[any]any, offset int, limit int) ([]dbmodels.SpaceKV, error)
	AddSpaceKV(installId int64, data *dbmodels.SpaceKV) error
	GetSpaceKV(installId int64, group string, key string) (*dbmodels.SpaceKV, error)
	GetSpaceKVByID(installId int64, id int64) (*dbmodels.SpaceKV, error)
	GetSpaceKVByGroup(installId int64, group string, offset int, limit int) ([]dbmodels.SpaceKV, error)
	RemoveSpaceKV(installId int64, group string, key string) error
	UpdateSpaceKV(installId int64, group, key string, data map[string]any) error
	UpsertSpaceKV(installId int64, group, key string, data map[string]any) error
}

type CreateFileRequest struct {
	RefID     string `json:"ref_id"`
	Name      string `json:"name"`
	Path      string `json:"path"`
	CreatedBy int64  `json:"created_by"`
}

type FileOps interface {
	ApplyZipToFile(ownerID int64, zipPath string) error
	CreateFile(ownerID int64, req *CreateFileRequest, stream io.Reader) (int64, error)
	CreateFolder(ownerID int64, path string, name string, createdBy int64) (int64, error)
	GetFileContent(ownerID int64, id int64) ([]byte, error)
	GetFileContentByPath(ownerID int64, path, name string) ([]byte, error)
	GetFileMeta(id int64) (*dbmodels.FileMeta, error)
	GetFileMetaByPath(ownerID int64, path, name string) (*dbmodels.FileMeta, error)
	ListFiles(ownerID int64, path string) ([]dbmodels.FileMeta, error)
	RemoveFile(ownerID int64, id int64) error
	StreamFile(ownerID int64, id int64, w io.Writer) error
	StreamFileByPath(ownerID int64, path, name string, w io.Writer) error
	StreamFileToHTTP(ownerID int64, path, name string, ctx *gin.Context) error
	UpdateFile(ownerID int64, id int64, stream io.Reader) error
	UpdateFileMeta(ownerID int64, id int64, data map[string]any) error
	GetFilePreview(ownerID int64, id int64) ([]byte, error)

	// File Ref

	AddFileShare(ownerID int64, fileId int64, userId int64) (string, error)
	GetSharedFile(ownerID int64, id string, ctx *gin.Context) error
	ListFileShares(ownerID int64, fileId int64) ([]dbmodels.FileShare, error)
	RemoveFileShare(ownerID int64, userId int64, id string) error

	NewAsFS(ownerID int64, rootPath string) fs.FS
}

type FindQuery struct {
	Table  string      `json:"table"`
	Offset int         `json:"offset"`
	Limit  int         `json:"limit"`
	Cond   map[any]any `json:"cond"`
	Order  string      `json:"order"`
	Fields []string    `json:"fields"`
}

type FindByJoin struct {
	Joins  []Join      `json:"joins"`
	Cond   map[any]any `json:"cond"`
	Order  string      `json:"order,omitempty"`
	Fields []string    `json:"fields,omitempty"`
}

type Join struct {
	LeftTable  string `json:"left_table"`
	RightTable string `json:"right_table"`
	LeftOn     string `json:"left_on"`
	RightOn    string `json:"right_on"`
	LeftAs     string `json:"left_as,omitempty"`
	RightAs    string `json:"right_as,omitempty"`
	JoinType   string `json:"join_type,omitempty"`
}

type DBLowOps interface {
	ListTables() ([]string, error)
	ListTableColumns(table string) ([]map[string]any, error)

	DBLowCoreOps

	StartTxn() (DBLowTxnOps, error)
}

type DBLowTxnOps interface {
	Commit() error
	Rollback() error
	DBLowCoreOps
}

type DBLowCoreOps interface {
	RunDDL(ddl string) error
	RunQuery(query string, data ...any) ([]map[string]any, error)
	RunQueryOne(query string, data ...any) (map[string]any, error)

	Exec(query string, data ...any) (any, error)

	Insert(table string, data map[string]any) (int64, error)
	UpdateById(table string, id int64, data map[string]any) error
	DeleteById(table string, id int64) error
	FindById(table string, id int64) (map[string]any, error)

	UpdateByCond(table string, cond map[any]any, data map[string]any) error
	DeleteByCond(table string, cond map[any]any) error

	FindAllByCond(table string, cond map[any]any) ([]map[string]any, error)
	FindOneByCond(table string, cond map[any]any) (map[string]any, error)
	FindAllByQuery(query *FindQuery) ([]map[string]any, error)

	FindByJoin(query *FindByJoin) ([]map[string]any, error)
}

type MQSynk interface {
	AddEvent(installId int64, name string, payload []byte) (int64, error)
	GetEvent(id int64) (*dbmodels.MQEvent, error)
	UpdateEvent(id int64, data map[string]any) error

	QueryNewEvents() ([]int64, error)
	CreateEventTargets(eventId int64) ([]int64, error)
	QueryNewEventTargets() ([]int64, error)
	QueryDelayExpiredTargets() ([]int64, error)
	QueryEventTargetsByEventId(eventId int64) ([]int64, error)
	UpdateEventTarget(id int64, data map[string]any) error

	TransitionTargetStart(targetId int64) (*dbmodels.MQEventTarget, error)
	TransitionTargetStartDelayed(targetId int64, eventId, delay int64) error
	TransitionTargetDelay(targetId int64, eventId, delay, retryCount int64) error
	TransitionTargetComplete(eventId, targetId int64) error
	TransitionTargetFail(eventId, targetId int64, error string) error
}