sql_tools 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b2ba309601d1b52dd6da36b4a8bb360a36d626887c422670c8ab9baeac3613b1
-  data.tar.gz: aed2136db1548ecebf953dade273f0827eaebc5f9669b3739fae7a4f243e53f3
+  metadata.gz: bcf9039011a7ccbab22cfb0dc525581dddf1bf64f648ef49e6d0385487833a15
+  data.tar.gz: af79326d174f0a70fd1f53fea96e2858214a0ac1f57254f582cffd24990473db
 SHA512:
-  metadata.gz: 5659b85ea66765857217a48c74c115e24e30a57d292fa035c45f51565b7a0a478ce76e7750785ad11b1643eda23a40a05f683a83051b105eeaab14dbc133927d
-  data.tar.gz: 12b1b67b6273fb448ed28eeb74d4a755cab78edaa026cefe617426367b1a53657cc355dbbee1c012febded7c2b562879e01e5ef482e8620ab279053090604c73
+  metadata.gz: 5417f2bf4e359e91f580fdc941e8101dde4846ec4119ae81b443fea1a011a39ad29795294d76601bc3030029fa07995b73bd216e821167097e90a3c060ae393c
+  data.tar.gz: d87c01b0bf9d724631479af9d8c023f349d06b0937108556682ebae19bac2ee9c31b52ed37e6e0f22d2f14e1db9ff145ccc727e8a4c62f6fd3e65808afccd75a

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.0"
 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.0
 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