require 'generators/happy_seed/devise/devise_generator'

module HappySeed
  module Generators
    class DeviseInvitableGenerator < HappySeedGenerator
      source_root File.expand_path('../templates', __FILE__)

      def self.fingerprint
        gem_available? 'devise_invitable'
      end

      def install_device_invitable
        return if already_installed

        require_generator DeviseGenerator

        gem 'devise_invitable'

        Bundler.with_clean_env do
          run "bundle install --without production"
          run 'rake db:migrate'
        end

        run 'rails generate devise_invitable:install'
        run 'rails generate devise_invitable User'
        run 'rails generate devise_invitable:views'
        
        directory '.'

        if gem_available?( "haml-rails" )
          remove_file 'app/views/devise/invitations/new.html.erb'
          remove_file 'app/views/devise/invitations/edit.html.erb'
        end
      end

      private
        def gem_available?(name)
          Gem::Specification.find_by_name(name)
        rescue Gem::LoadError
          false
        rescue
          Gem.available?(name)
        end

    end
  end
end