formatter

type AssistantMessage struct {
	Content []ContentBlock `json:"content"`
}

type ContentBlock struct {
	Type     string `json:"type"`
	Text     string `json:"text"`
	Thinking string `json:"thinking"`
	// tool_use fields:
	ID    string          `json:"id"`
	Name  string          `json:"name"`
	Input json.RawMessage `json:"input"`
	// tool_result fields:
	ToolUseID string          `json:"tool_use_id"`
	IsError   *bool           `json:"is_error"`
	Content   json.RawMessage `json:"content"`
}

type Formatter interface {
	// ProcessStream reads newline-delimited stream-json from r until EOF or ctx cancellation.
	// For each line it writes the raw bytes (including the trailing newline) to rawWriter,
	// and a formatted human-readable line (with timestamp prefix, trailing newline) to formattedWriter.
	// Lines that are not valid JSON are written verbatim to both outputs.
	// ProcessStream returns nil on clean EOF. It returns a non-nil error only when rawWriter
	// or formattedWriter return a write error.
	ProcessStream(
		ctx context.Context,
		r io.Reader,
		rawWriter io.Writer,
		formattedWriter io.Writer,
	) error
}

type RateLimitInfo struct {
	// Status is the rate-limit status string, e.g. "allowed_warning" or "blocked".
	Status string `json:"status"`
	// ResetsAt is the unix epoch (seconds) when the bucket resets. Zero means unknown.
	ResetsAt int64 `json:"resetsAt"`
	// RateLimitType identifies the bucket, e.g. "seven_day" or "five_hour".
	RateLimitType string `json:"rateLimitType"`
	// Utilization is the fraction of the bucket consumed, 0.0–1.0.
	Utilization float64 `json:"utilization"`
	// IsUsingOverage indicates overage consumption.
	IsUsingOverage bool `json:"isUsingOverage"`
	// SurpassedThreshold indicates the threshold has been crossed.
	SurpassedThreshold bool `json:"surpassedThreshold"`
}

type StreamMessage struct {
	Type    string          `json:"type"`
	Subtype string          `json:"subtype"`
	Message json.RawMessage `json:"message"`
	// Result event fields (type == "result"):
	DurationMs   *float64 `json:"duration_ms"`
	TotalCostUSD *float64 `json:"total_cost_usd"`
	Usage        *Usage   `json:"usage"`
	Result       *string  `json:"result"`
	IsError      *bool    `json:"is_error"`
	// RateLimitInfo is populated when Type == "rate_limit_event".
	RateLimitInfo *RateLimitInfo `json:"rate_limit_info"`
	// System init fields (type == "system", subtype == "init"):
	SessionID string   `json:"session_id"`
	Model     string   `json:"model"`
	Cwd       string   `json:"cwd"`
	Tools     []string `json:"tools"`
}

type Usage struct {
	InputTokens          *int `json:"input_tokens"`
	OutputTokens         *int `json:"output_tokens"`
	CacheReadInputTokens *int `json:"cache_read_input_tokens"`
}

type UserMessage struct {
	Content []ContentBlock `json:"content"`
}