protocol

type Ack struct {
	CommandID uuid.UUID `json:"commandId"`
	Duplicate bool      `json:"duplicate"`
}

type ArtifactFinish struct {
	UploadID uuid.UUID `json:"uploadId"`
}

type ArtifactPart struct {
	UploadID uuid.UUID `json:"uploadId"`
	Offset   int64     `json:"offset"`
	Data     string    `json:"data"`
}

type ArtifactStart struct {
	UploadID    uuid.UUID `json:"uploadId"`
	ArtifactID  uuid.UUID `json:"artifactId"`
	TaskID      uuid.UUID `json:"taskId"`
	Name        string    `json:"name"`
	Kind        string    `json:"kind"`
	SHA256      string    `json:"sha256"`
	Size        int64     `json:"size"`
	MIME        string    `json:"mime"`
	Compression string    `json:"compression"`
}

type CodexAccountUsage struct {
	LifetimeTokens        *int64 `json:"lifetimeTokens,omitempty"`
	PeakDailyTokens       *int64 `json:"peakDailyTokens,omitempty"`
	CurrentStreakDays     *int64 `json:"currentStreakDays,omitempty"`
	LongestStreakDays     *int64 `json:"longestStreakDays,omitempty"`
	LongestRunningTurnSec *int64 `json:"longestRunningTurnSec,omitempty"`
}

type CodexCredits struct {
	Balance    *string `json:"balance,omitempty"`
	HasCredits bool    `json:"hasCredits"`
	Unlimited  bool    `json:"unlimited"`
}

type CodexMetadata struct {
	Available   bool               `json:"available"`
	RefreshedAt time.Time          `json:"refreshedAt"`
	Models      []CodexModel       `json:"models"`
	RateLimits  *CodexRateLimits   `json:"rateLimits,omitempty"`
	Usage       *CodexAccountUsage `json:"usage,omitempty"`
	Message     string             `json:"message,omitempty"`
}

type CodexModel struct {
	ID                        string                 `json:"id"`
	Model                     string                 `json:"model"`
	DisplayName               string                 `json:"displayName"`
	Description               string                 `json:"description"`
	IsDefault                 bool                   `json:"isDefault"`
	DefaultReasoningEffort    string                 `json:"defaultReasoningEffort"`
	SupportedReasoningEfforts []CodexReasoningEffort `json:"supportedReasoningEfforts"`
	DefaultServiceTier        *string                `json:"defaultServiceTier,omitempty"`
	ServiceTiers              []CodexServiceTier     `json:"serviceTiers"`
}

type CodexRateLimit struct {
	LimitID   *string               `json:"limitId,omitempty"`
	LimitName *string               `json:"limitName,omitempty"`
	PlanType  *string               `json:"planType,omitempty"`
	Primary   *CodexRateLimitWindow `json:"primary,omitempty"`
	Secondary *CodexRateLimitWindow `json:"secondary,omitempty"`
	Credits   *CodexCredits         `json:"credits,omitempty"`
}

type CodexRateLimitWindow struct {
	UsedPercent        int    `json:"usedPercent"`
	ResetsAt           *int64 `json:"resetsAt,omitempty"`
	WindowDurationMins *int64 `json:"windowDurationMins,omitempty"`
}

type CodexRateLimits struct {
	Default   CodexRateLimit            `json:"default"`
	ByLimitID map[string]CodexRateLimit `json:"byLimitId,omitempty"`
}

type CodexReasoningEffort struct {
	ID          string `json:"id"`
	Description string `json:"description"`
}

type CodexServiceTier struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
}

type Dispatch struct {
	CommandID uuid.UUID       `json:"commandId"`
	TaskID    *uuid.UUID      `json:"taskId,omitempty"`
	Type      string          `json:"type"`
	Payload   json.RawMessage `json:"payload"`
}

type Envelope struct {
	ProtocolVersion string          `json:"protocolVersion"`
	MessageID       uuid.UUID       `json:"messageId"`
	Type            string          `json:"type"`
	SentAt          time.Time       `json:"sentAt"`
	CorrelationID   *uuid.UUID      `json:"correlationId,omitempty"`
	Payload         json.RawMessage `json:"payload"`
}

type EventPayload struct {
	TaskID         uuid.UUID       `json:"taskId"`
	Type           string          `json:"type"`
	Data           json.RawMessage `json:"data"`
	SourceMessage  uuid.UUID       `json:"sourceMessageId"`
	CodexTimestamp *time.Time      `json:"codexTimestamp,omitempty"`
}

type Heartbeat struct {
	Status      string         `json:"status"`
	Load        map[string]any `json:"load"`
	ActiveTasks []uuid.UUID    `json:"activeTasks"`
	CodexStatus string         `json:"codexStatus"`
	Codex       *CodexMetadata `json:"codex,omitempty"`
}

type Hello struct {
	RunnerID         uuid.UUID      `json:"runnerId"`
	NodeID           string         `json:"nodeId"`
	Name             string         `json:"name"`
	Hostname         string         `json:"hostname"`
	Platform         string         `json:"platform"`
	Arch             string         `json:"arch"`
	Version          string         `json:"version"`
	CodexVersion     string         `json:"codexVersion"`
	CodexStatus      string         `json:"codexStatus"`
	Capabilities     map[string]any `json:"capabilities"`
	LastAckCommandID *uuid.UUID     `json:"lastAckCommandId,omitempty"`
}

type Result struct {
	CommandID uuid.UUID       `json:"commandId"`
	Success   bool            `json:"success"`
	Data      json.RawMessage `json:"data,omitempty"`
	Error     string          `json:"error,omitempty"`
}

type Snapshot struct {
	Tasks      []SnapshotTask      `json:"tasks"`
	Commands   []SnapshotCommand   `json:"commands"`
	Workspaces []SnapshotWorkspace `json:"workspaces"`
}

type SnapshotCommand struct {
	CommandID uuid.UUID       `json:"commandId"`
	Status    string          `json:"status"`
	Result    json.RawMessage `json:"result,omitempty"`
	Error     string          `json:"error,omitempty"`
	UpdatedAt time.Time       `json:"updatedAt"`
}

type SnapshotTask struct {
	TaskID       uuid.UUID `json:"taskId"`
	ThreadID     string    `json:"threadId,omitempty"`
	ActiveTurnID string    `json:"activeTurnId,omitempty"`
	RelayTurnID  uuid.UUID `json:"relayTurnId,omitempty"`
	WorkspaceID  uuid.UUID `json:"workspaceId"`
	UpdatedAt    time.Time `json:"updatedAt"`
}

type SnapshotWorkspace struct {
	WorkspaceID   uuid.UUID `json:"workspaceId"`
	CanonicalPath string    `json:"canonicalPath"`
	UpdatedAt     time.Time `json:"updatedAt"`
}

type Welcome struct {
	RunnerID         uuid.UUID `json:"runnerId"`
	HeartbeatSeconds int       `json:"heartbeatSeconds"`
	ServerTime       time.Time `json:"serverTime"`
	Message          string    `json:"message,omitempty"`
}