resources.go

// Package resources serves read-only MCP resources.
package resources

import (
	gomcp "github.com/modelcontextprotocol/go-sdk/mcp"
	"github.com/radimsem/remindb/pkg/logbuf"
	"github.com/radimsem/remindb/pkg/mcp/ledger"
	"github.com/radimsem/remindb/pkg/mcp/rescanstat"
	"github.com/radimsem/remindb/pkg/mcp/session"
	"github.com/radimsem/remindb/pkg/store"
)

const mimeJSON = "application/json"

var Subscribable = map[string]string{
	"graph":       GraphURI,
	"snapshots":   SnapshotsURI,
	"temperature": TemperatureURI,
	"tree":        TreeURI,
	"files":       FilesURI,
	"logs":        LogsURI,
	"rescan":      RescanURI,
}

type Deps struct {
	Store         *store.Store
	HotThreshold  float64
	ColdThreshold float64
	LogBuffer     *logbuf.Buffer
	Sessions      *session.Registry
	Ledger        *ledger.Ledger
	RescanStatus  *rescanstat.Status
	SessionLogDir string
}

// Register every read-only resource on the server.
func Register(srv *gomcp.Server, d *Deps) {
	srv.AddResource(&gomcp.Resource{
		Name:        "overview",
		URI:         OverviewURI,
		MIMEType:    mimeJSON,
		Description: "Database introspection overview — node, snapshot, temperature, and relation counts as stable JSON. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleOverview)

	srv.AddResource(&gomcp.Resource{
		Name:        "files",
		URI:         FilesURI,
		MIMEType:    mimeJSON,
		Description: "Compiled source files grouped by compile root, with per-file node and token counts as stable JSON — the JSON twin of `remindb inspect --files`. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleFiles)

	srv.AddResource(&gomcp.Resource{
		Name:        "tree",
		URI:         TreeURI,
		MIMEType:    mimeJSON,
		Description: "Full parent/child node hierarchy as nested JSON — the structured twin of `MemoryTree`'s text. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleTree)

	srv.AddResource(&gomcp.Resource{
		Name:        "graph",
		URI:         GraphURI,
		MIMEType:    mimeJSON,
		Description: "Relations knowledge graph — resolved edges, pending/unresolved edges, and the referenced node set as stable JSON, for a UI drawing the brain graph. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleGraph)

	srv.AddResourceTemplate(&gomcp.ResourceTemplate{
		Name:        "tree-by-root",
		URITemplate: TreeByRootTemplate,
		MIMEType:    mimeJSON,
		Description: "Node hierarchy rooted at {rootId}, optionally depth-bounded via ?depth=N (omitted = full subtree). Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleTreeByRoot)

	srv.AddResource(&gomcp.Resource{
		Name:        "snapshots",
		URI:         SnapshotsURI,
		MIMEType:    mimeJSON,
		Description: "Version history — every snapshot newest-first with parent links, compile root, and the HEAD marker, as stable JSON for an interactive timeline. Mirrors MemoryHistory. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSnapshots)

	srv.AddResourceTemplate(&gomcp.ResourceTemplate{
		Name:        "snapshots-limited",
		URITemplate: SnapshotsLimitTemplate,
		MIMEType:    mimeJSON,
		Description: "Version history bounded to the newest ?limit=N snapshots (omitted = full history). Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSnapshotsLimited)

	srv.AddResource(&gomcp.Resource{
		Name:        "temperature",
		URI:         TemperatureURI,
		MIMEType:    mimeJSON,
		Description: "Per-node temperature for a heatmap — every node's id, label, temperature, and pinned flag in one array, plus an aggregate summary with the cold/hot cut points. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleTemperature)

	srv.AddResourceTemplate(&gomcp.ResourceTemplate{
		Name:        "snapshot-diffs",
		URITemplate: SnapshotDiffsTemplate,
		MIMEType:    mimeJSON,
		Description: "Per-snapshot diff records for snapshot {id} (op, node_id, old/new hash + content), the data behind MemoryDelta. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSnapshotDiffs)

	srv.AddResource(&gomcp.Resource{
		Name:        "doctor",
		URI:         DoctorURI,
		MIMEType:    mimeJSON,
		Description: "Health-check report — overall worst-wins status header plus every check's name/status/detail as stable JSON, byte-equivalent to `remindb doctor --json`. Read-only (never applies `--fix`). Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleDoctor)

	srv.AddResource(&gomcp.Resource{
		Name:        "logs",
		URI:         LogsURI,
		MIMEType:    mimeJSON,
		Description: "Recent server log records from the in-memory ring buffer, newest last, with an overflow count, as stable JSON for a log console. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleLogs)

	srv.AddResource(&gomcp.Resource{
		Name:        "sessions",
		URI:         SessionsURI,
		MIMEType:    mimeJSON,
		Description: "Active MCP client sessions on the bound database — db_path plus per-session id, transport, listen address (http only), connect/last-activity timestamps, and tool-call count, as stable JSON for a 'who's attached' view. Membership mirrors the SDK's live session set. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSessions)

	srv.AddResource(&gomcp.Resource{
		Name:        "sessions-history",
		URI:         SessionsHistoryURI,
		MIMEType:    mimeJSON,
		Description: "Durable per-client session ledger — every MCP client that has ever attached to this database, with stable hash id, last-seen metadata, session count, summed connection lifetime, last-disconnect time, and total tool calls, as stable JSON. Accumulates across reconnects and serve restarts. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSessionsHistory)

	srv.AddResourceTemplate(&gomcp.ResourceTemplate{
		Name:        "sessions-history-by-hash",
		URITemplate: SessionsHistoryByHashTemplate,
		MIMEType:    mimeJSON,
		Description: "Durable session ledger for a single client identified by {hash} (the id surfaced in remindb://sessions/history). Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSessionsHistoryByHash)

	srv.AddResource(&gomcp.Resource{
		Name:        "sessions-logs",
		URI:         SessionsLogsURI,
		MIMEType:    mimeJSON,
		Description: "Index of per-session logfiles under .remindb/logs/ — db_path plus one entry per session with session_id, size_bytes, rotated (a .1 generation exists), and modified_at, as stable JSON. Empty array when session logging is disabled. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSessionsLogs)

	srv.AddResourceTemplate(&gomcp.ResourceTemplate{
		Name:        "sessions-log-by-id",
		URITemplate: SessionsLogByIDTemplate,
		MIMEType:    mimeJSON,
		Description: "One session's captured tool-call + Warn/Error trace ({id} is the session id surfaced in remindb://sessions/logs) as structured JSON entries (time, level, msg, fields) in append order, active file only. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleSessionsLogByID)

	srv.AddResource(&gomcp.Resource{
		Name:        "rescan",
		URI:         RescanURI,
		MIMEType:    mimeJSON,
		Description: "Latest source-rescan tick — configured interval plus last_meta: run timestamp, error, add/modify/remove counts, and the per-file purge list, as stable JSON for a live rescan-activity panel. Passive read: does not boost temperature or create a snapshot.",
	}, d.HandleRescan)
}