protocol

type AuthErrorMeta struct {
	StatusCode          int    `json:"status_code,omitempty"`
	WWWAuthenticate     string `json:"www_authenticate,omitempty"`
	ResourceMetadataURL string `json:"resource_metadata_url,omitempty"`
}

type BashArgs struct {
	Command string            `json:"command"`
	Workdir string            `json:"workdir,omitempty"`
	Timeout int               `json:"timeout,omitempty"` // seconds
	Env     map[string]string `json:"env,omitempty"`
}

type BashResult struct {
	Stdout   string `json:"stdout"`
	Stderr   string `json:"stderr"`
	ExitCode int    `json:"exit_code"`

	// Per-stream truncation metadata. A stream is "truncated" when either
	// processLargeOutput trimmed it OR the underlying LimitedWriter hit its
	// hard cap mid-execution (the captured buffer carries an inline marker).
	TruncatedStdout bool   `json:"truncated_stdout,omitempty"`
	TruncatedStderr bool   `json:"truncated_stderr,omitempty"`
	StdoutFilePath  string `json:"stdout_file_path,omitempty"`
	StderrFilePath  string `json:"stderr_file_path,omitempty"`
	StdoutTotalSize int64  `json:"stdout_total_size,omitempty"`
	StderrTotalSize int64  `json:"stderr_total_size,omitempty"`

	// Legacy stdout-only mirrors. Kept so older safari clients that only
	// parse these still see truncation. New code should read the stream-
	// specific fields instead.
	Truncated bool   `json:"truncated,omitempty"`
	FilePath  string `json:"file_path,omitempty"`
	TotalSize int64  `json:"total_size,omitempty"`
}

type DeleteKnowledgeFilesArgs struct {
	RelPaths []string `json:"rel_paths"`
}

type DeleteKnowledgeFilesResult struct {
	Success bool `json:"success"`
}

type EnvironmentInfo struct {
	OS            string `json:"os"`                        // e.g., "darwin", "linux", "windows"
	OSVersion     string `json:"os_version"`                // e.g., "24.6.0" for macOS
	Arch          string `json:"arch"`                      // e.g., "amd64", "arm64"
	Hostname      string `json:"hostname"`                  // Machine hostname
	Shell         string `json:"shell"`                     // Default shell, e.g., "/bin/zsh"
	HomeDir       string `json:"home_dir"`                  // User home directory
	WorkspaceRoot string `json:"workspace_root"`            // Configured workspace root
	Username      string `json:"username"`                  // Current user
	NumCPU        int    `json:"num_cpu"`                   // Number of CPUs
	TotalMemoryMB int64  `json:"total_memory_mb,omitempty"` // Total memory in MB
	Timezone      string `json:"timezone"`                  // Timezone name, e.g., "Asia/Shanghai"
	UTCOffset     string `json:"utc_offset"`                // UTC offset, e.g., "+08:00"
}

type GlobArgs struct {
	Pattern string `json:"pattern"`
}

type GlobResult struct {
	Matches []string `json:"matches"`
}

type GrepArgs struct {
	Pattern string   `json:"pattern"`
	Include []string `json:"include,omitempty"`

	// OutputMode controls the shape of the result: "content" (default — file:line:text rows),
	// "files_with_matches" (one path per match), or "count" (one path:count per file).
	OutputMode string `json:"output_mode,omitempty"`
	// ContextBefore (-B) and ContextAfter (-A) request N lines of context around each match.
	// Honored by ripgrep only when OutputMode is "content".
	ContextBefore int `json:"context_before,omitempty"`
	ContextAfter  int `json:"context_after,omitempty"`
	// HeadLimit caps the number of result rows returned (post-format). 0 = unlimited.
	HeadLimit int `json:"head_limit,omitempty"`
	// FileType maps to ripgrep's --type (e.g., "go", "py", "js"). Ignored by the Go fallback.
	FileType string `json:"file_type,omitempty"`
	// CaseSensitive overrides ripgrep's default --smart-case.
	// nil = smart-case; true = case-sensitive; false = case-insensitive.
	CaseSensitive *bool `json:"case_sensitive,omitempty"`
}

type GrepMatch struct {
	Path       string `json:"path"`
	LineNumber int    `json:"line_number"`
	Content    string `json:"content"`
}

type GrepResult struct {
	Matches   []GrepMatch `json:"matches"`
	Content   string      `json:"content"`              // Formatted content (may be truncated)
	Truncated bool        `json:"truncated,omitempty"`  // Whether content was truncated
	FilePath  string      `json:"file_path,omitempty"`  // Path to full content if truncated
	TotalSize int64       `json:"total_size,omitempty"` // Original content size
	// RegexCompileError surfaces a non-fatal fallback path: the requested pattern failed
	// to compile as a Go regexp, so the Go fallback walked files matching it as a literal.
	// Empty when ripgrep handled the request or compilation succeeded.
	RegexCompileError string `json:"regex_compile_error,omitempty"`
}

type HeartbeatMetrics struct {
	CPUPercent    float64 `json:"cpu_percent,omitempty"`
	MemoryPercent float64 `json:"memory_percent,omitempty"`
	DiskPercent   float64 `json:"disk_percent,omitempty"`
}

type HeartbeatPayload struct {
	EnvironmentID string            `json:"environment_id"`
	Name          string            `json:"name"`
	Labels        []string          `json:"labels"`
	Version       string            `json:"version"`
	Environment   *EnvironmentInfo  `json:"environment,omitempty"`
	Metrics       *HeartbeatMetrics `json:"metrics,omitempty"`
}

type KnowledgeFile struct {
	RelPath    string `json:"rel_path"`
	Checksum   string `json:"checksum"`
	ContentB64 string `json:"content_b64"`
}

type KnowledgeFileStatus struct {
	RelPath string `json:"rel_path"`
	Success bool   `json:"success"`
	Error   string `json:"error,omitempty"`
}

type KnowledgeManifestEntry struct {
	RelPath  string `json:"rel_path"`
	Checksum string `json:"checksum"`
}

type ListArgs struct {
	Path      string   `json:"path"`
	Recursive bool     `json:"recursive,omitempty"`
	Ignore    []string `json:"ignore,omitempty"`
}

type ListEntry struct {
	Path  string `json:"path"`
	IsDir bool   `json:"is_dir"`
	Size  int64  `json:"size"`
}

type ListResult struct {
	Entries []ListEntry `json:"entries"`
}

type MCPCallArgs struct {
	Server   MCPServerConfig `json:"server"`
	ToolName string          `json:"tool_name"`
	Args     json.RawMessage `json:"args"`
	Timeout  int             `json:"timeout,omitempty"` // seconds
}

type MCPCallPayload struct {
	CallID    string          `json:"call_id"`
	Server    MCPServerConfig `json:"server"`
	ToolName  string          `json:"tool_name"`
	Arguments json.RawMessage `json:"arguments"`
}

type MCPCallResult struct {
	Content   string `json:"content"`
	IsError   bool   `json:"is_error,omitempty"`   // Whether the tool returned an error
	Truncated bool   `json:"truncated,omitempty"`  // Whether content was truncated
	FilePath  string `json:"file_path,omitempty"`  // Path to full content if truncated
	TotalSize int64  `json:"total_size,omitempty"` // Original content size
}

type MCPListToolsArgs struct {
	Server MCPServerConfig `json:"server"`
}

type MCPListToolsResult struct {
	Tools []MCPToolInfo `json:"tools"`
}

type MCPResultPayload struct {
	CallID  string          `json:"call_id"`
	Success bool            `json:"success"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   string          `json:"error,omitempty"`

	// P1: structured auth-error metadata (additive; old senders omit).
	ErrorType    string         `json:"error_type,omitempty"`    // "auth_error" | ""
	AuthMetadata *AuthErrorMeta `json:"auth_metadata,omitempty"` // non-nil iff ErrorType=="auth_error"
}

type MCPServerConfig struct {
	Name           string            `json:"name"`
	Transport      string            `json:"transport"` // stdio, sse
	Command        string            `json:"command,omitempty"`
	Args           []string          `json:"args,omitempty"`
	Env            map[string]string `json:"env,omitempty"`
	URL            string            `json:"url,omitempty"`
	Headers        map[string]string `json:"headers,omitempty"`
	DynamicHeaders map[string]string `json:"dynamic_headers,omitempty"`
}

type MCPToolInfo struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	InputSchema any    `json:"input_schema,omitempty"`
}

type Message struct {
	ID        string          `json:"id"`
	Type      MessageType     `json:"type"`
	Payload   json.RawMessage `json:"payload"`
	Timestamp int64           `json:"timestamp"`
}

type MessageType string

type PackFolderArgs struct {
	Path      string `json:"path"`       // absolute path inside workspace
	SkillRoot bool   `json:"skill_root"` // when true, also exclude evals/ at the root
	MaxBytes  int64  `json:"max_bytes"`  // 0 = no cap; >0 = error if zip exceeds
}

type PackFolderResult struct {
	ZipB64    string `json:"zip_b64"` // base64-encoded zip bytes
	SizeBytes int64  `json:"size_bytes"`
	FileCount int    `json:"file_count"`
}

type ReadArgs struct {
	Path   string `json:"path"`
	Offset int64  `json:"offset,omitempty"`
	Limit  int64  `json:"limit,omitempty"`
}

type ReadResult struct {
	Content   string `json:"content"` // base64 encoded
	TotalSize int64  `json:"total_size"`
}

type ReconcileKnowledgeManifestArgs struct {
	Files []KnowledgeManifestEntry `json:"files"`
}

type ReconcileKnowledgeManifestResult struct {
	Pruned     []string `json:"pruned,omitempty"`
	KeptCount  int      `json:"kept_count"`
	StaleCount int      `json:"stale_count"`
	NeedsStage []string `json:"needs_stage,omitempty"`
}

type StageKnowledgeFilesArgs struct {
	Files []KnowledgeFile `json:"files"`
}

type StageKnowledgeFilesResult struct {
	Files []KnowledgeFileStatus `json:"files"`
}

type SyncSkillArgs struct {
	SkillName string `json:"skill_name"`
	SkillDir  string `json:"skill_dir"`
	ZipData   string `json:"zip_data"` // base64 encoded zip
	Checksum  string `json:"checksum"`
}

type SyncSkillResult struct {
	Success bool   `json:"success"`
	Path    string `json:"path"`             // Local path where skill is materialized
	Cached  bool   `json:"cached,omitempty"` // True iff cache hit
}

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

type TaskOperation string

type TaskOutputPayload struct {
	TaskID string `json:"task_id"`
	Stream string `json:"stream"` // stdout, stderr
	Data   string `json:"data"`
}

type TaskRequestPayload struct {
	TaskID           string          `json:"task_id"`
	TraceID          string          `json:"trace_id,omitempty"`
	SourceInstanceID string          `json:"source_instance_id,omitempty"` // Safari instance ID
	Operation        TaskOperation   `json:"operation"`
	Args             json.RawMessage `json:"args"`
}

type TaskResultPayload struct {
	TaskID           string          `json:"task_id"`
	SourceInstanceID string          `json:"source_instance_id,omitempty"` // Safari instance ID
	Success          bool            `json:"success"`
	Result           json.RawMessage `json:"result,omitempty"`
	Error            string          `json:"error,omitempty"`
	ExitCode         int             `json:"exit_code,omitempty"`
}

type WebFetchArgs struct {
	URL     string `json:"url"`
	Format  string `json:"format,omitempty"`  // markdown, text, html (default: markdown)
	Timeout int    `json:"timeout,omitempty"` // seconds (default: 30, max: 120)
}

type WebFetchResult struct {
	Content   string `json:"content"`
	URL       string `json:"url"`                  // Final URL after redirects
	Truncated bool   `json:"truncated,omitempty"`  // Whether content was truncated
	FilePath  string `json:"file_path,omitempty"`  // Path to full content if truncated
	TotalSize int64  `json:"total_size,omitempty"` // Original content size
}

type WelcomePayload struct {
	EnvironmentID string   `json:"environment_id"`
	Name          string   `json:"name"`
	Labels        []string `json:"labels"`
}

type WriteArgs struct {
	Path    string `json:"path"`
	Content string `json:"content"` // base64 encoded
}