let mcp = PrismMCPServer(name: "my-app", version: "1.0.0")

// Register a tool
await mcp.registerTool(
    "greet",
    description: "Generate a greeting for a person",
    inputSchema: ["name": "string"],
    handler: { args in
        let name = args["name"] as? String ?? "World"
        return PrismMCPToolResult(
            content: [.text("Hello, \(name)!")],
            isError: false
        )
    }
)

// Register a resource
await mcp.registerResource(
    PrismMCPResource(uri: "app://config", name: "App Config", description: "Current configuration"),
    handler: { "environment: production\nversion: 1.0.0" }
)

// Start the server (stdio transport for CLI tools)
let transport = PrismMCPStdioTransport(server: mcp)
await transport.start()

let db = try PrismDatabase(path: "tasks.db")
let mcp = PrismMCPServer(name: "task-manager", version: "1.0.0")

// Tool: List tasks
await mcp.registerTool(
    "list_tasks",
    description: "List all tasks, optionally filtered by status",
    inputSchema: ["status": "string"],
    handler: { args in
        let status = args["status"] as? String
        var sql = "SELECT * FROM tasks"
        if let status { sql += " WHERE status = '\(status)'" }
        
        let rows = try await db.query(sql)
        let tasks = rows.map { "- [\($0["status"])] \($0["title"])" }
        return PrismMCPToolResult(
            content: [.text(tasks.joined(separator: "\n"))],
            isError: false
        )
    }
)

// Tool: Create task
await mcp.registerTool(
    "create_task",
    description: "Create a new task",
    inputSchema: ["title": "string", "priority": "string"],
    handler: { args in
        let title = args["title"] as? String ?? ""
        let priority = args["priority"] as? String ?? "medium"
        
        try await db.execute(
            "INSERT INTO tasks (title, priority, status) VALUES (?, ?, 'pending')",
            parameters: [.text(title), .text(priority)]
        )
        return PrismMCPToolResult(
            content: [.text("Created task: \(title)")],
            isError: false
        )
    }
)

// Resource: Project README
await mcp.registerResource(
    PrismMCPResource(uri: "project://readme", name: "README", description: "Project documentation"),
    handler: { try String(contentsOfFile: "README.md", encoding: .utf8) }
)