Refs #5525; Refactor

1 files changed, 10 insertions, 5 deletions

diff --git a/config/initializers/simple_form/safe_submit.rb b/config/initializers/simple_form/safe_submit.rb
index 8d10929eb..64b9f1e6e 100644
--- a/config/initializers/simple_form/safe_submit.rb
+++ b/config/initializers/simple_form/safe_submit.rb
@@ -1,7 +1,12 @@
-AF83::SimpleForm::SafeSubmit.decorate_simple_form
-
-if Rails.env.development?
-  ActionDispatch::Reloader.to_prepare do
-    AF83::SimpleForm::SafeSubmit.decorate_simple_form
+::SimpleForm::FormBuilder.class_eval do
+  def button_with_safe_submit(type, *args, &block)
+    options = args.extract_options!.dup
+    if type == :submit
+      options[:data] ||= {}
+      options[:data][:disable_with] ||= I18n.t('actions.wait_for_submission')
+    end
+    args << options
+    button_without_safe_submit type, *args, &block
   end
+  alias_method_chain :button, :safe_submit
 end