rails-pg-extras-mcp 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 51a70e772cecedb41e510b12575566f177b9e70fd374ae62dfb8f99bc8c04f87
-  data.tar.gz: b26aef4c74f6bde11aa077435f174ea7325cf8642b86c354ad9b09210849abcb
+  metadata.gz: 72d363ce5cf74c8c22e8a636c3b908f27be10cc5a21691acd480b791ab9e8c33
+  data.tar.gz: 36f3cb7fd5c1784c727def8ea7ad4c6f572f21c301ce99440440e4d5843f1e62
 SHA512:
-  metadata.gz: bdbcd8cdd983d36e0db3eef92c2a7db89168bc581d2e98fcb6b0d131713f7dc8fbb531a23a90d3f009c089a97bb21273e61313daa6951507c8005b2078b9f83c
-  data.tar.gz: 6e413fbfd0edd16253d5b2196a2608a42a00e31513fa68c50c33cf9a9959aa88a455bf62a922a49c782cf2bf41058268f6af8e253bed5655110892a15ce47da4
+  metadata.gz: 5565add5c44d77aa22c20f238c5f87d1f6e537f03f2b1d40fee5a5bbde03ed46690d11ecfecdecf791d65c0d748a26eaae5940d8808163e2c335fc4bad3c29ef
+  data.tar.gz: 8f7235f0b1e8ec213379c2f98895522ca54eed4c3c4912d7c5773a377fb9e0207a638ace52f235b7a6012161ed3f543afa565ecb5281e1f5fc3a7f09039f04b5

data/lib/rails-pg-extras-mcp.rb

@@ -75,23 +75,18 @@ class ExplainBaseTool < FastMcp::Tool
     create,
     grant,
     begin,
-    commit,
-    ;
+    commit
   ]
 
-  arguments do
-    required(:query).filled(:string).description("The query to debug")
-  end
-
-  def call(query:)
+  def call(sql_query:)
     connection = RailsPgExtras.connection
 
-    if DENYLIST.any? { |deny| query.downcase.include?(deny) }
+    if DENYLIST.any? { |deny| sql_query.downcase.include?(deny) }
       raise "This query is not allowed. It contains a denied keyword. Denylist: #{DENYLIST.join(", ")}"
     end
 
     connection.execute("BEGIN")
-    result = connection.execute("#{query}")
+    result = connection.execute("#{sql_query}")
     connection.execute("ROLLBACK")
 
     result.to_a
@@ -101,28 +96,84 @@ end
 class ExplainTool < ExplainBaseTool
   description "EXPLAIN a query. It must be an SQL string, without the EXPLAIN prefix"
 
+  arguments do
+    required(:sql_query).filled(:string).description("The SQL query to debug")
+  end
+
   def self.name
     "explain"
   end
 
-  def call(query:)
-    if query.downcase.include?("analyze")
+  def call(sql_query:)
+    if sql_query.downcase.include?("analyze")
       raise "This query is not allowed. It contains a denied ANALYZE keyword."
     end
 
-    super(query: "EXPLAIN #{query}")
+    super(sql_query: "EXPLAIN #{sql_query}")
   end
 end
 
 class ExplainAnalyzeTool < ExplainBaseTool
   description "EXPLAIN ANALYZE a query. It must be an SQL string, without the EXPLAIN ANALYZE prefix"
 
+  arguments do
+    required(:sql_query).filled(:string).description("The SQL query to debug")
+  end
+
   def self.name
     "explain_analyze"
   end
 
-  def call(query:)
-    super(query: "EXPLAIN ANALYZE #{query}")
+  def call(sql_query:)
+    super(sql_query: "EXPLAIN ANALYZE #{sql_query}")
+  end
+end
+
+class IndexInfoTool < FastMcp::Tool
+  description "Shows information about table indexes: name, table name, columns, index size, index scans, null frac"
+
+  arguments do
+    required(:table_name).filled(:string).description("The table name to get index info for")
+  end
+
+  def call(table_name:)
+    RailsPgExtras.index_info(args: { table_name: table_name }, in_format: :hash)
+  end
+
+  def self.name
+    "index_info"
+  end
+end
+
+class TableInfoTool < FastMcp::Tool
+  description "Shows information about a table: name, size, cache hit, estimated rows, sequential scans, indexes scans"
+
+  arguments do
+    required(:table_name).filled(:string).description("The table name to get info for")
+  end
+
+  def call(table_name:)
+    RailsPgExtras.table_info(args: { table_name: table_name }, in_format: :hash)
+  end
+
+  def self.name
+    "table_info"
+  end
+end
+
+class TableSchemaTool < FastMcp::Tool
+  description "Shows the schema of a table"
+
+  arguments do
+    required(:table_name).filled(:string).description("The table name to get schema for")
+  end
+
+  def call(table_name:)
+    RailsPgExtras.table_schema(args: { table_name: table_name }, in_format: :hash)
+  end
+
+  def self.name
+    "table_schema"
   end
 end
 
@@ -163,6 +214,9 @@ module RailsPgExtrasMcp
         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"

data/lib/rails_pg_extras_mcp/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsPgExtrasMcp
-  VERSION = "0.2.1"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-pg-extras-mcp
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - pawurb