installer

type AgentManager interface {
	// GetAgentByHostID returns the agent connection for a host
	// GetAgentByHostID 返回主机的 Agent 连接
	GetAgentByHostID(hostID uint) (agentID string, connected bool)

	// SendInstallCommand sends an installation command to an agent
	// SendInstallCommand 向 Agent 发送安装命令
	SendInstallCommand(ctx context.Context, agentID string, params map[string]string) (commandID string, err error)

	// GetCommandStatus returns the status of a command
	// GetCommandStatus 返回命令的状态
	GetCommandStatus(commandID string) (status string, progress int, message string, err error)

	// SendCommand sends a command to an agent and returns the result
	// SendCommand 向 Agent 发送命令并返回结果
	SendCommand(ctx context.Context, agentID string, commandType string, params map[string]string) (success bool, output string, err error)

	// SendTransferPackageCommand sends a package transfer chunk to an agent
	// SendTransferPackageCommand 向 Agent 发送安装包传输块
	SendTransferPackageCommand(ctx context.Context, agentID string, version string, fileName string, chunk []byte, offset int64, totalSize int64, isLast bool, checksum string) (success bool, receivedBytes int64, localPath string, err error)
}

type AvailableVersions struct {
	Versions            []string                                 `json:"versions"`
	RecommendedVersion  string                                   `json:"recommended_version"`
	LocalPackages       []PackageInfo                            `json:"local_packages"`
	VersionCapabilities map[string]seatunnel.VersionCapabilities `json:"version_capabilities"`
}

type CheckStatus string

type CheckpointConfig struct {
	StorageType      CheckpointStorageType `json:"storage_type"`
	Namespace        string                `json:"namespace"`
	HDFSNameNodeHost string                `json:"hdfs_namenode_host,omitempty"`
	HDFSNameNodePort int                   `json:"hdfs_namenode_port,omitempty"`
	// HDFS Kerberos authentication / HDFS Kerberos 认证
	KerberosPrincipal      string `json:"kerberos_principal,omitempty"`
	KerberosKeytabFilePath string `json:"kerberos_keytab_file_path,omitempty"`
	// HDFS HA mode configuration / HDFS HA 模式配置
	HDFSHAEnabled             bool   `json:"hdfs_ha_enabled,omitempty"`
	HDFSNameServices          string `json:"hdfs_name_services,omitempty"`
	HDFSHANamenodes           string `json:"hdfs_ha_namenodes,omitempty"`
	HDFSNamenodeRPCAddress1   string `json:"hdfs_namenode_rpc_address_1,omitempty"`
	HDFSNamenodeRPCAddress2   string `json:"hdfs_namenode_rpc_address_2,omitempty"`
	HDFSFailoverProxyProvider string `json:"hdfs_failover_proxy_provider,omitempty"`
	// HdfsSitePath 是可选的 hdfs-site.xml 路径，对应 SeaTunnel hdfs_site_path。
	// HdfsSitePath is the optional hdfs-site.xml path, mapped to SeaTunnel hdfs_site_path.
	HdfsSitePath string `json:"hdfs_site_path,omitempty"`
	// DisableCache 对应 plugin-config.disable.cache；nil 表示沿用默认（hdfs 类型默认关闭缓存）。
	// DisableCache maps to plugin-config.disable.cache; nil keeps the default (cache off for hdfs type).
	DisableCache *bool `json:"disable_cache,omitempty"`
	// OSS/S3 configuration / OSS/S3 配置
	StorageEndpoint  string `json:"storage_endpoint,omitempty"`
	StorageAccessKey string `json:"storage_access_key,omitempty"`
	StorageSecretKey string `json:"storage_secret_key,omitempty"`
	StorageBucket    string `json:"storage_bucket,omitempty"`
	// S3CredentialsProvider 对应 fs.s3a.aws.credentials.provider；空则使用 SimpleAWSCredentialsProvider。
	// S3CredentialsProvider maps to fs.s3a.aws.credentials.provider; empty uses SimpleAWSCredentialsProvider.
	S3CredentialsProvider string `json:"s3_credentials_provider,omitempty"`
}

type CheckpointStorageType string

type ConfigInitializer interface {
	// InitClusterConfigs initializes cluster configs by pulling from a host
	// InitClusterConfigs 通过从主机拉取来初始化集群配置
	InitClusterConfigs(ctx context.Context, clusterID uint, hostID uint, installDir string, userID uint) error
}

type ConnectorConfig struct {
	InstallConnectors bool         `json:"install_connectors"`
	Connectors        []string     `json:"connectors,omitempty"`
	PluginRepo        MirrorSource `json:"plugin_repo,omitempty"`
	// SelectedPlugins is the list of plugin names to install during SeaTunnel setup
	// SelectedPlugins 是 SeaTunnel 安装过程中要安装的插件名称列表
	SelectedPlugins []string `json:"selected_plugins,omitempty"`
	// SelectedPluginProfiles stores selected profile keys for profile-aware plugins
	// SelectedPluginProfiles 保存带画像插件选择的画像键
	SelectedPluginProfiles map[string][]string `json:"selected_plugin_profiles,omitempty"`
}

type DeletePackageResponse struct {
	ErrorMsg string `json:"error_msg"`
	Data     any    `json:"data"`
}

type DeploymentMode string

type DownloadListResponse struct {
	ErrorMsg string          `json:"error_msg"`
	Data     []*DownloadTask `json:"data"`
}

type DownloadRequest struct {
	Version    string       `json:"version" binding:"required"`
	Mirror     MirrorSource `json:"mirror"`
	WithSource *bool        `json:"with_source,omitempty"`
}

type DownloadResponse struct {
	ErrorMsg string        `json:"error_msg"`
	Data     *DownloadTask `json:"data"`
}

type DownloadStatus string

type DownloadTask struct {
	ID              string         `json:"id"`
	Version         string         `json:"version"`
	Mirror          MirrorSource   `json:"mirror"`
	DownloadURL     string         `json:"download_url"`
	Status          DownloadStatus `json:"status"`
	Progress        int            `json:"progress"` // 0-100
	DownloadedBytes int64          `json:"downloaded_bytes"`
	TotalBytes      int64          `json:"total_bytes"`
	Speed           int64          `json:"speed"` // bytes per second
	Message         string         `json:"message,omitempty"`
	Error           string         `json:"error,omitempty"`
	StartTime       time.Time      `json:"start_time"`
	EndTime         *time.Time     `json:"end_time,omitempty"`
	ExecutionID     string         `json:"execution_id,omitempty"`
	SourceRequested bool           `json:"source_requested"`
	SourceStatus    DownloadStatus `json:"source_status,omitempty"`
	SourceURL       string         `json:"source_url,omitempty"`
	SourceProgress  int            `json:"source_progress,omitempty"`
	SourceBytes     int64          `json:"source_downloaded_bytes,omitempty"`
	SourceTotal     int64          `json:"source_total_bytes,omitempty"`
	SourceChecksum  string         `json:"source_checksum,omitempty"`
	SourceError     string         `json:"source_error,omitempty"`
	OwnerUserID     uint64         `json:"-"`
	// contains filtered or unexported fields
}

type ExecutionRequest struct {
	RequestID      string
	IdempotencyKey string
	RequestHash    string
	Confirmed      bool
	ConfirmationID string
	ClientType     string
}

type GetPackageInfoResponse struct {
	ErrorMsg string       `json:"error_msg"`
	Data     *PackageInfo `json:"data"`
}

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

type HostInfo struct {
	ID          uint       `json:"id"`
	Name        string     `json:"name,omitempty"`
	AgentID     string     `json:"agent_id"`
	AgentStatus string     `json:"agent_status"`
	LastSeen    *time.Time `json:"last_seen"`
}

type HostProvider interface {
	// GetHostByID returns host information by ID
	// GetHostByID 根据 ID 返回主机信息
	GetHostByID(ctx context.Context, hostID uint) (*HostInfo, error)
}

type IMAPConfig struct {
	StorageType IMAPStorageType `json:"storage_type"`
	Namespace   string          `json:"namespace"`
	// HDFS configuration / HDFS 配置
	HDFSNameNodeHost string `json:"hdfs_namenode_host,omitempty"`
	HDFSNameNodePort int    `json:"hdfs_namenode_port,omitempty"`
	// HDFS Kerberos authentication / HDFS Kerberos 认证
	KerberosPrincipal      string `json:"kerberos_principal,omitempty"`
	KerberosKeytabFilePath string `json:"kerberos_keytab_file_path,omitempty"`
	// HDFS HA mode configuration / HDFS HA 模式配置
	HDFSHAEnabled             bool   `json:"hdfs_ha_enabled,omitempty"`
	HDFSNameServices          string `json:"hdfs_name_services,omitempty"`
	HDFSHANamenodes           string `json:"hdfs_ha_namenodes,omitempty"`
	HDFSNamenodeRPCAddress1   string `json:"hdfs_namenode_rpc_address_1,omitempty"`
	HDFSNamenodeRPCAddress2   string `json:"hdfs_namenode_rpc_address_2,omitempty"`
	HDFSFailoverProxyProvider string `json:"hdfs_failover_proxy_provider,omitempty"`
	// HdfsSitePath 是可选的 hdfs-site.xml 路径，对应 SeaTunnel hdfs_site_path。
	// HdfsSitePath is the optional hdfs-site.xml path, mapped to SeaTunnel hdfs_site_path.
	HdfsSitePath string `json:"hdfs_site_path,omitempty"`
	// DisableCache 对应 plugin-config.disable.cache；nil 表示沿用默认。
	// DisableCache maps to plugin-config.disable.cache; nil keeps the default.
	DisableCache *bool `json:"disable_cache,omitempty"`
	// OSS/S3 configuration / OSS/S3 配置
	StorageEndpoint  string `json:"storage_endpoint,omitempty"`
	StorageAccessKey string `json:"storage_access_key,omitempty"`
	StorageSecretKey string `json:"storage_secret_key,omitempty"`
	StorageBucket    string `json:"storage_bucket,omitempty"`
	// S3CredentialsProvider 对应 fs.s3a.aws.credentials.provider；空则使用 SimpleAWSCredentialsProvider。
	// S3CredentialsProvider maps to fs.s3a.aws.credentials.provider; empty uses SimpleAWSCredentialsProvider.
	S3CredentialsProvider string `json:"s3_credentials_provider,omitempty"`
}

type IMAPStorageType string

type InstallMode string

type InstallResponse struct {
	ErrorMsg string              `json:"error_msg"`
	Data     *InstallationStatus `json:"data"`
}

type InstallStep string

type InstallationRequest struct {
	HostID                  string                 `json:"host_id"`
	ClusterID               string                 `json:"cluster_id"`
	Version                 string                 `json:"version" binding:"required"`
	InstallDir              string                 `json:"install_dir"`
	InstallMode             InstallMode            `json:"install_mode"`
	Mirror                  MirrorSource           `json:"mirror,omitempty"`
	PackagePath             string                 `json:"package_path,omitempty"`
	DeploymentMode          DeploymentMode         `json:"deployment_mode"`
	NodeRole                NodeRole               `json:"node_role"`
	MasterAddresses         []string               `json:"master_addresses,omitempty"`
	WorkerAddresses         []string               `json:"worker_addresses,omitempty"` // Worker addresses for separated mode / 分离模式的 worker 地址
	ClusterPort             int                    `json:"cluster_port,omitempty"`     // Master hazelcast port / Master Hazelcast 端口
	WorkerPort              int                    `json:"worker_port,omitempty"`      // Worker hazelcast port / Worker Hazelcast 端口
	HTTPPort                int                    `json:"http_port,omitempty"`        // SeaTunnel HTTP API 端口
	JavaProxyPort           int                    `json:"java_proxy_port,omitempty"`  // Managed stx-java-proxy listen port / 托管 stx-java-proxy 监听端口
	EnableHTTP              *bool                  `json:"enable_http,omitempty"`      // 是否开启 SeaTunnel HTTP / Web UI
	DynamicSlot             *bool                  `json:"dynamic_slot,omitempty"`
	SlotNum                 *int                   `json:"slot_num,omitempty"`
	SlotAllocationStrategy  SlotAllocationStrategy `json:"slot_allocation_strategy,omitempty"`
	JobScheduleStrategy     JobScheduleStrategy    `json:"job_schedule_strategy,omitempty"`
	HistoryJobExpireMinutes *int                   `json:"history_job_expire_minutes,omitempty"`
	ScheduledDeletionEnable *bool                  `json:"scheduled_deletion_enable,omitempty"`
	JobLogMode              JobLogMode             `json:"job_log_mode,omitempty"`
	JVM                     *JVMConfig             `json:"jvm,omitempty"`
	Checkpoint              *CheckpointConfig      `json:"checkpoint,omitempty"`
	IMAP                    *IMAPConfig            `json:"imap,omitempty"`
	Connector               *ConnectorConfig       `json:"connector,omitempty"`
}

type InstallationStatus struct {
	ID          string      `json:"id"`
	HostID      string      `json:"host_id"`
	ClusterID   string      `json:"cluster_id,omitempty"`
	Status      StepStatus  `json:"status"`
	CurrentStep InstallStep `json:"current_step"`
	Steps       []StepInfo  `json:"steps"`
	Progress    int         `json:"progress"`
	Message     string      `json:"message,omitempty"`
	Error       string      `json:"error,omitempty"`
	Warnings    []string    `json:"warnings,omitempty"`
	StartTime   time.Time   `json:"start_time"`
	EndTime     *time.Time  `json:"end_time,omitempty"`
}

type JVMConfig struct {
	HybridHeapSize int `json:"hybrid_heap_size"`
	MasterHeapSize int `json:"master_heap_size"`
	WorkerHeapSize int `json:"worker_heap_size"`
}

type JobLogMode string

type JobScheduleStrategy string

type ListPackagesResponse struct {
	ErrorMsg string             `json:"error_msg"`
	Data     *AvailableVersions `json:"data"`
}

type MirrorSource string

type NodeJVMResolver interface {
	// ResolveNodeJVMByClusterAndHostAndRole resolves effective JVM config for one logical node.
	// ResolveNodeJVMByClusterAndHostAndRole 解析一个逻辑节点的生效 JVM 配置。
	ResolveNodeJVMByClusterAndHostAndRole(ctx context.Context, clusterID uint, hostID uint, role string) (*JVMConfig, error)
}

type NodeRole string

type NodeStarter interface {
	// EnsureNodeForInstallation ensures the installed node exists in cluster metadata before startup.
	// EnsureNodeForInstallation 在启动前确保已安装节点存在于集群元数据中。
	EnsureNodeForInstallation(ctx context.Context, clusterID uint, hostID uint, role string, installDir string, hazelcastPort int, apiPort int, workerPort int) error
	// StartNodeByClusterAndHost starts a node by cluster ID and host ID
	// StartNodeByClusterAndHost 根据集群 ID 和主机 ID 启动节点
	StartNodeByClusterAndHost(ctx context.Context, clusterID uint, hostID uint) (bool, string, error)
	// StartNodeByClusterAndHostAndRole starts a node by cluster ID, host ID and role (for separated mode: same host can have master + worker)
	// StartNodeByClusterAndHostAndRole 根据集群 ID、主机 ID 和角色启动节点（分离模式下同一主机可有 master+worker）
	StartNodeByClusterAndHostAndRole(ctx context.Context, clusterID uint, hostID uint, role string) (bool, string, error)
}

type NodeStatusUpdater interface {
	// UpdateNodeStatusByClusterAndHost updates the node status by cluster ID and host ID
	// UpdateNodeStatusByClusterAndHost 根据集群 ID 和主机 ID 更新节点状态
	UpdateNodeStatusByClusterAndHost(ctx context.Context, clusterID uint, hostID uint, status string) error
}

type PackageChunkUploadRequest struct {
	Version     string
	UploadID    string
	ChunkIndex  int
	TotalChunks int
	TotalSize   int64
	FileName    string
}

type PackageChunkUploadResult struct {
	UploadID       string       `json:"upload_id"`
	Completed      bool         `json:"completed"`
	ReceivedChunks int          `json:"received_chunks"`
	TotalChunks    int          `json:"total_chunks"`
	Package        *PackageInfo `json:"package,omitempty"`
}

type PackageInfo struct {
	Version            string                  `json:"version"`
	FileName           string                  `json:"file_name"`
	FileSize           int64                   `json:"file_size"`
	Checksum           string                  `json:"checksum,omitempty"`
	DownloadURLs       map[MirrorSource]string `json:"download_urls"`
	IsLocal            bool                    `json:"is_local"`
	LocalPath          string                  `json:"local_path,omitempty"`
	UploadedAt         *time.Time              `json:"uploaded_at,omitempty"`
	HasSource          bool                    `json:"has_source"`
	SourceStatus       DownloadStatus          `json:"source_status,omitempty"`
	SourceFileName     string                  `json:"source_file_name,omitempty"`
	SourceFileSize     int64                   `json:"source_file_size,omitempty"`
	SourceChecksum     string                  `json:"source_checksum,omitempty"`
	SourceUploadedAt   *time.Time              `json:"source_uploaded_at,omitempty"`
	SourceDownloadURLs map[MirrorSource]string `json:"source_download_urls"`
	SourceError        string                  `json:"source_error,omitempty"`
}

type PluginInstallInfo struct {
	// Name is the plugin name / Name 是插件名称
	Name string `json:"name"`
	// Category is the plugin category (source/sink/transform) / Category 是插件分类
	Category string `json:"category"`
	// Version is the plugin version / Version 是插件版本
	Version string `json:"version"`
	// Status is the installation status / Status 是安装状态
	Status PluginInstallStatus `json:"status"`
	// Progress is the installation progress (0-100) / Progress 是安装进度
	Progress int `json:"progress"`
	// Message is the status message / Message 是状态消息
	Message string `json:"message,omitempty"`
	// Error is the error message if failed / Error 是失败时的错误信息
	Error string `json:"error,omitempty"`
}

type PluginInstallStatus string

type PluginTransferer interface {
	// TransferPluginToAgent transfers a plugin to an agent
	// TransferPluginToAgent 将插件传输到 Agent
	TransferPluginToAgent(ctx context.Context, agentID, pluginName, version, installDir string, profileKeys []string) error

	// GetPluginArtifactID returns the Maven artifact ID for a plugin name
	// GetPluginArtifactID 返回插件名称对应的 Maven artifact ID
	GetPluginArtifactID(pluginName string) string

	// IsPluginDownloaded checks if a plugin is downloaded locally
	// IsPluginDownloaded 检查插件是否已下载到本地
	IsPluginDownloaded(name, version string) bool

	// DownloadPluginSync downloads a plugin synchronously (blocking)
	// DownloadPluginSync 同步下载插件（阻塞）
	DownloadPluginSync(ctx context.Context, pluginName, version, mirror string, profileKeys []string) error

	// GetPluginPreparationFingerprint returns a stable fingerprint for the plugin's effective dependency set.
	// GetPluginPreparationFingerprint 返回插件当前生效依赖集合的稳定指纹。
	GetPluginPreparationFingerprint(ctx context.Context, pluginName, version string, profileKeys []string) (string, error)

	// RecordInstalledPlugin records a plugin as installed for a cluster
	// RecordInstalledPlugin 记录插件已安装到集群
	RecordInstalledPlugin(ctx context.Context, clusterID uint, pluginName, version string) error
}

type PrecheckItem struct {
	Name    string                 `json:"name"`
	Status  CheckStatus            `json:"status"`
	Message string                 `json:"message"`
	Details map[string]interface{} `json:"details,omitempty"`
}

type PrecheckRequest struct {
	MinMemoryMB    int64  `json:"min_memory_mb"`
	MinCPUCores    int    `json:"min_cpu_cores"`
	MinDiskSpaceMB int64  `json:"min_disk_space_mb"`
	InstallDir     string `json:"install_dir"`
	Ports          []int  `json:"ports"`
}

type PrecheckResponse struct {
	ErrorMsg string          `json:"error_msg"`
	Data     *PrecheckResult `json:"data"`
}

type PrecheckResult struct {
	Items         []PrecheckItem `json:"items"`
	OverallStatus CheckStatus    `json:"overall_status"`
	Summary       string         `json:"summary"`
}

type RefreshVersionsResponse struct {
	ErrorMsg string   `json:"error_msg"`
	Data     []string `json:"data"`
}

type RetryStepRequest struct {
	Step string `json:"step" binding:"required"`
}

type RuntimeStorageValidationHostResult struct {
	HostID   uint              `json:"host_id"`
	HostName string            `json:"host_name,omitempty"`
	Success  bool              `json:"success"`
	Message  string            `json:"message"`
	Details  map[string]string `json:"details,omitempty"`
}

type RuntimeStorageValidationKind string

type RuntimeStorageValidationRequest struct {
	HostIDs    []uint                       `json:"host_ids" binding:"required,min=1"`
	Kind       RuntimeStorageValidationKind `json:"kind" binding:"required"`
	Checkpoint *CheckpointConfig            `json:"checkpoint,omitempty"`
	IMAP       *IMAPConfig                  `json:"imap,omitempty"`
}

type RuntimeStorageValidationResponse struct {
	ErrorMsg string                          `json:"error_msg"`
	Data     *RuntimeStorageValidationResult `json:"data"`
}

type RuntimeStorageValidationResult struct {
	Success bool                                  `json:"success"`
	Kind    RuntimeStorageValidationKind          `json:"kind"`
	Warning string                                `json:"warning,omitempty"`
	Hosts   []*RuntimeStorageValidationHostResult `json:"hosts"`
}

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

type SlotAllocationStrategy string

type SourceFetchRequest struct {
	Mirror MirrorSource `json:"mirror"`
}

type StepInfo struct {
	Step        InstallStep `json:"step"`
	Name        string      `json:"name"`
	Description string      `json:"description"`
	Status      StepStatus  `json:"status"`
	Progress    int         `json:"progress"`
	Message     string      `json:"message,omitempty"`
	Error       string      `json:"error,omitempty"`
	StartTime   *time.Time  `json:"start_time,omitempty"`
	EndTime     *time.Time  `json:"end_time,omitempty"`
	Retryable   bool        `json:"retryable"`
}

type StepStatus string

type UploadChunkResponse struct {
	ErrorMsg string                    `json:"error_msg"`
	Data     *PackageChunkUploadResult `json:"data"`
}

type UploadPackageResponse struct {
	ErrorMsg string       `json:"error_msg"`
	Data     *PackageInfo `json:"data"`
}