podemos-info / iban_bic

require "rails/generators"

require "rails/generators/active_record"

module IbanBic

  # Installs Iban+Bic in a rails app.

  class InstallGenerator < ::Rails::Generators::Base

    include ::Rails::Generators::Migration

    source_root File.expand_path("../templates", __FILE__)

    class_option(

      :with_static_bics,

      type: :boolean,

      default: false,

      desc: "Don't create bics table, use static bics."

    )

    class_option(

      :bics_table_name,

      type: :string,

      default: "bics",

      desc: "BICs table name, `bics` by default."

    )

    desc "Generates an initializer file for configuring IbanBic." \

         " Also can generate a migration to add a bics table."

    def create_migration_file

      add_iban_bic_migration "create_bics" unless options.with_static_bics?

    end

    def create_initializer

      initializer_dir = File.expand_path("config/initializers")

      if File.exist?(File.join(initializer_dir, "iban_bic.rb"))

        ::Kernel.warn "Initializer already exists"

      else

        template "iban_bic.rb.erb", "config/initializers/iban_bic.rb"

      end

    end

    def self.next_migration_number(dirname)

      ::ActiveRecord::Generators::Base.next_migration_number(dirname)

    end

    def migration_version

      "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"

    end

    protected

    def add_iban_bic_migration(template)

      migration_dir = File.expand_path("db/migrate")

      if self.class.migration_exists?(migration_dir, template)

        ::Kernel.warn "Migration already exists: #{template}"

      else

        migration_template "#{template}.rb.erb", "db/migrate/#{template}.rb"

      end

    end

  end

end