# frozen_string_literal: true

require 'rails/generators'

module Rails
  class DecoratorGenerator < Rails::Generators::NamedBase

    source_root File.expand_path('../templates', __FILE__)
    check_class_collision suffix: 'Decorator'

    def copy_files
      path = File.join('app', 'decorators', class_path, "#{file_name}_decorator.rb")
      template('decorator.rb.tt', path)
    end

    private

    def file_name
      @_file_name ||= remove_possible_suffix(super)
    end

    def remove_possible_suffix(name)
      name.sub(/_?decorator$/i, '')
    end

  end
end