demo_mode 3.7.1 → 3.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 268e07b92704ae904b434f1c2865a8a21ece4adfc51775339aa2f5243a7e4624
-  data.tar.gz: fe8974ba779ad030bf120811158e19f767c736c3c4faecf1c59a40dd019c7298
+  metadata.gz: b39607da6e06571e9a6b0d57b9efa1ab59f4109836e5c87776892fa30e239f95
+  data.tar.gz: 86c5e89834dff8123aa4b22f15d883d63eaa3209fc6d3236a261d625f98de2b6
 SHA512:
-  metadata.gz: f0da8528e1c63ea16b3ca1f9c1e3a6fb1731ea4e5e406bb93e5512422e311139a4e27c1fe6fc38ec45dc9b9be10223625ad15455b73765025d31480a4b8f0fe4
-  data.tar.gz: 75382c2d121e8308a9155280320ce94e3a7a04e0021161c3145a78ec6b6f8ec31815c15562197031cc55a757791cde01be774d4bc1a8a2312853dd6efa8d1bad
+  metadata.gz: 31b38ae765035a33cd107ea2340ce33c5875880102248c8bf269b55244aee007ca960c54c3ec79edd6a4a8b6e862811297b5ce4cd2c77c54bca07b620ad8ce41
+  data.tar.gz: f016a43e1afa88f64119ca71b548cd0a4c7bb4d07c5d2fb2bc57830508f26fcb6138fc13e4c7dddf9a3383a402457ed1cadc870b14e6b0055696ae91fb064290

data/app/jobs/demo_mode/account_generation_job.rb

@@ -4,12 +4,19 @@ module DemoMode
   class AccountGenerationJob < DemoMode.base_job_name.constantize
     def perform(session, **options)
       session.with_lock do
+        session.update!(status: 'processing') if session.failed?
         persona = session.persona
         raise "Unknown persona: #{session.persona_name}" if persona.blank?
 
         signinable = persona.generate!(variant: session.variant, password: session.signinable_password, options: options)
+        session.update!(signinable: signinable, persona_checksum: persona.file_checksum)
+
+        if session.claimed_at?
+          persona.effective_at_claim_callback(session.variant)&.call(signinable)
+        end
+
         new_status = session.claimed_at? ? 'in_use' : 'available'
-        session.update!(signinable: signinable, status: new_status, persona_checksum: persona.file_checksum)
+        session.update!(status: new_status)
       end
     rescue StandardError => e
       session.update!(status: 'failed')

data/app/models/demo_mode/session.rb

@@ -11,10 +11,10 @@ module DemoMode
     attr_accessor :pool_session
 
     steady_state :status do
-      state 'processing', default: true
+      state 'processing', default: true, from: 'failed'
       state 'available', from: 'processing'
       state 'in_use', from: %w(processing available)
-      state 'failed', from: 'processing'
+      state 'failed', from: %w(processing available)
     end
 
     scope :unclaimed, -> { where(claimed_at: nil) }
@@ -56,13 +56,15 @@ module DemoMode
     end
 
     def self.claim_for(persona_name:, variant: DEFAULT_VARIANT, **generation_opts)
+      persona = DemoMode.personas.find { |p| p.name.to_s == persona_name.to_s && p.variants.key?(variant) }
       pool_hit = false
       session = transaction do
         existing = available_for(persona_name, variant).lock.first
         pool_hit = existing.present?
-        (existing || new(persona_name: persona_name, variant: variant)).tap do |s|
-          s.claim!
-          AccountGenerationJob.perform_later(s, **generation_opts) if s.signinable.blank?
+        if existing
+          claim_pool_session(existing, persona, variant)
+        else
+          new_claimed_session(persona_name, variant, generation_opts)
         end
       end
       ActiveSupport::Notifications.instrument('demo_mode.session.claimed',
@@ -70,6 +72,28 @@ module DemoMode
       session
     end
 
+    class << self
+      private
+
+      def claim_pool_session(session, persona, variant)
+        transaction(requires_new: true) do
+          persona&.effective_at_claim_callback(variant)&.call(session.signinable)
+          session.claim!
+        rescue StandardError
+          raise ActiveRecord::Rollback
+        end
+        session.update!(status: 'failed') unless session.claimed_at?
+        session
+      end
+
+      def new_claimed_session(persona_name, variant, generation_opts)
+        new(persona_name: persona_name, variant: variant).tap do |s|
+          s.claim!
+          AccountGenerationJob.perform_later(s, **generation_opts) if s.signinable.blank?
+        end
+      end
+    end
+
     def claim!
       if new_record?
         self.claimed_at = Time.zone.now

data/lib/demo_mode/persona.rb

@@ -97,6 +97,16 @@ module DemoMode
       @enabled_condition ? @enabled_condition.call : true
     end
 
+    def at_claim(&block)
+      @at_claim_callback = block
+    end
+
+    attr_reader :at_claim_callback
+
+    def effective_at_claim_callback(variant_name)
+      variants[variant_name]&.at_claim_callback || @at_claim_callback
+    end
+
     def callout(callout = true) # rubocop:disable Style/OptionalBooleanParameter
       @callout = callout
     end
@@ -154,10 +164,15 @@ module DemoMode
         @enabled_condition ? @enabled_condition.call : true
       end
 
+      def at_claim(&block)
+        @at_claim_callback = block
+      end
+
       def title
         name.is_a?(Symbol) ? name.to_s.titleize : name.to_s
       end
 
+      attr_reader :at_claim_callback
       attr_reader :signinable_generator
     end
   end

data/lib/demo_mode/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DemoMode
-  VERSION = '3.7.1'
+  VERSION = '3.8.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: demo_mode
 version: !ruby/object:Gem::Version
-  version: 3.7.1
+  version: 3.8.0
 platform: ruby
 authors:
 - Nathan Griffith
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-17 00:00:00.000000000 Z
+date: 2026-04-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: actionpack