1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
|
#-- vim:sw=2:et
#++
#
# :title: Markov plugin
#
# Author:: Tom Gilbert <tom@linuxbrit.co.uk>
# Copyright:: (C) 2005 Tom Gilbert
#
# Contribute to chat with random phrases built from word sequences learned
# by listening to chat
class MarkovPlugin < Plugin
def initialize
super
@registry.set_default([])
@registry['enabled'] = false unless @registry.has_key?('enabled')
@lastline = false
end
def generate_string(word1, word2)
# limit to max of 50 words
output = word1 + " " + word2
# 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
end
49.times do
wordlist = @registry["#{word1} #{word2}"]
break if wordlist.empty?
word3 = wordlist[rand(wordlist.length)]
break if word3 == :nonword
output = output + " " + word3
word1, word2 = word2, word3
end
return output
end
def help(plugin, topic="")
"markov plugin: listens to chat to build a markov chain, with which it can (perhaps) attempt to (inanely) contribute to 'discussion'. Sort of.. Will get a *lot* better after listening to a lot of chat. usage: 'markov' to attempt to say something relevant to the last line of chat, if it can. other options to markov: 'ignore' => ignore a hostmask (accept no input), 'status' => show current status, 'probability [<chance>]' => set the % chance of rbot responding to input, or display the current probability, 'chat' => try and say something intelligent, 'chat about <foo> <bar>' => riff on a word pair (if possible)"
end
def clean_str(s)
str = s.dup
str.gsub!(/^\S+[:,;]/, "")
str.gsub!(/\s{2,}/, ' ') # fix for two or more spaces
return str.strip
end
def probability?
prob = @registry['probability']
prob = 25 if prob.kind_of? Array;
prob = 0 if prob < 0
prob = 100 if prob > 100
return prob
end
def status(m,params)
enabled = @registry['enabled']
if (enabled)
m.reply "markov is currently enabled, #{probability?}% chance of chipping in"
else
m.reply "markov is currently disabled"
end
end
def ignore?(user=nil)
return false unless user
@registry['ignore_users'].each do |mask|
return true if user.matches?(mask)
end
return false
end
def ignore(m, params)
if @registry['ignore_users'].nil?
@registry['ignore_users'] = []
end
action = params[:action]
user = params[:option]
case action
when 'remove':
if @registry['ignore_users'].include? user
s = @registry['ignore_users']
s.delete user
@registry['ignore_users'] = s
m.reply "#{user} removed"
else
m.reply "not found in list"
end
when 'add':
if user
if @registry['ignore_users'].include?(user)
m.reply "#{user} already in list"
else
@registry['ignore_users'] = @registry['ignore_users'].push user
m.reply "#{user} added to markov ignore list"
end
else
m.reply "give the name of a person to ignore"
end
when 'list':
m.reply "I'm ignoring #{@registry['ignore_users'].join(", ")}"
else
m.reply "have markov ignore the input from a hostmask. usage: markov ignore add <mask>; markov ignore remove <mask>; markov ignore list"
end
end
def enable(m, params)
@registry['enabled'] = true
m.okay
end
def probability(m, params)
if params[:probability]
@registry['probability'] = params[:probability].to_i
m.okay
else
m.reply _("markov has a %{prob}% chance of chipping in") % { :prob => probability? }
end
end
def disable(m, params)
@registry['enabled'] = false
m.okay
end
def should_talk
return false unless @registry['enabled']
prob = probability?
return true if prob > rand(100)
return false
end
def delay
1 + rand(5)
end
def random_markov(m, message)
return unless should_talk
word1, word2 = message.split(/\s+/)
line = generate_string(word1, word2)
return unless line
return if line == message
@bot.timer.add_once(delay) {
m.reply line
}
end
def chat(m, params)
line = generate_string(params[:seed1], params[:seed2])
if line != "#{params[:seed1]} #{params[:seed2]}"
m.reply line
else
m.reply "I can't :("
end
end
def rand_chat(m, params)
# pick a random pair from the db and go from there
word1, word2 = :nonword, :nonword
output = Array.new
50.times do
wordlist = @registry["#{word1} #{word2}"]
break if wordlist.empty?
word3 = wordlist[rand(wordlist.length)]
break if word3 == :nonword
output << word3
word1, word2 = word2, word3
end
if output.length > 1
m.reply output.join(" ")
else
m.reply "I can't :("
end
end
def listen(m)
return unless m.kind_of?(PrivMessage) && m.public?
return if m.address?
return if ignore? m.source
# in channel message, the kind we are interested in
message = clean_str m.message
if m.action?
message = "#{m.sourcenick} #{message}"
end
wordlist = message.split(/\s+/)
return unless wordlist.length >= 2
@lastline = message
word1, word2 = :nonword, :nonword
wordlist.each do |word3|
@registry["#{word1} #{word2}"] = @registry["#{word1} #{word2}"].push(word3)
word1, word2 = word2, word3
end
@registry["#{word1} #{word2}"] = @registry["#{word1} #{word2}"].push(:nonword)
return if m.replied?
random_markov(m, message)
end
end
plugin = MarkovPlugin.new
plugin.map 'markov ignore :action :option', :action => "ignore"
plugin.map 'markov ignore :action', :action => "ignore"
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', :action => "rand_chat"
plugin.map 'markov probability [:probability]', :action => "probability",
:requirements => {:probability => /^\d+%?$/}
|
