sqliterate 0.1.0

data/lib/sqliterate/grammar/literals.treetop

@@ -0,0 +1,40 @@
+module SQLiterate
+  grammar Literals
+    include Space
+    include Identifier
+    include Keywords
+    include String
+
+    rule literal_value
+      multi_string / numeric_literal / boolean_literal
+    end
+
+    rule positional_param
+      '$' digits <Node::PositionalParam>
+    end
+
+    rule numeric_literal
+      decimal_literal exponent_literal <Node::NumericLiteral>
+      /
+      decimal_literal
+    end
+
+    rule decimal_literal
+      ( digits ( '.' digits? )? / '.' digits ) <Node::DecimalLiteral>
+    end
+
+    rule exponent_literal
+      'e' s:( '+' / '-' )? digits <Node::ExponentLiteral>
+    end
+
+    rule digits
+      [0-9]+ <Node::Digits>
+    end
+
+    rule boolean_literal
+      true_kw <Node::Boolean::True>
+      /
+      false_kw <Node::Boolean::False>
+    end
+  end
+end

data/lib/sqliterate/grammar/operator.treetop

@@ -0,0 +1,65 @@
+module SQLiterate
+  grammar Operator
+    include Space
+    include Comment
+    include Keywords
+
+    rule gen_operator
+      symbol_operator / keyword_operator
+    end
+
+    rule keyword_operator
+      ( and_kw / or_kw / not_kw ) <Node::KeywordOperator>
+    end
+
+    rule symbol_operator
+      sign_ending_operator
+      /
+      nonsign_ending_operator
+      /
+      ambiguous_operator
+      /
+      single_char_operator
+    end
+
+    rule single_char_operator
+      c:op_char <Node::SingleCharOperator>
+    end
+
+    rule sign_ending_operator
+      b:( r:( basic_op_char !special_op_char )* basic_op_char )? special_op_char
+      o:( r:( op_char !(sign_op_char !op_char) )* op_char )? sign_op_char !op_char
+      <Node::SignEndingOperator>
+    end
+
+    rule nonsign_ending_operator
+      r:( op_char !(nonsign_op_char !op_char) )* op_char nonsign_op_char !op_char
+      <Node::NonsignEndingOperator>
+    end
+
+    rule ambiguous_operator
+      r:( op_char !(sign_op_char !op_char) )* op_char
+      <Node::AmbiguousOperator>
+    end
+
+    rule sign_op_char
+      c:[+\-] comment* <Node::OperatorChar>
+    end
+
+    rule basic_op_char
+      c:[+\-*/<>=] comment* <Node::OperatorChar>
+    end
+
+    rule special_op_char
+      c:[~!@#%^&|`?] comment* <Node::OperatorChar>
+    end
+
+    rule nonsign_op_char
+      c:[*/<>=~!@#%^&|`?] comment* <Node::OperatorChar>
+    end
+
+    rule op_char
+      c:[+\-*/<>=~!@#%^&|`?] comment* <Node::OperatorChar>
+    end
+  end
+end

data/lib/sqliterate/grammar/query.treetop

@@ -0,0 +1,25 @@
+module SQLiterate
+  grammar Query
+    include Space
+    include Keywords
+    include Expression
+
+    rule query
+      space with_section space query_expression space <Node::Query>
+    end
+
+    rule with_section
+      (with_kw required_space with_queries / space) <Node::WithSection>
+    end
+
+    rule with_queries
+      with_query r:( space ',' space with_query )* <Node::WithQueries>
+    end
+
+    rule with_query
+      identifier required_space as_kw required_space '('
+        space query_expression space
+      ')' <Node::WithQuery>
+    end
+  end
+end

data/lib/sqliterate/grammar/space.treetop

@@ -0,0 +1,17 @@
+module SQLiterate
+  grammar Space
+    include Comment
+
+    rule required_space
+      spacing+
+    end
+
+    rule space
+      spacing*
+    end
+
+    rule spacing
+      [ \n] / comment
+    end
+  end
+end

data/lib/sqliterate/grammar/string.treetop

@@ -0,0 +1,47 @@
+module SQLiterate
+  grammar String
+    include Identifier
+
+    rule multi_string
+      single_string t:( ' '* "\n" ' '* single_string )* <Node::MultiString>
+    end
+
+    rule single_string
+      std_string_constant / ext_string_constant
+    end
+
+    rule std_string_constant
+      "'" c:( std_character )* "'" <Node::StringConstant>
+    end
+
+    rule std_character
+      "''" <Node::Character::Quote>
+      /
+      [^'] <Node::Character::Text>
+    end
+
+    rule ext_string_constant
+      [eE] "'" c:( ext_character )* "'" <Node::StringConstant>
+    end
+
+    rule ext_character
+      "''" <Node::Character::Quote>
+      /
+      "\\" c:( ext_string_escape ) <Node::Character::ExtEscape>
+      /
+      [^\\'] <Node::Character::Text>
+    end
+
+    rule ext_string_escape
+      [bfnrt] <Node::Character::NotImplemented>
+      /
+      [0-7] 1..3 <Node::Character::NotImplemented>
+      /
+      'x' [0-9A-F] 1..2 <Node::Character::NotImplemented>
+      /
+      [uU] ([0-9A-F] 8..8 / [0-9A-F] 4..4) <Node::Character::NotImplemented>
+      /
+      [^xuU] <Node::Character::Text>
+    end
+  end
+end

data/lib/sqliterate/node/control.rb

@@ -0,0 +1,3 @@
+# coding: utf-8
+require 'sqliterate/node/control/expression'
+require 'sqliterate/node/control/query'

data/lib/sqliterate/node/control/expression.rb

@@ -0,0 +1,192 @@
+module SQLiterate
+  module Node
+    module GenExpression
+      def tables
+        gen_value.tables + r.elements.flat_map { |e| e.gen_value.tables }
+      end
+    end
+
+    module GenValue
+      module Literal
+        def tables
+          []
+        end
+      end
+      module Subscript
+        def tables
+          field_selection.tables + r.elements.flat_map do |e|
+            e.range_expression.tables
+          end
+        end
+      end
+    end
+
+    module FunctionCall
+      def tables
+        function_params.tables
+      end
+    end
+
+    module TypeCast
+      def tables
+        scalar_expression.tables
+      end
+    end
+
+    module VoidExpression
+      def tables
+        []
+      end
+    end
+
+    module AggregateExpression
+      def tables
+        expressions_list.tables + order_by_clause.tables
+      end
+    end
+
+    module OrderByClause
+      def tables
+        respond_to?(:expressions_list) ? expressions_list.tables : []
+      end
+    end
+
+    module ExpressionsList
+      def tables
+        gen_expression.tables + r.elements.flat_map do |e|
+          e.gen_expression.tables
+        end
+      end
+    end
+
+    module NamedExpressionsList
+      def tables
+        named_expression.tables + r.elements.flat_map do |e|
+          e.named_expression.tables
+        end
+      end
+    end
+
+    module NamedExpression
+      def tables
+        gen_expression.tables
+      end
+    end
+
+    module AllColumns
+      def tables
+        []
+      end
+    end
+
+    module RangeExpression
+      def tables
+        b.tables + e.tables
+      end
+    end
+
+    module FieldSelection
+      def tables
+        row_value.tables
+      end
+    end
+
+    module RowValue
+      module Query
+        def tables
+          query_expression.tables
+        end
+      end
+      module Expression
+        def tables
+          gen_expression.tables
+        end
+      end
+      module Identifier
+        def tables
+          []
+        end
+      end
+    end
+
+    module QueryExpression
+      def tables
+        query_value.tables + r.elements.flat_map { |e| e.query_value.tables }
+      end
+    end
+
+    module QueryValue
+      module Expression
+        def tables
+          query_expression.tables
+        end
+      end
+      module Select
+      end
+    end
+
+    module SelectQuery
+      def tables
+        select_list.tables + table_expression.tables
+      end
+    end
+
+    module TableExpression
+      def tables
+        from_clause.tables
+      end
+    end
+
+    module FromClause
+      def tables
+        respond_to?(:table_references) ? table_references.tables : []
+      end
+    end
+
+    module TableReferences
+      def tables
+        table_joins.tables + r.elements.flat_map { |e| e.table_joins.tables }
+      end
+    end
+
+    module TableJoins
+      def tables
+        [table_reference.table] + r.elements.map { |e| e.joined_table.table }
+      end
+    end
+
+    module TableReference
+      def table
+        if respond_to?(:table_spec)
+          [ table_name.name, table_spec.tables ]
+        else
+          [ name, [name] ]
+        end
+      end
+    end
+
+    module TableSpec
+      module Query
+        def tables
+          query_expression.tables
+        end
+      end
+      module Joins
+        def tables
+          table_joins.tables
+        end
+      end
+      module Table
+        def tables
+          [name]
+        end
+      end
+    end
+
+    module Join
+      def table
+        table_reference.table
+      end
+    end
+  end
+end

data/lib/sqliterate/node/control/query.rb

@@ -0,0 +1,39 @@
+module SQLiterate
+  module Node
+    module Query
+      def common_tables
+        Hash[ with_section.tables.map { |t| [t.first, expand_table(t)] } ]
+      end
+
+      def tables
+        cte = common_tables
+        query_expression.tables.flat_map { |t| expand_table(t) }
+                               .flat_map { |t| cte[t] || [t] }
+                               .sort.uniq
+      end
+
+      def expand_table(t)
+        a, b = t
+        b ? b.flat_map { |x| expand_table(x) } : [a]
+      end
+    end
+
+    module WithSection
+      def tables
+        respond_to?(:with_queries) ? with_queries.tables : []
+      end
+    end
+
+    module WithQueries
+      def tables
+        [with_query.table] + r.elements.map { |e| e.with_query.table }
+      end
+    end
+
+    module WithQuery
+      def table
+        [ identifier.name, query_expression.tables ]
+      end
+    end
+  end
+end

data/lib/sqliterate/node/evaluate.rb

@@ -0,0 +1,5 @@
+# coding: utf-8
+require 'sqliterate/node/evaluate/literals'
+require 'sqliterate/node/evaluate/string'
+require 'sqliterate/node/evaluate/operator'
+require 'sqliterate/node/evaluate/expression'