package model

import (
	"time"

	"github.com/lib/pq"
)

type ApiKey_PG struct {
	ID             int64          `gorm:"column:id;primaryKey;autoIncrement"`
	Name           string         `gorm:"column:name;not null;unique;index:idx_apikey_name"`
	ApiType        string         `gorm:"column:apitype;not null;unique;index:idx_apikey_apitype"`
	ApiKey         string         `gorm:"column:apikey;not null;unique;index:idx_apikey_apikey"`
	Status         int            `gorm:"type:int;default:0"` // enabled 0, disabled 1
	Endpoint       string         `gorm:"column:endpoint"`
	ResourceNmae   string         `gorm:"column:resource_name"`
	DeploymentName string         `gorm:"column:deployment_name"`
	ApiSecret      string         `gorm:"column:api_secret"`
	ModelPrefix    string         `gorm:"column:model_prefix"`
	ModelAlias     string         `gorm:"column:model_alias"`
	SupportModels  pq.StringArray `gorm:"type:text[]"`
	UpdatedAt      time.Time      `json:"updatedAt,omitempty"`
	CreatedAt      time.Time      `json:"createdAt,omitempty"`
}

func (ApiKey_PG) TableName() string {
	return "apikeys"
}

type ApiKey struct {
	ID             int64     `gorm:"column:id;primaryKey;autoIncrement"`
	Name           string    `gorm:"column:name;not null;unique;index:idx_apikey_name"`
	ApiType        string    `gorm:"column:apitype;not null;unique;index:idx_apikey_apitype"`
	ApiKey         string    `gorm:"column:apikey;not null;unique;index:idx_apikey_apikey"`
	Status         int       `json:"status" gorm:"type:int;default:0"` // enabled 0, disabled 1
	Endpoint       string    `gorm:"column:endpoint"`
	ResourceNmae   string    `gorm:"column:resource_name"`
	DeploymentName string    `gorm:"column:deployment_name"`
	ApiSecret      string    `gorm:"column:api_secret"`
	ModelPrefix    string    `gorm:"column:model_prefix"`
	ModelAlias     string    `gorm:"column:model_alias"`
	SupportModels  []string  `gorm:"type:json"`
	CreatedAt      time.Time `json:"created_at,omitempty" gorm:"autoUpdateTime"`
	UpdatedAt      time.Time `json:"updated_at,omitempty" gorm:"autoCreateTime"`
}

func (ApiKey) TableName() string {
	return "apikeys"
}