selfupgrade

type AgentFleet interface {
	// ListNodes 返回所有已安装 Agent 的节点（含离线）。/ ListNodes returns every host with an installed Agent, including offline ones.
	ListNodes(ctx context.Context) ([]FleetNode, error)
	// Node 返回单个节点的最新连接信息。/ Node returns the latest connection info of one host.
	Node(ctx context.Context, hostID uint) (FleetNode, bool)
	// TransferFile 让 Agent 通过现有 DownloadPackage 流拉取文件，返回 Agent 上的路径。
	// TransferFile has the Agent pull a file over the existing DownloadPackage stream and returns its path on the Agent.
	TransferFile(ctx context.Context, agentID, key, localPath, checksum string) (string, error)
	// SendUpgradeCommand 发送 UPGRADE 子命令并返回成功输出。/ SendUpgradeCommand sends an UPGRADE sub-command and returns its output.
	SendUpgradeCommand(ctx context.Context, agentID string, params map[string]string, timeout time.Duration) (string, error)
}

type AgentNodeUpgrade struct {
	HostID         uint   `json:"host_id"`
	HostName       string `json:"host_name"`
	IPAddress      string `json:"ip_address,omitempty"`
	AgentID        string `json:"agent_id,omitempty"`
	FromVersion    string `json:"from_version,omitempty"`
	TargetVersion  string `json:"target_version"`
	CurrentVersion string `json:"current_version,omitempty"`
	Status         string `json:"status"`
	Message        string `json:"message,omitempty"`
	ProxyStatus    string `json:"proxy_status,omitempty"`
	ProxyMessage   string `json:"proxy_message,omitempty"`
	// ProxyRestart 是该节点运行中 stx-java-proxy 的重启结果：restarted | not_running | rolled_back | failed | skipped。
	// ProxyRestart is the restart result of the node's running stx-java-proxy.
	ProxyRestart        string     `json:"proxy_restart,omitempty"`
	ProxyRestartMessage string     `json:"proxy_restart_message,omitempty"`
	UpdatedAt           *time.Time `json:"updated_at,omitempty"`
}

type AgentVersionInfo struct {
	HostID    uint   `json:"host_id"`
	HostName  string `json:"host_name"`
	IPAddress string `json:"ip_address,omitempty"`
	Version   string `json:"version"`
	Online    bool   `json:"online"`
	UpToDate  bool   `json:"up_to_date"`
}

type AssetSpec struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
	Required    bool   `json:"required"`
	Glob        bool   `json:"glob"`
}

type AssetState struct {
	AssetSpec
	Status   string `json:"status"`
	Size     int64  `json:"size,omitempty"`
	Path     string `json:"path,omitempty"`
	Message  string `json:"message,omitempty"`
	Progress int    `json:"progress,omitempty"`
}

type CheckRequest struct {
	SimulateFailure string `json:"simulate_failure,omitempty"`
	ForceLatest     string `json:"force_latest,omitempty"`
}

type CheckResult struct {
	CurrentVersion  string `json:"current_version"`
	LatestVersion   string `json:"latest_version"`
	UpdateAvailable bool   `json:"update_available"`
	ReleaseURL      string `json:"release_url,omitempty"`
	CheckedAt       string `json:"checked_at"`
	Message         string `json:"message,omitempty"`
	ErrorCode       string `json:"error_code,omitempty"`
	// StagedVersion 是检查后仍保留的已预置版本（可能与 LatestVersion 不同）。
	// StagedVersion is the staged version kept after the check (may differ from LatestVersion).
	StagedVersion string `json:"staged_version,omitempty"`
	// StagePreserved 表示本次检查保留了已预置包，可直接执行升级。
	// StagePreserved reports that this check kept the staged package, so execution can proceed.
	StagePreserved bool `json:"stage_preserved,omitempty"`
}

type ExecuteRequest struct {
	SimulateFailure string `json:"simulate_failure,omitempty"`
	// Mode 显式选择 "apply"（真实应用：替换并重启控制面、Agent、proxy、随包前端）或 "dry_run"（演练，不做任何变更）。
	// 省略时回退到旧字段 apply；两者都省略时为 dry_run（安全默认）。
	// Mode explicitly selects "apply" (really replace and restart the control plane, Agents, proxies and bundled
	// frontend) or "dry_run" (rehearse, change nothing). When omitted the legacy apply flag decides; when both are
	// omitted the request is a dry run (safe default).
	Mode string `json:"mode,omitempty"`
	// Apply 是旧字段，等价于 mode=apply；与 mode 冲突时报错。
	// Apply is the legacy flag, equivalent to mode=apply; conflicting with mode is rejected.
	Apply bool `json:"apply,omitempty"`
}

type FleetNode struct {
	HostID      uint
	HostName    string
	IPAddress   string
	AgentID     string
	Version     string
	OS          string
	Arch        string
	Online      bool
	ConnectedAt time.Time
	// InstallDirs 是该节点上 SeaTunnel 集群节点的安装目录，用于定位运行中的 stx-java-proxy 状态目录。
	// InstallDirs are the SeaTunnel install dirs of cluster nodes on this host, used to find running stx-java-proxy state.
	InstallDirs []string
}

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

type ProcessRestart struct {
	// Component: control_plane | frontend
	Component string `json:"component"`
	// Strategy: standalone | systemd | supervisor | pm2 | hook | none
	Strategy string     `json:"strategy,omitempty"`
	Status   string     `json:"status"`
	Message  string     `json:"message,omitempty"`
	OldPID   int        `json:"old_pid,omitempty"`
	NewPID   int        `json:"new_pid,omitempty"`
	At       *time.Time `json:"at,omitempty"`
}

type Response struct {
	ErrorMsg  string      `json:"error_msg"`
	ErrorCode string      `json:"error_code,omitempty"`
	Data      interface{} `json:"data"`
}

type RestartPlanInfo struct {
	ControlPlaneStrategy string `json:"control_plane_strategy"`
	ControlPlaneDetail   string `json:"control_plane_detail,omitempty"`
	// FrontendDetected 表示检测到随包的 Next standalone 前端进程，会一并重启。
	// FrontendDetected reports a bundled Next standalone frontend process that is restarted too.
	FrontendDetected bool   `json:"frontend_detected"`
	FrontendStrategy string `json:"frontend_strategy,omitempty"`
	// RestartsAgents / RestartsProxies 恒为 true：Agent 换二进制后原地重启，运行中的 stx-java-proxy 换 jar 后重启。
	// RestartsAgents / RestartsProxies are always true: Agents restart in place after the swap, running proxies after the jar swap.
	RestartsAgents  bool `json:"restarts_agents"`
	RestartsProxies bool `json:"restarts_proxies"`
	// SeaTunnelAffected 恒为 false：自升级从不停止、重启或改动 SeaTunnel/Hazelcast 进程与作业。
	// SeaTunnelAffected is always false: self-upgrade never stops, restarts or modifies SeaTunnel/Hazelcast processes or jobs.
	SeaTunnelAffected bool `json:"seatunnel_affected"`
}

type RolloutOptions struct {
	Concurrency      int
	CommandTimeout   time.Duration
	ReconnectTimeout time.Duration
	PollInterval     time.Duration
}

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

type StageDownloadRequest struct {
	TargetVersion   string `json:"target_version,omitempty"`
	SimulateFailure string `json:"simulate_failure,omitempty"`
}

type StageState struct {
	Mode          string       `json:"mode,omitempty"`
	Status        string       `json:"status"`
	Progress      int          `json:"progress"`
	TargetVersion string       `json:"target_version,omitempty"`
	Assets        []AssetState `json:"assets"`
	Message       string       `json:"message,omitempty"`
	ErrorCode     string       `json:"error_code,omitempty"`
	Cancellable   bool         `json:"cancellable"`
}

type StatusResponse struct {
	CurrentVersion  string       `json:"current_version"`
	GitCommit       string       `json:"git_commit,omitempty"`
	BuildTime       string       `json:"build_time,omitempty"`
	LatestVersion   string       `json:"latest_version,omitempty"`
	UpdateAvailable bool         `json:"update_available"`
	Checked         bool         `json:"checked"`
	OverallStatus   string       `json:"overall_status"`
	Stage           StageState   `json:"stage"`
	ActiveTask      *TaskState   `json:"active_task,omitempty"`
	LastCheck       *CheckResult `json:"last_check,omitempty"`
	MockMode        bool         `json:"mock_mode"`
	// DeploymentType distinguishes container vs host deployments ("docker" | "host").
	// DeploymentType 区分容器化与物理宿主机部署 ("docker" | "host")。
	DeploymentType string `json:"deployment_type"`
	// DockerComposeUpgradeCmd provides the suggested container upgrade command.
	// DockerComposeUpgradeCmd 提供推荐的容器升级命令。
	DockerComposeUpgradeCmd string `json:"docker_compose_upgrade_cmd,omitempty"`
	// Architecture reflects the host operating system and CPU architecture.
	// Architecture 反映当前宿主机操作系统与架构 (如 linux/arm64)。
	Architecture string `json:"architecture"`
	// Agents 列出各节点当前 Agent 版本及是否与控制面一致。
	// Agents lists each node's current Agent version and whether it matches the control plane.
	Agents []AgentVersionInfo `json:"agents,omitempty"`
	// AgentsOutdated 是版本与控制面不一致的 Agent 数量。
	// AgentsOutdated counts Agents whose version differs from the control plane.
	AgentsOutdated int `json:"agents_outdated"`
	// RestartPlan 描述真实应用时会重启的 STX 进程（SeaTunnel 不受影响）。
	// RestartPlan describes the STX processes an apply restarts (SeaTunnel is unaffected).
	RestartPlan *RestartPlanInfo `json:"restart_plan,omitempty"`
}

type StepState struct {
	Code      string     `json:"code"`
	Label     string     `json:"label"`
	Status    string     `json:"status"`
	Message   string     `json:"message,omitempty"`
	Error     string     `json:"error,omitempty"`
	StartedAt *time.Time `json:"started_at,omitempty"`
	EndedAt   *time.Time `json:"ended_at,omitempty"`
}

type TaskState struct {
	ID            string      `json:"id"`
	Kind          string      `json:"kind"` // stage | execute
	Status        string      `json:"status"`
	TargetVersion string      `json:"target_version,omitempty"`
	CurrentStep   string      `json:"current_step,omitempty"`
	FailureStep   string      `json:"failure_step,omitempty"`
	FailureReason string      `json:"failure_reason,omitempty"`
	ErrorCode     string      `json:"error_code,omitempty"`
	Steps         []StepState `json:"steps,omitempty"`
	// AgentNodes 是各 Worker 节点 Agent/Proxy 的升级状态。
	// AgentNodes is the per-node Agent/Proxy upgrade status.
	AgentNodes []AgentNodeUpgrade `json:"agent_nodes,omitempty"`
	// ExecutionMode 是 apply 或 dry_run；DryRun 便于直接判断。
	// ExecutionMode is apply or dry_run; DryRun is a convenience flag.
	ExecutionMode string `json:"execution_mode,omitempty"`
	DryRun        bool   `json:"dry_run"`
	// Restarts 是控制面与随包前端的重启结果。/ Restarts holds control plane and bundled frontend restart results.
	Restarts    []ProcessRestart `json:"restarts,omitempty"`
	Cancellable bool             `json:"cancellable"`
	StartedAt   *time.Time       `json:"started_at,omitempty"`
	CompletedAt *time.Time       `json:"completed_at,omitempty"`
}

type UpgradeHistoryRecord struct {
	ID             string      `json:"id"`
	StartedAt      time.Time   `json:"started_at"`
	CompletedAt    time.Time   `json:"completed_at"`
	DurationMs     int64       `json:"duration_ms"`
	FromVersion    string      `json:"from_version"`
	TargetVersion  string      `json:"target_version"`
	Mode           string      `json:"mode"`            // online | offline
	DeploymentType string      `json:"deployment_type"` // host | docker
	Status         string      `json:"status"`          // succeeded | failed | rolled_back
	FailureStep    string      `json:"failure_step,omitempty"`
	FailureReason  string      `json:"failure_reason,omitempty"`
	ErrorCode      string      `json:"error_code,omitempty"`
	Steps          []StepState `json:"steps,omitempty"`
	// AgentNodes 是各节点 Agent/Proxy 的升级结果（离线节点重连后补升级会回写）。
	// AgentNodes is the per-node Agent/Proxy result (updated when offline nodes catch up after reconnecting).
	AgentNodes []AgentNodeUpgrade `json:"agent_nodes,omitempty"`
	// ExecutionMode 是 apply 或 dry_run（旧记录为空）。/ ExecutionMode is apply or dry_run (empty for older records).
	ExecutionMode string           `json:"execution_mode,omitempty"`
	DryRun        bool             `json:"dry_run"`
	Restarts      []ProcessRestart `json:"restarts,omitempty"`
}