added some debug stuff and made the -i flag use all of the fetched url instead of the first line

This commit is contained in:
2026-07-17 00:29:54 -06:00
parent 28110c7d94
commit 75d70d2b7e
+27 -5
View File
@@ -17,14 +17,17 @@ OptionParser.parse do |parser|
urlTimeout = timeout} urlTimeout = timeout}
parser.on("-f FILE", "--file=FILE", "Checks all urls in a txt file.") { |file| parser.on("-f FILE", "--file=FILE", "Checks all urls in a txt file.") { |file|
urls = File.read(file) urls = File.read(file)
checkUrlFromFile urls, output, urlTimeout.to_i, includesFlag, includes checkUrlFromFile urls, output, urlTimeout.to_i, includesFlag, includes, debug
exit 0} exit 0}
parser.on("-o FILENAME", "--output=FILENAME", "Sets the name of the outputted file. -f flag required") {|filename| parser.on("-o FILENAME", "--output=FILENAME", "Sets the name of the outputted file. -f flag required") {|filename|
output = filename} output = filename}
parser.on("-v", "--version", "Displays current version of rete.") { parser.on("-v", "--version", "Displays current version of rete.") {
puts "Rete 1.1 by koffee.zip" puts "Rete 1.1 by koffee.zip"
exit(0)} exit(0)}
parser.on("-d", "--debug", "Debug Mode") {debug = true} parser.on("-d", "--debug", "Debug Mode") do
debug = true
puts "Debug mode enabled!"
end
parser.on("-h", "--help", "Show this help") do parser.on("-h", "--help", "Show this help") do
puts parser puts parser
exit exit
@@ -36,7 +39,7 @@ OptionParser.parse do |parser|
end end
end end
def checkUrlFromFile(content, output, urlTimeout, includesFlag, includes) # Sorry for the crappy method name. def checkUrlFromFile(content, output, urlTimeout, includesFlag, includes, debug) # Sorry for the crappy method name.
repeat = 0 repeat = 0
working = 0 working = 0
notWorking = 0 notWorking = 0
@@ -44,6 +47,12 @@ def checkUrlFromFile(content, output, urlTimeout, includesFlag, includes) # Sorr
begin begin
config = File.read("config.txt") config = File.read("config.txt")
if debug == true
puts "=============================="
puts "Current Config:"
puts config
puts "=============================="
end
rescue File::NotFoundError rescue File::NotFoundError
puts "Oh No! It seems that your config.txt file does not exist!" puts "Oh No! It seems that your config.txt file does not exist!"
puts "This is required for rete to know what classifies if a url is working or not!" puts "This is required for rete to know what classifies if a url is working or not!"
@@ -69,7 +78,7 @@ def checkUrlFromFile(content, output, urlTimeout, includesFlag, includes) # Sorr
response = client.get("/") response = client.get("/")
if config.includes?(response.status_code.to_s) if config.includes?(response.status_code.to_s)
if includesFlag if includesFlag
if response.body.lines.first.includes?(includes) if response.body.lines.includes?(includes)
puts "#{urls[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}" puts "#{urls[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}"
File.open("#{output}", "a") do |file| File.open("#{output}", "a") do |file|
file.puts urls[repeat] file.puts urls[repeat]
@@ -113,6 +122,12 @@ exist = true
begin begin
config = File.read("config.txt") config = File.read("config.txt")
if debug == true
puts "=============================="
puts "Current Config:"
puts config
puts "=============================="
end
rescue File::NotFoundError rescue File::NotFoundError
puts "Oh No! It seems that your config.txt file does not exist!" puts "Oh No! It seems that your config.txt file does not exist!"
puts "This is required for rete to know what classifies if a url is working or not!" puts "This is required for rete to know what classifies if a url is working or not!"
@@ -135,8 +150,15 @@ while repeat < ARGV.size
response = client.get("/") response = client.get("/")
if config.includes?(response.status_code.to_s) if config.includes?(response.status_code.to_s)
if debug == true
puts "=============================="
puts "First line of Response from URL:"
puts response.body.lines.first
puts "=============================="
end
if includesFlag if includesFlag
if response.body.lines.first.includes?(includes) if response.body.lines.includes?(includes)
puts "#{ARGV[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}" puts "#{ARGV[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}"
else else
puts "#{ARGV[repeat]} is a valid URL, works but does not include \"#{includes}\" Status code: #{response.status_code}" puts "#{ARGV[repeat]} is a valid URL, works but does not include \"#{includes}\" Status code: #{response.status_code}"