sql_tools 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b2ba309601d1b52dd6da36b4a8bb360a36d626887c422670c8ab9baeac3613b1
-  data.tar.gz: aed2136db1548ecebf953dade273f0827eaebc5f9669b3739fae7a4f243e53f3
+  metadata.gz: f3b0a5d22da6c9be49114af27655dcffcfb1c0d6e2a045507f81947068dc4b84
+  data.tar.gz: 52003a39612fbb1cb6f778514235957e1024aa95ffd30f50b0f9bee48bba6ad5
 SHA512:
-  metadata.gz: 5659b85ea66765857217a48c74c115e24e30a57d292fa035c45f51565b7a0a478ce76e7750785ad11b1643eda23a40a05f683a83051b105eeaab14dbc133927d
-  data.tar.gz: 12b1b67b6273fb448ed28eeb74d4a755cab78edaa026cefe617426367b1a53657cc355dbbee1c012febded7c2b562879e01e5ef482e8620ab279053090604c73
+  metadata.gz: 94688c70de849c18bc490b236ad9242518cd659140a5978d6a20d05d619f0b3932d88299e831531a6b8f45d65ee91c1ef3cc6689bf1725bfc1f693f1240aed32
+  data.tar.gz: d825b27eea4c4ee7e96f64c6e4fa4a5cdc6e0fa65f7f46f4d17cbef7aa99c1c1f6b4d58db3df890b12a6a31d7f0ccbc8be3d386a6f94157d59883778cb6dc039

data/lib/sql_tools/column_definition.rb

@@ -0,0 +1,9 @@
+module SqlTools
+  class ColumnDefinition
+    def initialize(node)
+      @node = node
+    end
+
+    def name = @node.name.text
+  end
+end

data/lib/sql_tools/constraint.rb

@@ -0,0 +1,14 @@
+module SqlTools
+  class Constraint
+    class << self
+      def from(node)
+        if node.children.any? { |child| child.type == :keyword_key }
+          Index.new(node)
+        else
+          binding.b
+        end
+      end
+    end
+  end
+end
+

data/lib/sql_tools/index.rb

@@ -0,0 +1,28 @@
+module SqlTools
+  class Index
+    def initialize(node)
+      @node = node
+    end
+
+    def columns
+      @node.query(<<~QUERY).map do |match|
+        (ordered_columns (column name: (identifier) @column_name))
+      QUERY
+        match["column_name"].text
+      end
+    end
+
+    def primary_key?
+      @node.any? { |child| child.type == :keyword_primary } &&
+        @node.any? { |child| child.type == :keyword_key }
+    end
+
+    def name
+      return :primary_key if primary_key?
+
+      @node.name.text.delete("`")
+    end
+  end
+end
+
+

data/lib/sql_tools/schema.rb

@@ -0,0 +1,13 @@
+module SqlTools
+  class Schema
+    attr_accessor :create_table
+    attr_reader :columns, :constraints
+
+    def initialize
+      @columns = []
+      @constraints = []
+    end
+
+    def indices = @constraints.select { |c| c.is_a?(Index) }
+  end
+end

data/lib/sql_tools/schema_visitor.rb

@@ -0,0 +1,22 @@
+module SqlTools
+  class SchemaVisitor < TreeStand::Visitor
+    attr_reader :schema
+
+    def initialize(node)
+      super(node)
+      @schema = Schema.new
+    end
+
+    def on_create_table(node)
+      @schema.create_table = node
+    end
+
+    def on_column_definition(node)
+      @schema.columns << ColumnDefinition.new(node)
+    end
+
+    def on_constraint(node)
+      @schema.constraints << Constraint.from(node)
+    end
+  end
+end

data/lib/sql_tools/version.rb

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

data/lib/sql_tools.rb

@@ -10,6 +10,8 @@ module SqlTools
   class Error < StandardError; end
 
   class << self
+    def schema_from_sql(sql) = schema_from_tree(tree_from_sql(sql))
+    def schema_from_tree(tree) = SqlTools::SchemaVisitor.new(tree.root_node).visit.schema
     def query_from_sql(sql) = query_from_tree(tree_from_sql(sql))
     def query_from_tree(tree) = SqlTools::QueryVisitor.new(tree.root_node).visit.query
     def tree_from_sql(sql) = parser.parse_string(sql)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sql_tools
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - derekstride
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-07-03 00:00:00.000000000 Z
+date: 2024-07-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby_tree_sitter
@@ -52,7 +52,10 @@ files:
 - Rakefile
 - exe/install_tree_sitter_sql
 - lib/sql_tools.rb
+- lib/sql_tools/column_definition.rb
 - lib/sql_tools/common_table_expression.rb
+- lib/sql_tools/constraint.rb
+- lib/sql_tools/index.rb
 - lib/sql_tools/inner_join.rb
 - lib/sql_tools/left_join.rb
 - lib/sql_tools/predicate.rb
@@ -61,6 +64,8 @@ files:
 - lib/sql_tools/query.rb
 - lib/sql_tools/query_visitor.rb
 - lib/sql_tools/relation.rb
+- lib/sql_tools/schema.rb
+- lib/sql_tools/schema_visitor.rb
 - lib/sql_tools/selection.rb
 - lib/sql_tools/version.rb
 - rakelib/treesitter.rake