module MiscHelpers
  def store_translations(locale, translations, &block)
    I18n.backend.store_translations locale, translations
    yield
  ensure
    I18n.reload!
    I18n.backend.send :init_translations
  end

  def assert_no_select(selector, value = nil)
    assert_select(selector, :text => value, :count => 0)
  end

  def swap(object, new_values)
    old_values = {}
    new_values.each do |key, value|
      old_values[key] = object.send key
      object.send :"#{key}=", value
    end
    yield
  ensure
    old_values.each do |key, value|
      object.send :"#{key}=", value
    end
  end

  def swap_wrapper(name=:default, wrapper=self.custom_wrapper)
    old = SimpleForm.wrappers[name]
    SimpleForm.wrappers[name] = wrapper
    yield
  ensure
    SimpleForm.wrappers[name] = old
  end

  def custom_wrapper
    SimpleForm.build :tag => :section, :class => "custom_wrapper", :pattern => false do |b|
      b.use :pattern
      b.wrapper :another, :class => "another_wrapper" do |ba|
        ba.use :label
        ba.use :input
      end
      b.wrapper :error_wrapper, :tag => :div, :class => "error_wrapper" do |be|
        be.use :error, :wrap_with => { :tag => :span, :class => "omg_error" }
      end
      b.use :hint, :wrap_with => { :class => "omg_hint" }
    end
  end

  def custom_wrapper_with_wrapped_input
    SimpleForm.build :tag => :div, :class => "custom_wrapper" do |b|
      b.wrapper :tag => :div, :class => 'elem' do |component|
        component.use :label
        component.use :input, :wrap_with => { :tag => :div, :class => 'input' }
      end
    end
  end

  def custom_wrapper_with_wrapped_label
    SimpleForm.build :tag => :div, :class => "custom_wrapper" do |b|
      b.wrapper :tag => :div, :class => 'elem' do |component|
        component.use :label, :wrap_with => { :tag => :div, :class => 'label' }
        component.use :input
      end
    end
  end

  def custom_wrapper_without_top_level
    SimpleForm.build :tag => false, :class => 'custom_wrapper_without_top_level' do |b|
      b.use :label_input
      b.use :hint,  :wrap_with => { :tag => :span, :class => :hint }
      b.use :error, :wrap_with => { :tag => :span, :class => :error }
    end
  end

  def custom_wrapper_without_class
    SimpleForm.build :tag => :div, :wrapper_html => { :id => 'custom_wrapper_without_class' } do |b|
      b.use :label_input
    end
  end

  def custom_wrapper_with_label_html_option
    SimpleForm.build :tag => :div, :class => "custom_wrapper", :label_html => { :class => 'extra-label-class' } do |b|
      b.use :label_input
    end
  end

  def custom_form_for(object, *args, &block)
    simple_form_for(object, *(args << { :builder => CustomFormBuilder }), &block)
  end

  def custom_mapping_form_for(object, *args, &block)
    simple_form_for(object, *(args << { :builder => CustomMapTypeFormBuilder }), &block)
  end

  def with_concat_form_for(*args, &block)
    concat simple_form_for(*args, &(block || proc {}))
  end

  def with_concat_fields_for(*args, &block)
    concat simple_fields_for(*args, &block)
  end

  def with_concat_custom_form_for(*args, &block)
    concat custom_form_for(*args, &block)
  end

  def with_concat_custom_mapping_form_for(*args, &block)
    concat custom_mapping_form_for(*args, &block)
  end

  def with_form_for(object, *args, &block)
    with_concat_form_for(object) do |f|
      f.input(*args, &block)
    end
  end

  def with_input_for(object, attribute_name, type, options={})
    with_concat_form_for(object) do |f|
      f.input(attribute_name, options.merge(:as => type))
    end
  end
end

class CustomFormBuilder < SimpleForm::FormBuilder
  def input(attribute_name, *args, &block)
    super(attribute_name, *(args << { :input_html => { :class => 'custom' } }), &block)
  end
end

class CustomMapTypeFormBuilder < SimpleForm::FormBuilder
  map_type :custom_type, :to => SimpleForm::Inputs::StringInput
end