feat: add request header

client/client.go

@@ -108,7 +108,6 @@ func (c *Client) sendRequest(
 		Method:  method,
 		Params:  params,
 	}
-
 	response, err := c.transport.SendRequest(ctx, request)
 	if err != nil {
 		return nil, fmt.Errorf("transport error: %w", err)

client/transport/inprocess.go

@@ -34,7 +34,7 @@ func (c *InProcessTransport) SendRequest(ctx context.Context, request JSONRPCReq
 	}
 	requestBytes = append(requestBytes, '\n')
 
-	respMessage := c.server.HandleMessage(ctx, requestBytes)
+	respMessage := c.server.HandleMessage(ctx, map[string]string{}, requestBytes)
 	respByte, err := json.Marshal(respMessage)
 	if err != nil {
 		return nil, fmt.Errorf("failed to marshal response message: %w", err)
@@ -54,7 +54,7 @@ func (c *InProcessTransport) SendNotification(ctx context.Context, notification
 		return fmt.Errorf("failed to marshal notification: %w", err)
 	}
 	notificationBytes = append(notificationBytes, '\n')
-	c.server.HandleMessage(ctx, notificationBytes)
+	c.server.HandleMessage(ctx, map[string]string{}, notificationBytes)
 
 	return nil
 }

mcp/types.go

@@ -151,8 +151,9 @@ func (m *Meta) UnmarshalJSON(data []byte) error {
 }
 
 type Request struct {
-	Method string        `json:"method"`
-	Params RequestParams `json:"params,omitempty"`
+	Header map[string]string `json:"header"`
+	Method string            `json:"method"`
+	Params RequestParams     `json:"params,omitempty"`
 }
 
 type RequestParams struct {

server/internal/gen/request_handler.go.tmpl

@@ -14,6 +14,7 @@ import (
 // HandleMessage processes an incoming JSON-RPC message and returns an appropriate response
 func (s *MCPServer) HandleMessage(
 	ctx context.Context,
+	header map[string]string,
 	message json.RawMessage,
 ) mcp.JSONRPCMessage {
 	// Add server to context
@@ -90,6 +91,7 @@ func (s *MCPServer) HandleMessage(
 				err:  &UnparsableMessageError{message: message, err: unmarshalErr, method: baseMessage.Method},
 			}
 		} else {
+		    request.Header = header
 			s.hooks.before{{.HookName}}(ctx, baseMessage.ID, &request)
 			result, err = s.{{.HandlerFunc}}(ctx, baseMessage.ID, request)
 		}

server/resource_test.go

@@ -59,7 +59,8 @@ func TestMCPServer_RemoveResource(t *testing.T) {
 				)
 
 				// First, verify we have two resources
-				response := server.HandleMessage(context.Background(), []byte(`{
+				header := map[string]string{"Authorization": "Bearer test"}
+				response := server.HandleMessage(context.Background(), header, []byte(`{
 					"jsonrpc": "2.0",
 					"id": 1,
 					"method": "resources/list"
@@ -205,9 +206,9 @@ func TestMCPServer_RemoveResource(t *testing.T) {
 				"1.0.0",
 				WithResourceCapabilities(true, true),
 			)
-
+			header := map[string]string{"Authorization": "Bearer test"}
 			// Initialize the server
-			_ = server.HandleMessage(ctx, []byte(`{
+			_ = server.HandleMessage(ctx, header, []byte(`{
 				"jsonrpc": "2.0",
 				"id": 1,
 				"method": "initialize"
@@ -244,7 +245,7 @@ func TestMCPServer_RemoveResource(t *testing.T) {
 				"id": 1,
 				"method": "resources/list"
 			}`
-			resourcesList := server.HandleMessage(ctx, []byte(listMessage))
+			resourcesList := server.HandleMessage(ctx, header, []byte(listMessage))
 
 			// Validate the results
 			tt.validate(t, notifications, resourcesList)