microrag

type BuiltinResolver interface {
	Resolve(name string) (*BuiltinSpecJSON, error)
}

type BuiltinSpecJSON struct {
	Name      string `json:"name"`
	Module    string `json:"module"`
	Signature string `json:"signature"`
	IsPure    bool   `json:"is_pure"`
	Effect    string `json:"effect,omitempty"`
	Metadata  *struct {
		Description string `json:"description,omitempty"`
		Examples    []struct {
			Code        string `json:"code"`
			Description string `json:"description,omitempty"`
		} `json:"examples,omitempty"`
		Since string `json:"since,omitempty"`
	} `json:"metadata,omitempty"`
}

type CLIBuiltinResolver struct {
	Binary string
}

type CLISearcher struct {
	Binary  string        // path to ailang binary; "" → "ailang" on PATH
	Timeout time.Duration // per-call timeout; 0 → no timeout
}

type Config struct {
	Enabled       bool        `yaml:"enabled"`
	Routes        []Route     `yaml:"routes"`
	Dedup         DedupConfig `yaml:"dedup"`
	SessionBudget int         `yaml:"session_budget"`
	MarkerStyle   string      `yaml:"marker_style"`
}

type ContextResult struct {
	Injection *Injection `json:"injection,omitempty"`
	State     string     `json:"microrag_state"`
	Reason    string     `json:"reason,omitempty"` // why no injection (disabled, no_route, dedup, ...)
}

type DedupConfig struct {
	Windows          map[string]int     `yaml:"windows"`
	RelevanceBypass  map[string]float64 `yaml:"relevance_bypass"`
	WallClockMaxSecs int                `yaml:"wall_clock_max"`
}

type Engine struct {
	Cfg        *Config
	Searcher   Searcher
	SessionDir string
	Now        func() time.Time // injectable for tests
}

type Injection struct {
	InjectionText string  `json:"injection_text"`
	SnippetID     string  `json:"snippet_id"`
	Tokens        int     `json:"tokens"`
	Namespace     string  `json:"ns"`
	Score         float64 `json:"score"`
	State         string  `json:"microrag_state"` // "on" | "dryrun" | "disabled"
}

type LedgerEntry struct {
	TS        int64   `json:"ts"`
	SnippetID string  `json:"snippet_id"`
	Tokens    int     `json:"tokens"`
	FilePath  string  `json:"file_path"`
	Namespace string  `json:"ns"`
	State     string  `json:"microrag_state"`
	Score     float64 `json:"score"`
}

type LintNudge struct {
	Name          string `json:"name"`
	Module        string `json:"module"`
	Signature     string `json:"signature"`
	Description   string `json:"description,omitempty"`
	Example       string `json:"example,omitempty"`
	InjectionText string `json:"injection_text"`
	Tokens        int    `json:"tokens"`
}

type LintRequest struct {
	FilePath string
	Code     string
}

type LintResult struct {
	Nudges []LintNudge `json:"nudges,omitempty"`
	State  string      `json:"microrag_state"`
	Reason string      `json:"reason,omitempty"`
}

type Linter struct {
	Resolver   BuiltinResolver
	SessionDir string
	MaxNudges  int // hard cap per call; 0 → 2
	MaxTokens  int // per-nudge budget; 0 → 80
}

type Request struct {
	ToolName string // Edit | Write | Read
	FilePath string
	Content  string // for Edit/Write — first ~2KB used for query enrichment
}

type Route struct {
	Glob                  string  `yaml:"glob"`
	KB                    string  `yaml:"kb"`
	MaxTokensPerInjection int     `yaml:"max_tokens_per_injection"`
	RelevanceFloor        float64 `yaml:"relevance_floor"`
}

type SearchEnvelope struct {
	Scope   string      `json:"scope"`
	Count   int         `json:"count"`
	QueryMs int64       `json:"query_ms"`
	Results []SearchHit `json:"results"`
}

type SearchHit struct {
	Tier      string  `json:"tier"`
	Namespace string  `json:"namespace"`
	Key       string  `json:"key"`
	Score     float64 `json:"score"`
	Content   string  `json:"content"`
	UpdatedAt int64   `json:"updated_at_ms"`
	Source    string  `json:"source"`
}

type Searcher interface {
	Search(query, namespace string, limit int) ([]SearchHit, error)
}

type UserPromptRequest struct {
	Prompt     string
	Namespaces []string // empty → defaultUserPromptNamespaces
}

type UserPromptResult struct {
	Injection *Injection `json:"injection,omitempty"`
	State     string     `json:"microrag_state"`
	Reason    string     `json:"reason,omitempty"`
}