# frozen_string_literal: true

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

      site = Bridgetown::Current.site

      I18n.reload! # make sure any locale files get read again
      Bridgetown::Hooks.trigger :site, :pre_reload, site
      Bridgetown::Hooks.clear_reloadable_hooks
      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"
      class_option :"server-config",
                   aliases: "-s",
                   type: :boolean,
                   desc: "Load server configurations"
      class_option :verbose,
                   aliases: "-V",
                   type: :boolean,
                   desc: "Print verbose output."

      def console # rubocop:disable Metrics
        require "irb"
        new_history_behavior = false
        begin
          require "irb/ext/save-history"
        rescue LoadError
          # Code path for Ruby 3.3+
          new_history_behavior = true
        end
        require "amazing_print" unless options[:"bypass-ap"]

        Bridgetown.logger.adjust_verbosity(options)

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

        config_options = configuration_with_overrides(options)
        if options[:"server-config"]
          require "puma"
          require "bridgetown-core/rack/boot"
          Bridgetown::Rack.boot
        else
          config_options.run_initializers! context: :console
        end
        site = Bridgetown::Site.new(config_options)

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

        IRB::ExtendCommandBundle.include ConsoleMethods
        IRB.setup(nil)
        workspace = IRB::WorkSpace.new
        workspace.main.define_singleton_method(:site) { Bridgetown::Current.site }
        workspace.main.define_singleton_method(:collections) { site.collections }
        irb = IRB::Irb.new(workspace)
        IRB.conf[:IRB_RC]&.call(irb.context)
        IRB.conf[:MAIN_CONTEXT] = irb.context
        irb.context.io.load_history if new_history_behavior
        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)
          irb.context.io.save_history if new_history_behavior
        end
      end
    end
  end
end