demo_mode 3.7.2 → 3.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b855f5dbc8f1a8ff5a37f7ee07c8780f5da7bec9aedaa1558164a8f7bb18d769
-  data.tar.gz: 92aaf8d2e014423e8e37be02356a5efc1d70c0e9ecc740d1436b05f956782d50
+  metadata.gz: b39607da6e06571e9a6b0d57b9efa1ab59f4109836e5c87776892fa30e239f95
+  data.tar.gz: 86c5e89834dff8123aa4b22f15d883d63eaa3209fc6d3236a261d625f98de2b6
 SHA512:
-  metadata.gz: dd54802030d6570054be23f529fe02e7358f23468474341390ff35580208a3bc12b49ce6cf21e3c0f898d6fcc5496e45b278d42d7cfe00fc7e4befa2677aa002
-  data.tar.gz: 4a906c94e4c1a8541ba612927c0b5a936c24223976583dbdf9123099e9a0599c9736d8aec00a1284b93dadf3162c0b40f1950fbfc06c175fe66eea0b1871dcbf
+  metadata.gz: 31b38ae765035a33cd107ea2340ce33c5875880102248c8bf269b55244aee007ca960c54c3ec79edd6a4a8b6e862811297b5ce4cd2c77c54bca07b620ad8ce41
+  data.tar.gz: f016a43e1afa88f64119ca71b548cd0a4c7bb4d07c5d2fb2bc57830508f26fcb6138fc13e4c7dddf9a3383a402457ed1cadc870b14e6b0055696ae91fb064290

data/app/jobs/demo_mode/account_generation_job.rb

@@ -9,8 +9,14 @@ module DemoMode
         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

@@ -14,7 +14,7 @@ module DemoMode
       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.2'
+  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.2
+  version: 3.8.0
 platform: ruby
 authors:
 - Nathan Griffith
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-22 00:00:00.000000000 Z
+date: 2026-04-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: actionpack