schreifuchs/pierre-bot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: gitea client

Browse Source
This commit is contained in:
schreifuchs
2026-02-12 20:58:55 +01:00
parent 8583ab48ce
commit 378d008a91
9 changed files with 225 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
cmd/pierre/main.go
View File

@@ -9,18 +9,26 @@ import (
"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/gitadapters/gitea"
"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"`
BaseURL string `help:"Bitbucket Base URL (e.g. https://bitbucket.example.com)" 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 GiteaConfig struct {
BaseURL string `help:"Gitea Base URL (e.g. https://gitea.com)" env:"GITEA_URL"`
Token string `help:"API Token" env:"GITEA_TOKEN"`
}
type RepoArgs struct {
Owner string `arg:"" help:"Project Key or Owner" env:"PIERRE_OWNER"`
Repo string `arg:"" help:"Repository Slug" env:"PIERRE_REPO"`
PRID int `arg:"" help:"Pull Request ID" name:"pr"`
}
type LLMConfig struct {
@@ -31,9 +39,12 @@ type LLMConfig struct {
}
type Config struct {
Bitbucket BitbucketConfig `embed:"" prefix:"bitbucket-"`
LLM LLMConfig `embed:"" prefix:"llm-"`
Config kong.ConfigFlag `help:"Path to a YAML config file"`
GitProvider string `help:"Git provider (bitbucket or gitea)" env:"GIT_PROVIDER"`
Bitbucket BitbucketConfig `embed:"" prefix:"bitbucket-"`
Gitea GiteaConfig `embed:"" prefix:"gitea-"`
Repo RepoArgs `embed:""`
LLM LLMConfig `embed:"" prefix:"llm-"`
Config kong.ConfigFlag `help:"Path to a YAML config file"`
}
func main() {
@@ -46,25 +57,54 @@ func main() {
defaultConfig := filepath.Join(home, ".config", "pierre", "config.yaml")
// Parse flags, env vars, and config files
kong.Parse(cfg,
ctx := kong.Parse(cfg,
kong.Name("pierre"),
kong.Description("AI-powered Pull Request reviewer for Bitbucket"),
kong.Description("AI-powered Pull Request reviewer"),
kong.UsageOnError(),
kong.Configuration(kongyaml.Loader, "config.yaml", defaultConfig),
)
// Initialize Bitbucket Adapter
bitbucket := gitadapters.NewBitbucket(cfg.Bitbucket.BaseURL, cfg.Bitbucket.Token)
// Auto-detect provider
provider := cfg.GitProvider
if provider == "" {
if cfg.Bitbucket.BaseURL != "" && cfg.Gitea.BaseURL == "" {
provider = "bitbucket"
} else if cfg.Gitea.BaseURL != "" && cfg.Bitbucket.BaseURL == "" {
provider = "gitea"
} else if cfg.Bitbucket.BaseURL != "" && cfg.Gitea.BaseURL != "" {
log.Fatal("Multiple git providers configured. Please specify one using --git-provider.")
} else {
log.Fatal("No git provider configured. Please provide Bitbucket or Gitea configuration.")
}
}
// Fetch Diff using positional args
diff, err := bitbucket.GetDiff(cfg.Bitbucket.Project, cfg.Bitbucket.Repo, cfg.Bitbucket.PRID)
var adapter gitadapters.Adapter
switch provider {
case "bitbucket":
if cfg.Bitbucket.BaseURL == "" {
log.Fatal("Bitbucket Base URL is required when using bitbucket provider.")
}
adapter = gitadapters.NewBitbucket(cfg.Bitbucket.BaseURL, cfg.Bitbucket.Token)
case "gitea":
if cfg.Gitea.BaseURL == "" {
log.Fatal("Gitea Base URL is required when using gitea provider.")
}
adapter, err = gitea.New(cfg.Gitea.BaseURL, cfg.Gitea.Token)
if err != nil {
log.Fatalf("Error initializing Gitea adapter: %v", err)
}
default:
log.Fatalf("Unknown git provider: %s", provider)
}
// Fetch Diff using positional args from shared RepoArgs
diff, err := adapter.GetDiff(cfg.Repo.Owner, cfg.Repo.Repo, cfg.Repo.PRID)
if err != nil {
log.Fatalf("Error fetching diff: %v", err)
}
// Initialize AI Adapter
var ai pierre.ChatAdapter
var ai chatter.ChatAdapter
switch cfg.LLM.Provider {
case "gemini":
@@ -91,4 +131,6 @@ func main() {
fmt.Printf("File: %s\nLine: %d\nMessage: %s\n%s\n",
c.File, c.Line, c.Message, "---")
}
_ = ctx
}