#!/usr/bin/env ruby

require 'memory_profiler' # https://github.com/SamSaffron/memory_profiler
require 'wpscan'

report = MemoryProfiler.report(top: 15) do
  WPScan::Scan.new do |s|
    s.controllers <<
      WPScan::Controller::CustomDirectories.new <<
      WPScan::Controller::InterestingFindings.new <<
      WPScan::Controller::WpVersion.new <<
      WPScan::Controller::MainTheme.new <<
      WPScan::Controller::Enumeration.new <<
      WPScan::Controller::PasswordAttack.new <<
      WPScan::Controller::Aliases.new

    s.run
  end
end

# scale_bytes option not yet supported in the latest stable
# See https://github.com/SamSaffron/memory_profiler/issues/68
report.pretty_print($stdout, scale_bytes: true, detailed_report: true, to_file: 'memprof.report')