groovy 0.6.0 → 0.6.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e3e87c857a45ab045967ed1af60d619ec7340fc751c950032645657049bd1c68
-  data.tar.gz: 4886e8db94b7fa3627312f82842d141b9c24ea7954e59955f4840dab7f70f776
+  metadata.gz: a622cdd3453e0fee1feeebf798cbcec9f0ef9f68759aec64f65c3d4a500f52af
+  data.tar.gz: be1220642bf299a61ad5900bea1fffa3359725575def843ce233290e26a5b8e2
 SHA512:
-  metadata.gz: df4c68551089cf9ade2f20ac70436962fff82c0a83db5caeb0208b107a1215443386ab429d644c939200fef844a70db315c502cfb4c901aff7c809f578a07b67
-  data.tar.gz: 8eb292ae1c4365040a62c4f9feb8e04b27119ae5b4d84b1adf0640cb7f1d46cbea29bafd57ad076ca0d0f593fae504f4bb6eb4afac392b530f725f82948b8ad1
+  metadata.gz: 9943e2715a0099f8b1b2795ab66259cbb5b71d5a782e4669b503e61f30a8df56fa57332f88c3592e3b69c0c63638da5d1ea471b102e62511028178886cd5b968
+  data.tar.gz: fa02339637f33c69bb37301f63d8b14b77d61b185f0b3c23d78712fb26159d2d5594dc2bb9bba7bb225211c4ab6a5adef167ddade4ea5b5999dae8ce2d5e5f07

data/lib/groovy/model.rb

@@ -13,6 +13,10 @@ module Groovy
   class Error < StandardError; end
   class RecordNotPersisted < Error; end
 
+  def self.models
+    @models ||= {}
+  end
+
   module Model
 
     def self.initialize_from_record(obj)
@@ -26,13 +30,23 @@ module Groovy
     # end
 
     def self.model_from_table(table_name)
-      Kernel.const_get(table_name.to_s.sub(/ies$/, 'y').sub(/s$/, '').classify)
+      get_class(table_name.to_s.sub(/ies$/, 'y').sub(/s$/, ''))
     end
 
     def self.included(base)
       base.extend(ClassMethods)
       base.include(Forwardable)
       base.table_name = base.name.sub(/y$/, 'ie') + 's'
+
+      # add to global model list
+      # Groovy.models[base.name] = base
+      Groovy.models[base.table_name] = base
+    end
+
+    def self.get_class(table_name)
+      # classify method
+      classified = table_name.gsub(/([A-Z])/, '_\1').split('_').collect! { |w| w.capitalize }.join
+      Kernel.const_get(classified)
     end
 
     module ClassMethods
@@ -86,6 +100,12 @@ module Groovy
         add_accessors_for(name)
       end
 
+      def add_reference(name, table_name, options = {})
+        schema.reference(name, table_name, options)
+        schema.sync
+        add_accessors_for(name)
+      end
+
       def add_accessors_for(col, s = schema)
         if s.attribute_columns.include?(col)
           add_attr_accessors(col)
@@ -242,10 +262,14 @@ module Groovy
         @callbacks ||= {}
       end
 
-      [:before_create, :after_create].each do |event|
-        define_method(event) do |*method_names|
-          callbacks[:before_create] ||= []
-          callbacks[:before_create].push(*method_names)
+      [:before_create, :after_create, :before_update, :after_update].each do |event|
+        define_method(event) do |*args, &block|
+          callbacks[event] ||= []
+          if block
+            callbacks[event].push(block)
+          else
+            callbacks[event].push(*args)
+          end
         end
       end
 
@@ -295,6 +319,7 @@ module Groovy
 
     def initialize(attrs = nil, record = nil, key = nil)
       @attributes, @vectors, @_key = {}, {}, key # key is used on creation only
+      @foreign_refs_to_update = {}
 
       if set_record(record)
         # load_attributes_from_record(record)
@@ -333,6 +358,10 @@ module Groovy
       # _key.nil?
     end
 
+    def persisted?
+      !new_record?
+    end
+
     def [](key)
       k = key.to_sym
       @attributes[k] = get_record_attribute(k) unless @attributes.key?(k)
@@ -366,6 +395,18 @@ module Groovy
       obj.each { |k,v| public_send("#{k}=", v) }
     end
 
+    def set_reverse_reference(key, obj, removing = false)
+      if self.class.schema.singular_references.include?(key.to_sym)
+        # puts "setting #{obj} as #{key}"
+        set_ref(key, removing ? nil : obj, true)
+      elsif self.class.schema.plural_references.include?(key.to_sym)
+        # puts "adding #{obj} to #{key}"
+        public_send(key).set_ref(obj, removing)
+      else
+        raise "Invalid reference name: #{name}"
+      end
+    end
+
     def update_attributes(obj)
       set_attributes(obj)
       return false if respond_to?(:invalid?) and invalid?
@@ -378,7 +419,11 @@ module Groovy
 
     def save(options = {})
       return false if respond_to?(:invalid?) and invalid?
-      new_record? ? create : update
+      if new_record?
+        create && true
+      else
+        update && true
+      end
     end
 
     def save!(options = {})
@@ -407,6 +452,7 @@ module Groovy
 
       @attributes = {}
       @changes = {}
+      @foreign_refs_to_update = {}
       self
     end
 
@@ -424,6 +470,7 @@ module Groovy
 
     private
 
+
     def get_record_attribute(key)
       return if record.nil?
       val = record[key]
@@ -460,12 +507,26 @@ module Groovy
       end
     end
 
-    def set_ref(name, obj)
-      if record.nil?
+
+    def set_ref(name, obj, from_foreign = false)
+      # puts "Setting ref #{name} -> #{obj} (#{obj.class})"
+
+      if !from_foreign && obj.is_a?(Model) && !obj.new_record?
+        if foreign_ref = self.class.schema.reverse_reference_of(name.to_sym)
+          # puts "Found circular ref (#{name} -> #{foreign_ref})"
+          if current = get_ref(name)
+            current.set_reverse_reference(foreign_ref, self, true)
+          end
+
+          @foreign_refs_to_update[name] = [foreign_ref, obj]
+        end
+      end
+
+      if record.nil? # not persisted yet
         set_attribute(name, obj.id) # obj should be a groovy model or groonga record
       else
-        obj = obj.record if obj.respond_to?(:record)
-        record[name] = obj
+        val = obj.respond_to?(:record) ? obj.record : obj
+        record[name] = val
       end
     end
 
@@ -477,17 +538,28 @@ module Groovy
     def create
       fire_callbacks(:before_create)
       set_record(self.class.insert(@attributes, @_key))
+      update_foreign_refs
       fire_callbacks(:after_create)
       self
     end
 
     def fire_callbacks(name)
+      # puts "Firing #{name} callbacks"
       if arr = self.class.callbacks[name] and arr.any?
-        arr.each { |fn| send(fn) }
+        arr.each { |fn| fire_callback(fn) }
+      end
+    end
+
+    def fire_callback(fn)
+      if fn.is_a?(Proc)
+        fn.call(self)
+      else
+        send(fn)
       end
     end
 
     def update
+      fire_callbacks(:before_update)
       ensure_persisted!
       changes.each do |key, values|
         # puts "Updating #{key} from #{values[0]} to #{values[1]}"
@@ -495,9 +567,22 @@ module Groovy
       end
       self.class.set_timestamp(record, :updated_at)
       changes = {}
+      update_foreign_refs
+      fire_callbacks(:after_update)
       self
     end
 
+    def update_foreign_refs
+      return if @foreign_refs_to_update.empty?
+
+      @foreign_refs_to_update.each do |attr, (ref, obj)|
+        # puts "setting #{obj.class}##{ref} to #{self}"
+        obj.set_reverse_reference(ref, self)
+      end
+
+      @foreign_refs_to_update = {}
+    end
+
     def ensure_persisted!
       raise "Not persisted" if new_record?
     end

data/lib/groovy/query.rb

@@ -238,7 +238,6 @@ module Groovy
     end
 
     def in_batches(of: 1000, from: nil, &block)
-      puts "of: #{of}"
       sorting[:limit] = of
       sorting[:offset] = from || 0
 

data/lib/groovy/schema.rb

@@ -33,6 +33,11 @@ module Groovy
       @table ||= context[table_name]
     end
 
+    # def [](key)
+    #   # @spec[key.to_sym]
+    #   table.columns.select { |c| c.column? && c.name.to_s == "#{table_name}.#{key.to_s}" }
+    # end
+
     def search_table
       @cache[:search_table] ||= context[SEARCH_TABLE_NAME]
     end
@@ -69,6 +74,22 @@ module Groovy
       get_names(table.columns.select { |c| c.column? && c.range.name == type })
     end
 
+    def refs_by_table
+      @refs_by_table ||= {}
+    end
+
+    def refs_by_name
+      @refs_by_name ||= {}
+    end
+
+    def reverse_reference_of(ref_name)
+      if ref_table = refs_by_name[ref_name.to_sym]
+        if foreign_model = Groovy.models[ref_table]
+          foreign_model.schema.refs_by_table[@table_name]
+        end
+      end
+    end
+
     def reload
       @cache = {}
     end
@@ -95,6 +116,14 @@ module Groovy
 
     def reference(name, table_name = nil, options = {})
       table_name = "#{name}s" if table_name.nil?
+
+      unless context[table_name]
+        log "Table #{table_name} doesn't exist yet! Creating now..."
+        create_table!(table_name)
+      end
+
+      refs_by_table[table_name] = name.to_sym
+      refs_by_name[name.to_sym] = table_name
       @spec[name.to_sym] = { type: :reference, args: [table_name, options] }
     end
 
@@ -108,9 +137,11 @@ module Groovy
 
       ensure_created!
       remove_columns_not_in(@spec.keys)
+
       @spec.each do |col, spec|
         check_and_add_column(col, spec[:type], spec[:args])
       end
+
       @index_columns.each do |col|
         add_index_on(col)
       end
@@ -186,15 +217,15 @@ module Groovy
       end
     end
 
-    def create_table!
-      log "Creating table!"
-      Groonga::Schema.create_table(table_name, context: context)
+    def create_table!(name = table_name)
+      log "Creating table #{name}!"
+      Groonga::Schema.create_table(name, context: context)
     end
 
-    def remove_table!
-      log "Removing table!"
-      Groonga::Schema.remove_table(table_name, context: context)
-      @table = nil
+    def remove_table!(name = table_name)
+      log "Removing table #{name}!"
+      Groonga::Schema.remove_table(name, context: context)
+      @table = nil if name == table_name
     end
 
     def get_names(columns)

data/lib/groovy/vector.rb

@@ -50,7 +50,9 @@ module Groovy
 
     def set(new_items)
       check_parent!
-      obj.record[key] = new_items
+      obj.record[key] = new_items.map do |obj|
+        obj.is_a?(Model) ? obj.record : obj
+      end
     end
 
     def has?(item)
@@ -61,17 +63,36 @@ module Groovy
     def push(item)
       raise "Invalid item type: #{item.class}" unless item.is_a?(Model)
       check_parent!
-      raise "Already in list!" if has?(item)
+      if has?(item)
+        # puts "Item already present #{item.inspect}"
+        return false
+      end
       item.save unless item.record
       push_record(item.record)
+      # size
+    end
+
+    def push!(item)
+      push or raise "Already in list!"
     end
 
     def remove(item)
       check_parent!
-      raise "Item not saved: #{record}" unless item.record
+      raise "Item not saved: #{item.inspect}" unless item.record
       remove_record(item.record)
     end
 
+    def set_ref(item, removing = false)
+      check_parent!
+      raise "Item not saved: #{item.inspect}" unless item.record
+
+      if removing
+        remove_record(item.record)
+      elsif !has?(item)
+        push_record(item.record)
+      end
+    end
+
     alias_method :<<, :push
 
     [:first, :last, :find_by, :search, :where, :not, :sort_by, :limit, :offset, :paginate, :in_batches].each do |scope_method|
@@ -80,6 +101,10 @@ module Groovy
       end
     end
 
+    def ==(arr)
+      to_a == arr
+    end
+
     private
     attr_reader :obj, :key
 

data/lib/groovy/version.rb

@@ -1,3 +1,3 @@
 module Groovy
-  VERSION = '0.6.0'.freeze
+  VERSION = '0.6.4'.freeze
 end

data/spec/callbacks_spec.rb

@@ -0,0 +1,58 @@
+require_relative './spec_helper'
+
+describe Groovy::Model do
+
+  before :all do
+    Groovy.open('tmp/callbacks', 'callbacks_spec')
+    load_callbacks_schema!
+  end
+
+  after :all do
+    Groovy.close('callbacks_spec')
+  end
+
+  def load_callbacks_schema!
+    klass = Class.new
+    Object.const_set("User", klass)
+    User.class_eval do
+      include Groovy::Model
+
+      after_create :created!
+      after_update do |record|
+        record.updated!
+      end
+
+      schema(context: 'callbacks_spec') do |t|
+        t.string :name
+        t.timestamps
+      end
+
+      def created!
+        # puts "created"
+      end
+
+      def updated!
+        # puts "updated"
+      end
+    end
+
+    User.add_reference :comments, "Comments", type: :vector
+  end
+
+  it 'fires created callbacks' do
+    user = User.new(name: 'John')
+    expect(user).to receive(:created!)
+    expect(user).not_to receive(:updated!)
+    expect(user.save).to eq(true)
+  end
+
+  it 'fires updated callbacks' do
+    user = User.new(name: 'John')
+    user.save
+    user.name = 'Josh'
+    expect(user).not_to receive(:created!)
+    expect(user).to receive(:updated!)
+    expect(user.save).to eq(true)
+  end
+
+end

data/spec/vector_spec.rb

@@ -0,0 +1,68 @@
+require_relative './spec_helper'
+
+describe Groovy::Model do
+
+  before :all do
+    Groovy.open('tmp/vector', 'vector_spec')
+    load_schema! 'vector_spec'
+  end
+
+  after :all do
+    Groovy.close('vector_spec')
+  end
+
+  def load_vector_schema!
+    klass = Class.new
+    Object.const_set("Comment", klass)
+    Comment.class_eval do
+      include Groovy::Model
+      schema(context: 'vector_spec') do |t|
+        t.string :content
+        t.timestamps
+      end
+    end
+
+    klass = Class.new
+    Object.const_set("User", klass)
+    User.class_eval do
+      include Groovy::Model
+      schema(context: 'vector_spec') do |t|
+        t.string :name
+        t.timestamps
+      end
+    end
+
+    Comment.add_reference :author, "Users"
+    User.add_reference :comments, "Comments", type: :vector
+  end
+
+  describe 'relations' do
+
+    before do
+      load_vector_schema!
+    end
+
+    it 'loads records and adds reverse relations automatically' do
+      user = User.new(name: 'John')
+      expect(user.save).to eq(true)
+      comment = Comment.new(author: user, content: 'Hello there!')
+      expect(comment.save).to eq(true)
+      # user.comments << comment
+      expect(user.reload.comments).to eq([comment])
+
+      user2 = User.new(name: 'Paul')
+      expect(user2.save).to eq(true)
+
+      comment.author = user2
+      comment.save
+
+      expect(user.reload.comments).to eq([])
+      expect(user2.reload.comments).to eq([comment])
+
+      comment.delete
+      expect(user2.reload.comments).to eq([])
+    end
+
+  end
+
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: groovy
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.6.4
 platform: ruby
 authors:
 - Tomás Pollak
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-11-27 00:00:00.000000000 Z
+date: 2023-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rroonga
@@ -97,11 +97,13 @@ files:
 - lib/groovy/types.rb
 - lib/groovy/vector.rb
 - lib/groovy/version.rb
+- spec/callbacks_spec.rb
 - spec/groovy_spec.rb
 - spec/model_spec.rb
 - spec/query_spec.rb
 - spec/search_spec.rb
 - spec/spec_helper.rb
+- spec/vector_spec.rb
 homepage: https://github.com/tomas/groovy
 licenses: []
 metadata: {}