standard_procedure_fabrik 0.2.5 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c159a340077c82edfeac1780155f1cd1b5756672c7d0791ed4a662935d7f71fc
-  data.tar.gz: 5bab132dfe25fdcdc856169970742e179e0cafec96b03d5689f65620db4ef709
+  metadata.gz: 3a9563aec3880d3a7863b0c4d177abac5a3e681063d6a92f507de0aedd4861d5
+  data.tar.gz: ce1e4af35a3b7f8e454c680cee91c1fd79f063e341ae78f721ec6d14cd6976b2
 SHA512:
-  metadata.gz: fb2948cee905f209d413903c65c419e1b115cfdc3912fb7a65df9fb037d46aed8c5655dd50d5b8c048c6fd83ee0fc542079cf67d196be958287863d0fab5af6e
-  data.tar.gz: 1f349571d8e50270cf8fb5be9a331cba8e1cb3dfbdede04287bcd56b4fe31c500199bdd78b5b599a5b5a6e1122097f5d736ba219a34a989b7ba9c73590412fa3
+  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.3.0.gem.sha512

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

data/lib/fabrik/database.rb

@@ -22,6 +22,8 @@ module Fabrik
 
     def after_create_for(klass) = proxy_for(klass).callback
 
+    def functions_for(klass) = proxy_for(klass).functions
+
     def method_missing(method_name, *, &block)
       @proxies[method_name] || ((klass = class_from(method_name)).nil? ? super : register(klass))
     end
@@ -48,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
 
@@ -58,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
@@ -66,18 +77,20 @@ 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)
-      @klass = klass
-      @default_attributes = {}
-      @unique_keys = []
-    end
-
     def configure(&configuration)
       instance_eval(&configuration) unless configuration.nil?
     end
   end
 
   class Proxy < SimpleDelegator
+    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
@@ -91,9 +104,9 @@ 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)"
 
@@ -101,8 +114,6 @@ module Fabrik
       @blueprint.configure(&configuration) unless configuration.nil?
     end
 
-    attr_accessor :blueprint
-
     def unique_keys = @blueprint.unique_keys
 
     def klass = @blueprint.klass
@@ -111,12 +122,7 @@ module Fabrik
 
     def callback = @blueprint.callback
 
-    def initialize(db, blueprint)
-      @db = db
-      @blueprint = blueprint
-      @records = {}
-      super(klass)
-    end
+    def functions = @blueprint.functions
 
     private def find_or_create_record(attributes, callback:)
       attributes = attributes_with_defaults(attributes)
@@ -137,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.5"
+  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.5
+  version: 0.3.0
 platform: ruby
 authors:
 - Rahoul Baruah
 bindir: exe
 cert_chain: []
-date: 2025-02-17 00:00:00.000000000 Z
+date: 2025-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -61,6 +61,7 @@ files:
 - 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