init first last step

zombocom · Dec 3, 2023 · 280b4ac · 280b4ac
1 parent faa8c11
commit 280b4ac
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/wicked/wizard.rb b/lib/wicked/wizard.rb
@@ -49,10 +49,17 @@ def wizard_value(step_name)
       @wicked_redirect_params = nil
     end
 
+    private def check_first_last_step!(step)
+      return steps.first if step.to_s == Wicked::FIRST_STEP
+      return steps.last if step.to_s == Wicked::LAST_STEP
+      return step
+    end
+
     private def setup_step_from(the_step)
       return if steps.nil?
 
       the_step ||= steps.first
+      the_step = check_first_last_step!(the_step)
 
       valid_steps = steps + self.class::PROTECTED_STEPS
       resolved_step = valid_steps.detect { |stp| stp.to_s == the_step }