package llm

import (
	"reflect"

	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/gotypes"
	"yunion.io/x/pkg/util/sets"

	"yunion.io/x/onecloud/pkg/apis"
	computeapi "yunion.io/x/onecloud/pkg/apis/compute"
)

var (
	LLM_SKU_BASE_NETWORK_TYPES = sets.NewString(
		string(computeapi.NETWORK_TYPE_HOSTLOCAL),
		string(computeapi.NETWORK_TYPE_GUEST),
	)
)

func IsLLMSkuBaseNetworkType(t string) bool {
	return LLM_SKU_BASE_NETWORK_TYPES.Has(t)
}

type HostInfo struct {
	HostId       string `json:"host_id"`
	Host         string `json:"host"`
	HostAccessIp string `json:"host_access_ip"`
	HostEIP      string `json:"host_eip"`
}

func init() {
	gotypes.RegisterSerializable(reflect.TypeOf(&PortMappings{}), func() gotypes.ISerializable {
		return &PortMappings{}
	})
	gotypes.RegisterSerializable(reflect.TypeOf(&Devices{}), func() gotypes.ISerializable {
		return &Devices{}
	})
	gotypes.RegisterSerializable(reflect.TypeOf(&PortMappingEnvs{}), func() gotypes.ISerializable {
		return &PortMappingEnvs{}
	})
	gotypes.RegisterSerializable(reflect.TypeOf(&Envs{}), func() gotypes.ISerializable {
		return &Envs{}
	})
}

type PortMappingEnv struct {
	Key       string `json:"key"`
	ValueFrom string `json:"value_from"`
}

type PortMappingEnvs []PortMappingEnv

func (pm PortMappingEnvs) String() string {
	return jsonutils.Marshal(pm).String()
}

func (pm PortMappingEnvs) IsZero() bool {
	return len(pm) == 0
}

type PortMapping struct {
	Protocol        string                           `json:"protocol"`
	ContainerPort   int                              `json:"container_port"`
	RemoteIps       []string                         `json:"remote_ips"`
	FirstPortOffset *int                             `json:"first_port_offset"`
	Envs            []computeapi.GuestPortMappingEnv `json:"envs"`
}

type PortMappings []PortMapping

func (s PortMappings) String() string {
	return jsonutils.Marshal(s).String()
}

func (s PortMappings) IsZero() bool {
	return len(s) == 0
}

type Device struct {
	DevType    string `json:"dev_type"`
	Model      string `json:"model"`
	DevicePath string `json:"device_path"`
}

type Devices []Device

func (s Devices) String() string {
	return jsonutils.Marshal(s).String()
}

func (s Devices) IsZero() bool {
	return len(s) == 0
}

type Env struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type Envs []Env

func (s Envs) String() string {
	return jsonutils.Marshal(s).String()
}

func (s Envs) IsZero() bool {
	return len(s) == 0
}

type LLMSkuDetails struct {
	apis.SharableVirtualResourceDetails
	// 当前大模型套餐包含的实例个数。
	LLMCapacity int    `json:"llm_capacity"`
	Image       string `json:"image"`
	ImageLabel  string `json:"image_label"`
	ImageName   string `json:"image_name"`

	MountedModelDetails []MountedModelInfo `json:"mounted_model_details"`

	Template string `json:"template"`

	// LLMType 为 SKU 类型（ollama/vllm/dify），与 llm_spec 一起带出
	LLMType string `json:"llm_type"`
	// LLMSpec 从 SKU 持久化字段带出，保证 list/show 与 create 一致
	LLMSpec *LLMSpec `json:"llm_spec,omitempty"`
}

type MountedAppResourceDetails struct {
	MountedModels []string `json:"mounted_models"`
}

type LLMSKuBaseCreateInput struct {
	apis.SharableVirtualResourceCreateInput

	Cpu       int `json:"cpu"`
	Memory    int `json:"memory"`
	Bandwidth int `json:"bandwidth"`

	Volumes      *Volumes          `json:"volumes"`
	PortMappings *PortMappings     `json:"port_mappings"`
	Devices      *Devices          `json:"devices"`
	Envs         *Envs             `json:"envs"`
	Properties   map[string]string `json:"properties"`
}

type LLMSkuBaseUpdateInput struct {
	apis.SharableVirtualResourceBaseUpdateInput

	Cpu    *int `json:"cpu"`
	Memory *int `json:"memory"`

	// RequstSyncImage *bool `json:"request_sync_image"`

	DiskSize    *int     `json:"disk_size" yunion-deprecated-by:"disk_size_mb"`
	DiskSizeMB  *int     `json:"disk_size_mb"`
	TemplateId  *string  `json:"template_id"`
	StorageType *string  `json:"storage_type"`
	Volumes     *Volumes `json:"volumes"`

	Bandwidth    *int              `json:"bandwidth"`
	PortMappings *PortMappings     `json:"port_mappings"`
	Devices      *Devices          `json:"devices"`
	Envs         *Envs             `json:"envs"`
	Properties   map[string]string `json:"properties"`
}

type LLMSkuListInput struct {
	apis.SharableVirtualResourceListInput
	MountedModelResourceListInput

	LLMType  string   `json:"llm_type"`
	LLMTypes []string `json:"llm_types"`
}

type LLMSkuCreateInput struct {
	LLMSKuBaseCreateInput
	MountedModelResourceCreateInput

	LLMImageId string `json:"llm_image_id"`
	LLMType    string `json:"llm_type"`

	// LLMSpec:
	// - ollama/vllm: backend builds llm_spec from llm_image_id + mounted_models; for vllm preferred model should be set in llm_spec.vllm.preferred_model.
	// - dify: client must send llm_spec with type "dify" and dify payload.
	LLMSpec *LLMSpec `json:"llm_spec,omitempty"`
}

type LLMSkuUpdateInput struct {
	LLMSkuBaseUpdateInput
	MountedModelResourceUpdateInput

	LLMImageId string `json:"llm_image_id"`

	// LLMSpec:
	// - dify: send full spec to update image ids.
	// - ollama/vllm: backend may build from llm_image_id/mounted_models; for vllm preferred model should be set in llm_spec.vllm.preferred_model.
	LLMSpec *LLMSpec `json:"llm_spec,omitempty"`
}

// type LLMModelCloneInput struct {
// 	Name string `json:"name"`
// }

// type LLMModelSyncImageRequestTaskInput struct {
// 	Request bool `json:"request"`
// }

// type DifySkulListInput struct {
// 	apis.SharableVirtualResourceListInput
// 	MountedModelResourceListInput
// }

// type DifySkuCreateInput struct {
// 	LLMSKuBaseCreateInput

// 	PostgresImageId     string `json:"postgres_image_id"`
// 	RedisImageId        string `json:"redis_image_id"`
// 	NginxImageId        string `json:"nginx_image_id"`
// 	DifyApiImageId      string `json:"dify_api_image_id"`
// 	DifyPluginImageId   string `json:"dify_plugin_image_id"`
// 	DifyWebImageId      string `json:"dify_web_image_id"`
// 	DifySandboxImageId  string `json:"dify_sandbox_image_id"`
// 	DifySSRFImageId     string `json:"dify_ssrf_image_id"`
// 	DifyWeaviateImageId string `json:"dify_weaviate_image_id"`
// }

// type DifySkuUpdateInput struct {
// 	LLMSkuBaseUpdateInput

// 	PostgresImageId     string `json:"postgres_image_id"`
// 	RedisImageId        string `json:"redis_image_id"`
// 	NginxImageId        string `json:"nginx_image_id"`
// 	DifyApiImageId      string `json:"dify_api_image_id"`
// 	DifyPluginImageId   string `json:"dify_plugin_image_id"`
// 	DifyWebImageId      string `json:"dify_web_image_id"`
// 	DifySandboxImageId  string `json:"dify_sandbox_image_id"`
// 	DifySSRFImageId     string `json:"dify_ssrf_image_id"`
// 	DifyWeaviateImageId string `json:"dify_weaviate_image_id"`
// }