@@ -5,8 +5,11 @@ import (
55"fmt"
66"io"
77"log"
8+ "net/http"
9+ "net/url"
810"os"
911"os/signal"
12+ "strings"
1013"syscall"
1114
1215"github.com/github/github-mcp-server/pkg/github"
@@ -15,6 +18,7 @@ import (
1518gogithub"github.com/google/go-github/v69/github"
1619"github.com/mark3labs/mcp-go/mcp"
1720"github.com/mark3labs/mcp-go/server"
21+ "github.com/shurcooL/githubv4"
1822"github.com/sirupsen/logrus"
1923)
2024
@@ -44,25 +48,43 @@ type MCPServerConfig struct {
4448}
4549
4650func NewMCPServer (cfg MCPServerConfig ) (* server.MCPServer ,error ) {
47- ghClient := gogithub .NewClient (nil ).WithAuthToken (cfg .Token )
48- ghClient .UserAgent = fmt .Sprintf ("github-mcp-server/%s" ,cfg .Version )
49-
50- if cfg .Host != "" {
51- var err error
52- ghClient ,err = ghClient .WithEnterpriseURLs (cfg .Host ,cfg .Host )
53- if err != nil {
54- return nil ,fmt .Errorf ("failed to create GitHub client with host: %w" ,err )
55- }
51+ apiHost ,err := parseAPIHost (cfg .Host )
52+ if err != nil {
53+ return nil ,fmt .Errorf ("failed to parse API host: %w" ,err )
5654}
5755
56+ // Construct our REST client
57+ restClient := gogithub .NewClient (nil ).WithAuthToken (cfg .Token )
58+ restClient .UserAgent = fmt .Sprintf ("github-mcp-server/%s" ,cfg .Version )
59+ restClient .BaseURL = apiHost .baseRESTURL
60+ restClient .UploadURL = apiHost .uploadURL
61+
62+ // Construct our GraphQL client
63+ // We're using NewEnterpriseClient here unconditionally as opposed to NewClient because we already
64+ // did the necessary API host parsing so that github.com will return the correct URL anyway.
65+ gqlHTTPClient := & http.Client {
66+ Transport :& bearerAuthTransport {
67+ transport :http .DefaultTransport ,
68+ token :cfg .Token ,
69+ },
70+ }// We're going to wrap the Transport later in beforeInit
71+ gqlClient := githubv4 .NewEnterpriseClient (apiHost .graphqlURL .String (),gqlHTTPClient )
72+
5873// When a client send an initialize request, update the user agent to include the client info.
5974beforeInit := func (_ context.Context ,_ any ,message * mcp.InitializeRequest ) {
60- ghClient . UserAgent = fmt .Sprintf (
75+ userAgent : =fmt .Sprintf (
6176"github-mcp-server/%s (%s/%s)" ,
6277cfg .Version ,
6378message .Params .ClientInfo .Name ,
6479message .Params .ClientInfo .Version ,
6580)
81+
82+ restClient .UserAgent = userAgent
83+
84+ gqlHTTPClient .Transport = & userAgentTransport {
85+ transport :gqlHTTPClient .Transport ,
86+ agent :userAgent ,
87+ }
6688}
6789
6890hooks := & server.Hooks {
@@ -83,14 +105,19 @@ func NewMCPServer(cfg MCPServerConfig) (*server.MCPServer, error) {
83105}
84106
85107getClient := func (_ context.Context ) (* gogithub.Client ,error ) {
86- return ghClient ,nil // closing over client
108+ return restClient ,nil // closing over client
109+ }
110+
111+ getGQLClient := func (_ context.Context ) (* githubv4.Client ,error ) {
112+ return gqlClient ,nil // closing over client
87113}
88114
89115// Create default toolsets
90116toolsets ,err := github .InitToolsets (
91117enabledToolsets ,
92118cfg .ReadOnly ,
93119getClient ,
120+ getGQLClient ,
94121cfg .Translator ,
95122)
96123if err != nil {
@@ -213,3 +240,141 @@ func RunStdioServer(cfg StdioServerConfig) error {
213240
214241return nil
215242}
243+
244+ type apiHost struct {
245+ baseRESTURL * url.URL
246+ graphqlURL * url.URL
247+ uploadURL * url.URL
248+ }
249+
250+ func newDotcomHost () (apiHost ,error ) {
251+ baseRestURL ,err := url .Parse ("https://api.github.com/" )
252+ if err != nil {
253+ return apiHost {},fmt .Errorf ("failed to parse dotcom REST URL: %w" ,err )
254+ }
255+
256+ gqlURL ,err := url .Parse ("https://api.github.com/graphql" )
257+ if err != nil {
258+ return apiHost {},fmt .Errorf ("failed to parse dotcom GraphQL URL: %w" ,err )
259+ }
260+
261+ uploadURL ,err := url .Parse ("https://uploads.github.com" )
262+ if err != nil {
263+ return apiHost {},fmt .Errorf ("failed to parse dotcom Upload URL: %w" ,err )
264+ }
265+
266+ return apiHost {
267+ baseRESTURL :baseRestURL ,
268+ graphqlURL :gqlURL ,
269+ uploadURL :uploadURL ,
270+ },nil
271+ }
272+
273+ func newGHECHost (hostname string ) (apiHost ,error ) {
274+ u ,err := url .Parse (hostname )
275+ if err != nil {
276+ return apiHost {},fmt .Errorf ("failed to parse GHEC URL: %w" ,err )
277+ }
278+
279+ // Unsecured GHEC would be an error
280+ if u .Scheme == "http" {
281+ return apiHost {},fmt .Errorf ("GHEC URL must be HTTPS" )
282+ }
283+
284+ restURL ,err := url .Parse (fmt .Sprintf ("https://api.%s/" ,u .Hostname ()))
285+ if err != nil {
286+ return apiHost {},fmt .Errorf ("failed to parse GHEC REST URL: %w" ,err )
287+ }
288+
289+ gqlURL ,err := url .Parse (fmt .Sprintf ("https://api.%s/graphql" ,u .Hostname ()))
290+ if err != nil {
291+ return apiHost {},fmt .Errorf ("failed to parse GHEC GraphQL URL: %w" ,err )
292+ }
293+
294+ uploadURL ,err := url .Parse (fmt .Sprintf ("https://uploads.%s" ,u .Hostname ()))
295+ if err != nil {
296+ return apiHost {},fmt .Errorf ("failed to parse GHEC Upload URL: %w" ,err )
297+ }
298+
299+ return apiHost {
300+ baseRESTURL :restURL ,
301+ graphqlURL :gqlURL ,
302+ uploadURL :uploadURL ,
303+ },nil
304+ }
305+
306+ func newGHESHost (hostname string ) (apiHost ,error ) {
307+ u ,err := url .Parse (hostname )
308+ if err != nil {
309+ return apiHost {},fmt .Errorf ("failed to parse GHES URL: %w" ,err )
310+ }
311+
312+ restURL ,err := url .Parse (fmt .Sprintf ("%s://%s/api/v3/" ,u .Scheme ,u .Hostname ()))
313+ if err != nil {
314+ return apiHost {},fmt .Errorf ("failed to parse GHES REST URL: %w" ,err )
315+ }
316+
317+ gqlURL ,err := url .Parse (fmt .Sprintf ("%s://%s/api/graphql" ,u .Scheme ,u .Hostname ()))
318+ if err != nil {
319+ return apiHost {},fmt .Errorf ("failed to parse GHES GraphQL URL: %w" ,err )
320+ }
321+
322+ uploadURL ,err := url .Parse (fmt .Sprintf ("%s://%s/api/uploads/" ,u .Scheme ,u .Hostname ()))
323+ if err != nil {
324+ return apiHost {},fmt .Errorf ("failed to parse GHES Upload URL: %w" ,err )
325+ }
326+
327+ return apiHost {
328+ baseRESTURL :restURL ,
329+ graphqlURL :gqlURL ,
330+ uploadURL :uploadURL ,
331+ },nil
332+ }
333+
334+ // Note that this does not handle ports yet, so development environments are out.
335+ func parseAPIHost (s string ) (apiHost ,error ) {
336+ if s == "" {
337+ return newDotcomHost ()
338+ }
339+
340+ u ,err := url .Parse (s )
341+ if err != nil {
342+ return apiHost {},fmt .Errorf ("could not parse host as URL: %s" ,s )
343+ }
344+
345+ if u .Scheme == "" {
346+ return apiHost {},fmt .Errorf ("host must have a scheme (http or https): %s" ,s )
347+ }
348+
349+ if strings .HasSuffix (u .Hostname (),"github.com" ) {
350+ return newDotcomHost ()
351+ }
352+
353+ if strings .HasSuffix (u .Hostname (),"ghe.com" ) {
354+ return newGHECHost (s )
355+ }
356+
357+ return newGHESHost (s )
358+ }
359+
360+ type userAgentTransport struct {
361+ transport http.RoundTripper
362+ agent string
363+ }
364+
365+ func (t * userAgentTransport )RoundTrip (req * http.Request ) (* http.Response ,error ) {
366+ req = req .Clone (req .Context ())
367+ req .Header .Set ("User-Agent" ,t .agent )
368+ return t .transport .RoundTrip (req )
369+ }
370+
371+ type bearerAuthTransport struct {
372+ transport http.RoundTripper
373+ token string
374+ }
375+
376+ func (t * bearerAuthTransport )RoundTrip (req * http.Request ) (* http.Response ,error ) {
377+ req = req .Clone (req .Context ())
378+ req .Header .Set ("Authorization" ,"Bearer " + t .token )
379+ return t .transport .RoundTrip (req )
380+ }