require 'rails/generators'
module Authorization
  class InstallGenerator < Rails::Generators::Base
    
    include Rails::Generators::Migration
    source_root File.expand_path('../templates', __FILE__)

    argument :name, type: :string, default: "User"
    argument :attributes, type: :array, default: ['name:string'], banner: "field[:type] field[:type]"
    class_option :create_user, type: :boolean, default: false, desc: "Creates the defined User model with attributes given."
    class_option :commit, type: :boolean, default: false, desc: "Performs rake tasks such as migrate and seed."
    class_option :user_belongs_to_role, type: :boolean, default: false, desc: "Users have only one role, which can inherit others roles."

    def self.next_migration_number(dirname)
      if ActiveRecord::Base.timestamped_migrations
        Time.now.utc.strftime("%Y%m%d%H%M%S")
      else
        "%.3d" % (current_migration_number(dirname) + 1)
      end
    end

    def install_decl_auth
      habtm_table_name  = "#{name.pluralize}" <= "Roles" ? "#{name.pluralize}Roles" : "Roles#{name.pluralize}" unless options[:user_belongs_to_role]
      habtm_file_glob  = "#{name.pluralize}" <= "Roles" ? 'db/migrate/*create_*_roles*' : 'db/migrate/*create_roles_*' unless options[:user_belongs_to_role]

      generate 'model', "#{name} #{attributes.join(' ')}" if options[:create_user]
      generate 'model', 'Role title:string'

      if options[:user_belongs_to_role]
        inject_into_file "app/models/#{name.singularize.downcase}.rb", "  belongs_to :role\n", after: "ActiveRecord::Base\n"
        generate 'migration', "AddRoleIdTo#{name.camelcase} role_id:integer"
      else
        generate 'migration', "Create#{habtm_table_name} #{name.downcase}:integer role:integer"
        gsub_file Dir.glob(habtm_file_glob).last, 'integer', 'references'
        inject_into_file Dir.glob(habtm_file_glob).last, ", id: false", before: ' do |t|'
        inject_into_file "app/models/role.rb", "  has_and_belongs_to_many :#{name.downcase.pluralize}\n", after: "ActiveRecord::Base\n"
        inject_into_file "app/models/#{name.singularize.downcase}.rb", "  has_and_belongs_to_many :roles\n", after: "ActiveRecord::Base\n"
      end

      rake 'db:migrate' if options[:commit]

      if options[:user_belongs_to_role]
        inject_into_file "app/models/#{name.singularize.downcase}.rb", before: "\nend" do <<-'RUBY'


  def role_symbols
    [role.title.to_sym]
  end
        RUBY
        end
      else
        inject_into_file "app/models/#{name.singularize.downcase}.rb", before: "\nend" do <<-'RUBY'


  def role_symbols
    (roles || []).map {|r| r.title.to_sym}
  end
        RUBY
        end
      end

      inject_into_file 'db/seeds.rb', after: ".first)\n" do <<-'RUBY'

roles = Role.create([
  {title: 'admin'},
  {title: 'user'}
]) if Role.count == 0
RUBY
      end

      rake 'db:seed' if options[:commit]

      generate 'authorization:rules'
      puts "Please run `rake db:migrate` and `rake db:seed` to finish installing." unless options[:commit]
    end
  end
end