queryable_pstore 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6c5824998d9e38d575dd1dcf003ee15a3bcbb3a5
-  data.tar.gz: 21ac5f9b08055b31647061d83724d1b24b3b9341
+  metadata.gz: f5720955066adad1544e426f2f2294e8be6a7583
+  data.tar.gz: 7c4efd3bbecd436764f2e7ffd67edb8d0b9f8a8c
 SHA512:
-  metadata.gz: 911b154063bca79eefc44859cbfa7214d5d16dbf120e7c4d15d9671f9e67604773fc15e9a8aa9b06979ddfb8e9789203821bd70c64a85c21c86813c71c1316f3
-  data.tar.gz: 5e32dfea85ab83ca7e8c72e9f2c1b5a9883553bc928cd57f5cb7fc9079182adab0e6450fcbd42aa5210e59e95cc1c39409cf909163504bf87e6df6e9ea33b3e6
+  metadata.gz: 3f5165b82bac9c9abb2605ad517241df8a47b86d8c13bfce626f4211bfb821c3cbd64b3cd2a461dcb208e46df8757e9557749fc52e88d2d770cf819507e6616d
+  data.tar.gz: e2e59afff0482a82d3831484ca12d3759239702a80af0401b7c57a7ac95d262009e6446add04ca7d606800be54ba17e20d0142f43f76c7977860cf042aa49f6e

data/lib/csv_converter.rb

@@ -0,0 +1,20 @@
+  class CSVConverter
+    def initialize(convert)
+      @key = convert.keys.first
+      @conversion = convert[@key]
+    end
+
+    def convert(hash)
+      case @conversion
+      when :integer
+        hash[@key] = hash[@key].to_i
+      when :float
+        hash[@key] = hash[@key].to_f
+      else
+        raise ArgumentError.new("Unknown converter: `#{@conversion}`")
+      end
+
+      hash
+    end
+
+  end

data/lib/query.rb

@@ -0,0 +1,68 @@
+  class Query
+    attr_reader :attribute, :condition, :argument
+    TERMINATING_FUNCTIONS = [:pluck]
+
+    def initialize(attribute, condition, argument)
+      @attribute = attribute
+      @condition = condition
+      @argument = argument
+    end
+
+    def valid?(records)
+      valid = attribute_present?(records) || argument_is_block? || terminating_function?
+      raise ArgumentError.new("The attribute `#{@attribute}` is not present in the PStore.") unless valid
+      valid
+    end
+
+    def attribute_present?(records)
+      records.any? { |record| record.keys.include? @attribute }
+    end
+
+    def argument_is_block?
+      @argument.respond_to?(:call)
+    end
+
+    def terminating_function?
+      TERMINATING_FUNCTIONS.include?(@condition)
+    end
+
+    def filter(results)
+      if @attribute.empty?
+        handle_terminating_function(results)
+      else
+        results.select do |result|
+          conditional(result, @condition, @argument)
+        end
+      end
+    end
+
+    def handle_terminating_function(results)
+      case @condition
+      when :pluck
+        results.map { |r| r[@argument] }
+      else
+        raise "Unknown Terminating Function: #{@condition}"
+      end
+    end
+
+    def conditional(row, condition, argument)
+      thing_to_check = row[@attribute] # for direct comparison methods
+
+      case condition
+      when :gt
+        thing_to_check > argument
+      when :lt
+        thing_to_check < argument
+      when :eq
+        thing_to_check == argument
+      when :between
+        argument.include?(thing_to_check) 
+      when :ilike
+        thing_to_check.downcase.include? argument.downcase
+      when :lambda
+        argument.call(OpenStruct.new(row)) # lambda takes the whole row
+      else
+        raise "Unknown Conditional: #{condition}"
+      end
+    end
+  end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: queryable_pstore
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Jacob Smith
@@ -18,6 +18,8 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/csv_converter.rb
+- lib/query.rb
 - lib/queryable_pstore.rb
 homepage: https://github.com/jacobsmith/queryable_pstore
 licenses: