taskflow

type ApplyWebClientIPReq struct {
	ClientIP string `json:"client_ip"`
}

type AskUserQuestionResponse struct {
	TaskId      string `json:"task_id,omitempty"`
	RequestId   string `json:"request_id,omitempty"`
	AnswersJson string `json:"answers_json,omitempty"`
	Cancelled   bool   `json:"cancelled,omitempty"`
}

type CheckTokenReq struct {
	Token     string `json:"token" validate:"required"`
	MachineID string `json:"machine_id"`
}

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

type Clienter interface {
	VirtualMachiner() VirtualMachiner
	Host() Hoster
	FileManager() FileManager
	TaskManager() TaskManager
	PortForwarder() PortForwarder
	Stats(ctx context.Context) (*Stats, error)
	TaskLive(ctx context.Context, taskID string, flush bool, fn func(*TaskChunk) error) error
}

type ClosePortForward struct {
	ID        string `json:"id" query:"id" validate:"required"`
	ForwardID string `json:"forward_id"`
}

type CloseTerminalReq struct {
	ID         string `json:"id" query:"id"`
	TerminalID string `json:"terminal_id" query:"terminal_id"`
}

type CodingAgent int

type CodingConfig struct{}

type Condition struct {
	Type               string          `json:"type,omitempty"`
	Status             ConditionStatus `json:"status,omitempty"`
	Reason             string          `json:"reason,omitempty"`
	Message            string          `json:"message,omitempty"`
	LastTransitionTime int64           `json:"last_transition_time,omitempty"`
	Progress           *int32          `json:"progress,omitempty"`
}

type ConditionStatus int32

type ConfigFile struct {
	Path    string `json:"path"`
	Content string `json:"content"`
}

type CreatePortForward struct {
	ID           string   `json:"id" query:"id" validate:"required"`
	UserID       string   `json:"user_id"`
	LocalPort    int32    `json:"local_port"`
	WhitelistIPs []string `json:"whitelist_ips"`
}

type CreateTaskReq struct {
	ID           uuid.UUID         `json:"id"`
	VMID         string            `json:"vm_id"`
	SystemPrompt string            `json:"system_prompt,omitempty"`
	Text         string            `json:"text,omitempty"`
	LLM          LLM               `json:"llm,omitzero"`
	CodingAgent  CodingAgent       `json:"coding_agent,omitempty"`
	Configs      []ConfigFile      `json:"configs,omitzero"`
	McpConfigs   []McpServerConfig `json:"mcp_configs,omitzero"`
	Env          map[string]string `json:"env,omitempty"`
}

type CreateVirtualMachineReq struct {
	UserID              string         `json:"user_id" validate:"required"`
	HostID              string         `json:"host_id" validate:"required"`
	HostName            string         `json:"hostname"`
	Git                 Git            `json:"git"`
	ZipUrl              string         `json:"zip_url"`
	ImageURL            string         `json:"image_url"`
	ProxyURL            string         `json:"proxy_url"`
	TaskID              uuid.UUID      `json:"task_id"`
	LLM                 LLMProviderReq `json:"llm"`
	Cores               string         `json:"cores"`
	Memory              uint64         `json:"memory"`
	InstallCodingAgents bool           `json:"install_coding_agents"`
	Envs                []string       `json:"envs,omitempty"`
}

type DeleteVirtualMachineReq struct {
	HostID string `json:"host_id" query:"host_id" validate:"required"`
	UserID string `json:"user_id" query:"user_id" validate:"required"`
	ID     string `json:"id" query:"id" validate:"required"`
}

type File struct {
	Name          string   `json:"name"`
	User          string   `json:"user"`
	Size          uint64   `json:"size"`
	Kind          FileKind `json:"kind"`
	SymlinkTarget string   `json:"symlink_target,omitempty"`
	SymlinkKind   FileKind `json:"symlink_kind,omitempty"`
	UnixMode      uint32   `json:"unix_mode"`
	CreatedAt     int64    `json:"created_at"`
	AccessedAt    int64    `json:"accessed_at"`
	UpdatedAt     int64    `json:"updated_at"`
}

type FileKind string

type FileManager interface {
	Operate(ctx context.Context, req FileReq) ([]*File, error)
	Download(ctx context.Context, req FileReq, fn func(uint64, []byte) error) error
	Upload(ctx context.Context, req FileReq, data <-chan []byte) error
}

type FileOP string

type FileReq struct {
	Operate FileOP `json:"operate" query:"operate"`
	UserID  string `json:"user_id" query:"user_id"`
	ID      string `json:"id" query:"id"`
	Path    string `json:"path,omitempty" query:"path"`
	Source  string `json:"source,omitempty" query:"source"`
	Target  string `json:"target,omitempty" query:"target"`
	Content string `json:"content,omitempty" query:"content"`
}

type GetCodingConfigReq struct {
	VmID string `json:"vm_id"`
}

type GetTaskStreamIPsReq struct {
	TaskID string `json:"task_id"`
}

type GetTaskStreamIPsResp struct {
	IPs []string `json:"ips"`
}

type Git struct {
	URL      string `json:"url"`
	Token    string `json:"token"`
	ProxyURL string `json:"proxy_url"`
	Branch   string `json:"branch,omitempty"`
	Username string `json:"username,omitempty"`
	Email    string `json:"email,omitempty"`
}

type GitCredentialRequest struct {
	TaskID   string `json:"task_id"`  // 任务 id
	VMID     string `json:"vm_id"`    // 虚拟机 id
	Protocol string `json:"protocol"` // git 协议 (e.g., "https")
	Host     string `json:"host"`     // git server host (e.g., "gitlab.example.com")
	Path     string `json:"path"`     // 仓库路径 (e.g., "owner/repo.git")
}

type GitCredentialResponse struct {
	Username *string `json:"username,omitempty"`
	Password *string `json:"password,omitempty"`
	Error    *string `json:"error,omitempty"`
}

type HibernateVirtualMachineReq struct {
	HostID        string `json:"host_id" query:"host_id" validate:"required"` // 宿主机 id
	UserID        string `json:"user_id" query:"user_id" validate:"required"` // 用户id
	ID            string `json:"id" query:"id" validate:"required"`           // 虚拟机 id
	EnvironmentID string `json:"environment_id" query:"environment_id"`       // environment id
}

type Host struct {
	ID         string `json:"id"`
	UserID     string `json:"user_id"`
	Hostname   string `json:"hostname"`
	Arch       string `json:"arch"`
	OS         string `json:"os"`
	Name       string `json:"name"`
	Cores      int32  `json:"cores"`
	Memory     uint64 `json:"memory"`
	Disk       uint64 `json:"disk"`
	PublicIP   string `json:"public_ip"`
	InternalIP string `json:"internal_ip"`
	TTL        TTL    `json:"ttl"`
	CreatedAt  int64  `json:"created_at"`
	Version    string `json:"version"`
}

type Hoster interface {
	List(ctx context.Context, userID string) (map[string]*Host, error)
	IsOnline(ctx context.Context, req *IsOnlineReq[string]) (*IsOnlineResp, error)
}

type IsOnlineReq[T any] struct {
	IDs []T `json:"ids"`
}

type IsOnlineResp struct {
	OnlineMap map[string]bool `json:"online_map"`
}

type LLM struct {
	ApiKey      string   `json:"api_key"`
	BaseURL     string   `json:"base_url"`
	Model       string   `json:"model"`
	ApiType     string   `json:"api_type,omitempty"` // 接口类型 anthropic | openai
	Temperature *float32 `json:"temperature,omitempty"`
}

type LLMInfo struct {
	ApiKey      string   `json:"api_key"`
	BaseURL     string   `json:"base_url"`
	Model       string   `json:"model"`
	Temperature *float32 `json:"temperature,omitempty"`
}

type LLMProvider string

type LLMProviderReq struct {
	Provider    LLMProvider `json:"provider"`
	ApiKey      string      `json:"api_key"`
	BaseURL     string      `json:"base_url"`
	Model       string      `json:"model"`
	Temperature *float32    `json:"temperature,omitempty"`
}

type ListLLMReq struct {
	VmID   string `json:"vm_id" validate:"required"`
	HostID string `json:"host_id" validate:"required"`
}

type ListPortforwadReq struct {
	ID        string `json:"id" query:"id" validate:"required"` // 虚拟机 id
	RequestId string `json:"request_id,omitempty"`
}

type ListPortforwadResp struct {
	RequestId string             `json:"request_id,omitempty"`
	Ports     []*PortForwardInfo `json:"ports"`
}

type McpHttpHeader struct {
	Name  string `json:"name,omitempty"`
	Value string `json:"value,omitempty"`
}

type McpServerConfig struct {
	Type    string            `json:"type,omitempty"`
	Name    string            `json:"name,omitempty"`
	Url     *string           `json:"url,omitempty"`
	Headers []*McpHttpHeader  `json:"headers,omitempty"`
	Command *string           `json:"command,omitempty"`
	Args    []string          `json:"args,omitempty"`
	Env     map[string]string `json:"env,omitempty"`
}

type Opt func(*Client)

type PortForwardInfo struct {
	Port         int32    `json:"port"`
	Status       string   `json:"status"`
	Process      string   `json:"process"`
	ForwardID    *string  `json:"forward_id"`
	AccessURL    *string  `json:"access_url"`
	CreatedAt    int64    `json:"created_at"`
	Success      bool     `json:"success"`
	ErrorMessage string   `json:"error_message,omitempty"`
	WhitelistIPs []string `json:"whitelist_ips"`
}

type PortForwarder interface {
	List(ctx context.Context, req ListPortforwadReq) (*ListPortforwadResp, error)
	Create(ctx context.Context, req CreatePortForward) (*PortForwardInfo, error)
	Close(ctx context.Context, req ClosePortForward) error
	Update(ctx context.Context, req UpdatePortForward) (*PortForwardInfo, error)
}

type RepoEntryMode int32

type RepoFileChangeInfo struct {
	Path      string  `json:"path,omitempty"`
	Status    string  `json:"status,omitempty"`
	Additions *int32  `json:"additions,omitempty"`
	Deletions *int32  `json:"deletions,omitempty"`
	OldPath   *string `json:"old_path,omitempty"`
}

type RepoFileChanges struct {
	TaskId     string                `json:"task_id,omitempty"`
	RequestId  string                `json:"request_id,omitempty"`
	Changes    []*RepoFileChangeInfo `json:"changes,omitempty"`
	Branch     *string               `json:"branch,omitempty"`
	CommitHash *string               `json:"commit_hash,omitempty"`
	Success    bool                  `json:"success"`
	Error      *string               `json:"error,omitempty"`
}

type RepoFileChangesReq struct {
	TaskId    string `json:"task_id,omitempty"`
	RequestId string `json:"request_id"`
}

type RepoFileDiff struct {
	TaskId    string  `json:"task_id,omitempty"`
	RequestId string  `json:"request_id"`
	Path      string  `json:"path,omitempty"`
	Diff      string  `json:"diff,omitempty"`
	Success   bool    `json:"success"`
	Error     *string `json:"error,omitempty"`
}

type RepoFileDiffReq struct {
	TaskId       string `json:"task_id,omitempty"`
	RequestId    string `json:"request_id"`
	Path         string `json:"path"`
	Unified      *bool  `json:"unified"`
	ContextLines *int32 `json:"context_lines"`
}

type RepoFileInfo struct {
	Name          string        `json:"name,omitempty"`
	Path          string        `json:"path,omitempty"`
	EntryMode     RepoEntryMode `json:"entry_mode,omitempty"`
	Size          int64         `json:"size,omitempty"`
	ModifiedAt    int64         `json:"modified_at,omitempty"`
	Mode          *uint32       `json:"mode,omitempty"`
	SymlinkTarget *string       `json:"symlink_target,omitempty"`
}

type RepoListFiles struct {
	TaskId    string          `json:"task_id,omitempty"`
	RequestId string          `json:"request_id"`
	Path      string          `json:"path,omitempty"`
	Files     []*RepoFileInfo `json:"files,omitempty"`
	Success   bool            `json:"success,omitempty"`
	Error     *string         `json:"error,omitempty"`
}

type RepoListFilesReq struct {
	TaskId        string  `json:"task_id,omitempty"`
	RequestId     string  `json:"request_id"`
	Path          string  `json:"path"`
	GlobPattern   *string `json:"glob_pattern"`
	IncludeHidden bool    `json:"include_hidden"`
}

type RepoReadFile struct {
	TaskId      string  `json:"task_id,omitempty"`
	RequestId   string  `json:"request_id"`
	Path        string  `json:"path,omitempty"`
	Content     []byte  `json:"content,omitempty"`
	TotalSize   int64   `json:"total_size,omitempty"`
	Offset      int64   `json:"offset,omitempty"`
	Length      int64   `json:"length,omitempty"`
	IsTruncated bool    `json:"is_truncated,omitempty"`
	Success     bool    `json:"success"`
	Error       *string `json:"error,omitempty"`
}

type RepoReadFileReq struct {
	TaskId    string `json:"task_id,omitempty"`
	RequestId string `json:"request_id"`
	Path      string `json:"path"`
	Offset    *int64 `json:"offset"`
	Length    *int64 `json:"length"`
}

type ReportEntry struct {
	ID     string `json:"id"`
	Source string `json:"source"`
	Ts     int64  `json:"ts"`
	Data   []byte `json:"data"`
}

type ReportSubscribeReq struct {
	ID      string `json:"id" query:"id" validate:"required"`
	History int64  `json:"history" query:"history"`
	FromID  string `json:"from_id" query:"from_id"`
}

type Reporter interface {
	Stop()
	BlockRead(fn func(ReportEntry)) error
}

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

type Resp[T any] struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    T      `json:"data,omitempty"`
}

type RestartTaskReq struct {
	ID          uuid.UUID `json:"id"`
	RequestId   string    `json:"request_id,omitempty"`
	LoadSession bool      `json:"load_session"`
}

type RestartTaskResp struct {
	ID        uuid.UUID `json:"id"`
	RequestId string    `json:"request_id,omitempty"`
	Success   bool      `json:"success"`
	Message   string    `json:"message"`
	SessionID string    `json:"session_id"`
}

type ResumeVirtualMachineReq struct {
	HostID        string `json:"host_id" query:"host_id" validate:"required"` // 宿主机 id
	UserID        string `json:"user_id" query:"user_id" validate:"required"` // 用户id
	ID            string `json:"id" query:"id" validate:"required"`           // 虚拟机 id
	EnvironmentID string `json:"environment_id" query:"environment_id"`       // environment id
}

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

type Sheller interface {
	Write(TerminalData) error
	Stop()
	BlockRead(fn func(TerminalData)) error
}

type Stats struct {
	OnlineHostCount        int `json:"online_host_count"`
	OnlineTaskCount        int `json:"online_task_count"`
	OnlineVMCount          int `json:"online_vm_count"`
	OnlineTerminalCount    int `json:"online_terminal_count"`
	UsingTerminalUserCount int `json:"using_terminal_user_count"`
}

type TTL struct {
	Kind    TTLKind `json:"kind"`
	Seconds int64   `json:"seconds"`
}

type TTLKind uint8

type Task struct {
	ID    uuid.UUID `json:"id"`
	Text  string    `json:"text"`
	Image string    `json:"image"`
}

type TaskApproveReq struct {
	ID          uuid.UUID `json:"id"`
	AutoApprove *bool     `json:"auto_approve,omitempty"`
}

type TaskChunk struct {
	Data      []byte `json:"data,omitempty"`
	Event     string `json:"event"`
	Kind      string `json:"kind"`
	Timestamp int64  `json:"timestamp,omitempty"`
}

type TaskManager interface {
	Create(ctx context.Context, req CreateTaskReq) error
	Stop(ctx context.Context, req TaskReq) error
	Restart(ctx context.Context, req RestartTaskReq) (*RestartTaskResp, error)
	Cancel(ctx context.Context, req TaskReq) error
	Continue(ctx context.Context, req TaskReq) error
	AutoApprove(ctx context.Context, req TaskApproveReq) error
	AskUserQuestion(ctx context.Context, req AskUserQuestionResponse) error
	ListFiles(ctx context.Context, req RepoListFilesReq) (*RepoListFiles, error)
	ReadFile(ctx context.Context, req RepoReadFileReq) (*RepoReadFile, error)
	FileDiff(ctx context.Context, req RepoFileDiffReq) (*RepoFileDiff, error)
	FileChanges(ctx context.Context, req RepoFileChangesReq) (*RepoFileChanges, error)
}

type TaskReq struct {
	VirtualMachine *VirtualMachine `json:"virtual_machine,omitempty"`
	Task           *Task           `json:"task,omitempty"`
}

type Terminal struct {
	ID             string `json:"id"`
	Title          string `json:"title"`
	ConnectedCount uint32 `json:"connected_count"`
	CreatedAt      int64  `json:"created_at"`
}

type TerminalData struct {
	Data      []byte        `json:"data,omitempty"`
	Connected bool          `json:"connected"`
	Resize    *TerminalSize `json:"resize,omitempty"`
	Error     *string       `json:"error,omitempty"`
}

type TerminalMode uint8

type TerminalReq struct {
	ID         string       `json:"id" query:"id"`
	Mode       TerminalMode `json:"mode" query:"mode"`
	TerminalID string       `json:"terminal_id" query:"terminal_id" validate:"required"`
	Exec       string       `json:"exec" query:"exec"`
	TerminalSize
}

type TerminalSize struct {
	Col uint32 `json:"col" query:"col"`
	Row uint32 `json:"row" query:"row"`
}

type Token struct {
	Kind        TokenKind  `json:"kind"`
	User        *TokenUser `json:"user"`
	ParentToken string     `json:"parent_token"`
	Token       string     `json:"token"`
	TaskID      uuid.UUID  `json:"task_id"`
	SessionID   string     `json:"session_id"`
	RemoteIP    string     `json:"remote_ip"`
	Version     string     `json:"version"`
}

type TokenKind string

type TokenTeam struct {
	ID   string `json:"id"`
	Name string `json:"name"`
}

type TokenUser struct {
	ID        string     `json:"id"`
	Name      string     `json:"name"`
	AvatarURL string     `json:"avatar_url"`
	Email     string     `json:"email"`
	Team      *TokenTeam `json:"team"`
}

type UpdatePortForward struct {
	ID           string   `json:"id" query:"id" validate:"required"`
	ForwardID    string   `json:"forward_id"`
	WhitelistIPs []string `json:"whitelist_ips"`
}

type VirtualMachine struct {
	ID            string               `json:"id"`
	EnvironmentID string               `json:"environment_id"`
	HostID        string               `json:"host_id"`
	Hostname      string               `json:"hostname"`
	Arch          string               `json:"arch"`
	OS            string               `json:"os"`
	Name          string               `json:"name"`
	Repository    string               `json:"repository"`
	Status        VirtualMachineStatus `json:"status"`
	StatusMessage string               `json:"status_message"`
	Cores         int32                `json:"cores"`
	Memory        uint64               `json:"memory"`
	Disk          uint64               `json:"disk"`
	TTL           TTL                  `json:"ttl"`
	ExternalIP    string               `json:"external_ip"`
	CreatedAt     int64                `json:"created_at"`
	Version       string               `json:"version"`
}

type VirtualMachineCondition struct {
	EnvID      string       `json:"env_id"`
	Conditions []*Condition `json:"conditions,omitempty"`
}

type VirtualMachineInfoReq struct {
	ID     string `json:"id" query:"id"`
	UserID string `json:"user_id" query:"user_id"`
}

type VirtualMachineStatus string

type VirtualMachiner interface {
	Create(ctx context.Context, req *CreateVirtualMachineReq) (*VirtualMachine, error)
	Delete(ctx context.Context, req *DeleteVirtualMachineReq) error
	Hibernate(ctx context.Context, req *HibernateVirtualMachineReq) error
	Resume(ctx context.Context, req *ResumeVirtualMachineReq) error
	List(ctx context.Context, id string) ([]*VirtualMachine, error)
	Info(ctx context.Context, req VirtualMachineInfoReq) (*VirtualMachine, error)
	Terminal(ctx context.Context, req *TerminalReq) (Sheller, error)
	Reports(ctx context.Context, req ReportSubscribeReq) (Reporter, error)
	TerminalList(ctx context.Context, id string) ([]*Terminal, error)
	CloseTerminal(ctx context.Context, req *CloseTerminalReq) error
	IsOnline(ctx context.Context, req *IsOnlineReq[string]) (*IsOnlineResp, error)
}