gitea-mcp/cmd/gitea-mcp/main.go

package main

import (
	"context"
	"log/slog"
	"net/http"
	"os"
	"strings"

	chassisauth "git.d-ma.be/mathias/mcp-chassis/auth"

	"git.d-ma.be/mathias/gitea-mcp/internal/allowlist"
	"git.d-ma.be/mathias/gitea-mcp/internal/auth"
	"git.d-ma.be/mathias/gitea-mcp/internal/config"
	"git.d-ma.be/mathias/gitea-mcp/internal/gitea"
	"git.d-ma.be/mathias/gitea-mcp/internal/mcp"
	"git.d-ma.be/mathias/gitea-mcp/internal/registry"
	"git.d-ma.be/mathias/gitea-mcp/internal/tools"
)

// version is the build version, overridable via -ldflags "-X main.version=<tag>".
// Defaults to "dev" for local / un-stamped builds (was a hardcoded, drifting
// "0.1.0" — it now tells the truth instead of a stale literal).
var version = "dev"

func main() {
	logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
	// Route the chassis's package-level slog (auth audit logs, gitea-mcp#9) through
	// the same structured handler as the rest of the server.
	slog.SetDefault(logger)

	cfg, err := config.Load()
	if err != nil {
		logger.Error("load config", "err", err)
		os.Exit(1)
	}

	ctx := context.Background()

	// Build the trusted-issuer list. Authentik (DEX_ISSUER_URL) for interactive/
	// web clients; the in-cluster k8s OIDC issuer for ServiceAccount tokens
	// (ADR-0011 D1). The k8s issuer is ADDITIVE and best-effort: if its client
	// can't be built (not in a pod) or its discovery is unreachable at startup,
	// it is dropped so existing Authentik/static auth is never taken down.
	var issuers []chassisauth.IssuerConfig
	if cfg.DexIssuerURL != "" {
		issuers = append(issuers, chassisauth.IssuerConfig{IssuerURL: cfg.DexIssuerURL, Audience: cfg.MCPAudience})
	}
	if cfg.K8sIssuerURL != "" {
		if client, cerr := k8sOIDCClient(); cerr != nil {
			logger.Warn("k8s OIDC client init failed; SA-token auth disabled (other auth unaffected)", "err", cerr)
		} else {
			issuers = append(issuers, chassisauth.IssuerConfig{IssuerURL: cfg.K8sIssuerURL, Audience: cfg.K8sAudience, HTTPClient: client})
		}
	}

	// No retry-without-the-k8s-issuer dance any more, and no "JWT auth degraded"
	// warning to act on. Since mcp-chassis v0.6.0 an unreachable issuer does not
	// fail construction: each issuer resolves independently and lazily, so one
	// down issuer degrades only itself and recovers on its own. The fallback this
	// replaced existed precisely because a single unreachable issuer used to take
	// the whole validator with it.
	jwtValidator, err := chassisauth.NewMultiJWTValidator(ctx, issuers)
	if err != nil {
		// Only a configuration error can reach here now.
		logger.Error("jwt validator config invalid", "err", err)
		os.Exit(1)
	}

	giteaClient := gitea.NewClient(cfg.GiteaBaseURL, cfg.DefaultToken)
	ownerAllow := allowlist.New(cfg.AllowedOwners)

	reg := registry.New()
	tools.RegisterAll(reg, giteaClient, ownerAllow, cfg.GiteaBaseURL, "mathias", "template-go-web")

	mcpSrv := mcp.NewServer(mcp.ServerOptions{
		Registry: reg,
		Sessions: mcp.NewSessionStore(),
	})

	// resourceMetadataURL is only emitted in the WWW-Authenticate challenge
	// when both MCPResourceURL and a Dex issuer are wired; empty disables
	// the challenge so static-only clients aren't pushed into OAuth discovery.
	var resourceMetadataURL string
	if cfg.MCPResourceURL != "" && cfg.DexIssuerURL != "" {
		resourceMetadataURL = strings.TrimRight(cfg.MCPResourceURL, "/") + "/.well-known/oauth-protected-resource"
	}

	mux := http.NewServeMux()
	mux.Handle("/mcp", mcp.OriginAllowlist(cfg.OriginAllowlist)(
		auth.PassthroughMiddleware(giteaClient, mcpSrv,
			chassisauth.BearerMiddleware(cfg.StaticToken, jwtValidator, "gitea", resourceMetadataURL,
				auth.CallerMiddleware(logger, mcpSrv),
			),
		),
	))
	// Readiness is evaluated per request, and evaluating it is what retries a
	// pending issuer — so the kubelet's probe cadence is the recovery loop.
	dexConfigured := cfg.DexIssuerURL != ""
	mux.Handle("/healthz", newHealthzHandler(dexConfigured, jwtValidator.Ready))
	mux.Handle("/readyz", newReadyzHandler(dexConfigured, jwtValidator.Ready))
	if cfg.DexIssuerURL != "" {
		mux.HandleFunc("GET /.well-known/oauth-protected-resource",
			chassisauth.ProtectedResourceHandler(cfg.MCPResourceURL, cfg.DexIssuerURL))
	}

	addr := ":" + cfg.Port
	logger.Info("gitea-mcp starting", "addr", addr, "version", version)
	if err := http.ListenAndServe(addr, mux); err != nil {
		logger.Error("server stopped", "err", err)
		os.Exit(1)
	}
}