firestore-odm 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6dbb42212308b32dcfd3d52e95f7b6d15a9dba8261ec9489541ecb94f46ba63
-  data.tar.gz: 0f7a0fa74de356f22fe7dceb6c83188b658b4dc566e4004b62498d40555a2793
+  metadata.gz: 9c19b9311e05a559ae948eb18b2c1dcd45bab13f779e856f55304928fffdfe9b
+  data.tar.gz: e25f77db855209bc069a7ca45aae3f776bbb03c47d983fd8e25abbed8e96b472
 SHA512:
-  metadata.gz: 29340ee23f122a2bc71cfd396d4c1aa33be49274990f1079fb4a81b599dd5d5cfbe014a76b2c6e3235e81de7f35891e8e84242a1b94b0fc913c0d04b59676714
-  data.tar.gz: d00dcfa5a1b1c2aacd2fe5d3e8696015bf8ab77e022a3ea57346aff787e98c36fa63e75ee2f3630bc10ddce3fbce22fb815c71b70cb862809011b230a44a34d2
+  metadata.gz: 04b5050dbb29a05f521b0bb82830808fc2cbf4af3beac1338d99b07daf4e754429317579b2c6f3c01912ac0fb42d865fe52bd5f81139c41ba091c6b8e8427c34
+  data.tar.gz: a87df1fadd1972422370c484f81f0550ffb1265e957763fa1338e51888ae3003ff84c6db8066e8f889234472065a3173307c4cfd13db1e43ea2a1014b8c8a5ce

data/lib/firestore-odm.rb

@@ -14,4 +14,68 @@ class Module
   def define_setter_method name, &block
     define_method("#{name}=", &block)
   end
+end
+
+module FirestoreODM
+  attr_reader :conversions
+
+  @conversions = {
+    String => :to_s,
+    Integer => :to_i,
+    Float => :to_f,
+    Date => :to_date,
+    DateTime => :to_datetime,
+    Time => :to_time
+  }
+
+  # Sets a conversion rule for a certain type.
+  # @param type [Class]
+  # @param target [Symbol, Proc]
+  def add_conversion type, target
+    @conversions[type] = target
+    return
+  end
+
+  # Gets a conversion.
+  # @param type [Class]
+  # @return [Proc]
+  def get_conversion type
+    case rule = conversions[type]
+    when Symbol
+      Proc.new { |value| value.method(rule).call }
+    when Proc
+      rule
+    else
+      raise "Conversion not found for type `#{type}`"
+    end
+  end
+
+  def more_than value
+    Proc.new { |o| value < o }
+  end
+
+  def min value
+    Proc.new { |o| value <= o }
+  end
+
+  def max value
+    Proc.new { |o| o <= value }
+  end
+
+  def less_than value
+    Proc.new { |o| o < value }
+  end
+
+  def check_option name, option
+    method = FirestoreODM.method name
+
+    case option
+    when Proc
+      check_option name, option.call
+    else
+      method.call option
+    end
+  end
+
+  extend self
 end

data/lib/firestore-odm/collection.rb

@@ -17,17 +17,21 @@ module FirestoreODM
     end
 
     # Creates a document.
-    # @return [FirestoreODM::Model] the document.
+    # @param opts [Hash] a hash with options.
+    # @yield [doc] 
+    # @return [FirestoreODM::Document] the document.
     def create opts = {}, &block
       block.call o = self.new
+
       doc = collection(opts[:relative_to]).add o.changes
       self.new doc
     end
 
-    # Creates a document.
+    # Creates a document with an specific id.
     # @param id [String] the document id.
-    # @param root [Google::Cloud::Firestore::DocumentReference]
-    # @return [FirestoreODM::Model] the document.
+    # @param opts [Hash]
+    # @yield [doc] 
+    # @return [FirestoreODM::Document] the document.
     def create_with_id id, opts = {}, &block
       block.call o = self.new
 
@@ -38,7 +42,8 @@ module FirestoreODM
 
     # Gets a document.
     # @param id [String] 
-    # @return [FirestoreODM::Model] the document.
+    # @param opts [Hash]
+    # @return [FirestoreODM::Document] the document.
     def from_id id, opts = {}
       doc = collection(opts[:relative_to]).doc id
       self.new doc

data/lib/firestore-odm/schema.rb

@@ -5,8 +5,27 @@ module FirestoreODM
     # @param type [Class] the field's type.
     # @param opts [Hash] the field's options.
     def field name, type, opts = {}
+      conversion = FirestoreODM.get_conversion type
+
+      list = []
+      
+      list << FirestoreODM.check_option(:min, opts[:min]) if opts[:min]
+      list << FirestoreODM.check_option(:max, opts[:max]) if opts[:max]
+      list << FirestoreODM.check_option(:less_than, opts[:less_than]) if opts[:less_than]
+      list << FirestoreODM.check_option(:more_than, opts[:more_than]) if opts[:more_than]
+
+      define_singleton_method("parse_#{name}") do |value|
+        list.reduce(true) { |result, o| result and o.call(value) }
+      end
+
       define_method(name) { self[name] }
-      define_setter_method(name) { |value| self[name] = value }
+      
+      define_method("#{name}=") do |value|
+        value = conversion.call value
+        raise "#{value} is not a valid value for field #{name}" unless self.class.method("parse_#{name}").call value
+        self[name] = value
+      end
+
       return
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: firestore-odm
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Felipe Cabrera
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-31 00:00:00.000000000 Z
+date: 2020-06-03 00:00:00.000000000 Z
 dependencies: []
 description: ''
 email: fecabrera0@outlook.com