glue_gun_dsl 0.1.23 → 0.1.26

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e38dfcee3bde86f31b597d6dccf5138fb7200050a05904a055b7d8f201192ab0
-  data.tar.gz: 9b031f91fd59f57d8a10edef069d0ec656e04b8440d55af1354dda0cbfa124e4
+  metadata.gz: ea68a31c6e7206e0aeeb14467b1335a39047b49d870fd79cde4ce334c7c6d860
+  data.tar.gz: 67eb48676fc5502ca0869041ecd6c5a7d653005ea14777fe315828eca035fd42
 SHA512:
-  metadata.gz: b25d5acc5cc955c9c3b20147c6bbf81c2911860f7e7f32b48b4f21273bf7c1f992f05c6b21ed2e6353b3ace8ac39023f87dcbd100fc9041731bea3a62b5bdf2b
-  data.tar.gz: 69736bb98acb9e1365879c0f68f7c401cc6b675e4c7c21af9fee87c0ea074dc76fc103413a386b5b76b1c8e44534f2c3e2f163ed2242389e14dc80fc7c02e923
+  metadata.gz: 6830bf72b7f05306b46eb9b987f2bd15facc0510c1e2983b2c0a51802807f39518a5d6829dc499a54138a47e797e433ea16ee2b7ef1131b594e50677a64fcefc
+  data.tar.gz: 450f8dd49be6d2b0cdc9d840f4c03f404c1b7aaf69664ed3666b7e0cbef948952754e71ee12484a634b4661cb2e3a42e99eabc2e3d246319f3cc75e7c7598969

data/lib/glue_gun/model.rb

@@ -3,6 +3,40 @@ module GlueGun
   module Model
     extend ActiveSupport::Concern
 
+    class ServiceRegistry
+      attr_accessor :services
+
+      def register(k, v)
+        @services ||= {}
+        @services[k.to_sym] = v
+      end
+
+      def options
+        services.keys
+      end
+
+      def []=(k, v)
+        register(k, v)
+      end
+
+      def [](k)
+        return nil if k.nil?
+
+        @services ||= {}
+        @services.dig(k.to_sym)
+      end
+
+      def default_key
+        return unless services.keys.count == 1
+
+        services.keys.first
+      end
+
+      def default
+        default_key.present? ? services[default_key] : nil
+      end
+    end
+
     included do
       include GlueGun::Shared
 
@@ -15,17 +49,31 @@ module GlueGun
 
       # Set default service attribute name based on the class name
       self.service_attribute_name = "#{name.demodulize.underscore}_service".to_sym
+
+      class_attribute :service_registry
+      self.service_registry = ServiceRegistry.new
+
+      # Set default option key based on the class name
+      class_attribute :option_key
+      self.option_key = "#{name.demodulize.underscore}_type".to_sym
+
+      def assign_attributes(attributes)
+        return if attributes.blank?
+
+        attributes = attributes.deep_symbolize_keys
+        db_attributes = self.class.extract_db_attributes(attributes)
+
+        # Assign database attributes
+        super(db_attributes)
+
+        assign_service_attributes(attributes)
+      end
+      alias_method :attributes=, :assign_attributes
     end
 
     class_methods do
-      def service(class_or_proc = nil, &block)
-        if class_or_proc.is_a?(Class)
-          self.service_class = class_or_proc
-        elsif block_given?
-          self.service_class_resolver = block
-        else
-          raise ArgumentError, "You must provide a service class, factory, or a block to resolve the service class."
-        end
+      def service(key, service_class)
+        service_registry[key] = service_class
       end
 
       def find_or_create_by!(attributes)
@@ -35,11 +83,7 @@ module GlueGun
 
         record = where(db_attributes).first_or_initialize(attributes)
 
-        if record.new_record?
-          record.save!
-        else
-          record.send(:build_service_object, attributes)
-        end
+        record.save! if record.new_record?
         yield record if block_given?
 
         record
@@ -52,11 +96,7 @@ module GlueGun
 
         record = where(db_attributes).first_or_initialize(attributes)
 
-        if record.new_record?
-          record.save
-        else
-          record.send(:build_service_object, attributes)
-        end
+        record.save if record.new_record?
         yield record if block_given?
 
         record
@@ -72,13 +112,30 @@ module GlueGun
     end
 
     def initialize(attributes = {})
-      attributes[:root_dir] = detect_root_dir
+      attributes[:root_dir] ||= detect_root_dir
       attributes = attributes.deep_symbolize_keys
+      attributes[option_key] ||= resolve_service_type(attributes, true)
       db_attributes = self.class.extract_db_attributes(attributes)
       super(db_attributes)
       build_service_object(attributes)
     end
 
+    def assign_service_attributes(attributes)
+      return if attributes.blank?
+
+      service_object = instance_variable_get("@#{service_attribute_name}")
+      return unless service_object.present?
+
+      extract_service_attributes(attributes, service_object.class).each do |attr_name, value|
+        unless service_object.respond_to?("#{attr_name}=")
+          raise NoMethodError, "Undefined attribute #{attr_name} for #{service_object.class.name}"
+        end
+
+        service_object.send("#{attr_name}=", value)
+        attribute_will_change!(attr_name.to_s)
+      end
+    end
+
     private
 
     def build_service_object(attributes)
@@ -98,14 +155,26 @@ module GlueGun
       instance_variable_set("@#{service_attribute_name}", service_instance)
     end
 
+    def resolve_service_type(attributes, initializing = false)
+      attrs = if initializing || !persisted? || attributes.key?(self.class.option_key)
+                attributes
+              else
+                { self.class.option_key => send(self.class.option_key) }
+              end
+      attrs[self.class.option_key] || self.class.service_registry.default_key
+    end
+
     def resolve_service_class(attributes)
-      if self.class.service_class
-        self.class.service_class
-      elsif self.class.service_class_resolver
-        self.class.service_class_resolver.call(attributes)
-      else
-        raise "Service class not defined for #{self.class.name}"
+      type = resolve_service_type(attributes)
+      service_class = self.class.service_registry[type] || self.class.service_registry.default
+
+      unless service_class
+        available_types = self.class.service_registry.options
+        raise ArgumentError,
+              "#{self.class} requires argument #{self.class.option_key}. Invalid option key received: #{type}. Allowed options are: #{available_types}"
       end
+
+      service_class
     end
 
     def extract_service_attributes(attributes, service_class)
@@ -158,8 +227,9 @@ module GlueGun
             raise "Don't know how to serialize dependency of type #{dep}, available options are #{opts.keys}. You didn't specify an option."
           end
 
+          serialized = this_dep.respond_to?(:serialize) ? this_dep.serialize : this_dep.attributes
           hash[dep] = {
-            selected_option => service_object.send(dep).attributes
+            selected_option => serialized
           }
         end
       end

data/lib/glue_gun/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GlueGun
-  VERSION = "0.1.23"
+  VERSION = "0.1.26"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: glue_gun_dsl
 version: !ruby/object:Gem::Version
-  version: 0.1.23
+  version: 0.1.26
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-23 00:00:00.000000000 Z
+date: 2024-10-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel