forgejo-mcp/cmd/cmd.go

package cmd

import (
	"context"
	"flag"
	"fmt"
	"net/url"
	"os"

	"codeberg.org/goern/forgejo-mcp/v2/operation"
	flagPkg "codeberg.org/goern/forgejo-mcp/v2/pkg/flag"
	"codeberg.org/goern/forgejo-mcp/v2/pkg/log"
)

var (
	transport string
	urlFlag   string
	ssePort   int
	httpPort  int
	token     string
	userAgent string

	debug bool
)

// isVersionRequest returns true for both the "version" subcommand and the
// GNU-standard --version / -version flags.  All three forms must exit before
// flag.Parse() runs so that --url is not required.
func isVersionRequest() bool {
	if len(os.Args) < 2 {
		return false
	}
	arg := os.Args[1]
	return arg == "version" || arg == "--version" || arg == "-version"
}

// initFlags registers and parses CLI flags using a dedicated FlagSet to avoid
// polluting the global flag.CommandLine (which breaks `go test`).
func initFlags() {
	fs := flag.NewFlagSet("forgejo-mcp", flag.ExitOnError)

	fs.StringVar(
		&transport,
		"t",
		"stdio",
		"Transport type (stdio, sse, or http)",
	)
	fs.StringVar(
		&transport,
		"transport",
		"stdio",
		"Transport type (stdio, sse, or http)",
	)
	fs.StringVar(
		&urlFlag,
		"url",
		"",
		"Forgejo instance URL (required, must start with http:// or https://)",
	)
	fs.IntVar(
		&ssePort,
		"sse-port",
		8080,
		"Port for SSE transport mode",
	)
	fs.IntVar(
		&httpPort,
		"http-port",
		8080,
		"Port for streamable HTTP transport mode",
	)
	fs.StringVar(
		&token,
		"token",
		"",
		"Your personal access token",
	)
	fs.StringVar(
		&userAgent,
		"user-agent",
		"",
		"User agent for HTTP requests (default: forgejo-mcp/<version>)",
	)
	fs.BoolVar(
		&debug,
		"d",
		true,
		"debug mode",
	)
	fs.BoolVar(
		&debug,
		"debug",
		true,
		"debug mode",
	)

	fs.Parse(os.Args[1:])

	flagPkg.URL = urlFlag
	flagPkg.UserAgent = userAgent
	initConfig()
}

// initConfig resolves URL, token, and debug from flags and environment variables.
func initConfig() {
	if flagPkg.URL == "" {
		flagPkg.URL = os.Getenv("FORGEJO_URL")
		if flagPkg.URL != "" {
			log.Debug("Using FORGEJO_URL environment variable")
		}
	}
	if flagPkg.URL == "" {
		// Fallback to deprecated GITEA_HOST with warning
		if giteaHost := os.Getenv("GITEA_HOST"); giteaHost != "" {
			log.Warn("Deprecated environment variable used",
				log.StringField("deprecated_var", "GITEA_HOST"),
				log.StringField("preferred_var", "FORGEJO_URL"),
				log.StringField("migration_help", "Please update your configuration to use FORGEJO_URL"),
			)
			flagPkg.URL = giteaHost
		}
	}
	if flagPkg.URL == "" {
		log.Fatal("Missing required configuration",
			log.StringField("missing", "url"),
			log.StringField("help", "Provide URL with -url flag or FORGEJO_URL environment variable"),
		)
	}

	// Validate URL has proper scheme
	log.Debug("Validating URL configuration",
		log.SanitizedURLField("url", flagPkg.URL),
	)
	if err := validateURL(flagPkg.URL); err != nil {
		log.Fatal("Invalid URL configuration",
			log.SanitizedURLField("url", flagPkg.URL),
			log.ErrorField(err),
		)
	}

	flagPkg.SSEPort = ssePort
	flagPkg.HTTPPort = httpPort
	flagPkg.Token = token
	if flagPkg.Token == "" {
		flagPkg.Token = os.Getenv("FORGEJO_ACCESS_TOKEN")
		if flagPkg.Token != "" {
			log.Debug("Using FORGEJO_ACCESS_TOKEN environment variable")
		}
	}
	if flagPkg.Token == "" {
		// Fallback to deprecated GITEA_ACCESS_TOKEN with warning
		if giteaToken := os.Getenv("GITEA_ACCESS_TOKEN"); giteaToken != "" {
			log.Warn("Deprecated environment variable used",
				log.StringField("deprecated_var", "GITEA_ACCESS_TOKEN"),
				log.StringField("preferred_var", "FORGEJO_ACCESS_TOKEN"),
				log.StringField("migration_help", "Please update your configuration to use FORGEJO_ACCESS_TOKEN"),
			)
			flagPkg.Token = giteaToken
		}
	}

	// User agent - CLI flag takes precedence, then environment variable, then default
	if flagPkg.UserAgent == "" {
		flagPkg.UserAgent = os.Getenv("FORGEJO_USER_AGENT")
		if flagPkg.UserAgent != "" {
			log.Debug("Using FORGEJO_USER_AGENT environment variable")
		}
	}

	if debug {
		flagPkg.Debug = debug
		log.Debug("Debug mode enabled via flag")
	}
	if !debug {
		flagPkg.Debug = os.Getenv("FORGEJO_DEBUG") == "true"
		if flagPkg.Debug {
			log.Debug("Debug mode enabled via FORGEJO_DEBUG environment variable")
		}
		if !flagPkg.Debug {
			// Fallback to deprecated GITEA_DEBUG with warning
			if os.Getenv("GITEA_DEBUG") == "true" {
				log.Warn("Deprecated environment variable used",
					log.StringField("deprecated_var", "GITEA_DEBUG"),
					log.StringField("preferred_var", "FORGEJO_DEBUG"),
					log.StringField("migration_help", "Please update your configuration to use FORGEJO_DEBUG"),
				)
				flagPkg.Debug = true
			}
		}
	}
}

func validateURL(urlStr string) error {
	parsedURL, err := url.Parse(urlStr)
	if err != nil {
		return fmt.Errorf("invalid URL format: %w", err)
	}

	if parsedURL.Scheme != "http" && parsedURL.Scheme != "https" {
		return fmt.Errorf("URL must start with http:// or https://, got: %s", parsedURL.Scheme)
	}

	if parsedURL.Host == "" {
		return fmt.Errorf("URL must include a host")
	}

	return nil
}

func Execute(version string) {
	if isVersionRequest() {
		fmt.Printf("forgejo-mcp %s\n", version)
		return
	}

	// CLI mode: detect --cli early, skip default flag parsing since CLI
	// has its own args (tool name, --args, --output) that would confuse it.
	cliMode = hasCLIFlag()
	if cliMode {
		initConfig()
	} else {
		initFlags()
	}

	// Set default user agent if not provided via CLI or env var
	if flagPkg.UserAgent == "" {
		flagPkg.UserAgent = "forgejo-mcp/" + version
		log.Debug("Using default user agent",
			log.StringField("user_agent", flagPkg.UserAgent),
		)
	}

	defer log.Default().Sync()

	if cliMode {
		RunCLI(version)
		return
	}

	log.Infof("Starting Forgejo MCP Server %s", version)
	log.Info("Server configuration loaded",
		log.SanitizedURLField("url", flagPkg.URL),
		log.StringField("transport", transport),
		log.IntField("sse-port", flagPkg.SSEPort),
		log.BoolField("debug", flagPkg.Debug),
		log.BoolField("token_configured", flagPkg.Token != ""),
		log.StringField("user_agent", flagPkg.UserAgent),
	)

	if err := operation.Run(transport, version); err != nil {
		if err == context.Canceled {
			log.Info("Server shutdown due to context cancellation")
			return
		}
		log.Fatalf("Run Forgejo MCP Server Error: %v", err)
	}
}