pi

type Block struct {
	Type string `json:"type"`

	// text.
	Text          string `json:"text"`
	TextSignature string `json:"textSignature"`

	// image: base64 data.
	Data     string `json:"data"`
	MimeType string `json:"mimeType"`

	// thinking.
	Thinking          string `json:"thinking"`
	ThinkingSignature string `json:"thinkingSignature"`
	Redacted          bool   `json:"redacted"`

	// toolCall.
	ID               string          `json:"id"`
	Name             string          `json:"name"`
	Arguments        json.RawMessage `json:"arguments"`
	ThoughtSignature string          `json:"thoughtSignature"`
	Namespace        string          `json:"namespace"`

	Members
	// contains filtered or unexported fields
}

type Content struct {
	// Text is the content when it was a string.
	Text string
	// IsText says the content was a string.
	IsText bool
	Blocks []*Block
}

type File struct {
	// Path is the file the session was read from; empty when it was read
	// from a reader.
	Path string
	// Lines are the lines that decoded as JSON, the header among them,
	// in file order.
	Lines []*Line
	// Errors lists the lines that could not be read, which the reader
	// skipped, as pi skips them.
	Errors []LineError
	// Surrogates lists the lines that held lone UTF-16 surrogate
	// escapes, which the reader removed before decoding; such a line
	// writes back without them.
	Surrogates []int
}

type Line struct {
	Type string `json:"type"`
	// ID is the session's id on the header, the entry's id otherwise.
	ID string `json:"id"`
	// ParentID is empty for a root entry; the input wrote null.
	ParentID  string `json:"parentId"`
	Timestamp string `json:"timestamp"`

	// session (the header). Version is a number, as JavaScript writes
	// it; [File.Version] reads it.
	Version       float64 `json:"version"`
	CWD           string  `json:"cwd"`
	ParentSession string  `json:"parentSession"`

	// message.
	Message *Message `json:"message"`

	// model_change and usage.
	Provider string `json:"provider"`
	ModelID  string `json:"modelId"`
	// thinking_level_change.
	ThinkingLevel string `json:"thinkingLevel"`

	// usage.
	Kind  string `json:"kind"`
	Model string `json:"model"`
	Note  string `json:"note"`
	// usage, compaction and branch_summary.
	Usage *Usage `json:"usage"`

	// compaction and branch_summary.
	Summary  string          `json:"summary"`
	Details  json.RawMessage `json:"details"`
	FromHook *bool           `json:"fromHook"`
	// compaction.
	FirstKeptEntryID string `json:"firstKeptEntryId"`
	// FirstKeptEntryIndex is version 1's form of FirstKeptEntryID: an
	// index into the file's entries, the header counted.
	FirstKeptEntryIndex *int     `json:"firstKeptEntryIndex"`
	TokensBefore        int      `json:"tokensBefore"`
	SystemMessage       *Message `json:"systemMessage"`
	// branch_summary.
	FromID string `json:"fromId"`

	// custom and custom_message.
	CustomType string `json:"customType"`
	// custom.
	Data json.RawMessage `json:"data"`
	// custom_message.
	Content *Content `json:"content"`
	Display bool     `json:"display"`

	// context_edit and label.
	TargetID string `json:"targetId"`
	// context_edit: null omits the target from context, an object
	// replaces its content.
	Replacement json.RawMessage `json:"replacement"`
	// label: absent clears the label.
	Label *string `json:"label"`
	// session_info.
	Name *string `json:"name"`

	Members

	// No is the line's 1-based number in its file.
	No int `json:"-"`
	// Raw is the line as read, without NUL bytes.
	Raw json.RawMessage `json:"-"`
	// Err is set when the line is JSON but does not decode as a line of
	// this package's types; only Type, No, Raw and, when they read, ID,
	// ParentID and Timestamp are then set.
	Err error `json:"-"`
	// Synthetic says ID and ParentID were given by [Migrate] to a
	// version 1 entry, which had none on disk.
	Synthetic bool `json:"-"`
}

type LineError struct {
	No  int
	Err error
}

type Members = jsonline.Members

type Message struct {
	Role    string   `json:"role"`
	Content *Content `json:"content"`
	// Timestamp is Unix milliseconds.
	Timestamp float64 `json:"timestamp"`

	// system.
	Sections     Sections    `json:"sections"`
	ToolsAdded   []*Tool     `json:"toolsAdded"`
	ToolsRemoved []*ToolName `json:"toolsRemoved"`
	Replace      bool        `json:"replace"`

	// assistant.
	API                   string          `json:"api"`
	Provider              string          `json:"provider"`
	Model                 string          `json:"model"`
	ResponseModel         string          `json:"responseModel"`
	ResponseID            string          `json:"responseId"`
	ThinkingLevel         string          `json:"thinkingLevel"`
	ProviderThinkingLevel string          `json:"providerThinkingLevel"`
	Diagnostics           json.RawMessage `json:"diagnostics"`
	StopReason            string          `json:"stopReason"`
	RawStopReason         string          `json:"rawStopReason"`
	Deferred              json.RawMessage `json:"deferred"`
	ErrorMessage          string          `json:"errorMessage"`
	EndTurn               *bool           `json:"endTurn"`
	// assistant and toolResult.
	Usage *Usage `json:"usage"`

	// toolResult.
	ToolCallID string `json:"toolCallId"`
	ToolName   string `json:"toolName"`
	IsError    bool   `json:"isError"`
	// toolResult and custom.
	Details json.RawMessage `json:"details"`

	// bashExecution.
	Command            string `json:"command"`
	Output             string `json:"output"`
	ExitCode           *int   `json:"exitCode"`
	Cancelled          bool   `json:"cancelled"`
	Truncated          bool   `json:"truncated"`
	FullOutputPath     string `json:"fullOutputPath"`
	ExcludeFromContext bool   `json:"excludeFromContext"`

	// custom.
	CustomType string `json:"customType"`
	Display    bool   `json:"display"`

	// branchSummary and compactionSummary, which pi derives in memory
	// and does not persist.
	Summary      string `json:"summary"`
	FromID       string `json:"fromId"`
	TokensBefore int    `json:"tokensBefore"`

	Members
}

type Migration struct {
	// From is the version the file declared; To is [CurrentVersion].
	From, To int
	// IDs counts the version 1 entries given an id and a parentId,
	// chaining them in file order.
	IDs int
	// FirstKept counts the version 1 compactions whose
	// firstKeptEntryIndex became a firstKeptEntryId.
	FirstKept int
	// HookMessages counts the messages whose version 2 role hookMessage
	// became custom.
	HookMessages int
	// KeptIDs says the file read as version 1 but its entries carry ids,
	// which were kept rather than replaced.
	KeptIDs bool
}

type Node struct {
	Line     *Line
	Parent   *Node
	Children []*Node
}

type Section struct {
	Name   string
	Text   string
	Remove bool
}

type Sections []Section

type Tool struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Parameters  json.RawMessage `json:"parameters"`

	Members
}

type ToolName struct {
	Name string `json:"name"`

	Members
}

type Tree struct {
	// Roots are the entries with no parent, in file order.
	Roots []*Node
	// Nodes holds every entry in file order.
	Nodes []*Node
	// Leaf is the last entry in the file, where pi resumes; nil when
	// the file has no entry.
	Leaf *Node
	// Dangling lists the entries whose parentId names no entry; each
	// starts a root.
	Dangling []*Line
	// Cycles lists the entries whose parent links led back to
	// themselves; the link was cut and each starts a root.
	Cycles []*Line
	// Duplicates lists the entries whose id an earlier entry had. pi
	// indexes by id and the later entry wins, so a parentId naming the
	// id names the later one; the earlier keeps its place in the tree.
	Duplicates []*Line
	// contains filtered or unexported fields
}

type Usage struct {
	Input        int64  `json:"input"`
	Output       int64  `json:"output"`
	CacheRead    int64  `json:"cacheRead"`
	CacheWrite   int64  `json:"cacheWrite"`
	CacheWrite1h *int64 `json:"cacheWrite1h"`
	Reasoning    *int64 `json:"reasoning"`
	TotalTokens  int64  `json:"totalTokens"`
	// Cost is in dollars, per kind of token and in total.
	Cost json.RawMessage `json:"cost"`

	Members
}