:validate => Proc.new { |v| v >= 0 },
:desc => "Time the learning thread spends sleeping after learning a line. If set to zero, learning from files can be very CPU intensive, but also faster.")
Config.register Config::IntegerValue.new('markov.delay',
- :default => true,
+ :default => 5,
:validate => Proc.new { |v| v >= 0 },
:desc => "Wait short time before contributing to conversation.")
Config.register Config::IntegerValue.new('markov.answer_addressed',
next
else
# intern after clearing leftover end-of-actions if present
- sym = w.chomp("\001").intern
+ sym = w.chomp("\001")
end
end
hash[sym] += 1
end
debug 'closing learning thread'
+ @learning_queue.clear
@learning_queue.push nil
@learning_thread.join
debug 'learning thread closed'
+ @chains.close
+ @rchains.close
+ super
end
- # if passed a pair, pick a word from the registry using the pair as key.
- # otherwise, pick a word from an given list
+ # pick a word from the registry using the pair as key.
def pick_word(word1, word2=MARKER, chainz=@chains)
- if word1.kind_of? Array
- wordlist = word1
- else
- k = "#{word1} #{word2}"
- return MARKER unless chainz.key? k
- wordlist = chainz[k]
- end
+ k = "#{word1} #{word2}"
+ return MARKER unless chainz.key? k
+ wordlist = chainz[k]
+ pick_word_from_list wordlist
+ end
+
+ # pick a word from weighted hash
+ def pick_word_from_list(wordlist)
total = wordlist.first
hash = wordlist.last
return MARKER if total == 0
output = word1
keys = []
@chains.each_key(output) do |key|
- if key.downcase.include? output
- keys << key
- else
- break
- end
+ if key.downcase.include? output
+ keys << key
+ else
+ break
+ end
end
return nil if keys.empty?
output = keys[rand(keys.size)].split(/ /)
- end
+ end
output = output.split(/ /) unless output.is_a? Array
input = [word1, word2]
while output.length < @bot.config['markov.max_words'] and (output.first != MARKER or output.last != MARKER) do
if output == input
nil
else
- output.join(" ")
- end
+ output.join(" ")
+ end
end
def help(plugin, topic="")
end
end
- def clean_str(s)
- str = s.dup
- str.gsub!(/^\S+[:,;]/, "")
+ def clean_message(m)
+ str = m.plainmessage.dup
+ str =~ /^(\S+)([:,;])/
+ if $1 and m.target.is_a? Irc::Channel and m.target.user_nicks.include? $1.downcase
+ str.gsub!(/^(\S+)([:,;])\s+/, "")
+ end
str.gsub!(/\s{2,}/, ' ') # fix for two or more spaces
return str.strip
end
m.okay
end
- def should_talk
+ def should_talk(m)
return false unless @bot.config['markov.enabled']
- prob = probability?
+ prob = m.address? ? @bot.config['markov.answer_addressed'] : probability?
return true if prob > rand(100)
return false
end
def reply_delay(m, line)
m.replied = true
if @bot.config['markov.delay'] > 0
- @bot.timer.add_once(@bot.config['markov.delay']) {
+ @bot.timer.add_once(1 + rand(@bot.config['markov.delay'])) {
m.reply line, :nick => false, :to => :public
}
else
end
def random_markov(m, message)
- return unless (should_talk or (m.address? and @bot.config['markov.answer_addressed'] > rand(100)))
+ return unless should_talk(m)
- words = clean_str(message).split(/\s+/)
+ words = clean_message(m).split(/\s+/)
if words.length < 2
line = generate_string words.first, nil
else
pairs = seq_pairs(words).sort_by { rand }
pairs.each do |word1, word2|
- line = generate_string(word1.intern, word2.intern)
+ line = generate_string(word1, word2)
if line and message.index(line) != 0
reply_delay m, line
return
end
random_markov(m, message) unless readonly? m or m.replied?
- learn message
+ learn clean_message(m)
end
def learn_line(message)
# debug "learning #{message.inspect}"
- wordlist = clean_str(message).split(/\s+/).reject do |w|
+ wordlist = message.strip.split(/\s+/).reject do |w|
@bot.config['markov.ignore_patterns'].map do |pat|
w =~ Regexp.new(pat.to_s)
end.select{|v| v}.size != 0
- end.map { |w| w.intern }
+ end
return unless wordlist.length >= 2
word1, word2 = MARKER, MARKER
wordlist << MARKER
wordlist.each do |word3|
- learn_triplet(word1, word2, word3)
+ learn_triplet(word1, word2, word3.to_sym)
word1, word2 = word2, word3
end
end
m.okay
end
+
+ def stats(m, params)
+ m.reply "Markov status: chains: #{@chains.length} forward, #{@rchains.length} reverse, queued phrases: #{@learning_queue.size}"
+ end
+
end
plugin = MarkovPlugin.new
plugin.map 'markov enable', :action => "enable"
plugin.map 'markov disable', :action => "disable"
plugin.map 'markov status', :action => "status"
+plugin.map 'markov stats', :action => "stats"
plugin.map 'chat about :seed1 [:seed2]', :action => "chat"
plugin.map 'chat', :action => "rand_chat"
plugin.map 'markov probability [:probability]', :action => "probability",