-- name: UpsertWorkspaceAgentContextSnapshot :one

INSERT INTO workspace_agent_context_snapshots (

workspace_agent_id,

version,

aggregate_hash,

snapshot_error,

received_at

) VALUES (

@workspace_agent_id,

@version,

@aggregate_hash,

@snapshot_error,

@received_at

)

ON CONFLICT (workspace_agent_id) DO UPDATE SET

version = EXCLUDED.version,

aggregate_hash = EXCLUDED.aggregate_hash,

snapshot_error = EXCLUDED.snapshot_error,

received_at = EXCLUDED.received_at

RETURNING *;

-- name: UpsertWorkspaceAgentContextResource :one

INSERT INTO workspace_agent_context_resources (

workspace_agent_id,

source,

body_kind,

body,

content_hash,

size_bytes,

status,

error,

source_path,

created_at,

updated_at

) VALUES (

@workspace_agent_id,

@source,

@body_kind,

@body,

@content_hash,

@size_bytes,

@status,

@error,

@source_path,

@now,

@now

)

ON CONFLICT (workspace_agent_id, source) DO UPDATE SET

body_kind = EXCLUDED.body_kind,

body = EXCLUDED.body,

content_hash = EXCLUDED.content_hash,

size_bytes = EXCLUDED.size_bytes,

status = EXCLUDED.status,

error = EXCLUDED.error,

source_path = EXCLUDED.source_path,

updated_at = EXCLUDED.updated_at

RETURNING *;

-- name: DeleteStaleWorkspaceAgentContextResources :exec

-- Deletes any resources for the agent whose source is not in the

-- supplied active set. Atomic alongside the snapshot upsert so the

-- stored snapshot and resource rows always agree.

DELETE FROM workspace_agent_context_resources

WHERE workspace_agent_id = @workspace_agent_id

AND NOT (source = ANY(@active_sources :: text[]));

-- name: GetLatestWorkspaceAgentContextSnapshot :one

SELECT * FROM workspace_agent_context_snapshots

WHERE workspace_agent_id = @workspace_agent_id;

-- name: ListWorkspaceAgentContextResources :many

SELECT * FROM workspace_agent_context_resources

WHERE workspace_agent_id = @workspace_agent_id

ORDER BY source ASC;