package state

import (
	"time"

	"xorm.io/xorm"
	"xorm.io/xorm/names"
)



type BaseModel struct {
	ID        int64      `xorm:"pk autoincr" json:"id"`
	CreatedAt time.Time  `xorm:"created" json:"created_at"`
	UpdatedAt time.Time  `xorm:"updated" json:"updated_at"`
	DeletedAt *time.Time `xorm:"deleted" json:"deleted_at,omitempty"`
}



type Repo struct {
	BaseModel   `xorm:"extends"`
	NamespaceID int64  `xorm:"notnull unique(cred)" json:"namespace_id"`
	Namespace   string `xorm:"varchar(255) notnull index" json:"namespace"`
	NameID      int64  `xorm:"notnull unique(cred)" json:"name_id"`
	Name        string `xorm:"varchar(255) notnull" json:"name"`
	IsPrivate   bool   `xorm:"default false" json:"is_private"`
}


type SSHKey struct {
	BaseModel   `xorm:"extends"`
	PublicKey   string     `xorm:"text notnull" json:"-"`
	Fingerprint string     `xorm:"varchar(255) unique notnull" json:"fingerprint"`
	KeyType     string     `xorm:"varchar(50) notnull" json:"key_type"`
	Comment     string     `xorm:"varchar(255)" json:"comment,omitempty"`
	LastUsedAt  *time.Time `xorm:"" json:"last_used_at,omitempty"`
}




type PAT struct {
	BaseModel  `xorm:"extends"`
	TokenHash  string     `xorm:"varchar(64) unique notnull" json:"-"`
	Prefix     string     `xorm:"varchar(20)" json:"prefix"`
	Name       string     `xorm:"varchar(255)" json:"name"`
	ExpiresAt  *time.Time `xorm:"" json:"expires_at,omitempty"`
	LastUsedAt *time.Time `xorm:"" json:"last_used_at,omitempty"`
}






type ACL struct {
	BaseModel  `xorm:"extends"`
	CredType   int    `xorm:"notnull unique(cred) index" json:"cred_type"`
	CredID     int64  `xorm:"notnull unique(cred)" json:"cred_id"`
	TargetType int    `xorm:"notnull unique(cred) index" json:"target_type"`
	TargetID   int64  `xorm:"notnull unique(cred)" json:"target_id"`
	Perm       string `xorm:"varchar(20) default 'read'" json:"perm"`
}

type CredType int


const (
	CredTypePAT CredType = 0
	CredTypeSSH CredType = 1
)

type TargetType int

const (
	TargetTypeRepo TargetType = 0
	TargetTypeNS   TargetType = 1
)




func InitEngine(driverName, dataSourceName string) (*xorm.Engine, error) {
	engine, err := xorm.NewEngine(driverName, dataSourceName)
	if err != nil {
		return nil, err
	}
	engine.SetTableMapper(names.GonicMapper{})
	engine.SetColumnMapper(names.GonicMapper{})
	if err := engine.Sync2(
		new(Repo),
		new(SSHKey),
		new(PAT),
		new(ACL),
	); err != nil {
		return nil, err
	}
	return engine, nil
}