rails-pg-extras-mcp 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6544b81c0de61c0ac92c52514874ec1672990525739d3ae4fe989d2f8f88f4eb
-  data.tar.gz: d9c1d471162694594c59165864ba48a2cd7bfcf0741e1dca4bd9246b769ea138
+  metadata.gz: 72d363ce5cf74c8c22e8a636c3b908f27be10cc5a21691acd480b791ab9e8c33
+  data.tar.gz: 36f3cb7fd5c1784c727def8ea7ad4c6f572f21c301ce99440440e4d5843f1e62
 SHA512:
-  metadata.gz: 950732591722d121efe13ee44003c6573dc5bbbedf7df3476c71324aa9b6dc9c26ff5c55fc4e7714ae4d34ae674d7e7f7f3cc48cdfc11ebf592145b56c601b6f
-  data.tar.gz: 9024824533e151b01dc9150f4daa850ac0b0cbbfb875ebf34bdce8a1a03d65c45125b6094f1cb61ab6cbf843cf537531e81a9db53b6d9f6137d16952ff1ff6c3
+  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,36 @@ 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
 
@@ -152,6 +155,10 @@ class TableInfoTool < FastMcp::Tool
   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

data/lib/rails_pg_extras_mcp/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsPgExtrasMcp
-  VERSION = "0.2.2"
+  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.2
+  version: 0.2.3
 platform: ruby
 authors:
 - pawurb