rudimentary enterprise support

SamMorrowDrums · SamMorrowDrums · commit cf5ee86cad61 · 2025-03-20T17:04:26.000+01:00
diff --git a/cmd/github-mcp-server/main.go b/cmd/github-mcp-server/main.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io"
 	stdlog "log"
+	"net/url"
 	"os"
 	"os/signal"
 	"syscall"
@@ -49,10 +50,12 @@ func init() {
 	// Add global flags that will be shared by all commands
 	rootCmd.PersistentFlags().String("log-file", "", "Path to log file")
 	rootCmd.PersistentFlags().Bool("enable-command-logging", false, "When enabled, the server will log all command requests and responses to the log file")
+	rootCmd.PersistentFlags().String("gh-host", "", "Specify the GitHub hostname (for GitHub Enterprise etc.)")
 
 	// Bind flag to viper
 	viper.BindPFlag("log-file", rootCmd.PersistentFlags().Lookup("log-file"))
 	viper.BindPFlag("enable-command-logging", rootCmd.PersistentFlags().Lookup("enable-command-logging"))
+	viper.BindPFlag("gh-host", rootCmd.PersistentFlags().Lookup("gh-host"))
 
 	// Add subcommands
 	rootCmd.AddCommand(stdioCmd)
@@ -92,6 +95,20 @@ func runStdioServer(logger *log.Logger, logCommands bool) error {
 		logger.Fatal("GITHUB_PERSONAL_ACCESS_TOKEN not set")
 	}
 	ghClient := gogithub.NewClient(nil).WithAuthToken(token)
+	if host := viper.GetString("gh-host"); host != "" {
+		url, err := url.Parse(fmt.Sprintf("https://api.%s/", host))
+		if err != nil {
+			return fmt.Errorf("failed to parse provided GitHub host URL: %w", err)
+		}
+
+		uploadUrl, err := url.Parse(fmt.Sprintf("https://uploads.%s/", host))
+		if err != nil {
+			return fmt.Errorf("failed to parse provided GitHub host URL: %w", err)
+		}
+
+		ghClient.BaseURL = url
+		ghClient.UploadURL = uploadUrl
+	}
 
 	// Create server
 	ghServer := github.NewServer(ghClient)
