demo_mode 3.7.2 → 3.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b855f5dbc8f1a8ff5a37f7ee07c8780f5da7bec9aedaa1558164a8f7bb18d769
-  data.tar.gz: 92aaf8d2e014423e8e37be02356a5efc1d70c0e9ecc740d1436b05f956782d50
+  metadata.gz: 7e73fd7fddd766bef9d1545a62dc2ddd66afc0b8b9c54520eaaf0a565133b475
+  data.tar.gz: a4738c449db124a9a0bcd78f75ef4f5d3a0f5de70bc3d841ef42b8e602f3d971
 SHA512:
-  metadata.gz: dd54802030d6570054be23f529fe02e7358f23468474341390ff35580208a3bc12b49ce6cf21e3c0f898d6fcc5496e45b278d42d7cfe00fc7e4befa2677aa002
-  data.tar.gz: 4a906c94e4c1a8541ba612927c0b5a936c24223976583dbdf9123099e9a0599c9736d8aec00a1284b93dadf3162c0b40f1950fbfc06c175fe66eea0b1871dcbf
+  metadata.gz: 74f400f23e3c529a3b928861a6870ac5b39e234cd0fa69174bd88176bb6ac416518a46b390d78dd4a671958b164dd2aa4fabc95a557c051b6835e70f4a555d2d
+  data.tar.gz: 06d3c2e5dd2dd757b11c3195124ec237839b56f68f46b693b8cde88c5d7875fdf21bd19333e732ca9c0cd49e8848c112cf9d95e8fa5abba6b57e09c09dc08fd9

data/app/jobs/demo_mode/account_generation_job.rb

@@ -3,14 +3,20 @@
 module DemoMode
   class AccountGenerationJob < DemoMode.base_job_name.constantize
     def perform(session, **options)
-      session.with_lock do
+      session.with_lock(requires_new: true) 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

@@ -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

@@ -68,13 +68,17 @@ module DemoMode
         variant = variants[variant]
         CleverSequence.reset! if defined?(CleverSequence)
         DemoMode.current_password = password if password
-        DemoMode.around_persona_generation.call(variant.signinable_generator, **options)
+        ActiveRecord::Base.transaction(requires_new: true) do
+          DemoMode.around_persona_generation.call(variant.signinable_generator, **options)
+        end
       rescue ActiveRecord::RecordNotUnique, ActiveRecord::RecordInvalid => e
         raise if retried || !should_retry_with_sequence_adjustment?(e)
 
         retried = true
         CleverSequence.with_sequence_adjustment do
-          DemoMode.around_persona_generation.call(variant.signinable_generator, **options)
+          ActiveRecord::Base.transaction(requires_new: true) do
+            DemoMode.around_persona_generation.call(variant.signinable_generator, **options)
+          end
         end
       ensure
         DemoMode.current_password = nil
@@ -97,6 +101,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 +168,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.1'
 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.1
 platform: ruby
 authors:
 - Nathan Griffith
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-22 00:00:00.000000000 Z
+date: 2026-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: actionpack