rasti-db 1.5.0 → 2.3.0

data/lib/rasti/db/data_source.rb

@@ -0,0 +1,18 @@
+module Rasti
+  module DB
+    class DataSource
+
+      attr_reader :db, :schema
+
+      def initialize(db, schema=nil)
+        @db = db
+        @schema = schema ? schema.to_sym : nil
+      end
+
+      def qualify(collection_name)
+        schema ? Sequel[schema][collection_name] : Sequel[collection_name]
+      end
+      
+    end
+  end
+end

data/lib/rasti/db/environment.rb

@@ -0,0 +1,32 @@
+module Rasti
+  module DB
+    class Environment
+
+      def initialize(data_sources)
+        @data_sources = data_sources
+      end
+
+      def data_source(name)
+        raise "Undefined data source #{name}" unless data_sources.key? name
+        data_sources[name]
+      end
+
+      def data_source_of(collection_class)
+        data_source collection_class.data_source_name
+      end
+
+      def qualify(data_source_name, collection_name)
+        data_source(data_source_name).qualify collection_name
+      end
+
+      def qualify_collection(collection_class)
+        data_source_of(collection_class).qualify collection_class.collection_name
+      end
+
+      private
+
+      attr_reader :data_sources
+
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/base.rb

@@ -0,0 +1,17 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        class Base
+
+          def filter_condition_for(comparison_name, identifier, argument)
+            type = type_for argument
+            raise UnsupportedTypeComparison.new(type, comparison_name) unless type.respond_to?(comparison_name)
+            type.public_send comparison_name, identifier, argument.value
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/postgres.rb

@@ -0,0 +1,21 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        class Postgres < Base
+
+          PG_TYPES = {
+            array: Types::PGArray
+          }
+
+          private
+
+          def type_for(argument)
+            PG_TYPES.fetch(argument.type, Types::Generic)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/sqlite.rb

@@ -0,0 +1,21 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        class SQLite < Base
+
+          SQLITE_TYPES = {
+            array: Types::SQLiteArray
+          }
+
+          private
+
+          def type_for(argument)
+            SQLITE_TYPES.fetch(argument.type, Types::Generic)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/types/generic.rb

@@ -0,0 +1,49 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        module Types
+          class Generic
+
+            def self.equal(identifier, value)
+              {identifier => value}
+            end
+
+            def self.not_equal(identifier, value)
+              Sequel.negate equal(identifier, value)
+            end
+
+            def self.greater_than(identifier, value)
+              identifier > value
+            end
+
+            def self.greater_than_or_equal(identifier, value)
+              identifier >= value
+            end
+
+            def self.less_than(identifier, value)
+              identifier < value
+            end
+
+            def self.less_than_or_equal(identifier, value)
+              identifier <= value
+            end
+
+            def self.like(identifier, value)
+              Sequel.ilike identifier, value
+            end
+
+            def self.include(identifier, value)
+              Sequel.ilike identifier, "%#{value}%"
+            end
+
+            def self.not_include(identifier, value)
+              ~include(identifier, value)
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/types/pg_array.rb

@@ -0,0 +1,32 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        module Types
+          class PGArray
+
+            def self.equal(identifier, values)
+              Sequel.&(
+                Sequel.pg_array(identifier).contains(Sequel.pg_array(values)),
+                Sequel.pg_array(identifier).contained_by(Sequel.pg_array(values))
+              )
+            end
+
+            def self.not_equal(identifier, values)
+              ~equal(identifier, values)
+            end
+
+            def self.include(identifier, values)
+              Sequel.pg_array(identifier).overlaps Sequel.pg_array(values)
+            end
+
+            def self.not_include(identifier, values)
+              ~include(identifier, values)
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/types/sqlite_array.rb

@@ -0,0 +1,34 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        module Types
+          class SQLiteArray
+
+            def self.equal(identifier, values)
+              array = values.map { |value| "\"#{value}\"" }.join(',')
+              {identifier => "[#{array}]"}
+            end
+
+            def self.not_equal(identifier, values)
+              Sequel.|(*values.map { |value| ~Sequel.like(identifier, "%\"#{value}\"%") })
+            end
+
+            def self.like(identifier, values)
+              Sequel.|(*values.map { |value| Sequel.like(identifier, "%#{value}%") })
+            end
+
+            def self.include(identifier, values)
+              Sequel.|(*values.map { |value| Sequel.like(identifier, "%\"#{value}\"%") })
+            end
+
+            def self.not_include(identifier, values)
+              Sequel.&(*values.map { |value| ~Sequel.like(identifier, "%\"#{value}\"%") })
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/filter_condition_strategies/unsupported_type_comparison.rb

@@ -0,0 +1,22 @@
+module Rasti
+  module DB
+    module NQL
+      module FilterConditionStrategies
+        class UnsupportedTypeComparison < StandardError
+
+          attr_reader :argument_type, :comparison_name
+
+          def initialize(argument_type, comparison_name)
+            @argument_type = argument_type
+            @comparison_name = comparison_name
+          end
+
+          def message
+            "Unsupported comparison #{comparison_name} for #{argument_type}"
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/nodes/array_content.rb

@@ -0,0 +1,21 @@
+module Rasti
+  module DB
+    module NQL
+      module Nodes
+        class ArrayContent < Treetop::Runtime::SyntaxNode
+
+          def values
+            [left.value] + right_value
+          end
+
+          private
+
+          def right_value
+            right.is_a?(self.class) ? right.values : [right.value]
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/nodes/attribute.rb

@@ -0,0 +1,37 @@
+module Rasti
+  module DB
+    module NQL
+      module Nodes
+        class Attribute < Treetop::Runtime::SyntaxNode
+
+          def identifier(collection_class)
+            if computed? collection_class
+              collection_class.computed_attributes[column].identifier
+            else
+              tables.empty? ? Sequel[column] : Sequel[tables.join('__').to_sym][column]
+            end
+          end
+
+          def tables
+            _tables.elements.map{ |e| e.table.text_value }
+          end
+
+          def column
+            _column.text_value.to_sym
+          end
+
+          def computed_attributes(collection_class)
+            computed?(collection_class) ? [column] : []
+          end
+
+          private
+
+          def computed?(collection_class)
+            collection_class.computed_attributes.key? column
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rasti/db/nql/nodes/binary_node.rb

@@ -8,6 +8,10 @@ module Rasti
             values.flat_map(&:dependency_tables)
           end
 
+          def computed_attributes(collection_class)
+            left.computed_attributes(collection_class) | right.computed_attributes(collection_class)
+          end
+
           def values
             @values ||= values_for(left) + values_for(right)
           end

data/lib/rasti/db/nql/nodes/comparisons/base.rb

@@ -6,7 +6,21 @@ module Rasti
           class Base < Treetop::Runtime::SyntaxNode
 
             def dependency_tables
-              field.tables.empty? ? [] : [field.tables.join('.')]
+              attribute.tables.empty? ? [] : [attribute.tables.join('.')]
+            end
+
+            def computed_attributes(collection_class)
+              attribute.computed_attributes(collection_class)
+            end
+
+            def filter_condition(collection_class)
+              DB.nql_filter_condition_for comparison_name, attribute.identifier(collection_class), argument
+            end
+
+            private
+
+            def comparison_name
+              Inflecto.underscore(Inflecto.demodulize(self.class)).to_sym
             end
 
           end

data/lib/rasti/db/nql/nodes/comparisons/equal.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class Equal < Base
 
-            def filter_condition
-              { field.identifier => argument.value }
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/greater_than.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class GreaterThan < Base
 
-            def filter_condition
-              field.identifier > argument.value
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/greater_than_or_equal.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class GreaterThanOrEqual < Base
 
-            def filter_condition
-              field.identifier >= argument.value
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/include.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class Include < Base
 
-            def filter_condition
-              Sequel.ilike(field.identifier, "%#{argument.value}%")
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/less_than.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class LessThan < Base
 
-            def filter_condition
-              field.identifier < argument.value
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/less_than_or_equal.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class LessThanOrEqual < Base
 
-            def filter_condition
-              field.identifier <= argument.value
-            end
-            
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/like.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class Like < Base
 
-            def filter_condition
-              Sequel.ilike(field.identifier, argument.value)
-            end
-
           end
         end
       end

data/lib/rasti/db/nql/nodes/comparisons/not_equal.rb

@@ -5,10 +5,6 @@ module Rasti
         module Comparisons
           class NotEqual < Base
 
-            def filter_condition
-              Sequel.negate(field.identifier => argument.value)
-            end
-
           end
         end
       end