93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
package compare
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"gitea.com/gitea/gitea-mcp/pkg/gitea"
|
|
"gitea.com/gitea/gitea-mcp/pkg/log"
|
|
"gitea.com/gitea/gitea-mcp/pkg/params"
|
|
"gitea.com/gitea/gitea-mcp/pkg/to"
|
|
"gitea.com/gitea/gitea-mcp/pkg/tool"
|
|
|
|
gitea_sdk "code.gitea.io/sdk/gitea"
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/mark3labs/mcp-go/server"
|
|
)
|
|
|
|
const (
|
|
CompareCommitsToolName = "compare_commits"
|
|
)
|
|
|
|
var Tool = tool.New()
|
|
|
|
var (
|
|
CompareCommitsTool = mcp.NewTool(
|
|
CompareCommitsToolName,
|
|
mcp.WithDescription("Compare commits between two branches/tags/commits"),
|
|
mcp.WithString("owner", mcp.Required(), mcp.Description("Repository owner")),
|
|
mcp.WithString("repo", mcp.Required(), mcp.Description("Repository name")),
|
|
mcp.WithString("base", mcp.Required(), mcp.Description("Base branch/tag/commit")),
|
|
mcp.WithString("head", mcp.Required(), mcp.Description("Head branch/tag/commit")),
|
|
)
|
|
)
|
|
|
|
func init() {
|
|
Tool.RegisterRead(server.ServerTool{
|
|
Tool: CompareCommitsTool,
|
|
Handler: compareCommitsFn,
|
|
})
|
|
}
|
|
|
|
func compareCommitsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
|
log.Debugf("[Compare] Called compareCommitsFn")
|
|
args := req.GetArguments()
|
|
owner, err := params.GetString(args, "owner")
|
|
if err != nil {
|
|
return to.ErrorResult(err)
|
|
}
|
|
repo, err := params.GetString(args, "repo")
|
|
if err != nil {
|
|
return to.ErrorResult(err)
|
|
}
|
|
base, err := params.GetString(args, "base")
|
|
if err != nil {
|
|
return to.ErrorResult(err)
|
|
}
|
|
head, err := params.GetString(args, "head")
|
|
if err != nil {
|
|
return to.ErrorResult(err)
|
|
}
|
|
|
|
client, err := gitea.ClientFromContext(ctx)
|
|
if err != nil {
|
|
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
|
}
|
|
|
|
compare, _, err := client.CompareCommits(owner, repo, base, head)
|
|
if err != nil {
|
|
return to.ErrorResult(fmt.Errorf("compare commits err: %v", err))
|
|
}
|
|
|
|
return to.TextResult(slimCompare(compare))
|
|
}
|
|
|
|
func slimCompare(c *gitea_sdk.Compare) map[string]interface{} {
|
|
result := map[string]interface{}{
|
|
"total_commits": c.TotalCommits,
|
|
}
|
|
|
|
if c.Commits != nil {
|
|
commits := make([]map[string]interface{}, 0, len(c.Commits))
|
|
for _, cmt := range c.Commits {
|
|
commits = append(commits, map[string]interface{}{
|
|
"sha": cmt.SHA,
|
|
"url": cmt.URL,
|
|
})
|
|
}
|
|
result["commits"] = commits
|
|
}
|
|
|
|
return result
|
|
}
|