standard_procedure_fabrik 0.2.4 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd5573ca4788578745a5a30a0bccaefda4dc491f8535ea52399e7b6ab595900d
-  data.tar.gz: 4ff630d03615fc92cd670c40d903807587805ca1d865ae61a1279df664a62c25
+  metadata.gz: 3a9563aec3880d3a7863b0c4d177abac5a3e681063d6a92f507de0aedd4861d5
+  data.tar.gz: ce1e4af35a3b7f8e454c680cee91c1fd79f063e341ae78f721ec6d14cd6976b2
 SHA512:
-  metadata.gz: 5dc48f0c99f743cdad6d48b1b18ea7ce162b1419b69e1966e66061d6a6ccc513333bdf655ef3dc6a9b3c15f06e2448c322abab5e44a615e4eb769a21e2fdc892
-  data.tar.gz: db7c8a0bc3ef925b400a9ef00a0106dce6e0adc5082aed88a0b4fb8c1853873ece84711e4a7db6dc5a06c8eeccfb1e6c6def6c050a8b048ea57d8c8a00a14a69
+  metadata.gz: b24fb0e5485bd707f2a4580dbbb7c5da45956141ce54e45ef6dedbcfc2e27f31312ca93e8db02dc832e2e3438983ded4e7f9bae5c3eb57687f33c6495dc0edda
+  data.tar.gz: a1f7fb3e608bea037480f6a5bde881c05415d177a18d094a3f911848007af635cb88d6ebb1d0ec9f8b4d4a09827bef6b1b7aaf8f2f90886b48ccaeac80a8778e

data/README.md

@@ -109,6 +109,32 @@ end
 puts Company.find_by(name: "MegaCorp").employees.size # => 1
 ```
 
+Or if you need more control, you can define functions that work within the scope of the factory.  
+
+```ruby
+@db.configure do
+  with Company do
+    name { Faker::Company.name }
+
+    function(:create_with_employees) do |name = nil, employee_count: 1|
+      companies.create(name: name).tap do |company|
+        employee_count.times { employees.create company: company }
+      end
+    end 
+
+  end
+  with Employee do
+    first_name { Faker::Name.first_name }
+    last_name  { Faker::Name.last_name }
+    role "Cleaner"
+  end
+end
+@db.companies.create name: "TinyCo"
+puts Company.find_by(name: "TinyCo").employees.size # => 0
+@db.companies.create_with_company name: "MegaCorp", employee_count: 4
+puts Company.find_by(name: "MegaCorp").employees.size # => 4
+```
+
 ## References
 
 You've created a load of models.  And you need to reference them to create more models.  You could search for them by hand, but that's for chumps.

data/checksums/standard_procedure_fabrik-0.2.5.gem.sha512

@@ -0,0 +1 @@
+53742d6fa70383e74ac73d440a51f06ce4083f885d2a1bd5e0dc7be4840f43f2d9818b573c686acfb496319e63f685f521d27ddb09f27109cd294dab0f8dbbd7

data/checksums/standard_procedure_fabrik-0.3.0.gem.sha512

@@ -0,0 +1 @@
+444a3a02f443dec303cd377e27191e1bcbf44797efe6836b4abc3e9dc74964ee43ba024dcc9f1d36d314dd240f1ef81dbc4cdab4985a0a5424c83d141f7094a3

data/lib/fabrik/database.rb

@@ -6,49 +6,40 @@ require "ostruct"
 
 module Fabrik
   class Database
-    def configure(&block) = instance_eval(&block)
+    def configure(&configuration) = instance_eval(&configuration)
 
-    def register(klass, as: nil, &block)
+    def register(klass, as: nil, &configuration)
       blueprint_name = (as.nil? ? blueprint_name_for(klass) : as.to_s.pluralize).to_sym
-      blueprint = Blueprint.new(klass, &block)
-      set_blueprint_for(blueprint_name, blueprint)
-
-      define_singleton_method blueprint_name do
-        proxy_for blueprint_name
-      end
-
-      proxy_for blueprint_name
+      @proxies[blueprint_name] ||= Proxy.new(self, Blueprint.new(klass))
+      @proxies[blueprint_name].configure(&configuration)
+      @proxies[blueprint_name]
     end
     alias_method :with, :register
 
-    def defaults_for(klass) = @blueprints[klass].default_attributes
+    def defaults_for(klass) = proxy_for(klass).default_attributes
 
-    def unique_keys_for(klass) = @blueprints[klass].unique_keys
+    def unique_keys_for(klass) = proxy_for(klass).unique_keys
 
-    def after_create_for(klass) = @blueprints[klass].callback
+    def after_create_for(klass) = proxy_for(klass).callback
 
-    def method_missing(method_name, *, &block) = (klass = class_from(method_name)).nil? ? super : register(klass)
+    def functions_for(klass) = proxy_for(klass).functions
 
-    def respond_to_missing?(method_name, include_private = false) = !class_from(method_name).nil? || super
+    def method_missing(method_name, *, &block)
+      @proxies[method_name] || ((klass = class_from(method_name)).nil? ? super : register(klass))
+    end
+
+    def respond_to_missing?(method_name, include_private = false)
+      @proxies.key?(method_name) || !class_from(method_name).nil? || super
+    end
 
     def initialize &config
-      @blueprints = {}
       @proxies = {}
       instance_eval(&config) unless config.nil?
     end
 
     private def blueprint_name_for(klass) = klass.name.split("::").map(&:underscore).join("_").pluralize
 
-    private def proxy_for(klass_or_blueprint_name)
-      blueprint_name = klass_or_blueprint_name.is_a?(Class) ? blueprint_name_for(klass_or_blueprint_name) : klass_or_blueprint_name
-      @proxies[blueprint_name.to_sym] ||= Proxy.new(self, @blueprints[blueprint_name])
-    end
-
-    private def set_blueprint_for(name, blueprint)
-      @blueprints[name] = blueprint
-      @blueprints[blueprint.klass] = blueprint
-      proxy_for(name).blueprint = blueprint
-    end
+    private def proxy_for(klass) = @proxies.values.find { |proxy| proxy.klass == klass }
 
     private def class_from(method_name)
       klass = nil
@@ -59,7 +50,14 @@ module Fabrik
   end
 
   class Blueprint
-    attr_reader :klass, :default_attributes, :unique_keys, :callback
+    def initialize(klass)
+      @klass = klass
+      @default_attributes = {}
+      @unique_keys = []
+      @functions = {}
+    end
+
+    attr_reader :klass, :default_attributes, :unique_keys, :callback, :functions
 
     def defaults(**default_attributes) = @default_attributes = default_attributes
 
@@ -69,6 +67,8 @@ module Fabrik
 
     def call_after_create(record, db) = @callback.nil? ? nil : db.instance_exec(record, &@callback)
 
+    def function(name, &implementation) = @functions[name.to_sym] = implementation
+
     def method_missing(method_name, *args, &block)
       @default_attributes[method_name.to_sym] = args.first.nil? ? block : ->(_) { args.first }
     end
@@ -77,17 +77,22 @@ module Fabrik
 
     def to_s = "#{klass} blueprint (#{object_id}) (#{default_attributes.keys.size} defaults, #{unique_keys.size} unique keys #{(!callback.nil?) ? "with callback" : ""})"
 
-    def initialize(klass, &block)
-      @klass = klass
-      @default_attributes = {}
-      @unique_keys = []
-      instance_eval(&block) unless block.nil?
+    def configure(&configuration)
+      instance_eval(&configuration) unless configuration.nil?
     end
   end
 
   class Proxy < SimpleDelegator
-    def create(label = nil, **attributes)
-      (@blueprint.unique_keys.any? ? find_or_create_record(attributes) : create_record(attributes)).tap do |record|
+    def initialize(db, blueprint)
+      @db = db
+      @blueprint = blueprint
+      @records = {}
+      super(klass)
+    end
+    attr_accessor :blueprint
+
+    def create(label = nil, after_create: true, **attributes)
+      (@blueprint.unique_keys.any? ? find_or_create_record(attributes, callback: after_create) : create_record(attributes, callback: after_create)).tap do |record|
         self[label] = record if label
       end
     end
@@ -99,37 +104,36 @@ module Fabrik
       @records[label.to_sym] = record
     end
 
-    def method_missing(method_name, *args, &block) = self[method_name.to_sym] || super
+    def method_missing(method_name, *args, &block) = self[method_name] || call_function(method_name, *args) || super
 
-    def respond_to_missing?(method_name, include_private = false) = !self[method_name].nil? || super
+    def respond_to_missing?(method_name, include_private = false) = !self[method_name].nil? || !functions[method_name].nil? || super
 
     def to_s = "Proxy #{object_id} for #{@blueprint} (#{@records.keys.size} records)"
 
-    def initialize(db, blueprint)
-      @db = db
-      @blueprint = blueprint
-      @records = {}
-      super(klass)
+    def configure(&configuration)
+      @blueprint.configure(&configuration) unless configuration.nil?
     end
 
-    attr_accessor :blueprint
+    def unique_keys = @blueprint.unique_keys
+
+    def klass = @blueprint.klass
 
-    private def unique_keys = @blueprint.unique_keys
+    def default_attributes = @blueprint.default_attributes
 
-    private def klass = @blueprint.klass
+    def callback = @blueprint.callback
 
-    private def default_attributes = @blueprint.default_attributes
+    def functions = @blueprint.functions
 
-    private def find_or_create_record(attributes)
+    private def find_or_create_record(attributes, callback:)
       attributes = attributes_with_defaults(attributes)
-      find_record(attributes) || create_record(attributes)
+      find_record(attributes) || create_record(attributes, callback: callback)
     end
 
     private def find_record(attributes) = attributes.slice(*unique_keys).empty? ? nil : klass.find_by(**attributes.slice(*unique_keys))
 
-    private def create_record(attributes)
+    private def create_record(attributes, callback:)
       klass.create!(**attributes_with_defaults(attributes)).tap do |record|
-        @blueprint.call_after_create(record, @db)
+        @blueprint.call_after_create(record, @db) if callback
       end
     end
 
@@ -139,5 +143,9 @@ module Fabrik
         generated_attributes[name] = default_attributes[name].nil? ? nil : @db.instance_exec(generated_attributes, &default_attributes[name])
       end.to_h.merge(attributes)
     end
+
+    private def call_function(method_name, *args)
+      functions[method_name].nil? ? nil : @db.instance_exec(*args, &functions[method_name])
+    end
   end
 end

data/lib/fabrik/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Fabrik
-  VERSION = "0.2.4"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: standard_procedure_fabrik
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.3.0
 platform: ruby
 authors:
 - Rahoul Baruah
 bindir: exe
 cert_chain: []
-date: 2025-02-10 00:00:00.000000000 Z
+date: 2025-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -60,6 +60,8 @@ files:
 - checksums/standard_procedure_fabrik-0.2.2.gem.sha512
 - checksums/standard_procedure_fabrik-0.2.3.gem.sha512
 - checksums/standard_procedure_fabrik-0.2.4.gem.sha512
+- checksums/standard_procedure_fabrik-0.2.5.gem.sha512
+- checksums/standard_procedure_fabrik-0.3.0.gem.sha512
 - lib/fabrik.rb
 - lib/fabrik/database.rb
 - lib/fabrik/version.rb