feat: Add analyze command #43

Author: TobiasGleiter

.gitignore

@@ -25,4 +25,8 @@ go.work.sum
 # env file
 .env
 
-/tmp
+/tmp
+
+# db
+analyze.db
+agent.db

cmd/cli/analyze.go

@@ -0,0 +1,108 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"flag"
+	"fmt"
+	"os"
+
+	"github.com/bit8bytes/gogantic/agents"
+	"github.com/bit8bytes/gogantic/llms/ollama"
+	"github.com/bit8bytes/gogantic/runner"
+	"github.com/bit8bytes/gogantic/stores/moderncsqlite"
+	"github.com/bit8bytes/gogantic/tools"
+)
+
+func analyze(ctx context.Context, db *sql.DB, wd string) error {
+	fs := flag.NewFlagSet("analyze", flag.ExitOnError)
+	model := fs.String("model", "gemma3:12b", "Ollama model to use")
+	numCtx := fs.Int("ctx", 16384, "Context window size")
+	verbose := fs.Bool("v", false, "Print thought messages as the agent reasons")
+	fs.Parse(os.Args[2:])
+
+	targetPath := fs.Arg(0)
+	if targetPath == "" {
+		return fmt.Errorf("usage: beago analyze [flags] <path>")
+	}
+
+	storage, err := moderncsqlite.New(ctx, db)
+	if err != nil {
+		return fmt.Errorf("storage init: %w", err)
+	}
+	defer storage.Close()
+
+	llm := ollama.New(ollama.Model{
+		Model:   *model,
+		Options: ollama.Options{NumCtx: *numCtx},
+		Stream:  false,
+		Format:  ollama.JSON,
+	})
+
+	// These goTools are specifically designed for Golang.
+	goTools := []agents.Tool{
+		tools.GitDiff{},
+		tools.ListDeclarations{},
+		tools.FindCalls{},
+		tools.FindUsages{},
+		tools.GetFunctionSignature{},
+		tools.GetStructFields{},
+		tools.RunGoBuild{},
+		tools.RunGoVet{},
+	}
+
+	agent, err := agents.NewReAct(ctx, llm, goTools, storage)
+	if err != nil {
+		return fmt.Errorf("agent init: %w", err)
+	}
+
+	task := fmt.Sprintf(`You are a Go code reviewer. Your job is to check if newly written code follows the same style and best practices as the rest of this codebase.
+
+Project root: %s
+Target file: %s
+
+Follow these steps:
+1. Call GitDiff with the target file path to get the exact changes.
+2. Call ListDeclarations with the project root to find comparable existing code.
+3. Call GetFunctionSignature on functions from the diff and on similar functions found in step 2.
+4. Compare them. Report ONLY concrete deviations in: error handling style, naming conventions, doc comments, receiver naming, function signature shape.
+
+Format your final answer exactly like this:
+
+REVIEW <target file>
+
+No issues found.
+
+Or if there are deviations:
+
+REVIEW <target file>
+
+ISSUE: <short title>
+  File:     <file>:<line>
+  Expected: <what the codebase does>
+  Found:    <what the new code does>
+
+Report only concrete deviations. No prose outside this format.`, wd, targetPath)
+
+	if err := agent.Task(ctx, task); err != nil {
+		return fmt.Errorf("task: %w", err)
+	}
+
+	r := runner.New(agent, *verbose)
+	if err := r.Run(ctx); err != nil {
+		return fmt.Errorf("run: %w", err)
+	}
+
+	finalAnswer, err := agent.Answer()
+	if errors.Is(err, agents.ErrNoFinalAnswer) {
+		fmt.Println("No final answer found")
+		return nil
+	}
+	if err != nil {
+		return fmt.Errorf("answer: %w", err)
+	}
+
+	fmt.Println(finalAnswer)
+	return nil
+}

cmd/cli/main.go

@@ -0,0 +1,72 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"os"
+	"time"
+)
+
+func main() {
+	if len(os.Args) < 2 {
+		fmt.Fprintln(os.Stderr, "usage: beago <command> [args]")
+		os.Exit(1)
+	}
+
+	wd, err := os.Getwd()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "error: %v\n", err)
+		os.Exit(1)
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*240)
+	defer cancel()
+
+	db, err := setupDatabase(ctx, "analyze.db")
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "database setup: %v\n", err)
+		os.Exit(1)
+	}
+	defer db.Close()
+
+	if err := migrateDatabase(ctx, db); err != nil {
+		fmt.Fprintf(os.Stderr, "database migration: %v\n", err)
+		os.Exit(1)
+	}
+
+	switch os.Args[1] {
+	case "analyze":
+		if err := analyze(ctx, db, wd); err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			os.Exit(1)
+		}
+	default:
+		fmt.Fprintf(os.Stderr, "unknown command: %s\n", os.Args[1])
+		os.Exit(1)
+	}
+}
+
+func setupDatabase(ctx context.Context, path string) (*sql.DB, error) {
+	db, err := sql.Open("sqlite", path)
+	if err != nil {
+		return nil, err
+	}
+
+	if err := db.PingContext(ctx); err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}
+
+func migrateDatabase(ctx context.Context, db *sql.DB) error {
+	_, err := db.ExecContext(ctx, `
+		CREATE TABLE IF NOT EXISTS messages (
+			id         INTEGER PRIMARY KEY AUTOINCREMENT,
+			role       TEXT NOT NULL,
+			content    TEXT NOT NULL,
+			created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+		);`)
+	return err
+}