support server-sent events

Author: ezh0v

Dockerfile

@@ -0,0 +1,23 @@
+FROM golang:1.24-alpine3.21 AS builder
+
+WORKDIR /app
+
+COPY go.mod go.sum ./
+
+RUN go mod download
+
+COPY . .
+
+RUN go build -o weather-mcp-server ./cmd/weather-mcp-server
+
+FROM gcr.io/distroless/base-debian12
+
+WORKDIR /app
+
+COPY --from=builder /app/weather-mcp-server .
+
+USER nonroot:nonroot
+
+EXPOSE 8000
+
+ENTRYPOINT ["./weather-mcp-server", "--address", "0.0.0.0:8000"]

README.md

@@ -1,11 +1,12 @@
 <div align="center">
-<img alt="example output template" src="./assets/weather.svg">
+<img alt="example output template" src="assets/weather.svg">
 
 <h1>Weather API MCP Server</h1>
 
 [![license](https://img.shields.io/badge/license-MIT-red.svg)](LICENSE)
 [![go version](https://img.shields.io/github/go-mod/go-version/TuanKiri/weather-mcp-server)](go.mod)
 [![go report](https://goreportcard.com/badge/github.com/TuanKiri/weather-mcp-server)](https://goreportcard.com/report/github.com/TuanKiri/weather-mcp-server)
+[![build](https://github.com/TuanKiri/weather-mcp-server/workflows/Build/badge.svg)](https://github.com/TuanKiri/weather-mcp-server/actions?workflow=Build)
 
 <strong>[Report Bug](https://github.com/TuanKiri/weather-mcp-server/issues/new?assignees=&labels=bug&projects=&template=bug_report.yml)</strong> | <strong>[Request Feature](https://github.com/TuanKiri/weather-mcp-server/issues/new?assignees=&labels=enhancement&projects=&template=feature_request.yml)</strong>
 
@@ -14,13 +15,15 @@
 A lightweight Model Context Protocol (MCP) server that enables AI assistants like Claude to retrieve and interpret real-time weather data.
 
 <div align="center">
-<img alt="demo example" src="./assets/weather.gif" width="480">
+<img alt="demo example" src="assets/weather.gif" width="480">
 </div>
 
 ## Installing on Claude Desktop
 
 To use your MCP server with Claude Desktop, add it to your Claude configuration:
 
+#### 1. Local mode
+
 ```json
 {
   "mcpServers": {
@@ -35,7 +38,19 @@ To use your MCP server with Claude Desktop, add it to your Claude configuration:
 }
 ```
 
-You can get your API key in your personal account at [weatherapi](https://www.weatherapi.com/my/).
+You can get an API key from your personal account on [WeatherAPI](https://www.weatherapi.com/my/).
+
+#### 2. Remote mode
+
+```json
+{
+  "mcpServers": {
+    "weather-mcp-server": {
+      "url": "http://host:port/sse"
+    }
+  }
+}
+```
 
 ## Build from source
 
@@ -45,6 +60,22 @@ You can use `go` to build the binary in the `cmd/github-mcp-server` directory.
 go build -o weather-mcp-server ./cmd/weather-mcp-server
 ```
 
+## Using MCP with Docker Containers
+
+#### 1. Build the Docker image:
+
+```shell
+docker build -t weather-mcp-server .
+```
+
+#### 2. Run the Docker Container:
+
+```shell
+docker run -e WEATHER_API_KEY=your-api-key -d --name weather-mcp-server -p 8000:8000 weather-mcp-server
+```
+
+Replace `your-api-key` with your actual [WeatherAPI](https://www.weatherapi.com/my/) API key.
+
 ## Tools
 
 - **current_weather** - Gets the current weather for a city
@@ -77,4 +108,4 @@ Please follow the [contribution guidelines](.github/CONTRIBUTING.md).
 
 ## License
 
-This MCP server is licensed under the MIT License.
+This MCP server is licensed under the [MIT License](LICENSE).

cmd/weather-mcp-server/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"log"
 	"os"
 	"time"
@@ -9,7 +10,11 @@ import (
 )
 
 func main() {
-	cfg := server.Config{
+	addr := flag.String("address", "", "The host and port to start the sse server")
+	flag.Parse()
+
+	cfg := &server.Config{
+		ListenAddr:        *addr,
 		WeatherAPIKey:     os.Getenv("WEATHER_API_KEY"),
 		WeatherAPITimeout: 1 * time.Second,
 	}
@@ -18,7 +23,7 @@ func main() {
 		log.Fatal(err)
 	}
 
-	if err := server.Run(&cfg); err != nil {
+	if err := server.Run(cfg); err != nil {
 		log.Fatal(err)
 	}
 }

internal/server/config.go

@@ -6,6 +6,7 @@ import (
 )
 
 type Config struct {
+	ListenAddr        string
 	WeatherAPIKey     string
 	WeatherAPITimeout time.Duration
 }

internal/server/server.go

@@ -1,8 +1,12 @@
 package server
 
 import (
+	"context"
 	"embed"
 	"html/template"
+	"log"
+	"os/signal"
+	"syscall"
 
 	"github.com/mark3labs/mcp-go/server"
 
@@ -30,13 +34,34 @@ func Run(cfg *Config) error {
 		server.WithLogging(),
 	)
 
-	serverTools := []tools.ToolFunc{
+	toolFuncs := []tools.ToolFunc{
 		tools.CurrentWeather,
 	}
 
-	for _, tool := range serverTools {
+	for _, tool := range toolFuncs {
 		s.AddTool(tool(svc))
 	}
 
+	if cfg.ListenAddr != "" {
+		return serveSSE(s, cfg.ListenAddr)
+	}
+
 	return server.ServeStdio(s)
 }
+
+func serveSSE(s *server.MCPServer, addr string) error {
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	srv := server.NewSSEServer(s)
+
+	go func() {
+		if err := srv.Start(addr); err != nil {
+			log.Fatal(err)
+		}
+	}()
+
+	<-ctx.Done()
+
+	return srv.Shutdown(context.TODO())
+}