summaryrefslogtreecommitdiff
path: root/data/rbot/plugins
diff options
context:
space:
mode:
authorGiuseppe Bilotta <giuseppe.bilotta@gmail.com>2009-02-21 20:03:53 +0100
committerGiuseppe Bilotta <giuseppe.bilotta@gmail.com>2009-02-22 01:58:05 +0100
commit27353b5177b4c4ea50b8b8c6617cd350519308f5 (patch)
tree2cd1204d265eb8b556261ba6c43168c431b323ec /data/rbot/plugins
parent8aeceff885261b662ab373483e1669fcfbc42a21 (diff)
markov: try harder when generating strings
generate_strings() is now able to work with a single word. Additionally, when the given lookup is not found in the database, it will try lookups that start, and failing that include, the given one.
Diffstat (limited to 'data/rbot/plugins')
-rw-r--r--data/rbot/plugins/markov.rb52
1 files changed, 40 insertions, 12 deletions
diff --git a/data/rbot/plugins/markov.rb b/data/rbot/plugins/markov.rb
index 62559658..9da95af0 100644
--- a/data/rbot/plugins/markov.rb
+++ b/data/rbot/plugins/markov.rb
@@ -60,23 +60,51 @@ class MarkovPlugin < Plugin
def generate_string(word1, word2)
# limit to max of markov.max_words words
- output = word1 + " " + word2
+ if word2
+ output = "#{word1} #{word2}"
+ else
+ output = word1.to_s
+ end
- # try to avoid :nonword in the first iteration
- wordlist = @registry["#{word1} #{word2}"]
- wordlist.delete(:nonword)
- if not wordlist.empty?
- word3 = wordlist[rand(wordlist.length)]
- output = output + " " + word3
- word1, word2 = word2, word3
+ if @registry.key? output
+ wordlist = @registry[output]
+ wordlist.delete(:nonword)
+ else
+ output.downcase!
+ keys = []
+ @registry.each_key(output) do |key|
+ if key.downcase.include? output
+ keys << key
+ else
+ break
+ end
+ end
+ if keys.empty?
+ keys = @registry.keys.select { |k| k.downcase.include? output }
+ end
+ return nil if keys.empty?
+ while key = keys.delete_one
+ wordlist = @registry[key]
+ wordlist.delete(:nonword)
+ unless wordlist.empty?
+ output = key
+ word1, word2 = output.split
+ break
+ end
+ end
end
+ return nil if wordlist.empty?
+
+ word3 = wordlist.pick_one
+ output << " #{word3}"
+ word1, word2 = word2, word3
(@bot.config['markov.max_words'] - 1).times do
wordlist = @registry["#{word1} #{word2}"]
break if wordlist.empty?
- word3 = wordlist[rand(wordlist.length)]
+ word3 = wordlist.pick_one
break if word3 == :nonword
- output = output + " " + word3
+ output << " #{word3}"
word1, word2 = word2, word3
end
return output
@@ -196,7 +224,7 @@ class MarkovPlugin < Plugin
def chat(m, params)
line = generate_string(params[:seed1], params[:seed2])
- if line != "#{params[:seed1]} #{params[:seed2]}"
+ if line and line != [params[:seed1], params[:seed2]].compact.join(" ")
m.reply line
else
m.reply "I can't :("
@@ -262,7 +290,7 @@ plugin.map 'markov ignore', :action => "ignore"
plugin.map 'markov enable', :action => "enable"
plugin.map 'markov disable', :action => "disable"
plugin.map 'markov status', :action => "status"
-plugin.map 'chat about :seed1 :seed2', :action => "chat"
+plugin.map 'chat about :seed1 [:seed2]', :action => "chat"
plugin.map 'chat', :action => "rand_chat"
plugin.map 'markov probability [:probability]', :action => "probability",
:requirements => {:probability => /^\d+%?$/}