Class: RailsPgExtrasMcp::App

Inherits:
Object
  • Object
show all
Defined in:
lib/rails-pg-extras-mcp.rb

Class Method Summary collapse

Class Method Details

.build(opts = {}) ⇒ Object 



186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
 
# File 'lib/rails-pg-extras-mcp.rb', line 186

def self.build(opts = {})
  app = lambda do |_env|
    [200, { "Content-Type" => "text/html" },
     ["<html><body><h1>Hello from Rack!</h1><p>This is a simple Rack app with MCP middleware.</p></body></html>"]]
  end

  default_opts = {
    name: "rails-pg-extras-mcp",
    version: RailsPgExtrasMcp::VERSION,
    path_prefix: "/pg_extras_mcp",
    logger: Logger.new($stdout),
  }

  opts = default_opts.merge(opts)

  rack_method_name = opts[:auth_token].present? ? :authenticated_rack_middleware : :rack_middleware

  # Create the MCP middleware
  FastMcp.public_send(rack_method_name,
                      app,
                      **opts) do |server|
    server.register_tools(DiagnoseTool)
    server.register_tools(MissingFkConstraintsTool)
    server.register_tools(MissingFkIndexesTool)
    server.register_tools(IndexInfoTool)
    server.register_tools(TableInfoTool)
    server.register_tools(TableSchemaTool)
    server.register_tools(*QUERY_TOOL_CLASSES)
    server.register_tools(ExplainTool) if ENV["PG_EXTRAS_MCP_EXPLAIN_ENABLED"] == "true"
    server.register_tools(ExplainAnalyzeTool) if ENV["PG_EXTRAS_MCP_EXPLAIN_ANALYZE_ENABLED"] == "true"

    server.register_resource(ReadmeResource)
  end
end