provenance

type Attribution struct {
	ID           string            `json:"id"`
	SessionKey   string            `json:"session_key"`
	RunID        string            `json:"run_id,omitempty"`
	WorkspaceID  string            `json:"workspace_id,omitempty"`
	AuthorType   AuthorType        `json:"author_type"`
	AuthorID     string            `json:"author_id"`
	FilePath     string            `json:"file_path,omitempty"`
	CommitHash   string            `json:"commit_hash,omitempty"`
	StepID       string            `json:"step_id,omitempty"`
	Source       AttributionSource `json:"source,omitempty"`
	LinesAdded   int               `json:"lines_added"`
	LinesRemoved int               `json:"lines_removed"`
	TokensUsed   TokenSummary      `json:"tokens_used"`
	CreatedAt    time.Time         `json:"created_at"`
}

type AttributionReport struct {
	SessionKey  string                 `json:"session_key"`
	ByAuthor    map[string]AuthorStats `json:"by_author"`
	ByFile      map[string]FileStats   `json:"by_file"`
	TotalTokens TokenSummary           `json:"total_tokens"`
	Checkpoints int                    `json:"checkpoints"`
	GeneratedAt time.Time              `json:"generated_at"`
}

type AttributionService struct {
	// contains filtered or unexported fields
}

type AttributionSource string

type AttributionStore interface {
	SaveAttribution(ctx context.Context, attr Attribution) error
	ListBySession(ctx context.Context, sessionKey string, limit int) ([]Attribution, error)
}

type AttributionView struct {
	Attributions   []Attribution           `json:"attributions"`
	ByAuthorTokens map[string]TokenSummary `json:"by_author_tokens,omitempty"`
	TotalTokens    TokenSummary            `json:"total_tokens"`
	Checkpoints    int                     `json:"checkpoints"`
}

type AuthorStats struct {
	AuthorType   AuthorType   `json:"author_type"`
	LinesAdded   int          `json:"lines_added"`
	LinesRemoved int          `json:"lines_removed"`
	TokensUsed   TokenSummary `json:"tokens_used"`
	FileCount    int          `json:"file_count"`
}

type AuthorType string

type BundleService struct {
	// contains filtered or unexported fields
}

type BundleSigner interface {
	Sign(ctx context.Context, payload []byte) ([]byte, error)
	Algorithm() string
}

type Checkpoint struct {
	ID         string            `json:"id"`
	SessionKey string            `json:"session_key"`
	RunID      string            `json:"run_id,omitempty"`
	Label      string            `json:"label"`
	Trigger    CheckpointTrigger `json:"trigger"`
	JournalSeq int64             `json:"journal_seq"`
	GitRef     string            `json:"git_ref,omitempty"`
	TokensUsed *TokenSummary     `json:"tokens_used,omitempty"`
	Metadata   map[string]string `json:"metadata,omitempty"`
	CreatedAt  time.Time         `json:"created_at"`
}

type CheckpointService struct {
	// contains filtered or unexported fields
}

type CheckpointStore interface {
	// SaveCheckpoint persists a new checkpoint.
	SaveCheckpoint(ctx context.Context, cp Checkpoint) error

	// GetCheckpoint returns a checkpoint by ID.
	GetCheckpoint(ctx context.Context, id string) (*Checkpoint, error)

	// ListByRun returns checkpoints for a run, ordered by journal_seq asc.
	ListByRun(ctx context.Context, runID string) ([]Checkpoint, error)

	// ListBySession returns checkpoints for a session, ordered by created_at desc.
	ListBySession(ctx context.Context, sessionKey string, limit int) ([]Checkpoint, error)

	// CountBySession returns the number of checkpoints for a session.
	CountBySession(ctx context.Context, sessionKey string) (int, error)

	// DeleteCheckpoint removes a checkpoint by ID.
	DeleteCheckpoint(ctx context.Context, id string) error
}

type CheckpointTrigger string

type EntAttributionStore struct {
	// contains filtered or unexported fields
}

type EntCheckpointStore struct {
	// contains filtered or unexported fields
}

type EntSessionTreeStore struct {
	// contains filtered or unexported fields
}

type FileStats struct {
	LinesAdded   int `json:"lines_added"`
	LinesRemoved int `json:"lines_removed"`
	AuthorCount  int `json:"author_count"`
}

type GitFileStat struct {
	FilePath     string
	LinesAdded   int
	LinesRemoved int
}

type MemoryAttributionStore struct {
	// contains filtered or unexported fields
}

type MemoryStore struct {
	// contains filtered or unexported fields
}

type MemoryTreeStore struct {
	// contains filtered or unexported fields
}

type PQBundleSigner interface {
	SignPQ(ctx context.Context, payload []byte) ([]byte, error)
	PQAlgorithm() string
	PQPublicKey() []byte
}

type ProvenanceBundle struct {
	Version              string             `json:"version"`
	Checkpoints          []Checkpoint       `json:"checkpoints"`
	SessionTree          []SessionNode      `json:"session_tree,omitempty"`
	Attributions         []Attribution      `json:"attributions,omitempty"`
	Report               *AttributionReport `json:"report,omitempty"`
	SignerDID            string             `json:"signer_did,omitempty"`
	SignatureAlgorithm   string             `json:"signature_algorithm,omitempty"`
	Signature            []byte             `json:"signature,omitempty"`
	PQSignerPublicKey    []byte             `json:"pq_signer_public_key,omitempty"` // embedded ML-DSA-65 pubkey for rotation-safe verification
	PQSignatureAlgorithm string             `json:"pq_signature_algorithm,omitempty"`
	PQSignature          []byte             `json:"pq_signature,omitempty"`
	RedactionLevel       RedactionLevel     `json:"redaction_level"`
}

type RedactionLevel string

type SessionNode struct {
	SessionKey  string        `json:"session_key"`
	ParentKey   string        `json:"parent_key,omitempty"`
	AgentName   string        `json:"agent_name"`
	Goal        string        `json:"goal,omitempty"`
	RunID       string        `json:"run_id,omitempty"`
	WorkspaceID string        `json:"workspace_id,omitempty"`
	Depth       int           `json:"depth"`
	Status      SessionStatus `json:"status"`
	CreatedAt   time.Time     `json:"created_at"`
	ClosedAt    *time.Time    `json:"closed_at,omitempty"`
}

type SessionStatus string

type SessionTree struct {
	// contains filtered or unexported fields
}

type SessionTreeStore interface {
	// SaveNode persists a session node.
	SaveNode(ctx context.Context, node SessionNode) error

	// GetNode returns a node by session key.
	GetNode(ctx context.Context, sessionKey string) (*SessionNode, error)

	// GetChildren returns direct children of a session.
	GetChildren(ctx context.Context, parentKey string) ([]SessionNode, error)

	// ListAll returns all session nodes, ordered by created_at desc.
	ListAll(ctx context.Context, limit int) ([]SessionNode, error)

	// UpdateStatus updates the status and optional closed_at of a node.
	UpdateStatus(ctx context.Context, sessionKey string, status SessionStatus, closedAt *time.Time) error
}

type SignatureVerifyFunc func(signerDID string, payload, signature []byte) error

type TokenSummary struct {
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
	TotalTokens  int64 `json:"total_tokens"`
}

type TokenUsageReader interface {
	QueryBySession(ctx context.Context, sessionKey string) ([]observability.TokenUsage, error)
}