@m16khb/llm-wiki 0.1.0 → 0.1.2

package/internal/sessionctx/sessionctx_test.go

@@ -0,0 +1,415 @@
+package sessionctx
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/m16khb/llm-wiki/internal/wiki"
+)
+
+func TestBuildContextDerivesProjectTermsAndSearchesWiki(t *testing.T) {
+	vaultRoot := t.TempDir()
+	writeFile(t, vaultRoot, "20-wiki/concepts/llm-wiki-hooks.md", `---
+title: LLM Wiki Hooks
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [llm-wiki, hooks]
+domain: agent-memory
+---
+
+# LLM Wiki Hooks
+
+Codex and Claude Code use portable hook commands for llm-wiki context.
+`)
+	projectRoot := filepath.Join(t.TempDir(), "llm-wiki")
+	writeFile(t, projectRoot, "go.mod", "module github.com/m16khb/llm-wiki\n")
+	writeFile(t, projectRoot, "README.md", "# LLM Wiki\n")
+
+	v, err := wiki.New(vaultRoot)
+	if err != nil {
+		t.Fatalf("wiki.New() error = %v", err)
+	}
+	result, err := BuildContext(v, Options{ProjectPath: projectRoot, Limit: 5})
+	if err != nil {
+		t.Fatalf("BuildContext() error = %v", err)
+	}
+	if result.Project.Name != "llm-wiki" || !contains(result.Project.Terms, "llm-wiki") {
+		t.Fatalf("unexpected project info: %+v", result.Project)
+	}
+	if len(result.Results) != 1 || result.Results[0].Slug != "llm-wiki-hooks" {
+		t.Fatalf("expected hook search hit, got %+v", result.Results)
+	}
+	text := FormatAdditionalContext(result)
+	if !strings.Contains(text, "llm-wiki session context") || !strings.Contains(text, "[[llm-wiki-hooks]]") {
+		t.Fatalf("unexpected context text:\n%s", text)
+	}
+}
+
+func TestBuildContextAddsTechResultsForGoAndSwaggerProject(t *testing.T) {
+	vaultRoot := t.TempDir()
+	writeFile(t, vaultRoot, "20-wiki/concepts/llm-wiki-hooks.md", `---
+title: LLM Wiki Hooks
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [llm-wiki, hooks]
+domain: agent-memory
+---
+
+# LLM Wiki Hooks
+
+Project-specific context for llm-wiki hook behavior.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/go-conventions.md", `---
+title: Go Conventions
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [go, golang]
+domain: programming
+---
+
+# Go Conventions
+
+Go and Golang style rules for agents working in Go modules.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/swagger-rules.md", `---
+title: Swagger Rules
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [swagger, openapi, swaggo]
+domain: api
+---
+
+# Swagger Rules
+
+Swagger and OpenAPI documentation rules for API projects.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/mcp-conventions.md", `---
+title: MCP Conventions
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [mcp]
+domain: protocol
+---
+
+# MCP Conventions
+
+Model Context Protocol rules for local tools.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/sqlite-conventions.md", `---
+title: SQLite Conventions
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [sqlite]
+domain: database
+---
+
+# SQLite Conventions
+
+SQLite WAL queue rules.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/llm-wiki-tech-map.md", `---
+title: LLM Wiki Tech Map
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [llm-wiki, go, golang, mcp, sqlite, swagger, openapi, swaggo]
+domain: agent-memory
+---
+
+# LLM Wiki Tech Map
+
+Project-specific map for llm-wiki Go, MCP, SQLite, and Swagger retrieval.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/gin-conventions.md", `---
+title: Gin Conventions
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [go, gin]
+domain: web
+---
+
+# Gin Conventions
+
+Go Gin framework rules. This should only appear when Gin is detected.
+`)
+	writeFile(t, vaultRoot, "20-wiki/concepts/repository-references/other-api/other-api-go-noise.md", `---
+title: Other API Go Noise
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [go]
+domain: programming
+---
+
+# Other API Go Noise
+
+	go go go go go go go go go go go go go go go go go go go go
+	golang golang golang golang golang golang golang golang golang golang
+`)
+	projectRoot := filepath.Join(t.TempDir(), "llm-wiki")
+	writeFile(t, projectRoot, "go.mod", `module github.com/m16khb/llm-wiki
+
+require (
+	github.com/modelcontextprotocol/go-sdk v1.6.0
+	github.com/swaggo/gin-swagger v1.6.0
+	modernc.org/sqlite v1.50.1
+)
+`)
+	writeFile(t, projectRoot, "README.md", "# LLM Wiki\n")
+
+	v, err := wiki.New(vaultRoot)
+	if err != nil {
+		t.Fatalf("wiki.New() error = %v", err)
+	}
+	result, err := BuildContext(v, Options{ProjectPath: projectRoot, Limit: 5})
+	if err != nil {
+		t.Fatalf("BuildContext() error = %v", err)
+	}
+	text := FormatAdditionalContext(result)
+	for _, want := range []string{"Detected tech terms:", "go", "golang", "mcp", "sqlite", "swagger", "openapi", "swaggo", "Suggested reads:", "[[go-conventions]]", "[[swagger-rules]]", "[[mcp-conventions]]", "[[sqlite-conventions]]"} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("expected formatted context to contain %q:\n%s", want, text)
+		}
+	}
+	if len(result.TechResults) == 0 || result.TechResults[0].Slug == "other-api-go-noise" {
+		t.Fatalf("expected unrelated repository-reference tech hit to be demoted, got %+v", result.TechResults)
+	}
+	if result.TechResults[0].Slug != "llm-wiki-tech-map" {
+		t.Fatalf("expected project-specific tech map first, got %+v", result.TechResults)
+	}
+	for _, hit := range result.TechResults {
+		if hit.Slug == "gin-conventions" {
+			t.Fatalf("did not expect Gin-specific page without Gin dependency, got %+v", result.TechResults)
+		}
+		if hit.Slug == "other-api-go-noise" {
+			t.Fatalf("did not expect unrelated repository-reference page, got %+v", result.TechResults)
+		}
+	}
+}
+
+func TestBuildContextAddsPackageWrapperTechResults(t *testing.T) {
+	vaultRoot := t.TempDir()
+	writeFile(t, vaultRoot, "20-wiki/concepts/npm-wrapper.md", `---
+title: npm Wrapper
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [node, npm, npx]
+domain: packaging
+---
+
+# npm Wrapper
+
+Node, npm, and npx wrapper conventions.
+`)
+	projectRoot := filepath.Join(t.TempDir(), "wrapped-cli")
+	writeFile(t, projectRoot, "package.json", `{
+  "name": "@example/wrapped-cli",
+  "bin": {
+    "wrapped-cli": "npm/bin/wrapped-cli.js"
+  }
+}
+`)
+
+	v, err := wiki.New(vaultRoot)
+	if err != nil {
+		t.Fatalf("wiki.New() error = %v", err)
+	}
+	result, err := BuildContext(v, Options{ProjectPath: projectRoot, Limit: 5})
+	if err != nil {
+		t.Fatalf("BuildContext() error = %v", err)
+	}
+	text := FormatAdditionalContext(result)
+	for _, want := range []string{"Detected tech terms:", "node", "npm", "npx", "Suggested reads:", "[[npm-wrapper]]"} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("expected formatted context to contain %q:\n%s", want, text)
+		}
+	}
+}
+
+func TestBuildContextAddsSwaggerTechResultsForOpenAPIFile(t *testing.T) {
+	vaultRoot := t.TempDir()
+	writeFile(t, vaultRoot, "20-wiki/concepts/swagger-rules.md", `---
+title: Swagger Rules
+type: concept
+status: active
+created: 2026-05-23
+updated: 2026-05-23
+tags: [swagger, openapi]
+domain: api
+---
+
+# Swagger Rules
+
+OpenAPI and Swagger documentation conventions.
+`)
+	projectRoot := filepath.Join(t.TempDir(), "api-project")
+	writeFile(t, projectRoot, "openapi.yaml", "openapi: 3.0.0\ninfo:\n  title: API\n")
+
+	v, err := wiki.New(vaultRoot)
+	if err != nil {
+		t.Fatalf("wiki.New() error = %v", err)
+	}
+	result, err := BuildContext(v, Options{ProjectPath: projectRoot, Limit: 5})
+	if err != nil {
+		t.Fatalf("BuildContext() error = %v", err)
+	}
+	text := FormatAdditionalContext(result)
+	for _, want := range []string{"Detected tech terms:", "swagger", "openapi", "Suggested reads:", "[[swagger-rules]]"} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("expected formatted context to contain %q:\n%s", want, text)
+		}
+	}
+}
+
+func TestFormatAdditionalContextRendersConciseReadList(t *testing.T) {
+	ctx := Context{
+		Project: Project{
+			Path:      "/workspace/llm-wiki",
+			Name:      "@m16khb/llm-wiki",
+			Terms:     []string{"llm-wiki", "wiki"},
+			TechTerms: []string{"go", "mcp"},
+		},
+		Query: "go hook context",
+		Results: []wiki.SearchResult{
+			{
+				Slug:    "index",
+				Path:    "00-meta/index.md",
+				Title:   "Wiki Index",
+				Type:    "index",
+				Score:   99,
+				Snippet: "# Wiki Index\n\nHuge catalog noise should not appear in hook output.",
+			},
+			{
+				Slug:    "llm-wiki-hooks",
+				Path:    "20-wiki/concepts/llm-wiki-hooks.md",
+				Title:   "LLM Wiki Hooks",
+				Type:    "concept",
+				Score:   80,
+				Snippet: "# LLM Wiki Hooks\n\nLong raw snippets make hook output noisy.",
+			},
+		},
+		TechResults: []wiki.SearchResult{
+			{
+				Slug:    "go-conventions",
+				Path:    "20-wiki/concepts/go-conventions.md",
+				Title:   "Go Conventions",
+				Type:    "concept",
+				Score:   30,
+				Snippet: "# Go Conventions\n\nDetailed page body should be read explicitly, not injected.",
+			},
+			{
+				Slug:    "api-swagger-openapi-conventions",
+				Path:    "20-wiki/concepts/api-swagger-openapi-conventions.md",
+				Title:   "Swagger / OpenAPI Conventions",
+				Type:    "concept",
+				Tags:    []string{"go", "swagger", "openapi"},
+				Score:   28,
+				Snippet: "# Swagger / OpenAPI Conventions\n\nOnly include when requested or detected.",
+			},
+		},
+	}
+
+	text := FormatAdditionalContext(ctx)
+	for _, want := range []string{
+		"Suggested reads:",
+		"- [[llm-wiki-hooks]] `20-wiki/concepts/llm-wiki-hooks.md` — LLM Wiki Hooks",
+		"- [[go-conventions]] `20-wiki/concepts/go-conventions.md` — Go Conventions",
+		"Hook context is search-only; read listed pages before relying on details.",
+	} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("expected concise context to contain %q:\n%s", want, text)
+		}
+	}
+	for _, notWant := range []string{
+		"Relevant wiki pages:",
+		"Relevant tech wiki pages:",
+		"[[index]]",
+		"api-swagger-openapi-conventions",
+		"Huge catalog noise",
+		"Long raw snippets",
+		"Detailed page body",
+		"Only include when requested",
+	} {
+		if strings.Contains(text, notWant) {
+			t.Fatalf("did not expect noisy hook output %q:\n%s", notWant, text)
+		}
+	}
+}
+
+func TestFormatAdditionalContextUsesQueryTermsWithoutTechAllowlist(t *testing.T) {
+	ctx := Context{
+		Project: Project{
+			Path:  "/workspace/polyglot",
+			Name:  "polyglot",
+			Terms: []string{"polyglot"},
+		},
+		Query: "k8s rust c++",
+		Results: []wiki.SearchResult{
+			{Slug: "k8s-patterns", Path: "20-wiki/concepts/k8s-patterns.md", Title: "K8s Patterns", Type: "concept", Score: 30},
+			{Slug: "rust-async-stack", Path: "20-wiki/concepts/rust-async-stack.md", Title: "Rust Async Stack", Type: "concept", Score: 30},
+			{Slug: "cpp-guide", Path: "20-wiki/concepts/cpp-guide.md", Title: "C++ Guide", Type: "concept", Score: 30},
+			{Slug: "unrelated-api", Path: "20-wiki/concepts/unrelated-api.md", Title: "Unrelated API", Type: "concept", Tags: []string{"k8s"}, Score: 30},
+		},
+	}
+
+	text := FormatAdditionalContext(ctx)
+	for _, want := range []string{"[[k8s-patterns]]", "[[rust-async-stack]]", "[[cpp-guide]]"} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("expected query-matched page %q without a hard-coded tech allowlist:\n%s", want, text)
+		}
+	}
+	if strings.Contains(text, "[[unrelated-api]]") {
+		t.Fatalf("did not expect a page to match only through tags:\n%s", text)
+	}
+}
+
+func TestExtractCaptureBlock(t *testing.T) {
+	msg := "done\n\n```llm-wiki-capture\ntitle: Hook decision\ntags: hooks, codex\n---\nPersist this decision.\n```\n"
+	block, ok := ExtractCaptureBlock(msg)
+	if !ok {
+		t.Fatalf("expected capture block")
+	}
+	if block.Title != "Hook decision" || block.Body != "Persist this decision." || len(block.Tags) != 2 {
+		t.Fatalf("unexpected block: %+v", block)
+	}
+}
+
+func contains(values []string, want string) bool {
+	for _, v := range values {
+		if v == want {
+			return true
+		}
+	}
+	return false
+}
+
+func writeFile(t *testing.T, root, rel, content string) {
+	t.Helper()
+	abs := filepath.Join(root, filepath.FromSlash(rel))
+	if err := os.MkdirAll(filepath.Dir(abs), 0o755); err != nil {
+		t.Fatalf("mkdir %s: %v", rel, err)
+	}
+	if err := os.WriteFile(abs, []byte(content), 0o644); err != nil {
+		t.Fatalf("write %s: %v", rel, err)
+	}
+}