rastating/wordpress-exploit-framework

# frozen_string_literal: true

module Wpxf
  module Cli
    # Methods for handling module loading and searching.
    module Modules
      def initialize
        super
        self.context_stack = []
      end

      def context
        context_stack.last
      end

      def back
        context_stack.pop
        context.module.active_workspace = active_workspace if context

        refresh_autocomplete_options
      end

      def reload
        return unless module_loaded?(false)

        begin
          mod = context.reload
          mod.event_emitter.subscribe(self)
          mod.active_workspace = active_workspace
          print_good "Reloaded module: #{mod}"
        rescue StandardError => e
          print_bad "Failed to reload module: #{e}"
          return
        end

        apply_global_options(mod)
      end

      def use(module_path)
        context = Context.new
        begin
          mod = context.load_module(module_path)
          mod.event_emitter.subscribe(self)
          mod.active_workspace = active_workspace
          print_good "Loaded module: #{mod}"
          mod.emit_usage_info
          context_stack.push(context)
        rescue StandardError => e
          print_bad "Failed to load module: #{e}"
          return
        end

        apply_global_options(mod)
        refresh_autocomplete_options
      end

      def module_name_from_class(klass)
        klass.new.module_name
      end

      def search_modules(args)
        modules = Wpxf::Models::Module.where(Sequel.ilike(:name, "%#{args.join(' ')}%"))
        modules.map { |m| { path: m.path, title: m.name } }
      end

      def print_module_table(modules)
        modules = modules.sort_by { |k| k[:path] }
        modules.unshift(path: 'Module', title: 'Title')
        puts
        indent_cursor 2 do
          print_table(modules)
        end
      end

      def search(*args)
        results = search_modules(args)

        if !results.empty?
          print_good "#{results.length} Results for \"#{args.join(' ')}\""
          print_module_table results
        else
          print_bad "No results for \"#{args.join(' ')}\""
        end
      end

      def reset_context_stack
        self.context_stack = []
        setup_auto_complete
      end

      attr_accessor :context_stack
    end
  end
end