protocol

type Command struct {
	ID      string            `json:"id"`
	Shell   string            `json:"shell"`
	Line    string            `json:"line"`
	User    string            `json:"user"`
	Group   string            `json:"group"`
	Env     map[string]string `json:"env"`
	AllowSh bool              `json:"allow_sh,omitempty"`
	Data    string            `json:"data,omitempty"`

	// Signature verification fields (populated by alpacon-server from AI server signing)
	Signature  string `json:"signature,omitempty"`
	AnalyzedAt string `json:"analyzed_at,omitempty"`
	KeyID      string `json:"key_id,omitempty"`
}

type CommandChunk struct {
	Seq     int    `json:"seq"`
	Content string `json:"content"`
}

type CommandData struct {
	SessionID               string                   `json:"session_id"`
	URL                     string                   `json:"url"`
	Rows                    uint16                   `json:"rows"`
	Cols                    uint16                   `json:"cols"`
	Username                string                   `json:"username"`
	Groupname               string                   `json:"groupname"`
	Groupnames              []string                 `json:"groupnames"`
	HomeDirectory           string                   `json:"home_directory"`
	HomeDirectoryPermission string                   `json:"home_directory_permission"`
	PurgeHomeDirectory      bool                     `json:"purge_home"`
	UID                     uint64                   `json:"uid"`
	GID                     uint64                   `json:"gid"`
	Comment                 string                   `json:"comment"`
	Shell                   string                   `json:"shell"`
	Groups                  []uint64                 `json:"groups"`
	IsServiceAccount        bool                     `json:"is_service_account,omitempty"`
	Type                    string                   `json:"type"`
	Content                 string                   `json:"content"`
	Path                    string                   `json:"path"`
	Paths                   []string                 `json:"paths"`
	Files                   []File                   `json:"files,omitempty"`
	AllowOverwrite          bool                     `json:"allow_overwrite,omitempty"`
	AllowUnzip              bool                     `json:"allow_unzip,omitempty"`
	UseBlob                 bool                     `json:"use_blob,omitempty"`
	Keys                    []string                 `json:"keys"`
	ChainName               string                   `json:"chain_name"`
	Method                  string                   `json:"method"`
	Chain                   string                   `json:"chain"`
	Protocol                string                   `json:"protocol"`
	PortStart               int                      `json:"port_start"`
	PortEnd                 int                      `json:"port_end"`
	DPorts                  []int                    `json:"dports"`
	ICMPType                string                   `json:"icmp_type"`
	Source                  string                   `json:"source"`
	Destination             string                   `json:"destination"`
	Target                  string                   `json:"target"`
	Description             string                   `json:"description"`
	Priority                int                      `json:"priority"`
	RuleType                string                   `json:"rule_type"`
	Rules                   []map[string]interface{} `json:"rules"`
	Operation               string                   `json:"operation"`
	RuleID                  string                   `json:"rule_id"`
	OldRuleID               string                   `json:"old_rule_id"`
	AssignmentID            string                   `json:"assignment_id"`
	ServerID                string                   `json:"server_id"`
	ChainNames              []string                 `json:"chain_names"`

	// Backend information
	Backend string `json:"backend"`
	Table   string `json:"table"`
	Family  string `json:"family"`

	// Firewalld specific fields
	Zone              string `json:"zone"`
	Service           string `json:"service"`
	FirewalldRuleType string `json:"firewalld_rule_type"`

	// UFW specific fields
	Direction string `json:"direction"`
	Interface string `json:"interface"`

	// Tunnel specific fields
	TargetPort int    `json:"target_port"`
	ClientType string `json:"client_type"` // cli, web, editor
}

type CommandResponse struct {
	Success     bool    `json:"success"`
	Result      string  `json:"result"`
	ElapsedTime float64 `json:"elapsed_time"`
	ExitCode    int     `json:"exit_code"`
}

type File struct {
	Username       string `json:"username"`
	Groupname      string `json:"groupname"`
	Type           string `json:"type"`
	Content        string `json:"content"`
	Path           string `json:"path"`
	AllowOverwrite bool   `json:"allow_overwrite"`
	AllowUnzip     bool   `json:"allow_unzip"`
	URL            string `json:"url"`
}

type Message struct {
	Query   MessageType     `json:"query"`
	Command *Command        `json:"command,omitempty"`
	Reason  string          `json:"reason,omitempty"`
	Raw     json.RawMessage `json:"-"` // Original raw message for debugging
}

type MessageType string

type PingResponse struct {
	Query     string    `json:"query"`
	Timestamp time.Time `json:"timestamp,omitempty"`
}

type Response struct {
	Query string `json:"query"`
}