Sam Morrow
f1d33d51f6
Replace inline GraphQL-Features header logic in bearerAuthTransport with the exported GraphQLFeaturesTransport. This removes code duplication and ensures the transport is actually used, not just exported.
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package github
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// GraphQLFeaturesTransport is an http.RoundTripper that adds GraphQL-Features
|
|
// header to requests based on context values. This is required for using
|
|
// non-GA GraphQL API features like the agent assignment API.
|
|
//
|
|
// This transport is used internally by the MCP server and is also exported
|
|
// for library consumers who need to build their own HTTP clients with
|
|
// GraphQL feature flag support.
|
|
//
|
|
// Usage:
|
|
//
|
|
// httpClient := &http.Client{
|
|
// Transport: &github.GraphQLFeaturesTransport{
|
|
// Transport: http.DefaultTransport,
|
|
// },
|
|
// }
|
|
// gqlClient := githubv4.NewClient(httpClient)
|
|
//
|
|
// Then use withGraphQLFeatures(ctx, "feature_name") when calling GraphQL operations.
|
|
type GraphQLFeaturesTransport struct {
|
|
// Transport is the underlying HTTP transport. If nil, http.DefaultTransport is used.
|
|
Transport http.RoundTripper
|
|
}
|
|
|
|
// RoundTrip implements http.RoundTripper.
|
|
func (t *GraphQLFeaturesTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
transport := t.Transport
|
|
if transport == nil {
|
|
transport = http.DefaultTransport
|
|
}
|
|
|
|
// Clone the request to avoid mutating the original
|
|
req = req.Clone(req.Context())
|
|
|
|
// Check for GraphQL-Features in context and add header if present
|
|
if features := GetGraphQLFeatures(req.Context()); len(features) > 0 {
|
|
req.Header.Set("GraphQL-Features", strings.Join(features, ", "))
|
|
}
|
|
|
|
return transport.RoundTrip(req)
|
|
}
|