From 2aa55c94934e3461310520d22f99e36e4db36502 Mon Sep 17 00:00:00 2001 From: Marrub Date: Wed, 9 Aug 2017 05:56:43 -0400 Subject: [PATCH] Add searching to .quote --- source/mod_fun.rb | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/source/mod_fun.rb b/source/mod_fun.rb index be721aa..e413553 100644 --- a/source/mod_fun.rb +++ b/source/mod_fun.rb @@ -57,12 +57,18 @@ class Mod_Fun < Vrobot4::Module::Module end def c_quote m, argv - check_args argv, "", "N" + argv = argv.join(" ") if argv.length == 0 - max = JSON.parse(open(QDB + "/interface.cgi").read)["numQuotes"] - 1 m.reply_b open(QDB + "/q/" + (rand(max) + 1).to_s).read + elsif Vrobot4.is_num? argv + m.reply_b open(QDB + "/q/" + argv.to_i.to_s).read else - m.reply_b open(QDB + "/q/" + argv[0].to_i.to_s).read + raise ArgumentError, "invalid query" if argv.include? "&" + q = URI.escape argv + info = JSON.parse(open(QDB + "/interface.cgi?query=" + q).read) + res = info["searchResults"].sample + raise ArgumentError, "no results found" unless res + m.reply_b open(QDB + "/q/" + res.to_s).read end end