internal/analyze/analyze_test.go

@@ -0,0 +1,50 @@
+package analyze
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDiscoverSourceFiles(t *testing.T) {
+	tmpDir := t.TempDir()
+
+	// Create some source files
+	if err := os.WriteFile(filepath.Join(tmpDir, "main.go"), []byte("package main"), 0644); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(tmpDir, "helper.go"), []byte("package main"), 0644); err != nil {
+		t.Fatal(err)
+	}
+	// Create a non-source file (should be ignored)
+	if err := os.WriteFile(filepath.Join(tmpDir, "notes.txt"), []byte("notes"), 0644); err != nil {
+		t.Fatal(err)
+	}
+
+	files, err := DiscoverSourceFiles(tmpDir)
+	assert.NoError(t, err)
+	assert.GreaterOrEqual(t, len(files), 2)
+
+	// Verify .txt was not included
+	for _, f := range files {
+		assert.False(t, strings.HasSuffix(f, ".txt"))
+	}
+}
+
+func TestBuildProjectContext(t *testing.T) {
+	tmpDir := t.TempDir()
+
+	files := []string{
+		filepath.Join(tmpDir, "main.go"),
+		filepath.Join(tmpDir, "cmd", "root.go"),
+	}
+
+	ctx := BuildProjectContext(tmpDir, files)
+
+	assert.Contains(t, ctx, "Project Root:")
+	assert.Contains(t, ctx, "Total source files discovered: 2")
+	assert.Contains(t, ctx, "Key files")
+}

internal/docs/docs_test.go

@@ -0,0 +1,30 @@
+package docs
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBuildDocsMessages(t *testing.T) {
+	tests := []struct {
+		language   string
+		code       string
+		styleCheck string
+	}{
+		{language: "Go", code: "func Foo() {}", styleCheck: "godoc"},
+		{language: "Python", code: "def foo(): pass", styleCheck: "PEP 257"},
+		{language: "JavaScript", code: "function foo() {}", styleCheck: "JSDoc"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.language, func(t *testing.T) {
+			msgs := BuildDocsMessages(tt.language, tt.code)
+
+			assert.Len(t, msgs, 2)
+			assert.Equal(t, "system", msgs[0]["role"])
+			assert.Equal(t, "user", msgs[1]["role"])
+			assert.Contains(t, msgs[0]["content"], tt.styleCheck)
+		})
+	}
+}

internal/mcp/server.go

@@ -362,33 +362,56 @@ func (s *Server) registerResources() {
 	logger.Info("MCP resources registered", "count", 3)
 }
 
-// Placeholder handlers for resources (to be implemented)
+// handleRecipesLocal returns local recipes as a resource
 func (s *Server) handleRecipesLocal(ctx context.Context, req mcp.ReadResourceRequest) ([]mcp.ResourceContents, error) {
+	recipes, err := recipe.ListRecipes(".")
+	if err != nil {
+		recipes = []string{} // fallback to empty list
+	}
+
+	json := fmt.Sprintf(`{"type": "local", "recipes": %v}`, recipes)
+
 	return []mcp.ResourceContents{
 		mcp.TextResourceContents{
 			URI:      "recipes://local",
 			MIMEType: "application/json",
-			Text:     `{"status": "placeholder - local recipes not yet implemented"}`,
+			Text:     json,
 		},
 	}, nil
 }
 
+// handleRecipesGlobal returns global recipes as a resource
 func (s *Server) handleRecipesGlobal(ctx context.Context, req mcp.ReadResourceRequest) ([]mcp.ResourceContents, error) {
+	home, _ := os.UserHomeDir()
+	globalPath := filepath.Join(home, ".config", "grokkit", "recipes")
+
+	recipes, err := recipe.ListRecipes(globalPath)
+	if err != nil {
+		recipes = []string{} // fallback to empty list
+	}
+
+	json := fmt.Sprintf(`{"type": "global", "recipes": %v}`, recipes)
+
 	return []mcp.ResourceContents{
 		mcp.TextResourceContents{
 			URI:      "recipes://global",
 			MIMEType: "application/json",
-			Text:     `{"status": "placeholder - global recipes not yet implemented"}`,
+			Text:     json,
 		},
 	}, nil
 }
 
+// handlePrompts returns available language prompts as a resource
 func (s *Server) handlePrompts(ctx context.Context, req mcp.ReadResourceRequest) ([]mcp.ResourceContents, error) {
+	promptsList := prompts.ListAvailablePrompts()
+
+	json := fmt.Sprintf(`{"type": "prompts", "prompts": %v}`, promptsList)
+
 	return []mcp.ResourceContents{
 		mcp.TextResourceContents{
 			URI:      "prompts://language",
-			MIMEType: "text/markdown",
-			Text:     `# Language Prompts\n\nPlaceholder - language prompts not yet implemented.`,
+			MIMEType: "application/json",
+			Text:     json,
 		},
 	}, nil
 }

internal/mcp/server_test.go

@@ -0,0 +1,78 @@
+package mcp
+
+import (
+	"context"
+	"testing"
+
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewServer(t *testing.T) {
+	srv, err := NewServer()
+	assert.NoError(t, err)
+	assert.NotNil(t, srv)
+	assert.NotNil(t, srv.mcpServer)
+	assert.NotNil(t, srv.grokClient)
+}
+
+func TestServer_RegistersTools(t *testing.T) {
+	srv, err := NewServer()
+	assert.NoError(t, err)
+
+	// We can't easily inspect registered tools without exposing internals,
+	// so we just verify server creation succeeds with tools.
+	assert.NotNil(t, srv)
+}
+
+func TestHandleLintCode(t *testing.T) {
+	srv, _ := NewServer()
+
+	// Use a file that exists and is lintable
+	req := mockCallToolRequest(map[string]any{
+		"file_path": "main.go",
+	})
+
+	result, err := srv.handleLintCode(context.Background(), req)
+	assert.NoError(t, err)
+	assert.NotNil(t, result)
+	// The test may fail if no linter is installed, so we just check it doesn't error
+	assert.NotNil(t, result.Content)
+}
+
+func TestHandleAnalyzeCode(t *testing.T) {
+	srv, _ := NewServer()
+
+	req := mockCallToolRequest(map[string]any{
+		"dir": ".",
+	})
+
+	result, err := srv.handleAnalyzeCode(context.Background(), req)
+	assert.NoError(t, err)
+	assert.NotNil(t, result)
+	// The test may fail if prompt files are missing, so we just check it doesn't error
+	assert.NotNil(t, result.Content)
+}
+
+func TestHandleGenerateDocs(t *testing.T) {
+	srv, _ := NewServer()
+
+	req := mockCallToolRequest(map[string]any{
+		"file_path": "main.go",
+	})
+
+	result, err := srv.handleGenerateDocs(context.Background(), req)
+	assert.NoError(t, err)
+	assert.NotNil(t, result)
+	// The test may fail if the file is not readable in test env, so we just check it doesn't error
+	assert.NotNil(t, result.Content)
+}
+
+// Helper to create mock CallToolRequest
+func mockCallToolRequest(args map[string]any) mcp.CallToolRequest {
+	return mcp.CallToolRequest{
+		Params: mcp.CallToolParams{
+			Arguments: args,
+		},
+	}
+}

internal/prompts/list.go

@@ -0,0 +1,53 @@
+package prompts
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// ListAvailablePrompts returns a list of available language prompts.
+// It scans both project-local and global prompt directories.
+func ListAvailablePrompts() []string {
+	var prompts []string
+
+	// Check project-local prompts
+	localDir := ".grokkit/prompts"
+	if entries, err := os.ReadDir(localDir); err == nil {
+		for _, entry := range entries {
+			if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".md") {
+				name := strings.TrimSuffix(entry.Name(), ".md")
+				prompts = append(prompts, name)
+			}
+		}
+	}
+
+	// Check global prompts
+	home := os.Getenv("HOME")
+	if home == "" {
+		home = os.Getenv("USERPROFILE")
+	}
+	if home != "" {
+		globalDir := filepath.Join(home, ".config", "grokkit", "prompts")
+		if entries, err := os.ReadDir(globalDir); err == nil {
+			for _, entry := range entries {
+				if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".md") {
+					name := strings.TrimSuffix(entry.Name(), ".md")
+					// Avoid duplicates
+					found := false
+					for _, p := range prompts {
+						if p == name {
+							found = true
+							break
+						}
+					}
+					if !found {
+						prompts = append(prompts, name)
+					}
+				}
+			}
+		}
+	}
+
+	return prompts
+}

internal/recipe/list.go

@@ -0,0 +1,45 @@
+package recipe
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// ListRecipes returns a list of recipe names in the given directory.
+// It looks for .md files that contain recipe frontmatter.
+func ListRecipes(dir string) ([]string, error) {
+	if dir == "" {
+		dir = "."
+	}
+
+	var recipes []string
+
+	err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if info.IsDir() {
+			// Skip hidden directories and common ignore dirs
+			name := info.Name()
+			if strings.HasPrefix(name, ".") && name != "." || name == "node_modules" || name == "vendor" {
+				return filepath.SkipDir
+			}
+			return nil
+		}
+
+		if strings.HasSuffix(strings.ToLower(info.Name()), ".md") {
+			// Simple check for recipe files (contains "Objective:" or "Instructions:")
+			data, err := os.ReadFile(path)
+			if err == nil && (strings.Contains(string(data), "**Objective:**") || strings.Contains(string(data), "Objective:")) {
+				name := strings.TrimSuffix(info.Name(), ".md")
+				recipes = append(recipes, name)
+			}
+		}
+
+		return nil
+	})
+
+	return recipes, err
+}

internal/testgen/testgen_test.go

@@ -0,0 +1,31 @@
+package testgen
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetTestPrompt(t *testing.T) {
+	tests := []struct {
+		lang     string
+		expected string
+	}{
+		{"Go", "expert Go test writer"},
+		{"Python", "pytest expert"},
+		{"C", "Check framework"},
+		{"C++", "Google Test"},
+		{"Unknown", ""},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.lang, func(t *testing.T) {
+			prompt := GetTestPrompt(tt.lang)
+			if tt.expected == "" {
+				assert.Empty(t, prompt)
+			} else {
+				assert.Contains(t, prompt, tt.expected)
+			}
+		})
+	}
+}