# frozen_string_literal: true

module Bridgetown
  module ConsoleMethods
    def site
      Bridgetown::Current.site
    end

    def collections
      site.collections
    end

    def reload!
      Bridgetown.logger.info "Reloading site..."

      I18n.reload! # make sure any locale files get read again
      Bridgetown::Hooks.trigger :site, :pre_reload, site
      Bridgetown::Hooks.clear_reloadable_hooks
      site.plugin_manager.reload_plugin_files
      site.loaders_manager.reload_loaders
      Bridgetown::Hooks.trigger :site, :post_reload, site

      ConsoleMethods.site_reset(site)
    end

    def self.site_reset(site)
      site.reset
      Bridgetown.logger.info "Reading files..."
      site.read
      Bridgetown.logger.info "", "done!"
      Bridgetown.logger.info "Running generators..."
      site.generate
      Bridgetown.logger.info "", "done!"
    end
  end

  module Commands
    class Console < Thor::Group
      extend Summarizable
      include ConfigurationOverridable

      Registrations.register do
        register(Console, "console", "console", Console.summary)
      end

      def self.banner
        "bridgetown console [options]"
      end
      summary "Invoke an IRB console with the site loaded"

      class_option :config,
                   type: :array,
                   banner: "FILE1 FILE2",
                   desc: "Custom configuration file(s)"
      class_option :"bypass-ap",
                   type: :boolean,
                   desc: "Don't load AmazingPrint when IRB opens"
      class_option :blank,
                   type: :boolean,
                   desc: "Skip reading content and running generators before opening console"

      def console
        require "irb"
        require "irb/ext/save-history"
        require "amazing_print" unless options[:"bypass-ap"]

        Bridgetown.logger.info "Starting:", "Bridgetown v#{Bridgetown::VERSION.magenta}" \
                                            " (codename \"#{Bridgetown::CODE_NAME.yellow}\")" \
                                            " console…"
        Bridgetown.logger.info "Environment:", Bridgetown.environment.cyan
        site = Bridgetown::Site.new(configuration_with_overrides(options))

        ConsoleMethods.site_reset(site) unless options[:blank]

        IRB::ExtendCommandBundle.include ConsoleMethods
        IRB.setup(nil)
        workspace = IRB::WorkSpace.new
        irb = IRB::Irb.new(workspace)
        IRB.conf[:IRB_RC]&.call(irb.context)
        IRB.conf[:MAIN_CONTEXT] = irb.context
        Bridgetown.logger.info "Console:", "Your site is now available as #{"site".cyan}"
        Bridgetown.logger.info "",
                               "You can also access #{"collections".cyan} or perform a" \
                               " #{"reload!".cyan}"

        trap("SIGINT") do
          irb.signal_handle
        end

        begin
          catch(:IRB_EXIT) do
            unless options[:"bypass-ap"]
              AmazingPrint.defaults = {
                indent: 2,
              }
              AmazingPrint.irb!
            end
            irb.eval_input
          end
        ensure
          IRB.conf[:AT_EXIT].each(&:call)
        end
      end
    end
  end
end