init

2026-02-12 16:02:38 +01:00
commit 3d4d908d54
14 changed files with 885 additions and 0 deletions
--- a/cmd/pierre/main.go
+++ b/cmd/pierre/main.go
@@ -0,0 +1,86 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+
+	"bitbucket.bit.admin.ch/scm/~u80859501/pierre-bot/internal/chatter"
+	"bitbucket.bit.admin.ch/scm/~u80859501/pierre-bot/internal/gitadapters"
+	"bitbucket.bit.admin.ch/scm/~u80859501/pierre-bot/internal/pierre"
+	"github.com/alecthomas/kong"
+	kongyaml "github.com/alecthomas/kong-yaml"
+)
+
+type BitbucketConfig struct {
+	BaseURL string `help:"Bitbucket Base URL (e.g. https://bitbucket.example.com)" required:"" env:"BITBUCKET_URL"`
+	Token   string `help:"Bearer Token" env:"BITBUCKET_TOKEN"`
+	// Positional arguments
+	Project string `arg:"" help:"Project Key (e.g. PROJ)" env:"BITBUCKET_PROJECT"`
+	Repo    string `arg:"" help:"Repository Slug" env:"BITBUCKET_REPO"`
+	PRID    int    `arg:"" help:"Pull Request ID" name:"pr"`
+}
+
+type LLMConfig struct {
+	Provider string `help:"Provider for llm (ollama or gemini)" required:"" env:"LLM_PROVIDER"`
+	Endpoint string `help:"Endpoint for provider (only for ollama)" env:"LLM_ENDPOINT"`
+	APIKey   string `help:"APIKey for provider" env:"LLM_API_KEY"`
+	Model    string `help:"Model to use" env:"LLM_MODEL"`
+}
+
+type Config struct {
+	Bitbucket BitbucketConfig `embed:"" prefix:"bitbucket-"`
+	LLM       LLMConfig       `embed:"" prefix:"llm-"`
+	Config    kong.ConfigFlag `help:"Path to a YAML config file"`
+}
+
+func main() {
+	cfg := &Config{}
+
+	// Parse flags, env vars, and config files
+	kong.Parse(cfg,
+		kong.Name("pierre"),
+		kong.Description("AI-powered Pull Request reviewer for Bitbucket"),
+		kong.UsageOnError(),
+		kong.Configuration(kongyaml.Loader, "config.yaml", "~/.config/pierre/config.yaml"),
+	)
+
+	// Initialize Bitbucket Adapter
+	bitbucket := gitadapters.NewBitbucket(cfg.Bitbucket.BaseURL, cfg.Bitbucket.Token)
+
+	// Fetch Diff using positional args
+	diff, err := bitbucket.GetDiff(cfg.Bitbucket.Project, cfg.Bitbucket.Repo, cfg.Bitbucket.PRID)
+	if err != nil {
+		log.Fatalf("Error fetching diff: %v", err)
+	}
+
+	// Initialize AI Adapter
+
+	var ai pierre.ChatAdapter
+
+	switch cfg.LLM.Provider {
+	case "gemini":
+		ai, err = chatter.NewGeminiAdapter(context.Background(), cfg.LLM.APIKey, cfg.LLM.Model)
+	case "ollama":
+		ai, err = chatter.NewOllamaAdapter(cfg.LLM.Endpoint, cfg.LLM.Model)
+	default:
+		log.Fatalf("%s is not a valid llm provider", cfg.LLM.Provider)
+	}
+
+	if err != nil {
+		log.Fatalf("Error initializing AI: %v", err)
+	}
+
+	// Run Logic
+	comments, err := pierre.JudgePR(context.Background(), ai, diff)
+	if err != nil {
+		log.Fatalf("Error judging PR: %v", err)
+	}
+
+	fmt.Printf("Analysis complete. Found %d issues.\n---\n", len(comments))
+
+	for _, c := range comments {
+		fmt.Printf("File: %s\nLine: %d\nMessage: %s\n%s\n",
+			c.File, c.Line, c.Message, "---")
+	}
+}