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
|
#-- vim:sw=2:et
#++
#
# :title: Factoids pluing
#
# Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
# Copyright:: (C) 2007 Giuseppe Bilotta
# License:: GPLv2
#
# Store (and retrieve) unstructured one-sentence factoids
class FactoidsPlugin < Plugin
class Factoid
def initialize(hash)
@hash = hash.reject { |k, val| val.nil? or val.empty? rescue false }
raise ArgumentError, "no fact!" unless @hash[:fact]
if String === @hash[:when]
@hash[:when] = Time.parse @hash[:when]
end
end
def to_s
@hash[:fact]
end
def [](*args)
@hash[*args]
end
def to_hsh
return @hash
end
end
class FactoidList < ArrayOf
def initialize(ar=[])
super(Factoid, ar)
end
def index(f)
fact = f.to_s
return if fact.empty?
self.map { |f| f[:fact] }.index(fact)
end
def delete(f)
idx = index(f)
return unless idx
self.delete_at(idx)
end
def grep(x)
self.find_all { |f|
x === f[:fact]
}
end
end
def initialize
super
# TODO config
@dir = File.join(@bot.botclass,"factoids")
@filename = "factoids.rbot"
@factoids = FactoidList.new
read_factfile
@changed = false
end
def read_factfile(name=@filename,dir=@dir)
fname = File.join(dir,name)
if File.exist?(fname)
factoids = File.readlines(fname)
return if factoids.empty?
firstline = factoids.shift
pattern = firstline.chomp.split(" | ")
if pattern.length == 1 and pattern.first != "fact"
factoids.unshift(firstline)
factoids.each { |f|
@factoids << Factoid.new( :fact => f.chomp )
}
else
pattern.map! { |p| p.intern }
raise ArgumentError, "fact must be the last field" unless pattern.last == :fact
factoids.each { |f|
ar = f.chomp.split(" | ", pattern.length)
@factoids << Factoid.new(Hash[*([pattern, ar].transpose.flatten)])
}
end
end
end
def save
return unless @changed
Dir.mkdir(@dir) unless FileTest.directory?(@dir)
fname = File.join(@dir,@filename)
ar = ["when | who | where | fact"]
@factoids.each { |f|
ar << "%s | %s | %s | %s" % [ f[:when], f[:who], f[:where], f[:fact]]
}
Utils.safe_save(fname) do |file|
file.puts ar
end
@changed = false
end
def help(plugin, topic="")
_("factoids plugin: learn that <factoid>, forget that <factoids>, facts about <words>")
end
def learn(m, params)
factoid = Factoid.new(
:fact => params[:stuff].to_s,
:when => Time.now,
:who => m.source.fullform,
:where => m.channel.to_s
)
if @factoids.index(factoid)
m.reply _("I already know that %{factoid}" % { :factoid => factoid })
else
@factoids << factoid
@changed = true
m.okay
end
end
def forget(m, params)
factoid = params[:stuff].to_s
if @factoids.delete(factoid)
@changed = true
m.okay
else
m.reply _("I didn't know that %{factoid}" % { :factoid => factoid })
end
end
def facts(m, params)
if params[:words].empty?
m.reply _("I know %{count} facts" % { :count => @factoids.length })
else
rx = Regexp.new(params[:words].to_s, true)
known = @factoids.grep(rx)
if known.empty?
m.reply _("I know nothing about %{words}" % params)
else
m.reply known.join(" | "), :split_at => /\s+\|\s+/
end
end
end
def fact(m, params)
fact = nil
idx = 0
total = @factoids.length
if params[:index]
idx = params[:index].to_i
if idx <= 0 or idx > total
m.reply _("please select a fact number between 1 and %{total}" % { :total => total })
return
end
fact = @factoids[idx-1]
else
known = nil
if params[:words].empty?
if @factoids.empty?
m.reply _("I know nothing")
return
end
known = @factoids
else
rx = Regexp.new(params[:words].to_s, true)
known = @factoids.grep(rx)
if known.empty?
m.reply _("I know nothing about %{words}" % params)
return
end
end
fact = known.pick_one
idx = @factoids.index(fact)+1
end
m.reply _("fact %{idx}/%{total}: %{fact}" % {
:idx => idx,
:total => total,
:fact => fact
})
end
end
plugin = FactoidsPlugin.new
plugin.default_auth('edit', false)
plugin.map 'learn that *stuff'
plugin.map 'forget that *stuff', :auth_path => 'edit'
plugin.map 'facts [about *words]'
plugin.map 'fact [about *words]'
plugin.map 'fact :index', :requirements => { :index => /\d+/ }
|
