commit 2dd4aa94ad29fc756666ff2e36df6e49ee34d5d3 Author: hiifong Date: Sun Mar 16 00:21:29 2025 +0800 Initialize commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..e0787ab --- /dev/null +++ b/.gitignore @@ -0,0 +1,7 @@ +.idea +.vscode + +gitea-mcp +gitea-mcp.exe + +*.log \ No newline at end of file diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..c307b88 --- /dev/null +++ b/Makefile @@ -0,0 +1,6 @@ +GO ?= go +EXECUTABLE := gitea-mcp + +.PHONY: build +build: + $(GO) build -v -ldflags '-s -w' -o $(EXECUTABLE) \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..355c519 --- /dev/null +++ b/README.md @@ -0,0 +1,23 @@ +# Gitea MCP Server + +## Usage + +**MCP Server Config** +```json +{ + "mcpServers": { + "gitea": { + "command": "gitea-mcp", + "args": { + "-t": "stdio", + "--host": "https://gitea.com", + "--token": "" + }, + "env": { + "GITEA_HOST": "https://gitea.com", + "GITEA_ACCESS_TOKEN": "" + } + } + } +} +``` \ No newline at end of file diff --git a/cmd/cmd.go b/cmd/cmd.go new file mode 100644 index 0000000..491dbc3 --- /dev/null +++ b/cmd/cmd.go @@ -0,0 +1,62 @@ +package cmd + +import ( + "flag" + + "gitea.com/gitea/gitea-mcp/operation" + flagPkg "gitea.com/gitea/gitea-mcp/pkg/flag" + "gitea.com/gitea/gitea-mcp/pkg/log" +) + +var ( + transport string + host string + token string + + debug bool +) + +func init() { + flag.StringVar( + &transport, + "t", + "stdio", + "Transport type (stdio or sse)", + ) + flag.StringVar( + &transport, + "transport", + "stdio", + "Transport type (stdio or sse)", + ) + flag.StringVar( + &host, + "host", + "https://gitea.com", + "Gitea host", + ) + flag.StringVar( + &token, + "token", + "", + "Your personal access token", + ) + flag.BoolVar( + &debug, + "debug", + false, + "debug mode", + ) + + flag.Parse() + + flagPkg.Host = host + flagPkg.Token = token +} + +func Execute(version string) { + defer log.Sync() + if err := operation.Run(transport, version); err != nil { + log.Fatalf("Run Gitea MCP Server Error: %v", err) + } +} diff --git a/config.json b/config.json new file mode 100644 index 0000000..da71873 --- /dev/null +++ b/config.json @@ -0,0 +1,16 @@ +{ + "mcpServers": { + "gitea": { + "command": "gitea-mcp", + "args": { + "-t": "stdio", + "--host": "https://gitea.com", + "--token": "" + }, + "env": { + "GITEA_HOST": "https://gitea.com", + "GITEA_ACCESS_TOKEN": "" + } + } + } +} \ No newline at end of file diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..ab36243 --- /dev/null +++ b/go.mod @@ -0,0 +1,24 @@ +module gitea.com/gitea/gitea-mcp + +go 1.24.0 + +require ( + code.gitea.io/sdk/gitea v0.20.0 + github.com/mark3labs/mcp-go v0.13.0 + github.com/stretchr/testify v1.9.0 + go.uber.org/zap v1.27.0 +) + +require ( + github.com/42wim/httpsig v1.2.2 // indirect + github.com/davecgh/go-spew v1.1.1 // indirect + github.com/davidmz/go-pageant v1.0.2 // indirect + github.com/go-fed/httpsig v1.1.0 // indirect + github.com/google/uuid v1.6.0 // indirect + github.com/hashicorp/go-version v1.7.0 // indirect + github.com/pmezard/go-difflib v1.0.0 // indirect + go.uber.org/multierr v1.11.0 // indirect + golang.org/x/crypto v0.36.0 // indirect + golang.org/x/sys v0.31.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..9baa6db --- /dev/null +++ b/go.sum @@ -0,0 +1,48 @@ +code.gitea.io/sdk/gitea v0.20.0 h1:Zm/QDwwZK1awoM4AxdjeAQbxolzx2rIP8dDfmKu+KoU= +code.gitea.io/sdk/gitea v0.20.0/go.mod h1:faouBHC/zyx5wLgjmRKR62ydyvMzwWf3QnU0bH7Cw6U= +github.com/42wim/httpsig v1.2.2 h1:ofAYoHUNs/MJOLqQ8hIxeyz2QxOz8qdSVvp3PX/oPgA= +github.com/42wim/httpsig v1.2.2/go.mod h1:P/UYo7ytNBFwc+dg35IubuAUIs8zj5zzFIgUCEl55WY= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davidmz/go-pageant v1.0.2 h1:bPblRCh5jGU+Uptpz6LgMZGD5hJoOt7otgT454WvHn0= +github.com/davidmz/go-pageant v1.0.2/go.mod h1:P2EDDnMqIwG5Rrp05dTRITj9z2zpGcD9efWSkTNKLIE= +github.com/go-fed/httpsig v1.1.0 h1:9M+hb0jkEICD8/cAiNqEB66R87tTINszBRTjwjQzWcI= +github.com/go-fed/httpsig v1.1.0/go.mod h1:RCMrTZvN1bJYtofsG4rd5NaO5obxQ5xBkdiS7xsT7bM= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/hashicorp/go-version v1.7.0 h1:5tqGy27NaOTB8yJKUZELlFAS/LTKJkrmONwQKeRZfjY= +github.com/hashicorp/go-version v1.7.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA= +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/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= +go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= +go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0= +go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y= +go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8= +go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20210513164829-c07d793c2f9a/go.mod h1:P+XmwS30IXTQdn5tA2iutPOUgjI07+tq3H3K9MVA1s8= +golang.org/x/crypto v0.36.0 h1:AnAEvhDddvBdpY+uR+MyHmuZzzNqXSe/GvuDeob5L34= +golang.org/x/crypto v0.36.0/go.mod h1:Y4J0ReaxCR1IMaabaSMugxJES1EpwhBHhv2bDHklZvc= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.31.0 h1:ioabZlmFYtWhL+TRYpcnNlLwhyxaM9kWTDEmfnprqik= +golang.org/x/sys v0.31.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y= +golang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..16d7b9d --- /dev/null +++ b/main.go @@ -0,0 +1,13 @@ +package main + +import ( + "gitea.com/gitea/gitea-mcp/cmd" +) + +var ( + Version = "dev" +) + +func main() { + cmd.Execute(Version) +} diff --git a/operation/operation.go b/operation/operation.go new file mode 100644 index 0000000..fe5fbc3 --- /dev/null +++ b/operation/operation.go @@ -0,0 +1,50 @@ +package operation + +import ( + "fmt" + + "gitea.com/gitea/gitea-mcp/operation/repo" + "gitea.com/gitea/gitea-mcp/operation/user" + "gitea.com/gitea/gitea-mcp/pkg/log" + "github.com/mark3labs/mcp-go/server" +) + +var ( + mcpServer *server.MCPServer +) + +func RegisterTool(s *server.MCPServer) { + // User Tool + s.AddTool(user.GetMyUserInfoTool, user.MyUserInfoFn) + + // Repo Tool + s.AddTool(repo.GetMyReposTool, repo.MyUserReposFn) +} + +func Run(transport, version string) error { + mcpServer = newMCPServer(version) + RegisterTool(mcpServer) + switch transport { + case "stdio": + if err := server.ServeStdio(mcpServer); err != nil { + return err + } + case "sse": + sseServer := server.NewSSEServer(mcpServer) + log.Infof("Gitea MCP SSE server listening on :8080") + if err := sseServer.Start(":8080"); err != nil { + return err + } + default: + return fmt.Errorf("invalid transport type: %s. Must be 'stdio' or 'sse'", transport) + } + return nil +} + +func newMCPServer(version string) *server.MCPServer { + return server.NewMCPServer( + "Gitea MCP Server", + version, + server.WithLogging(), + ) +} diff --git a/operation/repo/repo.go b/operation/repo/repo.go new file mode 100644 index 0000000..0659f77 --- /dev/null +++ b/operation/repo/repo.go @@ -0,0 +1,40 @@ +package repo + +import ( + "context" + "encoding/json" + + "code.gitea.io/sdk/gitea" + giteaPkg "gitea.com/gitea/gitea-mcp/pkg/gitea" + "github.com/mark3labs/mcp-go/mcp" +) + +const ( + ListMyReposToolName = "list_my_repos" +) + +var ( + GetMyReposTool = mcp.NewTool( + ListMyReposToolName, + mcp.WithDescription("List My Repositories"), + ) +) + +func MyUserReposFn(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { + opts := gitea.ListReposOptions{ + ListOptions: gitea.ListOptions{ + Page: 1, + PageSize: 100, + }, + } + repos, _, err := giteaPkg.Client().ListMyRepos(opts) + if err != nil { + return mcp.NewToolResultError("Get My User Info Error"), err + } + + result, err := json.Marshal(repos) + if err != nil { + return mcp.NewToolResultError("Get My User Info Error"), err + } + return mcp.NewToolResultText(string(result)), nil +} diff --git a/operation/user/user.go b/operation/user/user.go new file mode 100644 index 0000000..fdeed80 --- /dev/null +++ b/operation/user/user.go @@ -0,0 +1,34 @@ +package user + +import ( + "context" + "encoding/json" + + "gitea.com/gitea/gitea-mcp/pkg/gitea" + + "github.com/mark3labs/mcp-go/mcp" +) + +const ( + GetMyUserInfoToolName = "get_my_user_info" +) + +var ( + GetMyUserInfoTool = mcp.NewTool( + GetMyUserInfoToolName, + mcp.WithDescription("Get My User Info"), + ) +) + +func MyUserInfoFn(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { + user, _, err := gitea.Client().GetMyUserInfo() + if err != nil { + return mcp.NewToolResultError("Get My User Info Error"), err + } + + result, err := json.Marshal(user) + if err != nil { + return mcp.NewToolResultError("Get My User Info Error"), err + } + return mcp.NewToolResultText(string(result)), nil +} diff --git a/pkg/flag/flag.go b/pkg/flag/flag.go new file mode 100644 index 0000000..27014a6 --- /dev/null +++ b/pkg/flag/flag.go @@ -0,0 +1,9 @@ +package flag + +var ( + Host string + Token string + Version string + + Debug bool +) diff --git a/pkg/gitea/gitea.go b/pkg/gitea/gitea.go new file mode 100644 index 0000000..859104a --- /dev/null +++ b/pkg/gitea/gitea.go @@ -0,0 +1,40 @@ +package gitea + +import ( + "os" + "sync" + + "gitea.com/gitea/gitea-mcp/pkg/flag" + "gitea.com/gitea/gitea-mcp/pkg/log" + + "code.gitea.io/sdk/gitea" +) + +var ( + client *gitea.Client + clientOnce sync.Once +) + +func Client() *gitea.Client { + clientOnce.Do(func() { + if client == nil { + host, token := flag.Host, flag.Token + if host == "" { + host = os.Getenv("GITEA_HOST") + } + if host == "" { + host = "https://gitea.com" + } + if token == "" { + token = os.Getenv("GITEA_TOKEN") + } + + c, err := gitea.NewClient(host, gitea.SetToken(token)) + if err != nil { + log.Fatalf("create gitea client err: %v", err) + } + client = c + } + }) + return client +} diff --git a/pkg/log/log.go b/pkg/log/log.go new file mode 100644 index 0000000..7992346 --- /dev/null +++ b/pkg/log/log.go @@ -0,0 +1,100 @@ +package log + +import ( + "os" + "sync" + "time" + + "go.uber.org/zap" + "go.uber.org/zap/zapcore" +) + +var ( + defaultLoggerOnce sync.Once + defaultLogger *zap.Logger +) + +func Default() *zap.Logger { + defaultLoggerOnce.Do(func() { + if defaultLogger == nil { + ec := zap.NewProductionEncoderConfig() + ec.EncodeTime = zapcore.TimeEncoderOfLayout(time.DateTime) + ec.EncodeLevel = zapcore.CapitalColorLevelEncoder + + var ws zapcore.WriteSyncer + var wss []zapcore.WriteSyncer + + wss = append(wss, zapcore.AddSync(os.Stdout)) + ws = zapcore.NewMultiWriteSyncer(wss...) + + enc := zapcore.NewConsoleEncoder(ec) + core := zapcore.NewCore(enc, ws, zapcore.DebugLevel) + options := []zap.Option{ + zap.AddStacktrace(zapcore.ErrorLevel), + zap.AddCaller(), + zap.AddCallerSkip(1), + } + defaultLogger = zap.New(core, options...) + } + }) + + return defaultLogger +} + +func SetDefault(logger *zap.Logger) { + if logger != nil { + defaultLogger = logger + } +} + +func Logger() *zap.Logger { + return defaultLogger +} + +func Debug(msg string, fields ...zap.Field) { + Default().Debug(msg, fields...) +} + +func Info(msg string, fields ...zap.Field) { + Default().Info(msg, fields...) +} + +func Warn(msg string, fields ...zap.Field) { + Default().Warn(msg, fields...) +} + +func Error(msg string, fields ...zap.Field) { + Default().Error(msg, fields...) +} + +func Panic(msg string, fields ...zap.Field) { + Default().Panic(msg, fields...) +} + +func Debugf(format string, args ...interface{}) { + Default().Sugar().Debugf(format, args...) +} + +func Infof(format string, args ...interface{}) { + Default().Sugar().Infof(format, args...) +} + +func Warnf(format string, args ...interface{}) { + Default().Sugar().Warnf(format, args...) +} + +func Errorf(format string, args ...interface{}) { + Default().Sugar().Errorf(format, args...) +} + +func Fatalf(format string, args ...interface{}) { + Default().Sugar().Fatalf(format, args...) +} + +func Sync() { + err := defaultLogger.Sync() + if err != nil { + Error(err.Error()) + } + Info("logger has been synced") +}