WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

cli/internal/mcp/server.go

@@ -0,0 +1,82 @@
+// Package mcp wires the curated weknora tool set to an
+// modelcontextprotocol/go-sdk server. RunStdio is the entry point invoked
+// by `weknora mcp serve`.
+//
+// Design notes:
+//
+//   - Tool surface is hand-curated rather than auto-derived from the cobra
+//     tree (which would expose auth/link/completion/destructive verbs that
+//     don't belong on an agent-callable surface).
+//   - Long-running tools (chat / session_ask) accumulate the LLM SSE
+//     stream server-side and return a single CallToolResult - MCP spec
+//     2025-06-18 does not define streamed tool-result content, so the
+//     accumulate-and-return pattern is the canonical path.
+//   - Handlers receive ctx for cancellation; mid-LLM-stream cancellation
+//     propagates to the SDK via context, which closes the SSE connection.
+package mcp
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"io"
+	"strings"
+
+	mcpsdk "github.com/modelcontextprotocol/go-sdk/mcp"
+
+	"github.com/Tencent/WeKnora/cli/internal/build"
+	sdk "github.com/Tencent/WeKnora/client"
+)
+
+// ServiceClient bundles the SDK methods the tool registry needs. *sdk.Client
+// satisfies it; tests substitute a fake to exercise the tool handlers
+// in-process without standing up a real WeKnora server.
+//
+// Embedding the full SDK Client would couple every tool test to every SDK
+// method; declaring the narrow surface here keeps the seam tight.
+type ServiceClient interface {
+	knowledgeBaseService
+	knowledgeService
+	chatService
+	agentService
+	chunkListService
+}
+
+var _ ServiceClient = (*sdk.Client)(nil)
+
+// RunStdio constructs the MCP server, registers the curated 10 tools, and
+// blocks reading JSON-RPC from stdin until the client disconnects or ctx
+// is cancelled. Returns the underlying transport error (if any); the cobra
+// RunE caller maps it through the usual cmdutil exit-code path.
+func RunStdio(ctx context.Context, svc ServiceClient) error {
+	v, _, _ := build.Info()
+	server := mcpsdk.NewServer(
+		&mcpsdk.Implementation{
+			Name:    "weknora",
+			Version: v,
+		},
+		nil,
+	)
+	registerTools(server, svc)
+	if err := server.Run(ctx, &mcpsdk.StdioTransport{}); err != nil {
+		// An MCP host shuts the server down by closing stdin (EOF) or cancelling
+		// the context — that is normal termination, not a failure, so exit 0.
+		// Only a genuine transport fault is surfaced as an error.
+		if isCleanShutdown(err) {
+			return nil
+		}
+		return fmt.Errorf("mcp serve: %w", err)
+	}
+	return nil
+}
+
+// isCleanShutdown reports whether err is the ordinary "client went away"
+// signal: EOF on stdin or a cancelled/expired context. The go-sdk wraps the
+// stdin EOF in a formatted string ("server is closing: EOF") that does not
+// always unwrap to io.EOF, so we also match the message suffix as a fallback.
+func isCleanShutdown(err error) bool {
+	if errors.Is(err, io.EOF) || errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
+		return true
+	}
+	return strings.HasSuffix(strings.TrimSpace(err.Error()), "EOF")
+}