Fix .gitignore/.dockerignore matching cmd/ source dir
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
Pattern 'woodpecker-mcp' without a slash matched any file or directory with that name at any depth, silently excluding cmd/woodpecker-mcp/ from commits and Docker build contexts. Anchor to the repo root so only the compiled root binary is ignored.
This commit is contained in:
parent
e8892ce182
commit
66f3478c08
@ -1,6 +1,6 @@
|
|||||||
# Build artifacts
|
# Build artifacts
|
||||||
woodpecker-mcp
|
/woodpecker-mcp
|
||||||
woodpecker-mcp.exe
|
/woodpecker-mcp.exe
|
||||||
dist/
|
dist/
|
||||||
*.test
|
*.test
|
||||||
|
|
||||||
|
|||||||
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,6 +1,6 @@
|
|||||||
# Build artifacts
|
# Build artifacts
|
||||||
woodpecker-mcp
|
/woodpecker-mcp
|
||||||
woodpecker-mcp.exe
|
/woodpecker-mcp.exe
|
||||||
dist/
|
dist/
|
||||||
*.test
|
*.test
|
||||||
|
|
||||||
|
|||||||
123
cmd/woodpecker-mcp/main.go
Normal file
123
cmd/woodpecker-mcp/main.go
Normal file
@ -0,0 +1,123 @@
|
|||||||
|
// woodpecker-mcp is a Model Context Protocol (MCP) server that connects AI
|
||||||
|
// assistants to Woodpecker CI. It communicates over stdio by default; a
|
||||||
|
// remote Streamable HTTP transport is available with --transport=http.
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"log/slog"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"strings"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
mcpgo "github.com/mark3labs/mcp-go/server"
|
||||||
|
|
||||||
|
"gitea.fe.kaien.ru/kaien/woodpecker-mcp/internal/config"
|
||||||
|
"gitea.fe.kaien.ru/kaien/woodpecker-mcp/internal/server"
|
||||||
|
"gitea.fe.kaien.ru/kaien/woodpecker-mcp/internal/woodpecker"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
transportStdio transport = "stdio"
|
||||||
|
transportHTTP transport = "http"
|
||||||
|
|
||||||
|
defaultListenAddr = ":8080"
|
||||||
|
shutdownTimeout = 5 * time.Second
|
||||||
|
)
|
||||||
|
|
||||||
|
type transport string
|
||||||
|
|
||||||
|
func parseTransport(name string) (transport, error) {
|
||||||
|
switch transport(name) {
|
||||||
|
case transportStdio, transportHTTP:
|
||||||
|
return transport(name), nil
|
||||||
|
default:
|
||||||
|
return "", fmt.Errorf("unsupported transport %q (supported: stdio, http)", name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// All logs go to stderr; stdout is reserved for the MCP JSON-RPC protocol
|
||||||
|
// when running over stdio.
|
||||||
|
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
|
||||||
|
|
||||||
|
var transportName, listenAddr string
|
||||||
|
flag.StringVar(&transportName, "transport", string(transportStdio), "MCP transport: stdio (default) or http")
|
||||||
|
flag.StringVar(&listenAddr, "listen", defaultListenAddr, "HTTP listen address (used with --transport=http)")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
tr, err := parseTransport(transportName)
|
||||||
|
if err != nil {
|
||||||
|
logger.Error("invalid configuration", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
if tr == transportHTTP && strings.TrimSpace(listenAddr) == "" {
|
||||||
|
logger.Error("invalid configuration", "error", "--listen must not be empty with --transport=http")
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg, err := config.Load()
|
||||||
|
if err != nil {
|
||||||
|
logger.Error("invalid configuration", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
client, err := woodpecker.NewClient(cfg.APIBaseURL(), cfg.WoodpeckerToken)
|
||||||
|
if err != nil {
|
||||||
|
logger.Error("failed to create Woodpecker API client", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||||||
|
defer stop()
|
||||||
|
|
||||||
|
mcpServer := server.New(client)
|
||||||
|
|
||||||
|
switch tr {
|
||||||
|
case transportStdio:
|
||||||
|
runStdio(ctx, logger, mcpServer)
|
||||||
|
case transportHTTP:
|
||||||
|
runHTTP(ctx, logger, mcpServer, listenAddr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// runStdio serves the MCP protocol over stdin/stdout until stdin closes or
|
||||||
|
// the context is cancelled.
|
||||||
|
func runStdio(ctx context.Context, logger *slog.Logger, mcpServer *mcpgo.MCPServer) {
|
||||||
|
transport := mcpgo.NewStdioServer(mcpServer)
|
||||||
|
if err := transport.Listen(ctx, os.Stdin, os.Stdout); err != nil {
|
||||||
|
logger.Error("MCP server stopped with error", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// runHTTP serves the MCP protocol over Streamable HTTP at /mcp and shuts
|
||||||
|
// down gracefully when the context is cancelled.
|
||||||
|
func runHTTP(ctx context.Context, logger *slog.Logger, mcpServer *mcpgo.MCPServer, addr string) {
|
||||||
|
srv := mcpgo.NewStreamableHTTPServer(mcpServer)
|
||||||
|
|
||||||
|
errCh := make(chan error, 1)
|
||||||
|
go func() { errCh <- srv.Start(addr) }()
|
||||||
|
logger.Info("MCP server listening", "address", addr, "endpoint", "/mcp")
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
|
||||||
|
defer cancel()
|
||||||
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
||||||
|
logger.Error("graceful shutdown failed", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
case err := <-errCh:
|
||||||
|
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||||
|
logger.Error("MCP HTTP server failed", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user