go-wikijs-mcp/internal/server/server.go

// Package server builds the MCP server and registers the WikiJS tools.
package server

import (
	"log/slog"

	"wikijs-mcp/internal/wikijs"

	"github.com/modelcontextprotocol/go-sdk/mcp"
)

const instructions = `This server exposes a WikiJS knowledge base.
Use search_pages to find pages by full-text query, get_page_by_id or
get_page_by_path to read a page, and get_all_pages to browse the wiki.
Page results include absolute URLs that can be shared with the user.`

// Options configures the MCP server.
type Options struct {
	// Version is reported to MCP clients.
	Version string
	// EnableEdit exposes the create_page and edit_page tools.
	EnableEdit bool
	// Logger receives MCP protocol logs. It must write to stderr when the
	// stdio transport is used.
	Logger *slog.Logger
}

// New creates an MCP server with all WikiJS tools registered.
func New(client *wikijs.Client, opts Options) *mcp.Server {
	logger := opts.Logger
	if logger == nil {
		logger = slog.New(slog.DiscardHandler)
	}

	s := mcp.NewServer(&mcp.Implementation{
		Name:        "wikijs-mcp-server",
		Title:       "WikiJS MCP Server",
		Version:     opts.Version,
		Description: "Search, read, and optionally edit pages in a WikiJS instance.",
	}, &mcp.ServerOptions{
		Instructions: instructions,
		Logger:       logger,
	})

	h := &handlers{wiki: client}

	readOnly := &mcp.ToolAnnotations{ReadOnlyHint: true}
	mcp.AddTool(s, &mcp.Tool{
		Name:        "search_pages",
		Title:       "Search pages",
		Description: "Search for pages in WikiJS by query string.",
		Annotations: readOnly,
	}, h.searchPages)
	mcp.AddTool(s, &mcp.Tool{
		Name:        "get_page_by_id",
		Title:       "Get page by ID",
		Description: "Get a WikiJS page by its numeric ID.",
		Annotations: readOnly,
	}, h.pageByID)
	mcp.AddTool(s, &mcp.Tool{
		Name:        "get_page_by_path",
		Title:       "Get page by path",
		Description: "Get a WikiJS page by its path and locale.",
		Annotations: readOnly,
	}, h.pageByPath)
	mcp.AddTool(s, &mcp.Tool{
		Name:        "get_all_pages",
		Title:       "List pages",
		Description: "List pages in WikiJS, optionally filtered by limit, locale, and tags.",
		Annotations: readOnly,
	}, h.listPages)

	if opts.EnableEdit {
		notDestructive := false
		mcp.AddTool(s, &mcp.Tool{
			Name:        "create_page",
			Title:       "Create page",
			Description: "Create a new WikiJS page.",
			Annotations: &mcp.ToolAnnotations{DestructiveHint: &notDestructive},
		}, h.createPage)
		mcp.AddTool(s, &mcp.Tool{
			Name:        "edit_page",
			Title:       "Edit page",
			Description: "Edit an existing WikiJS page by its numeric ID.",
			Annotations: &mcp.ToolAnnotations{IdempotentHint: true},
		}, h.editPage)
	}

	return s
}