models

type Artifact struct {
	ID          uint32 `gorm:"primaryKey;autoIncrement"`
	Name        string `gorm:"unique"`
	ProfileName string `gorm:"index;"`

	CreatedAt time.Time `gorm:"->;<-:create;"`
	UpdatedAt time.Time `gorm:"->;<-:update;"`
	Target    string    // build target, like win64, win32, linux64
	Type      string    // beacon、pulse、modules、prelude
	//Type      string    // build type, pe, dll, shellcode
	Source string // docker 、 saas 、github action、upload...
	//CA            string // ca file , ca file content
	Path        string
	Profile     Profile `gorm:"foreignKey:ProfileName;references:Name;-:migration;"`
	Os          string
	Format      string // file extension like .exe/.dll/.so/.dylib or ""
	Arch        string
	Log         string
	Status      string
	ParamsData  string
	Params      *implanttypes.ProfileParams `gorm:"-"`
	ProfileByte []byte
	Comment     string
}

type AuthzRule struct {
	ID        uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`
	CreatedAt time.Time `gorm:"->;<-:create;"`
	Role      string    `gorm:"index;type:string;"`
	Method    string    `gorm:"type:string;"`
	Allow     bool      `gorm:"default:true"`
}

type Certificate struct {
	ID        uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`
	CreatedAt time.Time `gorm:"->;<-:create;"`
	Name      string    `gorm:"unique"`
	Type      string    // 证书类型: selfsigned, letsencrypt, imported
	Domain    string    // 证书绑定域名
	CertPEM   string    // 证书内容 PEM
	KeyPEM    string    // 私钥内容 PEM
	CACertPEM string    // CA 证书内容 PEM
	CAKeyPEM  string    // CA 私钥内容 PEM
}

type Context struct {
	ID         uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid"`
	CreatedAt  time.Time `gorm:"->;<-:create"`
	UpdatedAt  time.Time
	SessionID  string `gorm:"type:string;index;"`
	PipelineID string `gorm:"type:string;index;"`
	TaskID     string `gorm:"type:string;index;"`
	Type       string
	Nonce      string
	Value      []byte
	Context    output.Context `gorm:"-"`

	Session  *Session  `gorm:"foreignKey:SessionID;references:SessionID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
	Pipeline *Pipeline `gorm:"foreignKey:PipelineID;references:Name;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
	Task     *Task     `gorm:"foreignKey:TaskID;references:ID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
}

type Operator struct {
	ID               uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`
	CreatedAt        time.Time `gorm:"->;<-:create;"`
	Name             string    `gorm:"uniqueIndex"`
	Remote           string    `gorm:"type:string;"`
	Type             string    `gorm:"type:string;"`
	Role             string    `gorm:"type:string;default:'operator'"`
	Fingerprint      string    `gorm:"uniqueIndex;type:string;"`
	Revoked          bool      `gorm:"default:false"`
	CAType           int
	KeyType          string
	CaCertificatePEM string
	CertificatePEM   string
	PrivateKeyPEM    string
}

type Pipeline struct {
	ID                           uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`
	CreatedAt                    time.Time `gorm:"->;<-:create;"`
	ListenerId                   string    `gorm:"type:string;uniqueIndex:idx_pipelines_listener_name"`
	Name                         string    `gorm:"type:string;unique;uniqueIndex:idx_pipelines_listener_name"`
	IP                           string    `gorm:"type:string;default:''"`
	Host                         string    `config:"host"`
	Port                         uint32    `config:"port"`
	Type                         string    `gorm:"type:string;"`
	Enable                       bool      `gorm:"type:boolean;"`
	ParamsData                   string    `gorm:"column:params"`
	CertName                     string    `gorm:"type:string;"`
	*implanttypes.PipelineParams `gorm:"-"`
}

type Profile struct {
	ID uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`

	// build
	Name string `gorm:"unique"` // Ensuring Name is unique

	// params
	Params     *implanttypes.ProfileParams `gorm:"-"`             // 使用 interface{} 使其更灵活
	ParamsData string                      `gorm:"column:params"` // 改用更简洁的数据库字段名

	// BasicPipeline 和 PulsePipeline
	PipelineID string `gorm:"type:string;index;"`

	// BasicPipeline 和 PulsePipeline
	Pipeline *Pipeline `gorm:"foreignKey:PipelineID;references:Name;-:migration;"`

	CreatedAt time.Time `gorm:"->;<-:create;"`
}

type Session struct {
	SessionID   string `gorm:"primaryKey;->;<-:create;type:uuid;"`
	RawID       uint32
	CreatedAt   time.Time `gorm:"->;<-:create;"`
	Note        string
	GroupName   string
	Target      string
	Initialized bool
	Type        string
	PipelineID  string
	ListenerID  string
	IsAlive     bool
	LastCheckin int64
	IsRemoved   bool                   `gorm:"default:false"`
	Data        *client.SessionContext `gorm:"-"`
	DataString  string                 `gorm:"column:data"`

	ProfileName string  `gorm:"index;"`
	Profile     Profile `gorm:"foreignKey:ProfileName;references:Name;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
}

type Task struct {
	ID          string    `gorm:"primaryKey;->;<-:create;"`
	Created     time.Time `gorm:"->;<-:create;"`
	Deadline    time.Time
	CallBy      string
	Seq         uint32
	Type        string
	SessionID   string
	Session     Session `gorm:"foreignKey:SessionID;references:SessionID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
	Cur         int
	Total       int
	Description string
	ClientName  string
	FinishTime  time.Time
	LastTime    time.Time
}

type Timer struct {
	Expression string  `json:"expression"`
	Jitter     float64 `json:"jitter"`
}

type WebsiteContent struct {
	ID        uuid.UUID `gorm:"primaryKey;->;<-:create;type:uuid;"`
	CreatedAt time.Time `gorm:"->;<-:create;"`

	File        string `gorm:""`
	Path        string `gorm:""`
	Size        uint64 `gorm:""`
	Type        string `gorm:""`
	ContentType string `gorm:""`
	Auth        string `gorm:""` // "user:pass" or empty; "none" = skip website default

	Pipeline   *Pipeline `gorm:"foreignKey:PipelineID;references:Name;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
	PipelineID string    `gorm:"type:string;index;"`
}