working on adding a includes flag
This commit is contained in:
+26
-1
@@ -4,9 +4,14 @@ require "validator"
|
|||||||
|
|
||||||
output = "passed.txt"
|
output = "passed.txt"
|
||||||
urlTimeout = 1
|
urlTimeout = 1
|
||||||
|
includes = "hafjdhfkjlhadhlahfhl"
|
||||||
|
includesFlag = false
|
||||||
|
|
||||||
OptionParser.parse do |parser|
|
OptionParser.parse do |parser|
|
||||||
parser.banner = "Usage: rete [arguments] [file/urls]"
|
parser.banner = "Usage: rete [arguments] [file/urls]"
|
||||||
|
parser.on("-i STRING", "--includes=STRING", "Checks to see if a successful response") {|string|
|
||||||
|
includes = string
|
||||||
|
includesFlag = true}
|
||||||
parser.on("-t TIMEOUT", "--timeout=TIMEOUT", "Sets how many seconds until rete declares a timeout") {|timeout|
|
parser.on("-t TIMEOUT", "--timeout=TIMEOUT", "Sets how many seconds until rete declares a timeout") {|timeout|
|
||||||
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|
|
||||||
@@ -62,10 +67,21 @@ def checkUrlFromFile(content, output, urlTimeout) # Sorry for the crappy method
|
|||||||
|
|
||||||
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 response.body.lines.first.includes?(includes)
|
||||||
|
puts "#{urls[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}"
|
||||||
|
File.open("#{output}", "a") do |file|
|
||||||
|
file.puts urls[repeat]
|
||||||
|
end
|
||||||
|
else
|
||||||
|
puts "#{urls[repeat]} is a valid URL, works but does not include \"#{includes}\" Status code: #{response.status_code}"
|
||||||
|
end
|
||||||
|
else
|
||||||
puts "#{urls[repeat]} is a valid URL and works. Status code: #{response.status_code}"
|
puts "#{urls[repeat]} is a valid URL and works. 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]
|
||||||
end
|
end
|
||||||
|
end
|
||||||
working += 1
|
working += 1
|
||||||
else
|
else
|
||||||
puts "#{urls[repeat]} is a valid URL but does not work. Status Code: #{response.status_code}"
|
puts "#{urls[repeat]} is a valid URL but does not work. Status Code: #{response.status_code}"
|
||||||
@@ -119,8 +135,17 @@ while repeat < ARGV.size
|
|||||||
client.connect_timeout = urlTimeout.to_i.seconds
|
client.connect_timeout = urlTimeout.to_i.seconds
|
||||||
|
|
||||||
response = client.get("/")
|
response = client.get("/")
|
||||||
|
puts response.body.lines.first
|
||||||
if config.includes?(response.status_code.to_s)
|
if config.includes?(response.status_code.to_s)
|
||||||
puts "#{ARGV[repeat]} is a valid URL and works. Status code: #{response.status_code}"
|
if includesFlag
|
||||||
|
if response.body.lines.first.includes?(includes)
|
||||||
|
puts "#{ARGV[repeat]} is a valid URL, works and includes \"#{includes}\" Status code: #{response.status_code}"
|
||||||
|
else
|
||||||
|
puts "#{ARGV[repeat]} is a valid URL, works but does not include \"#{includes}\" Status code: #{response.status_code}"
|
||||||
|
end
|
||||||
|
else
|
||||||
|
puts "#{ARGV[repeat]} is a valid URL and works. Status code: #{response.status_code}"
|
||||||
|
end
|
||||||
working += 1
|
working += 1
|
||||||
else
|
else
|
||||||
puts "#{ARGV[repeat]} is a valid URL but does not work. Status Code: #{response.status_code}"
|
puts "#{ARGV[repeat]} is a valid URL but does not work. Status Code: #{response.status_code}"
|
||||||
|
|||||||
Reference in New Issue
Block a user