yaml-file-db 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dcccc03ffdf919d10a071489113df40c30a6bdc7895648b333ce3fc892a95322
-  data.tar.gz: 9a5b1b311ed1cf15c73ecdc5fe2e33bd2e3471973d7909569d25881f579081b8
+  metadata.gz: cdec368220bb1f0c3075bc77ce04deee5d71b589dedfe4768c12a91c963b5373
+  data.tar.gz: 64a96c084ff0d6a622c787c84632fb9163f8218151e1575a71b0ca02bc10413d
 SHA512:
-  metadata.gz: c8569acad2ff389a8827dfb4203aa56525de1465849f01be2fa3f1f04742015d8fc615bf7c778f463ad8e752ceee1254928d268e3e3ee7568e704a5f4f2cd340
-  data.tar.gz: c4189b21b1b971969a0f2dfe3d2bb48ca181689f827b8e4bedefdd1033bb263f531c7248da2a364ef7d23d195ef802a58f2bb1c8f4ebd6ac4e668212e96684a8
+  metadata.gz: 83182ca75c22109d1e0a8c117cfae4d75551b9d6f363b4a244d596864fb9b6e657e49b29312ea1b235a0e607ca656a69238acbfa88a4359d7303994d16e7da69
+  data.tar.gz: 2d7e1a93edefba627b2da0bb473398e35dd84343070268256eb3b198d60cfd3bb1b24ad0aa0f16fc621efb1fb8c117ee39f2c9b8b7e9dff2d6d4ac69611df987

data/lib/yaml-file-db/database.rb

@@ -14,6 +14,7 @@ module YDB
     def build()
       build_tables()
       build_relationships()
+      check_relationships()
       self
     end
 
@@ -43,17 +44,37 @@ module YDB
     end
 
     def build_relationships()
+      keywords = keywords()
+      iterate_over_rows do |row|
+        row.build_relationships(self, keywords)
+      end
+    end
+
+    def check_relationships()
+      keywords = keywords()
+      iterate_over_rows do |row|
+        row.check_relationships(self, keywords)
+      end
+    end
+
+    private
+
+    def keywords()
       keywords = []
       (self.instance_variables - INTERNAL_VARS).each do |var|
         keywords << var.to_s[1..-1]
         keywords << var.to_s[1..-1].singularize
       end
+      keywords
+    end
+
+    def iterate_over_rows(&block)
       self.instance_variables.each do |var|
         next if INTERNAL_VARS.include? var
         table = instance_variable_get var
         table.each do |id, row|
           begin
-            row.build_relationships(self, keywords)
+            block.call(row)
           rescue ValidationError => error
             @errors << "#{row.source}: #{error.to_s}"
           end

data/lib/yaml-file-db/row.rb

@@ -13,26 +13,41 @@ module YDB
     end
 
     def build_relationships(db, keywords)
-      self.instance_variables.each do |var|
-        next if INTERNAL_VARS.include? var
-        keyword = var.to_s[1..-1]
-
-        if keywords.include? keyword
-          if keyword.pluralize == keyword
-            array = instance_variable_get var
-            entities = db.public_send(keyword.to_sym)
-            value = []
-            array.each do |primary_key|
-              entity = entities[primary_key]
-              raise ValidationError.new("invalid primary_key #{primary_key}") if entity.nil?
-              value << entity
+      iterate_over_columns do |key, value|
+        if keywords.include?(key)
+          if key.pluralize == key
+            rows = []
+            table = db.public_send(key.to_sym)
+            value.each do |primary_key|
+              row = table[primary_key]
+              raise ValidationError.new("invalid primary_key: #{primary_key} is not part of #{key}") if row.nil?
+              rows << row
             end
-            instance_variable_set("@#{keyword}", value)
+            instance_variable_set("@#{key}", rows)
           else
-            primary_key = instance_variable_get var
-            entity = db.public_send(keyword.pluralize.to_sym)[primary_key]
-            raise ValidationError.new("invalid primary_key #{primary_key}") if entity.nil?
-            instance_variable_set("@#{keyword}", entity)
+            row = db.public_send(key.pluralize.to_sym)[value]
+            raise ValidationError.new("invalid primary_key: #{primary_key} is not part of #{key}") if row.nil?
+            instance_variable_set("@#{key}", row)
+          end
+        end
+      end
+    end
+
+    def check_relationships(db, keywords)
+      iterate_over_columns do |key, value|
+        if keywords.include?(key)
+          # move row into array to make it iterable
+          value = [value] if value.is_a?(YDB::Row)
+          value.each do |row|
+            if row.respond_to?(self.class.to_s.downcase.to_sym)
+              unless row.public_send(self.class.to_s.downcase.to_sym) == self
+                raise ValidationError.new("inconsistent relationship: #{row.id} doesn't link back to #{self.id}")
+              end
+            elsif row.respond_to?(self.class.to_s.downcase.pluralize.to_sym)
+              unless row.public_send(self.class.to_s.downcase.pluralize.to_sym).include?(self)
+                raise ValidationError.new("inconsistent relationship: #{row.id} doesn't link back to #{self.id}")
+              end
+            end
           end
         end
       end
@@ -53,11 +68,20 @@ module YDB
 
       doc.each do |name, value|
         instance_variable_set("@#{name}", value)
-        next if self.respond_to? name.to_sym
+        next if self.respond_to?(name.to_sym)
         self.class.send("attr_reader", name.to_sym)
       end
     end
 
+    def iterate_over_columns(&block)
+      self.instance_variables.each do |var|
+        next if INTERNAL_VARS.include?(var)
+        key = var.to_s[1..-1]
+        value = instance_variable_get(var)
+        block.call(key, value)
+      end
+    end
+
     def validate_filename()
       raise ValidationError.new("invalid filename") unless self.id =~ /^[\w-]+$/
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yaml-file-db
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Louis-Philippe Gauthier
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-02-15 00:00:00.000000000 Z
+date: 2021-02-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport