-) Code formatting
-) rspec tests
This commit is contained in:
84
wpstools.rb
84
wpstools.rb
@@ -32,50 +32,50 @@ begin
|
||||
# A better way to do that should be to create a wpstools_options.rb file like wpscan_options.rb
|
||||
# and a wps_options.rb with common options code
|
||||
options = GetoptLong.new(
|
||||
["--help", "-h", GetoptLong::NO_ARGUMENT],
|
||||
["--verbose", "-v", GetoptLong::NO_ARGUMENT],
|
||||
["--generate_plugin_list", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--generate_full_plugin_list", GetoptLong::NO_ARGUMENT],
|
||||
["--generate_theme_list", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--generate_full_theme_list", GetoptLong::NO_ARGUMENT],
|
||||
["--gpl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gfpl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gtl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gftl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--update", "-u", GetoptLong::NO_ARGUMENT]
|
||||
["--help", "-h", GetoptLong::NO_ARGUMENT],
|
||||
["--verbose", "-v", GetoptLong::NO_ARGUMENT],
|
||||
["--generate_plugin_list", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--generate_full_plugin_list", GetoptLong::NO_ARGUMENT],
|
||||
["--generate_theme_list", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--generate_full_theme_list", GetoptLong::NO_ARGUMENT],
|
||||
["--gpl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gfpl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gtl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--gftl", GetoptLong::OPTIONAL_ARGUMENT],
|
||||
["--update", "-u", GetoptLong::NO_ARGUMENT]
|
||||
)
|
||||
|
||||
options.each do |option, argument|
|
||||
case option
|
||||
when "--help"
|
||||
help()
|
||||
exit
|
||||
when "--verbose"
|
||||
@verbose = true
|
||||
when "--generate_plugin_list", "--gpl"
|
||||
if argument == ''
|
||||
puts "Number of pages not supplied, defaulting to 150 pages ..."
|
||||
@number_of_pages = 150
|
||||
else
|
||||
@number_of_pages = argument.to_i
|
||||
end
|
||||
when "--help"
|
||||
help()
|
||||
exit
|
||||
when "--verbose"
|
||||
@verbose = true
|
||||
when "--generate_plugin_list", "--gpl"
|
||||
if argument == ''
|
||||
puts "Number of pages not supplied, defaulting to 150 pages ..."
|
||||
@number_of_pages = 150
|
||||
else
|
||||
@number_of_pages = argument.to_i
|
||||
end
|
||||
|
||||
@generate_plugin_list = true
|
||||
when "--generate_theme_list", "--gtl"
|
||||
if argument == ''
|
||||
puts "Number of pages not supplied, defaulting to 150 pages ..."
|
||||
@number_of_pages = 150
|
||||
else
|
||||
@number_of_pages = argument.to_i
|
||||
end
|
||||
|
||||
@generate_theme_list = true
|
||||
when "--update"
|
||||
@update = true
|
||||
when "--generate_full_plugin_list", "--gfpl"
|
||||
@generate_full_plugin_list = true
|
||||
when "--generate_full_theme_list", "--gftl"
|
||||
@generate_full_theme_list = true
|
||||
@generate_plugin_list = true
|
||||
when "--generate_theme_list", "--gtl"
|
||||
if argument == ''
|
||||
puts "Number of pages not supplied, defaulting to 150 pages ..."
|
||||
@number_of_pages = 150
|
||||
else
|
||||
@number_of_pages = argument.to_i
|
||||
end
|
||||
|
||||
@generate_theme_list = true
|
||||
when "--update"
|
||||
@update = true
|
||||
when "--generate_full_plugin_list", "--gfpl"
|
||||
@generate_full_plugin_list = true
|
||||
when "--generate_full_theme_list", "--gftl"
|
||||
@generate_full_theme_list = true
|
||||
end
|
||||
end
|
||||
|
||||
@@ -84,19 +84,19 @@ begin
|
||||
puts
|
||||
Generate_List.new('plugins', @verbose).generate_popular_list(@number_of_pages)
|
||||
end
|
||||
|
||||
|
||||
if @generate_full_plugin_list
|
||||
puts "[+] Generating new full plugin list"
|
||||
puts
|
||||
Generate_List.new('plugins', @verbose).generate_full_list
|
||||
end
|
||||
|
||||
|
||||
if @generate_theme_list
|
||||
puts "[+] Generating new most popular theme list"
|
||||
puts
|
||||
Generate_List.new('themes', @verbose).generate_popular_list(@number_of_pages)
|
||||
end
|
||||
|
||||
|
||||
if @generate_full_theme_list
|
||||
puts "[+] Generating new full theme list"
|
||||
puts
|
||||
|
||||
Reference in New Issue
Block a user