require 'rails/generators'

module Spotlight
  ##
  # spotlight:install generator
  class Install < Rails::Generators::Base
    source_root File.expand_path('../templates', __FILE__)
    class_option :solr_update_class, type: :string, default: 'Spotlight::SolrDocument::AtomicUpdates'
    class_option :mailer_default_url_host, type: :string, default: '' # e.g. localhost:3000

    def inject_spotlight_routes
      route "mount Spotlight::Engine, at: 'spotlight'"
      gsub_file 'config/routes.rb', /^\s*root.*/ do |match|
        '#' + match + ' # replaced by spotlight root path'
      end
      route "root to: 'spotlight/exhibits#index'"
    end

    def friendly_id
      gem 'friendly_id'
      generate 'friendly_id'
    end

    def paper_trail
      generate 'paper_trail:install'
    end

    def sitemaps
      gem 'sitemap_generator'
      copy_file 'config/sitemap.rb', 'config/sitemap.rb'

      say <<-EOS.strip_heredoc, :red
       Added a default sitemap_generator configuration in config/sitemap.rb; please
       update the default host to match your environment
      EOS
    end

    def assets
      copy_file 'spotlight.scss', 'app/assets/stylesheets/spotlight.scss'
      copy_file 'spotlight.js', 'app/assets/javascripts/spotlight.js'
    end

    def add_roles_to_user
      inject_into_class 'app/models/user.rb', User, '  include Spotlight::User'
    end

    def add_controller_mixin
      inject_into_file 'app/controllers/application_controller.rb', after: 'include Blacklight::Controller' do
        "\n  include Spotlight::Controller\n"
      end
    end

    def add_helper
      copy_file 'spotlight_helper.rb', 'app/helpers/spotlight_helper.rb'
      inject_into_class 'app/helpers/application_helper.rb', ApplicationHelper, '  include SpotlightHelper'
    end

    def add_model_mixin
      if File.exist? 'app/models/solr_document.rb'
        inject_into_file 'app/models/solr_document.rb', after: 'include Blacklight::Solr::Document' do
          "\n  include Spotlight::SolrDocument\n"
        end
      else
        say 'Unable to find SolrDocument class; add `include Spotlight::SolrDocument` to the class manually'
      end
    end

    def add_solr_indexing_mixin
      if File.exist? 'app/models/solr_document.rb'
        inject_into_file 'app/models/solr_document.rb', after: "include Spotlight::SolrDocument\n" do
          "\n  include #{options[:solr_update_class]}\n"
        end
      else
        say "Unable to find SolrDocument class; add `include #{options[:solr_update_class]}` to the class manually"
      end
    end

    def add_search_builder_mixin
      if File.exist? 'app/models/search_builder.rb'
        inject_into_file 'app/models/search_builder.rb', after: "include Blacklight::Solr::SearchBuilderBehavior\n" do
          "\n  include Spotlight::Catalog::AccessControlsEnforcement::SearchBuilder\n"
        end
      else
        say 'Unable to find SearchBuilder class; add `include Spotlight::Catalog::AccessControlsEnforcement::SearchBuilder` to the class manually.'
      end
    end

    def add_example_catalog_controller
      copy_file 'catalog_controller.rb', 'app/controllers/catalog_controller.rb'
    end

    def add_osd_viewer
      gem 'blacklight-gallery', '>= 0.3.0'
      generate 'blacklight_gallery:install'
    end

    def add_oembed
      gem 'blacklight-oembed'
      generate 'blacklight_oembed:install'
    end

    def add_mailer_defaults
      if options[:mailer_default_url_host].present?
        say 'Injecting a placeholder config.action_mailer.default_url_options; be sure to update it for your environment', :yellow
        insert_into_file 'config/application.rb', after: "< Rails::Application\n" do
          <<-EOF
          config.action_mailer.default_url_options = { host: "#{options[:mailer_default_url_host]}", from: "noreply@example.com" }
        EOF
        end
      else
        say 'Please add a default configuration config.action_mailer.default_url_options for your environment', :red
      end
    end

    def generate_social_share_button_initializer
      gem 'social-share-button'
      directory 'config'
    end

    def add_solr_config_resources
      copy_file 'jetty.rake', 'lib/tasks/jetty.rake'
      directory 'solr_conf'
    end
  end
end