Add demo prompt support

rubiojr · rubiojr · commit 3e948c98697c · 2025-03-14T11:40:48.000+01:00
diff --git a/go.mod b/go.mod
@@ -4,7 +4,7 @@ go 1.23.7
 
 require (
 	github.com/google/go-github/v69 v69.2.0
-	github.com/mark3labs/mcp-go v0.11.2
+	github.com/mark3labs/mcp-go v0.13.0
 	github.com/sirupsen/logrus v1.9.3
 	github.com/spf13/cobra v1.9.1
 	github.com/spf13/viper v1.19.0
diff --git a/go.sum b/go.sum
@@ -28,6 +28,8 @@ github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0V
 github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
 github.com/mark3labs/mcp-go v0.11.2 h1:mCxWFUTrcXOtJIn9t7F8bxAL8rpE/ZZTTnx3PU/VNdA=
 github.com/mark3labs/mcp-go v0.11.2/go.mod h1:cjMlBU0cv/cj9kjlgmRhoJ5JREdS7YX83xeIG9Ko/jE=
+github.com/mark3labs/mcp-go v0.13.0 h1:HP+cJaE9KjWufUF9FxN/XgcXE6LVSebFZLiZYPmFbGU=
+github.com/mark3labs/mcp-go v0.13.0/go.mod h1:cjMlBU0cv/cj9kjlgmRhoJ5JREdS7YX83xeIG9Ko/jE=
 github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
 github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
 github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
diff --git a/pkg/github/prompts.go b/pkg/github/prompts.go
@@ -0,0 +1,48 @@
+package github
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/google/go-github/v69/github"
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+)
+
+func NewMePrompt(client *github.Client) (mcp.Prompt, server.PromptHandlerFunc) {
+	prompt := mcp.NewPrompt("github_me", mcp.WithPromptDescription("GitHub Prompt"))
+	prompt.Arguments = []mcp.PromptArgument{}
+
+	return prompt, func(ctx context.Context, request mcp.GetPromptRequest) (*mcp.GetPromptResult, error) {
+		user, resp, err := client.Users.Get(ctx, "")
+		if err != nil {
+			return nil, fmt.Errorf("failed to get user: %w", err)
+		}
+		defer func() { _ = resp.Body.Close() }()
+
+		if resp.StatusCode != 200 {
+			return nil, fmt.Errorf("failed to read response body: %w", err)
+		}
+
+		r, err := json.Marshal(user)
+		if err != nil {
+			return nil, fmt.Errorf("failed to marshal user: %w", err)
+		}
+
+		return &mcp.GetPromptResult{
+			Description: "Your GitHub Identity",
+			Messages: []mcp.PromptMessage{
+				{
+					Role: mcp.RoleUser,
+					Content: mcp.TextContent{
+						Type: "text",
+						Text: string(r),
+					},
+				},
+			},
+		}, nil
+
+	}
+
+}
diff --git a/pkg/github/server.go b/pkg/github/server.go
@@ -18,6 +18,8 @@ func NewServer(client *github.Client) *server.MCPServer {
 		"github-mcp-server",
 		"0.0.1",
 		server.WithResourceCapabilities(true, true),
+		server.WithPromptCapabilities(true),
+		server.WithToolCapabilities(true),
 		server.WithLogging())
 
 	// Add GitHub tools - Issues
@@ -51,6 +53,10 @@ func NewServer(client *github.Client) *server.MCPServer {
 	// Add GitHub tools - Users
 	s.AddTool(getMe(client))
 
+	// Add github_me prompt handler
+	prompt, handler := NewMePrompt(client)
+	s.AddPrompt(prompt, handler)
+
 	return s
 }
 
