axiom-sql-generator 0.1.0

data/config/yardstick.yml

@@ -0,0 +1,2 @@
+---
+threshold: 100

data/lib/axiom-sql-generator.rb

@@ -0,0 +1,3 @@
+# encoding: utf-8
+
+require 'axiom/sql/generator'

data/lib/axiom/sql/generator.rb

@@ -0,0 +1,61 @@
+# encoding: utf-8
+
+require 'date'
+require 'time'
+
+require 'axiom'
+
+require 'axiom/sql/generator/core_ext/date'
+require 'axiom/sql/generator/core_ext/date_time'
+
+require 'axiom/sql/generator/visitor'
+
+require 'axiom/sql/generator/identifier'
+require 'axiom/sql/generator/attribute'
+require 'axiom/sql/generator/direction'
+require 'axiom/sql/generator/literal'
+
+require 'axiom/sql/generator/function'
+require 'axiom/sql/generator/function/aggregate'
+require 'axiom/sql/generator/function/connective'
+require 'axiom/sql/generator/function/numeric'
+require 'axiom/sql/generator/function/predicate'
+require 'axiom/sql/generator/function/proposition'
+require 'axiom/sql/generator/function/string'
+
+require 'axiom/sql/generator/relation'
+require 'axiom/sql/generator/relation/unary'
+require 'axiom/sql/generator/relation/base'
+require 'axiom/sql/generator/relation/binary'
+require 'axiom/sql/generator/relation/materialized'
+require 'axiom/sql/generator/relation/set'
+
+require 'axiom/sql/generator/relation/insertion'
+
+require 'axiom/sql/generator/version'
+
+module Axiom
+  module SQL
+    module Generator
+
+      # Raised when an invalid relation is visited
+      class InvalidRelationError < StandardError; end
+
+      LEFT_PARENTHESIS  = '('.freeze
+      RIGHT_PARENTHESIS = ')'.freeze
+
+      # Return a parenthesized SQL statement (inline modification)
+      #
+      # @param [#to_s] sql
+      #
+      # @return [#to_s]
+      #   same instance as sql
+      #
+      # @api private
+      def self.parenthesize!(sql)
+        sql.insert(0, LEFT_PARENTHESIS) << RIGHT_PARENTHESIS
+      end
+
+    end # module Generator
+  end # module SQL
+end # module Axiom

data/lib/axiom/sql/generator/attribute.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module Axiom
+  module SQL
+    module Generator
+
+      # Generates an SQL statement for an attribute
+      module Attribute
+        include Identifier
+
+        # Visit an Attribute
+        #
+        # @param [Attribute] attribute
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def visit_axiom_attribute(attribute)
+          visit_identifier(attribute.name)
+        end
+
+      end # module Attribute
+    end # module Generator
+  end # module SQL
+end # module Axiom

data/lib/axiom/sql/generator/core_ext/date.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+# Extend Date with methods available in ruby 1.9
+class Date
+
+  ISO_8601_FORMAT = '%F'.freeze
+
+  # Return the Date in ISO8601 date format
+  #
+  # @return [#to_s]
+  #
+  # @todo Remove once backports adds this method
+  #
+  # @api private
+  def iso8601
+    date = frozen? ? dup : self
+    date.strftime(ISO_8601_FORMAT)
+  end
+
+end # class Date

data/lib/axiom/sql/generator/core_ext/date_time.rb

@@ -0,0 +1,46 @@
+# encoding: utf-8
+
+# Extend DateTime with methods available in ruby 1.9
+class DateTime
+
+  # TODO: remove the rbx guard when fractional seconds are handled properly in 1.9 mode
+  SEC_FRACTION_MULTIPLIER = RUBY_VERSION < '1.9' ? 60 * 60 * 24 : 1
+
+  # Return the DateTime in ISO8601 date-time format
+  #
+  # @param [Integer] time_scale
+  #   the number of significant digits to use for fractional seconds
+  #
+  # @return [#to_s]
+  #
+  # @todo Remove once backports adds this method
+  #
+  # @api private
+  def iso8601(time_scale = 0)
+    super() + iso8601_timediv(time_scale)
+  end
+
+private
+
+  # Return the time with fraction seconds
+  #
+  # @param [Integer] time_scale
+  #   the number of significant digits to use for fractional seconds
+  #
+  # @return [#to_s]
+  #
+  # @api private
+  def iso8601_timediv(time_scale)
+    date_time = frozen? ? dup : self
+
+    fractional_seconds = unless time_scale.zero?
+      '.%0*d' % [
+        time_scale,
+        date_time.sec_fraction * SEC_FRACTION_MULTIPLIER * 10**time_scale
+      ]
+    end
+
+    date_time.strftime("T%T#{fractional_seconds}%Z")
+  end
+
+end # class DateTime

data/lib/axiom/sql/generator/direction.rb

@@ -0,0 +1,38 @@
+# encoding: utf-8
+
+module Axiom
+  module SQL
+    module Generator
+
+      # Generates an SQL statement for a direction
+      module Direction
+        include Attribute
+
+        DESC = ' DESC'.freeze
+
+        # Visit an Ascending Direction
+        #
+        # @param [Relation::Operation::Order::Ascending] direction
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def visit_axiom_relation_operation_order_ascending(direction)
+          dispatch direction.attribute
+        end
+
+        # Visit an Descending Direction
+        #
+        # @param [Relation::Operation::Order::Descending] direction
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def visit_axiom_relation_operation_order_descending(direction)
+          dispatch(direction.attribute) << DESC
+        end
+
+      end # module Direction
+    end # module Generator
+  end # module SQL
+end # module Axiom

data/lib/axiom/sql/generator/function.rb

@@ -0,0 +1,55 @@
+# encoding: utf-8
+
+module Axiom
+  module SQL
+    module Generator
+
+      # Generates an SQL statement for a function expression
+      module Function
+        include Attribute, Literal
+
+      private
+
+        # Return the SQL for a umary prefix operation
+        #
+        # @param [#to_s] operator
+        #
+        # @param [Function::Unary] function
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def unary_prefix_operation_sql(operator, function)
+          "#{operator} (#{dispatch(function.operand)})"
+        end
+
+        # Return the SQL for a binary prefix operation
+        #
+        # @param [#to_s] operator
+        #
+        # @param [Function::Binary] function
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def binary_prefix_operation_sql(operator, function)
+          "#{operator} (#{dispatch(function.left)}, #{dispatch(function.right)})"
+        end
+
+        # Return the SQL for a binary infix operation
+        #
+        # @param [#to_s] operator
+        #
+        # @param [Function::Binary] function
+        #
+        # @return [#to_s]
+        #
+        # @api private
+        def binary_infix_operation_sql(operator, function)
+          "#{dispatch(function.left)} #{operator} #{dispatch(function.right)}"
+        end
+
+      end # module Function
+    end # module Generator
+  end # module SQL
+end # module Axiom

data/lib/axiom/sql/generator/function/aggregate.rb

@@ -0,0 +1,134 @@
+# encoding: utf-8
+
+module Axiom
+  module SQL
+    module Generator
+      module Function
+
+        # Generates an SQL statement for an aggregate function
+        module Aggregate
+          include Function
+
+          COUNT              = 'COUNT'.freeze
+          SUM                = 'SUM'.freeze
+          MINIMUM            = 'MIN'.freeze
+          MAXIMUM            = 'MAX'.freeze
+          MEAN               = 'AVG'.freeze
+          VARIANCE           = 'VAR_POP'.freeze
+          STANDARD_DEVIATION = 'STDDEV_POP'.freeze
+
+          # Visit a count aggregate function
+          #
+          # @param [Axiom::Aggregate::Count] count
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_count(count)
+            unary_prefix_operation_sql(COUNT, count)
+          end
+
+          # Visit a sum aggregate function
+          #
+          # @param [Axiom::Aggregate::Sum] sum
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_sum(sum)
+            aggregate_function_sql(SUM, sum)
+          end
+
+          # Visit a minimum aggregate function
+          #
+          # @param [Axiom::Aggregate::Minimum] minimum
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_minimum(minimum)
+            # TODO: wrap this in a coalesce operation once the default can be made sane
+            unary_prefix_operation_sql(MINIMUM, minimum)
+          end
+
+          # Visit a maximum aggregate function
+          #
+          # @param [Axiom::Aggregate::Maximum] maximum
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_maximum(maximum)
+            # TODO: wrap this in a coalesce operation once the default can be made sane
+            unary_prefix_operation_sql(MAXIMUM, maximum)
+          end
+
+          # Visit a mean aggregate function
+          #
+          # @param [Axiom::Aggregate::Mean] mean
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_mean(mean)
+            unary_prefix_operation_sql(MEAN, mean)
+          end
+
+          # Visit a variance aggregate function
+          #
+          # @param [Axiom::Aggregate::Variance] variance
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_variance(variance)
+            unary_prefix_operation_sql(VARIANCE, variance)
+          end
+
+          # Visit a standard deviation aggregate function
+          #
+          # @param [Axiom::Aggregate::StandardDeviation] standard_deviation
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_aggregate_standard_deviation(standard_deviation)
+            unary_prefix_operation_sql(STANDARD_DEVIATION, standard_deviation)
+          end
+
+        private
+
+          # Return the SQL
+          #
+          # @param [#to_s] operation
+          #
+          # @param [Axiom::Aggregate] aggregate
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def aggregate_function_sql(operation, aggregate)
+            default_when_null(
+              unary_prefix_operation_sql(operation, aggregate),
+              aggregate.finalize(aggregate.default)
+            )
+          end
+
+          # Specify a default value when SQL expression evaluates to NULL
+          #
+          # @param [#to_s] sql
+          #
+          # @param [Visitable] default
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def default_when_null(sql, default)
+            "COALESCE (#{sql}, #{dispatch(default)})"
+          end
+
+        end # module Aggregate
+      end # module Function
+    end # module Generator
+  end # module SQL
+end # module Axiom

data/lib/axiom/sql/generator/function/connective.rb

@@ -0,0 +1,53 @@
+# encoding: utf-8
+
+module Axiom
+  module SQL
+    module Generator
+      module Function
+
+        # Generates an SQL statement for a connective
+        module Connective
+          include Function
+
+          AND = 'AND'.freeze
+          OR  = 'OR'.freeze
+          NOT = 'NOT'.freeze
+
+          # Visit an Conjunction connective
+          #
+          # @param [Function::Connective::Conjunction] conjunction
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_function_connective_conjunction(conjunction)
+            Generator.parenthesize!(binary_infix_operation_sql(AND, conjunction))
+          end
+
+          # Visit an Disjunction connective
+          #
+          # @param [Function::Connective::Disjunction] disjunction
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_function_connective_disjunction(disjunction)
+            Generator.parenthesize!(binary_infix_operation_sql(OR, disjunction))
+          end
+
+          # Visit an Negation connective
+          #
+          # @param [Function::Connective::Negation] negation
+          #
+          # @return [#to_s]
+          #
+          # @api private
+          def visit_axiom_function_connective_negation(negation)
+            unary_prefix_operation_sql(NOT, negation)
+          end
+
+        end # module Connective
+      end # module Function
+    end # module Generator
+  end # module SQL
+end # module Axiom