#
# Copyright (c) 2006-2024 Wade Alcorn - wade@bindshell.net
# Browser Exploitation Framework (BeEF) - https://beefproject.com
# See the file 'doc/COPYING' for copying permission
#

module BeEF
  module Core
    # @note This class migrates and updates values in the database each time you restart BeEF.
    #       So for example, when you want to add a new command module, you stop BeEF,
    #       copy your command module into the framework and then restart BeEF.
    #       That class will take care of installing automatically the new command module in the db.
    class Migration
      include Singleton

      #
      # Updates the database.
      #
      def update_db!
        update_commands!
      end

      #
      # Checks for new command modules and updates the database.
      #
      def update_commands!
        config = BeEF::Core::Configuration.instance

        db_modules = BeEF::Core::Models::CommandModule.all.pluck(:name)

        config.get('beef.module').each do |k, v|
          BeEF::Core::Models::CommandModule.new(name: k, path: "#{v['path']}module.rb").save! unless db_modules.include? k
        end

        BeEF::Core::Models::CommandModule.all.each do |mod|
          unless config.get("beef.module.#{mod.name}").nil?
            config.set "beef.module.#{mod.name}.db.id", mod.id
            config.set "beef.module.#{mod.name}.db.path", mod.path
          end
        end

        # Call Migration method
        BeEF::API::Registrar.instance.fire BeEF::API::Migration, 'migrate_commands'
      end
    end
  end
end