Initial scaffold: API service, K8s operator, and CRDs
Forgebot is a K8s operator + API service for dispatching AI agent jobs from git forge commands. Includes: - CRDs: AgentPool, AgentTask, ProviderQueue, RepositoryBinding - API server with webhook handler, task queue, and comment proxy - Operator controllers for task scheduling and job management - Gitea provider with webhook parsing and signature verification - PostgreSQL database with auto-migration - Woodpecker CI pipelines and multi-stage Dockerfiles
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
|
||||
"git.unkin.net/unkin/forgebot/internal/database"
|
||||
)
|
||||
|
||||
type HealthHandler struct {
|
||||
db *database.DB
|
||||
}
|
||||
|
||||
func NewHealthHandler(db *database.DB) *HealthHandler {
|
||||
return &HealthHandler{db: db}
|
||||
}
|
||||
|
||||
func (h *HealthHandler) Health(w http.ResponseWriter, r *http.Request) {
|
||||
status := "ok"
|
||||
code := http.StatusOK
|
||||
|
||||
if !h.db.Healthy(r.Context()) {
|
||||
status = "degraded"
|
||||
code = http.StatusServiceUnavailable
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(code)
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": status})
|
||||
}
|
||||
@@ -0,0 +1,136 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/go-chi/chi/v5"
|
||||
|
||||
"git.unkin.net/unkin/forgebot/internal/database"
|
||||
"git.unkin.net/unkin/forgebot/internal/provider/gitea"
|
||||
"git.unkin.net/unkin/forgebot/pkg/models"
|
||||
)
|
||||
|
||||
type TasksHandler struct {
|
||||
db *database.DB
|
||||
provider *gitea.Client
|
||||
}
|
||||
|
||||
func NewTasksHandler(db *database.DB, provider *gitea.Client) *TasksHandler {
|
||||
return &TasksHandler{db: db, provider: provider}
|
||||
}
|
||||
|
||||
func (h *TasksHandler) List(w http.ResponseWriter, r *http.Request) {
|
||||
status := r.URL.Query().Get("status")
|
||||
repository := r.URL.Query().Get("repository")
|
||||
|
||||
tasks, err := h.db.ListTasks(r.Context(), status, repository)
|
||||
if err != nil {
|
||||
slog.Error("failed to list tasks", "error", err)
|
||||
http.Error(w, "internal error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
if tasks == nil {
|
||||
tasks = []models.Task{}
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(tasks)
|
||||
}
|
||||
|
||||
func (h *TasksHandler) Create(w http.ResponseWriter, r *http.Request) {
|
||||
var req models.CreateTaskRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid request body", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
if req.Command == "" || req.Repository == "" {
|
||||
http.Error(w, "command and repository are required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
if !models.ValidCommands[req.Command] {
|
||||
http.Error(w, "invalid command", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
task, err := h.db.CreateTask(r.Context(), req)
|
||||
if err != nil {
|
||||
slog.Error("failed to create task", "error", err)
|
||||
http.Error(w, "internal error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
json.NewEncoder(w).Encode(task)
|
||||
}
|
||||
|
||||
func (h *TasksHandler) Get(w http.ResponseWriter, r *http.Request) {
|
||||
id := chi.URLParam(r, "id")
|
||||
|
||||
task, err := h.db.GetTask(r.Context(), id)
|
||||
if err != nil {
|
||||
http.Error(w, "task not found", http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(task)
|
||||
}
|
||||
|
||||
func (h *TasksHandler) UpdateStatus(w http.ResponseWriter, r *http.Request) {
|
||||
id := chi.URLParam(r, "id")
|
||||
|
||||
var req models.UpdateTaskRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid request body", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.db.UpdateTaskStatus(r.Context(), id, req); err != nil {
|
||||
slog.Error("failed to update task", "error", err, "id", id)
|
||||
http.Error(w, "internal error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
|
||||
func (h *TasksHandler) PostComment(w http.ResponseWriter, r *http.Request) {
|
||||
id := chi.URLParam(r, "id")
|
||||
|
||||
var req models.CommentRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid request body", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
task, err := h.db.GetTask(r.Context(), id)
|
||||
if err != nil {
|
||||
http.Error(w, "task not found", http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
|
||||
parts := strings.SplitN(task.Repository, "/", 2)
|
||||
if len(parts) != 2 {
|
||||
http.Error(w, "invalid repository format", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
issueNum := task.IssueNumber
|
||||
if task.PRNumber > 0 {
|
||||
issueNum = task.PRNumber
|
||||
}
|
||||
|
||||
if err := h.provider.PostComment(parts[0], parts[1], issueNum, req.Body); err != nil {
|
||||
slog.Error("failed to post comment", "error", err, "task", id)
|
||||
http.Error(w, "failed to post comment", http.StatusBadGateway)
|
||||
return
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
@@ -0,0 +1,91 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"io"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"git.unkin.net/unkin/forgebot/internal/database"
|
||||
"git.unkin.net/unkin/forgebot/internal/provider/gitea"
|
||||
"git.unkin.net/unkin/forgebot/pkg/models"
|
||||
)
|
||||
|
||||
type WebhookHandler struct {
|
||||
db *database.DB
|
||||
provider *gitea.Client
|
||||
webhookSecret string
|
||||
}
|
||||
|
||||
func NewWebhookHandler(db *database.DB, provider *gitea.Client, secret string) *WebhookHandler {
|
||||
return &WebhookHandler{
|
||||
db: db,
|
||||
provider: provider,
|
||||
webhookSecret: secret,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *WebhookHandler) HandleGitea(w http.ResponseWriter, r *http.Request) {
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
http.Error(w, "failed to read body", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
signature := r.Header.Get("X-Gitea-Signature")
|
||||
if !gitea.VerifySignature(body, h.webhookSecret, signature) {
|
||||
http.Error(w, "invalid signature", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
event, err := h.provider.ParseWebhook(body, h.webhookSecret)
|
||||
if err != nil {
|
||||
slog.Error("failed to parse webhook", "error", err)
|
||||
http.Error(w, "failed to parse webhook", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if event == nil {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
|
||||
commands := models.ParseCommands(event.Body)
|
||||
if len(commands) == 0 {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
|
||||
parts := strings.SplitN(event.Repository, "/", 2)
|
||||
if len(parts) != 2 {
|
||||
http.Error(w, "invalid repository format", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
for _, cmd := range commands {
|
||||
task, err := h.db.CreateTask(r.Context(), models.CreateTaskRequest{
|
||||
Command: cmd.Name,
|
||||
Repository: event.Repository,
|
||||
Ref: event.Ref,
|
||||
IssueNumber: event.IssueNum,
|
||||
PRNumber: event.PRNum,
|
||||
CommentID: event.CommentID,
|
||||
Body: cmd.Args,
|
||||
Author: event.Author,
|
||||
})
|
||||
if err != nil {
|
||||
slog.Error("failed to create task", "error", err, "command", cmd.Name)
|
||||
continue
|
||||
}
|
||||
|
||||
slog.Info("task created from webhook",
|
||||
"id", task.ID,
|
||||
"command", cmd.Name,
|
||||
"repository", event.Repository,
|
||||
"author", event.Author,
|
||||
)
|
||||
|
||||
h.provider.AddReaction(parts[0], parts[1], event.CommentID, "eyes")
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusAccepted)
|
||||
}
|
||||
Reference in New Issue
Block a user