main.go

package main

import (
	"context"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/MegaGrindStone/go-mcp"
	"github.com/mikaelmorvan/mcp-task/task"
	"github.com/redis/go-redis/v9"
)

func main() {

	// Read the configuration file
	config, err := LoadConfig("./config.json")
	if err != nil {
		panic(err)
	}

	// Create a Redis client
	redisClient := redis.NewClient(&redis.Options{
		Addr:     config.RedisAddr,
		DB:       config.RedisDB,       // use default DB
		Password: config.RedisPassword, // no password set
	})

	// Create a task server
	taskServer := task.NewTaskServer(redisClient)

	// Choose a transport method
	// Option 1: Server-Sent Events (SSE)
	sseSrv := mcp.NewSSEServer("/message")
	srv := mcp.NewServer(mcp.Info{
		Name:    config.ToolName,
		Version: config.ToolVersion,
	}, sseSrv,
		mcp.WithToolServer(taskServer),
		// mcp.WithToolServer(taskServer),
		mcp.WithServerPingInterval(30*time.Second),
		// Add other capabilities as needed
	)

	// Set up HTTP handlers for SSE
	http.Handle("/sse", sseSrv.HandleSSE())
	http.Handle("/message", sseSrv.HandleMessage())
	go http.ListenAndServe(config.ServerBindAddr, nil)

	// To shutdown gracefully
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	sig := make(chan os.Signal, 2)
	signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
	go func() {
		<-sig
		srv.Shutdown(ctx)
	}()

	// Start the server - this blocks until shutdown
	srv.Serve()

}