feat(server): implement Kafka MCP server with CLI and tools integration

Signed-off-by: Tommy Nguyen <[email protected]>

Author: tuannvm

Makefile

@@ -1,7 +1,7 @@
 .PHONY: build test clean run-dev release-snapshot run-docker run docker-compose-up docker-compose-down lint
 
 # Variables
-BINARY_NAME=mcp-trino
+BINARY_NAME=kafka-mcp-server
 VERSION ?= $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
 BUILD_DIR=bin
 
@@ -52,4 +52,4 @@ lint:
 	@golangci-lint run --timeout=5m
 
 # Default target
-all: clean build 
+all: clean build 

cmd/server/main.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"context"
+	"log/slog"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/tuannvm/kafka-mcp-server/config"
+	"github.com/tuannvm/kafka-mcp-server/kafka"
+	"github.com/tuannvm/kafka-mcp-server/mcp"
+	"github.com/tuannvm/kafka-mcp-server/server"
+)
+
+// Version is set during build via -X ldflags
+var Version = "dev"
+
+func main() {
+	// Setup signal handling for graceful shutdown
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// Handle SIGINT and SIGTERM
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		sig := <-sigCh
+		slog.Info("Received signal, shutting down", "signal", sig)
+		cancel()
+	}()
+
+	// Load configuration
+	cfg := config.LoadConfig() // Changed from cfg, err := config.LoadConfig()
+	
+	// Initialize Kafka client
+	kafkaClient, err := kafka.NewClient(cfg)
+	if err != nil {
+		slog.Error("Failed to create Kafka client", "error", err)
+		os.Exit(1)
+	}
+	defer kafkaClient.Close()
+
+	// Create MCP server
+	s := server.NewMCPServer("kafka-mcp-server", Version)
+
+	// Register MCP resources and tools
+	mcp.RegisterResources(s, kafkaClient)
+	mcp.RegisterTools(s, kafkaClient, cfg)
+
+	// Start server
+	slog.Info("Starting Kafka MCP server", "version", Version, "transport", cfg.MCPTransport)
+	if err := server.Start(ctx, s, cfg); err != nil {
+		slog.Error("Server error", "error", err)
+		os.Exit(1)
+	}
+
+	slog.Info("Server shutdown complete")
+}

go.mod

@@ -5,7 +5,6 @@ go 1.24.2
 require (
 	github.com/mark3labs/mcp-go v0.21.1
 	github.com/stretchr/testify v1.10.0
-	github.com/testcontainers/testcontainers-go v0.36.0
 	github.com/testcontainers/testcontainers-go/modules/kafka v0.36.0
 	github.com/twmb/franz-go v1.18.1
 	github.com/twmb/franz-go/pkg/kmsg v1.9.0
@@ -49,6 +48,7 @@ require (
 	github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
 	github.com/shirou/gopsutil/v4 v4.25.1 // indirect
 	github.com/sirupsen/logrus v1.9.3 // indirect
+	github.com/testcontainers/testcontainers-go v0.36.0 // indirect
 	github.com/tklauser/go-sysconf v0.3.12 // indirect
 	github.com/tklauser/numcpus v0.6.1 // indirect
 	github.com/yosida95/uritemplate/v3 v3.0.2 // indirect