b427041097
Replace GetMyUserInfo() with ServerVersion() in VerifyConnection() and HealthCheck(). The /api/v1/version endpoint requires no authentication, so repo/org-scoped tokens (which Forgejo explicitly denies the read:user scope) no longer crash the server at startup. Thanks to @synath for the fix.
137 lines
3.6 KiB
Go
137 lines
3.6 KiB
Go
package forgejo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"codeberg.org/goern/forgejo-mcp/v2/pkg/flag"
|
|
"codeberg.org/goern/forgejo-mcp/v2/pkg/log"
|
|
|
|
"codeberg.org/mvdkleijn/forgejo-sdk/forgejo/v3"
|
|
)
|
|
|
|
var (
|
|
client *forgejo.Client
|
|
clientOnce sync.Once
|
|
)
|
|
|
|
// Client returns a Forgejo client configured to connect to a Forgejo instance
|
|
// We use the standard Forgejo SDK to ensure API compatibility
|
|
func Client() *forgejo.Client {
|
|
clientOnce.Do(func() {
|
|
if client == nil {
|
|
// Use configured user agent or default to forgejo-mcp/<version>
|
|
userAgent := flag.UserAgent
|
|
if userAgent == "" {
|
|
userAgent = "forgejo-mcp/" + flag.Version
|
|
}
|
|
|
|
c, err := forgejo.NewClient(flag.URL,
|
|
forgejo.SetToken(flag.Token),
|
|
forgejo.SetUserAgent(userAgent),
|
|
)
|
|
if err != nil {
|
|
log.Error("Failed to create Forgejo client",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.ErrorField(err),
|
|
)
|
|
log.Fatalf("create forgejo client err: %v", err)
|
|
}
|
|
client = c
|
|
log.Info("Successfully created Forgejo client",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.BoolField("token_configured", flag.Token != ""),
|
|
log.StringField("user_agent", userAgent),
|
|
)
|
|
}
|
|
})
|
|
return client
|
|
}
|
|
|
|
// GetBaseURL returns the base URL of the Forgejo instance.
|
|
func GetBaseURL() string {
|
|
return flag.URL
|
|
}
|
|
|
|
// VerifyConnection attempts to get basic information to verify
|
|
// that the client is properly connected.
|
|
// Uses the /version endpoint (no auth required) so that tokens scoped
|
|
// only to repo/issue — e.g. organisation tokens — are not rejected.
|
|
func VerifyConnection() error {
|
|
start := time.Now()
|
|
|
|
log.Debug("Starting connection verification",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
)
|
|
|
|
version, resp, err := Client().ServerVersion()
|
|
duration := time.Since(start)
|
|
|
|
if err != nil {
|
|
log.Error("Connection verification failed",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.DurationField("duration", duration),
|
|
log.ErrorField(err),
|
|
)
|
|
return fmt.Errorf("failed to connect to Forgejo instance at %s: %v", flag.URL, err)
|
|
}
|
|
|
|
log.Info("Connection verification successful",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.DurationField("duration", duration),
|
|
log.StringField("server_version", version),
|
|
log.IntField("response_status", resp.StatusCode),
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
// HealthCheck performs a lightweight health check
|
|
func HealthCheck() error {
|
|
start := time.Now()
|
|
|
|
log.Debug("Starting health check")
|
|
|
|
version, resp, err := Client().ServerVersion()
|
|
duration := time.Since(start)
|
|
|
|
if err != nil {
|
|
log.Error("Health check failed",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.DurationField("duration", duration),
|
|
log.ErrorField(err),
|
|
)
|
|
return fmt.Errorf("health check failed: %v", err)
|
|
}
|
|
|
|
log.Debug("Health check successful",
|
|
log.SanitizedURLField("url", flag.URL),
|
|
log.DurationField("duration", duration),
|
|
log.StringField("server_version", version),
|
|
log.IntField("response_status", resp.StatusCode),
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
// LogAPICall logs API call information with timing
|
|
func LogAPICall(ctx context.Context, method, endpoint string, duration time.Duration, statusCode int, err error) {
|
|
if err != nil {
|
|
log.ErrorCtx(ctx, "API call failed",
|
|
log.StringField("method", method),
|
|
log.StringField("endpoint", endpoint),
|
|
log.DurationField("duration", duration),
|
|
log.IntField("status_code", statusCode),
|
|
log.ErrorField(err),
|
|
)
|
|
} else {
|
|
log.DebugCtx(ctx, "API call completed",
|
|
log.StringField("method", method),
|
|
log.StringField("endpoint", endpoint),
|
|
log.DurationField("duration", duration),
|
|
log.IntField("status_code", statusCode),
|
|
)
|
|
}
|
|
}
|