veritas-sql-generator 0.0.3 → 0.0.4

data/lib/veritas/sql/generator/function/proposition.rb

@@ -0,0 +1,38 @@
+# encoding: utf-8
+
+module Veritas
+  module SQL
+    module Generator
+      module Function
+
+        # Generates an SQL statement for a proposition function
+        module Proposition
+          include Function
+
+          # Visit a Tautology
+          #
+          # @param [Function::Proposition::Tautology] _tautology
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_veritas_function_proposition_tautology(_tautology)
+            TRUE
+          end
+
+          # Visit a Contradiction
+          #
+          # @param [Function::Proposition::Contradiction] _contradiction
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_veritas_function_proposition_contradiction(_contradiction)
+            FALSE
+          end
+
+        end # module Proposition
+      end # module Function
+    end # module Generator
+  end # module SQL
+end # module Veritas

data/lib/veritas/sql/generator/function/string.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+
+module Veritas
+  module SQL
+    module Generator
+      module Function
+
+        # Generates an SQL statement for a string function
+        module String
+          include Function
+
+          LENGTH = 'LENGTH'.freeze
+
+          # Visit a Length function
+          #
+          # @param [Function::String::Length] length
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_veritas_function_string_length(length)
+            unary_prefix_operation_sql(LENGTH, length)
+          end
+
+        end # module String
+      end # module Function
+    end # module Generator
+  end # module SQL
+end # module Veritas

data/lib/veritas/sql/generator/identifier.rb

@@ -18,7 +18,8 @@ module Veritas
         #
         # @api private
         def visit_identifier(identifier)
-          "#{QUOTE}#{identifier.to_s.gsub(QUOTE, ESCAPED_QUOTE)}#{QUOTE}"
+          escaped = identifier.to_s.gsub(QUOTE, ESCAPED_QUOTE)
+          escaped.insert(0, QUOTE) << QUOTE
         end
 
       end # module Identifier

data/lib/veritas/sql/generator/literal.rb

@@ -7,15 +7,13 @@ module Veritas
       # Generates an SQL statement for a literal
       module Literal
 
-        TRUE            = 'TRUE'.freeze
-        FALSE           = 'FALSE'.freeze
-        NULL            = 'NULL'.freeze
-        QUOTE           = "'".freeze
-        ESCAPED_QUOTE   = "''".freeze
-        SEPARATOR       = ', '.freeze
-        DATE_FORMAT     = '%F'.freeze
-        DATETIME_FORMAT = "#{DATE_FORMAT}T%T.%N%Z".freeze
-        TIME_SCALE      = 9
+        TRUE          = 'TRUE'.freeze
+        FALSE         = 'FALSE'.freeze
+        NULL          = 'NULL'.freeze
+        QUOTE         = "'".freeze
+        ESCAPED_QUOTE = "''".freeze
+        SEPARATOR     = ', '.freeze
+        TIME_SCALE    = 9
 
         # Returns an unfrozen object
         #
@@ -41,12 +39,14 @@ module Veritas
         #
         # @api private
         def visit_enumerable(enumerable)
-          "(#{enumerable.map { |entry| dispatch entry }.join(SEPARATOR)})"
+          Generator.parenthesize!(
+            enumerable.map { |entry| dispatch entry }.join(SEPARATOR)
+          )
         end
 
         # Visit a String
         #
-        # @note The string must be utf-8 encoded
+        # @note The string must be UTF-8 encoded
         #
         # @param [String] string
         #
@@ -54,7 +54,8 @@ module Veritas
         #
         # @api private
         def visit_string(string)
-          "#{QUOTE}#{string.gsub(QUOTE, ESCAPED_QUOTE)}#{QUOTE}"
+          escaped = string.gsub(QUOTE, ESCAPED_QUOTE)
+          escaped.insert(0, QUOTE) << QUOTE
         end
 
         # Visit a Numeric
@@ -86,11 +87,9 @@ module Veritas
         #
         # @return [#to_s]
         #
-        # @todo use Date#iso8601 when added to 1.8.7 by backports
-        #
         # @api private
         def visit_date(date)
-          dispatch Literal.dup_frozen(date).strftime(DATE_FORMAT)
+          dispatch date.iso8601
         end
 
         # Visit a DateTime and return in ISO 8601 date-time format
@@ -101,11 +100,9 @@ module Veritas
         #
         # @return [#to_s]
         #
-        # @todo use DateTime#iso8601(TIME_SCALE) when added to 1.8.7 by backports
-        #
         # @api private
         def visit_date_time(date_time)
-          dispatch date_time.new_offset.strftime(DATETIME_FORMAT)
+          dispatch date_time.new_offset.iso8601(TIME_SCALE)
         end
 
         # Visit a Time

data/lib/veritas/sql/generator/relation.rb

@@ -11,7 +11,8 @@ module Veritas
 
         EMPTY_STRING = ''.freeze
         SEPARATOR    = ', '.freeze
-        ALL_COLUMNS  = '*'.freeze
+        STAR         = '*'.freeze
+        EMPTY_HASH   = {}.freeze
 
         # Return the alias name
         #
@@ -32,34 +33,21 @@ module Veritas
             when Veritas::Relation::Operation::Set    then self::Set
             when Veritas::Relation::Operation::Binary then self::Binary
             when Veritas::Relation::Operation::Unary  then self::Unary
-            when Veritas::BaseRelation                then self::Base
+            when Veritas::Relation::Base              then self::Base
             else
               raise InvalidRelationError, "#{relation.class} is not a visitable relation"
           end
           klass.new.visit(relation)
         end
 
-        # Return the subquery for the relation and identifier
-        #
-        # @param [#to_subquery] relation
-        #
-        # @param [#to_s] identifier
-        #   optional identifier, defaults to relation.name
-        #
-        # @return [#to_s]
-        #
-        # @api private
-        def self.subquery(relation, identifier = relation.name)
-          "(#{relation.to_subquery}) AS #{visit_identifier(identifier)}"
-        end
-
         # Initialize a Generator
         #
         # @return [undefined]
         #
         # @api private
         def initialize
-          @sql = EMPTY_STRING
+          @sql        = EMPTY_STRING
+          @extensions = {}
         end
 
         # Visit an object and generate SQL from each node
@@ -93,6 +81,29 @@ module Veritas
           @sql
         end
 
+        # Return the SQL for the unary relation
+        #
+        # @example
+        #   sql = unary_relation.to_s
+        #
+        # @return [#to_s]
+        #
+        # @api public
+        def to_s
+          return EMPTY_STRING unless visited?
+          generate_sql(query_columns)
+        end
+
+        # Return the SQL suitable for an subquery
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def to_subquery
+          return EMPTY_STRING unless visited?
+          Generator.parenthesize!(generate_sql(subquery_columns))
+        end
+
         # Test if a visitable object has been visited
         #
         # @example
@@ -105,6 +116,122 @@ module Veritas
           !@name.nil?
         end
 
+      private
+
+        # Return the columns to use in a query
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def query_columns
+          explicit_columns
+        end
+
+        # Return the columns to use in a subquery
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def subquery_columns
+          implicit_columns
+        end
+
+        # Return the implicit columns for the select list
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def implicit_columns
+          sql = [ STAR, column_list_for(@extensions) ]
+          sql.reject! { |fragment| fragment.empty? }
+          sql.join(SEPARATOR)
+        end
+
+        # Return the explicit columns for the select list
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def explicit_columns
+          @distinct.to_s + column_list_for(@extensions.merge(@columns || EMPTY_HASH))
+        end
+
+        # Return the list of columns
+        #
+        # @param [#map] columns
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def column_list_for(columns)
+          sql = columns.values_at(*@header)
+          sql.compact!
+          sql.join(SEPARATOR)
+        end
+
+        # Return a list of columns in a header
+        #
+        # @param [Veritas::Relation] relation
+        #
+        # @param [#[]] aliases
+        #   optional aliases for the columns
+        #
+        # @return [Hash]
+        #
+        # @api private
+        def columns_for(relation, aliases = EMPTY_HASH)
+          columns = {}
+          relation.header.each do |attribute|
+            columns[aliases.fetch(attribute, attribute)] = column_for(attribute, aliases)
+          end
+          columns
+        end
+
+        # Return the column for an attribute
+        #
+        # @param [Attribute] attribute
+        #
+        # @param [#[]] aliases
+        #   aliases for the columns
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def column_for(attribute, aliases)
+          if aliases.key?(attribute)
+            alias_for(attribute, aliases[attribute])
+          else
+            dispatch(attribute)
+          end
+        end
+
+        # Return the column alias for an attribute
+        #
+        # @param [#to_s] attribute
+        #
+        # @param [Attribute, nil] alias_attribute
+        #   attribute to use for the alias
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def alias_for(attribute, alias_attribute)
+          "#{dispatch(attribute)} AS #{dispatch(alias_attribute)}"
+        end
+
+        # Add extensions for extension and summarize queries
+        #
+        # @param [#each] extensions
+        #
+        # @return [undefined]
+        #
+        # @api private
+        def add_extensions(extensions)
+          extensions.each do |attribute, function|
+            @extensions[attribute] = alias_for(function, attribute)
+          end
+        end
+
       end # class Relation
     end # module Generator
   end # module SQL

data/lib/veritas/sql/generator/relation/binary.rb

@@ -13,17 +13,6 @@ module Veritas
           LEFT_NAME  = 'left'.freeze
           RIGHT_NAME = 'right'.freeze
 
-          # Return the subquery for the generator and identifier
-          #
-          # @param [#to_subquery] generator
-          #
-          # @return [#to_s]
-          #
-          # @api private
-          def self.subquery(generator, *)
-            generator.kind_of?(Base) ? generator.to_subquery : super
-          end
-
           # Visit an Join
           #
           # @param [Algebra::Join] join
@@ -32,6 +21,7 @@ module Veritas
           #
           # @api private
           def visit_veritas_algebra_join(join)
+            @header = join.header
             set_operation(JOIN)
             set_columns(join)
             set_operands(join)
@@ -47,6 +37,7 @@ module Veritas
           #
           # @api private
           def visit_veritas_algebra_product(product)
+            @header = product.header
             set_operation(PRODUCT)
             set_columns(product)
             set_operands(product)
@@ -54,27 +45,6 @@ module Veritas
             self
           end
 
-          # Return the SQL for the binary relation
-          #
-          # @example
-          #   sql = binary_relation.to_s
-          #
-          # @return [#to_s]
-          #
-          # @api public
-          def to_s
-            generate_sql(@columns)
-          end
-
-          # Return the SQL suitable for an subquery
-          #
-          # @return [#to_s]
-          #
-          # @api private
-          def to_subquery
-            generate_sql(ALL_COLUMNS)
-          end
-
         private
 
           # Generate the SQL using the supplied columns
@@ -85,26 +55,7 @@ module Veritas
           #
           # @api private
           def generate_sql(columns)
-            return EMPTY_STRING unless visited?
-            "SELECT #{columns} FROM #{left_subquery} #{@operation} #{right_subquery}"
-          end
-
-          # Return the left subquery
-          #
-          # @return [#to_s]
-          #
-          # @api private
-          def left_subquery
-            self.class.subquery(@left, LEFT_NAME)
-          end
-
-          # Return the right subquery
-          #
-          # @return [#to_s]
-          #
-          # @api private
-          def right_subquery
-            self.class.subquery(@right, RIGHT_NAME)
+            "SELECT #{columns} FROM #{@left.to_subquery} AS #{visit_identifier(LEFT_NAME)} #{@operation} #{@right.to_subquery} AS #{visit_identifier(RIGHT_NAME)}"
           end
 
           # Set the operation
@@ -151,20 +102,21 @@ module Veritas
             @name = [ @left.name, @right.name ].uniq.join(UNDERSCORE).freeze
           end
 
-          # Return a list of columns in a header
-          #
-          # @param [Veritas::Relation] relation
-          #
-          # @return [#to_s]
-          #
-          # @api private
-          def columns_for(relation)
-            relation.header.map { |attribute| dispatch(attribute) }.join(SEPARATOR)
-          end
-
           # Generates an SQL statement for base relation binary operands
           class Base < Relation::Base
 
+            # Return the SQL suitable for an subquery
+            #
+            # Does not parenthesize the query
+            #
+            # @return [#to_s]
+            #
+            # @api private
+            def to_subquery
+              return EMPTY_STRING unless visited?
+              generate_sql
+            end
+
           private
 
             # Generate the SQL for this base relation
@@ -173,7 +125,7 @@ module Veritas
             #
             # @api private
             def generate_sql(*)
-              visited? ? @from : EMPTY_STRING
+              @from
             end
 
           end # class Base