common

type Capabilities interface {
	ContainersEngine() bool
	ContainersNetworking() bool
	Volumes() bool
	VolumesZFS() bool
	VMM() bool
}

type ContainerState struct {
	ContainerID   string  `json:"container_id"`
	CPUUsage      float64 `json:"cpu_usage"` // Important (percentage)
	CPUSystem     float64 `json:"cpu_system"`
	CPUUser       float64 `json:"cpu_user"`
	MemoryUsage   float64 `json:"mem_usage"`
	MemoryLimit   float64 `json:"mem_limit"`
	MemoryPercent float64 `json:"mem_percentage"` // Important (percentage)
}

type ContainerStates map[string]ContainerState

type Deployment struct {
	ID         DeploymentID
	LoadName   string
	LoadDriver LoadDriver
	Status     DeploymentStatus
	Metadata   any
}

type DeploymentID = uuid.UUID

type DeploymentStatus struct {
	StatusCode DeploymentStatusCode `json:"status"`
	Reason     string               `json:"reason"`
}

type DeploymentStatusCode string

type DeploymentsRepository interface {
	Create(loadName string, driver LoadDriver, status DeploymentStatus, metadata any) (DeploymentID, error)
	UpdateStatus(deploymentID DeploymentID, status DeploymentStatus) error
	UpdateMetadata(deploymentID DeploymentID, updateFn func(metadata any) error) error

	GetAll() ([]Deployment, error)
	GetByLoad(loadName string) ([]Deployment, error)
	GetByLoadAndStatus(loadName string, statusCode DeploymentStatusCode) ([]Deployment, error)
	GetDeployment(deploymentID DeploymentID) (*Deployment, error)

	DeleteByLoad(loadName string) error
	DeleteDeployment(deploymentID uuid.UUID) error
}

type Hash [32]byte

type HashableLoadConfig struct {
	Name         string       `json:"name"`
	Node         string       `json:"node"`
	DependsOn    []string     `json:"depends_on"`
	Driver       LoadDriverID `json:"driver"`
	DriverConfig any          `json:"driver_config"`
}

type Job struct {
	Name   string
	Node   *Node
	Driver JobDriver
}

type JobConfig struct {
	Name         string      `json:"name"`
	Node         string      `json:"node"`
	Driver       JobDriverID `json:"driver"`
	DriverConfig any         `json:"driver_config"`
}

type JobDriver interface {
	ID() JobDriverID

	Info() JobDriverInfo

	Run(w JobResultWriter, args ...string) error

	Config() JobDriverConfig
}

type JobDriverConfig struct {
	Driver       JobDriverID `json:"driver"`
	DriverConfig any         `json:"driver_config,omitempty"`
}

type JobDriverError struct {
	Code     JobDriverErrorCode
	DriverID JobDriverID
	Err      error
}

type JobDriverErrorCode string

type JobDriverID string

type JobDriverInfo struct {
	ID  JobDriverID
	New func(config any) (JobDriver, error)
}

type JobResult struct {
	Value *string `json:"value,omitempty"`
	Err   *string `json:"err,omitempty"`
}

type JobResultWriter struct {
	// contains filtered or unexported fields
}

type Load struct {
	Name       string
	Node       *Node
	DependsOn  []*Load
	Driver     LoadDriver
	StartChain LoadChain
	StopChain  LoadChain
}

type LoadChain []*Load

type LoadConfig struct {
	Name           string       `json:"name"`
	Node           string       `json:"node"`
	DependsOn      []string     `json:"depends_on"`
	Driver         LoadDriverID `json:"driver"`
	DriverConfig   any          `json:"driver_config"`
	StartChainHash string       `json:"start_chain_hash"`
	StopChainHash  string       `json:"stop_chain_hash"`
}

type LoadDriver interface {
	// GetLoadDriverID returns the unique identifier for this load driver.
	ID() LoadDriverID

	// Info returns driver description for internal factory use.
	Info() LoadDriverInfo

	// Provision validates prerequisites and creates a deployment in "provisioned" status.
	// It shall check load requirements but it won't check depending loads.
	// This is invoked within the agent and does not affect client behavior.
	//
	// Returns the deployment ID for the provisioned deployment.
	Provision(node NodeDriver, repository DeploymentsRepository, loadName string) (DeploymentID, error)

	// Deprovision removes a provisioned deployment with cleanup
	// Only operates on deployments in "provisioned" status.
	Deprovision(repository DeploymentsRepository, deployment Deployment) error

	// Start starts the load execution for a provisioned deployment.
	// This has no effect when called from the client.
	//
	// LoadDriver is responsible of the consistency of the DeploymentsRepository
	Start(repository DeploymentsRepository, deployment Deployment) error

	// Stop stops a running load execution within the agent.
	// This has no effect when called from the client.
	//
	// LoadDriver is responsible of the consistency of the DeploymentsRepository
	Stop(repository DeploymentsRepository, deployment Deployment) error

	// Kill stops immediately a running load execution within the agent.
	// This has no effect when called from the client.
	//
	// LoadDriver is responsible of the consistency of the DeploymentsRepository
	Kill(repository DeploymentsRepository, deployment Deployment) error

	// UpdateStatus update and returns current status based on internal drivers factors.
	UpdateStatus(repository DeploymentsRepository, deployment Deployment) (DeploymentStatus, error)

	// Config returns the configuration for this load driver.
	Config() LoadDriverConfig

	// Stream load stdout to writer
	StreamStdout(repository DeploymentsRepository, deployment Deployment, w io.Writer) error

	// Stream load stderr to writer
	StreamStderr(repository DeploymentsRepository, deployment Deployment, w io.Writer) error

	// Read load stdout from offset, returns bytes read and end position
	ReadStdout(repository DeploymentsRepository, deployment Deployment, offset int64) ([]byte, int64, error)

	// Read load stderr from offset, returns bytes read and end position
	ReadStderr(repository DeploymentsRepository, deployment Deployment, offset int64) ([]byte, int64, error)
}

type LoadDriverConfig struct {
	Driver       LoadDriverID `json:"driver"`
	DriverConfig any          `json:"driver_config"`
}

type LoadDriverError struct {
	Code     LoadDriverErrorCode
	DriverID LoadDriverID
	Err      error
}

type LoadDriverErrorCode string

type LoadDriverID string

type LoadDriverInfo struct {
	ID  LoadDriverID
	New func(config any) (LoadDriver, error)
}

type NetworkInterface struct {
	Name      string   `json:"name"`
	HWAddr    string   `json:"hwaddr,omitempty"`
	Addresses []string `json:"addresses,omitempty"`
}

type NewNodeDriverInfoOpts func(i *NodeDriverInfo) error

type Node struct {
	Name       string
	Url        string
	CloudInit  *cloudinit.CloudInit
	Registries []RegistryConfig
	Driver     NodeDriver
}

type NodeConfig struct {
	Name         string               `json:"name"`
	Url          string               `json:"url"`
	CloudInit    *cloudinit.CloudInit `json:"cloud_init,omitempty"`
	Driver       NodeDriverID         `json:"driver,omitempty"`
	DriverConfig *any                 `json:"driver_config,omitempty"`
	Registries   []RegistryConfig     `json:"registries,omitempty"`
}

type NodeContexBuilder func(nodeName string) NodeContext

type NodeContext struct {
	Repository   NodesRepository // Only available when agent is running
	Capabilities Capabilities
	NodeName     string
	RunMode      RunMode
}

type NodeDriver interface {
	// ID returns the unique identifier for this node driver.
	ID() NodeDriverID

	// Info returns driver description for internal factory use.
	Info() (NodeDriverInfo, error)

	// Config returns the configuration for this node driver.
	Config() NodeDriverConfig

	// PowerOn starts the node
	PowerOn(cloudInit *cloudinit.CloudInit) error

	// PowerOff stops the node immediately
	PowerOff() error

	// Shutdown stops the node
	Shutdown(message string, time uint32) error

	// Restart restarts the node
	// Message will be shown to users before shutdown on the time
	// offset specified
	Restart(message string, time uint32) error

	// State returns current node state (e.g. cpu, mem, etc...)
	State() (NodeState, error)

	// UpdateStatus update and returns current status based on internal drivers factors
	//
	// nodeName as nil for self-node
	UpdateStatus() (NodeStatus, error)
}

type NodeDriverBuilder func(ctx NodeContext, config *any) (NodeDriver, error)

type NodeDriverConfig struct {
	Driver       NodeDriverID `json:"driver"`
	DriverConfig *any         `json:"driver_config"`
}

type NodeDriverID string

type NodeDriverInfo struct {
	ID        NodeDriverID
	New       NodeDriverBuilder
	GuestMode bool
}

type NodeEntry struct {
	NodeName   string
	NodeDriver NodeDriver
	Registries []RegistryConfig
	Metadata   any
}

type NodeState struct {
	NumCPU          int     `json:"ncpu"`
	UserCPU         uint64  `json:"cpu_user"`
	IdleCPU         uint64  `json:"cpu_idle"`
	SystemCPU       uint64  `json:"cpu_system"`
	TotalCPU        uint64  `json:"cpu_total"`
	UsageCPUPercent float64 `json:"cpu_usage_percentage"` // Important

	TotalMem       uint64  `json:"mem_total"`
	UsedMem        uint64  `json:"mem_used"`
	FreeMem        uint64  `json:"mem_free"`
	FreeMemPercent float64 `json:"mem_free_percentage"` // Important

	FreeStorage uint64 `json:"free_storage"`

	// Swap memory information
	SwapTotal uint64 `json:"swap_total"` // Total available swap memory in bytes
	SwapUsed  uint64 `json:"swap_used"`  // Total used swap memory in bytes
	SwapFree  uint64 `json:"swap_free"`  // Total free swap memory in bytes

	// CPU load average
	CpuLoadAvg float32 `json:"cpu_load_avg"` // CPU load average (from 0 to 1)

	// Process counts
	ProcTotalCount    uint32 `json:"proc_total_count"`    // Number of active processes
	ProcSleepingCount uint32 `json:"proc_sleeping_count"` // Number of sleeping processes
	ProcRunningCount  uint32 `json:"proc_running_count"`  // Number of running processes
	ProcZombieCount   uint32 `json:"proc_zombie_count"`   // Number of zombie processes
	ProcStoppedCount  uint32 `json:"proc_stopped_count"`  // Number of stopped processes
	ProcIdleCount     uint32 `json:"proc_idle_count"`     // Number of idle processes
	ProcThreadsCount  uint32 `json:"proc_threads_count"`  // Number of threads

	Containers []ContainerState `json:"containers,omitempty"`

	NetworkInterfaces []NetworkInterface `json:"network_interfaces,omitempty"`
}

type NodeStatus struct {
	StatusCode NodeStatusCode `json:"status"`
	Reason     string         `json:"reason"`
}

type NodeStatusCode string

type NodesRepository interface {
	// SetSelf creates or updates the node for the caller
	SetSelf(nodeName string, driver NodeDriver, registries []RegistryConfig) error

	// GetSelf return nodeentry for the caller node
	GetSelf() (NodeEntry, error)

	// DeleteSelf caller node from repository
	DeleteSelf() error

	// Update self node metadata
	UpdateSelfMetadata(updateFn func(metadataPtr any) error) error

	// Retrieve node entry by agent ID
	GetByAgentId(agentId string) (NodeEntry, error)

	// SetGuestNode creates or update the guest node entry for the caller host node
	SetGuestNode(guestNodeName string, guestDriver NodeDriver) error

	// GetGuestNode returns a guest node of the caller host node
	GetGuestNode(guestNodeName string) (NodeEntry, error)

	// GetAllGuestNodes returns all guest nodes of the self node
	GetAllGuestNodes() ([]NodeEntry, error)

	// DeleteGuestNode deletes the guest node entry for the caller host node
	DeleteGuestNode(guestNodeName string, guestDriver NodeDriver) error

	// Update guest node metadata
	UpdateGuestMetadata(guestNodeName string, updateFn func(metadataPtr any) error) error
}

type RegistryAuth struct {
	Username string `json:"username,omitempty"`
	Password string `json:"password,omitempty"`
	Token    string `json:"token,omitempty"`
}

type RegistryConfig struct {
	Host     string       `json:"host"` // Registry host (e.g., "ghcr.io", "docker.io", "registry.example.com:5000")
	Auth     RegistryAuth `json:"auth"`
	Insecure bool         `json:"insecure,omitempty"` // Allow HTTP instead of HTTPS

	// Skip verification of the registry's TLS certificate.
	SkipTLSVerify bool `json:"skip_tls_verify,omitempty"`

	// Path to a PEM bundle with additional CAs trusted for this registry.
	CAFile string `json:"ca_file,omitempty"`
}

type RunMode int