# frozen_string_literal: true
#
# ronin-payloads - A Ruby micro-framework for writing and running exploit
# payloads.
#
# Copyright (c) 2007-2023 Hal Brodigan (postmodern.mod3 at gmail.com)
#
# ronin-payloads is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# ronin-payloads is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with ronin-payloads.  If not, see <https://www.gnu.org/licenses/>.
#

require 'ronin/payloads/encoders/registry'
require 'ronin/payloads/encoders/html_encoder'
require 'ronin/payloads/encoders/javascript_encoder'
require 'ronin/payloads/encoders/shell_encoder'
require 'ronin/payloads/encoders/powershell_encoder'
require 'ronin/payloads/encoders/sql_encoder'
require 'ronin/payloads/encoders/exceptions'
require 'ronin/core/params/exceptions'

module Ronin
  module Payloads
    class CLI
      #
      # Common methods for handling encoders.
      #
      module EncoderMethods
        #
        # Returns the encoder type for the encoder class.
        #
        # @param [Class<Encoders::Encoder>] encoder_class
        #   The encoder class.
        #
        # @return [String, nil]
        #
        def encoder_type(encoder_class)
          if    encoder_class < Encoders::HTMLEncoder       then 'html'
          elsif encoder_class < Encoders::JavaScriptEncoder then 'javascript'
          elsif encoder_class < Encoders::ShellEncoder      then 'shell'
          elsif encoder_class < Encoders::PowerShellEncoder then 'powershell'
          elsif encoder_class < Encoders::SQLEncoder        then 'sql'
          end
        end

        #
        # Loads a encoder class.
        #
        # @param [String] name
        #   The encoder name to load.
        #
        # @return [Class<Encoders::Encoder>]
        #   The loaded encoder class.
        #
        def load_encoder(name)
          Payloads::Encoders.load_class(name)
        rescue Payloads::ClassNotFound => error
          print_error error.message
          exit(1)
        rescue => error
          print_exception(error)
          print_error "an unhandled exception occurred while loading encoder #{name}"
          exit(-1)
        end

        #
        # Loads the encoder from a given file.
        #
        # @param [String] file
        #   The file to load the encoder class from.
        #
        # @return [Class<Encoders::Encoder>]
        #   The loaded encoder class.
        #
        def load_encoder_from(file)
          Payloads::Encoders.load_class_from_file(file)
        rescue Payloads::ClassNotFound => error
          print_error error.message
          exit(1)
        rescue => error
          print_exception(error)
          print_error "an unhandled exception occurred while loading encoder from file #{file}"
          exit(-1)
        end

        #
        # Initializes an encoder.
        #
        # @param [Class<Encoders::Encoder>] encoder_class
        #   The encoder class.
        #
        # @param [Hash{Symbol => Object}] kwargs
        #   Additional keyword arguments for {Ronin::Payloads::Encoders::Encoder#initialize}.
        #
        def initialize_encoder(encoder_class,**kwargs)
          encoder_class.new(**kwargs)
        rescue Core::Params::ParamError => error
          print_error error.message
          exit(1)
        rescue => error
          print_exception(error)
          print_error "an unhandled exception occurred while initializing encoder #{encoder_class.id}"
          exit(-1)
        end

        #
        # Validates the loaded encoder.
        #
        # @param [Encoders::Encoder] encoder
        #   The encoder to validate.
        #
        # @raise [Ronin::Core::Params::RequiredParam]
        #   One of the required params was not set.
        #
        # @raise [ValidationError]
        #   Another encoder validation error occurred.
        #
        def validate_encoder(encoder)
          encoder.validate
        rescue Core::Params::ParamError, Encoders::ValidationError => error
          print_error "failed to validate the encoder #{encoder.class_id}: #{error.message}"
          exit(1)
        rescue => error
          print_exception(error)
          print_error "an unhandled exception occurred while validating the encoder #{encoder.class_id}"
          exit(-1)
        end
      end
    end
  end
end