end
end
- def to_s
- @hash[:fact]
+ def to_s(opts={})
+ show_meta = opts[:meta]
+ fact = @hash[:fact]
+ if !show_meta
+ return fact
+ end
+ meta = ""
+ metadata = []
+ if @hash[:who]
+ metadata << _("from %{who}" % @hash)
+ end
+ if @hash[:when]
+ metadata << _("on %{when}" % @hash)
+ end
+ if @hash[:where]
+ metadata << _("in %{where}" % @hash)
+ end
+ unless metadata.empty?
+ meta << _(" [%{data}]" % {:data => metadata.join(" ")})
+ end
+ return fact+meta
end
def [](*args)
@hash[*args]
end
+ def []=(*args)
+ @hash.send(:[]=,*args)
+ end
+
def to_hsh
return @hash
end
+ alias :to_hash :to_hsh
end
class FactoidList < ArrayOf
@dir = File.join(@bot.botclass,"factoids")
@filename = "factoids.rbot"
@factoids = FactoidList.new
- read_factfile
+ begin
+ read_factfile
+ rescue
+ debug $!
+ end
@changed = false
end
def read_factfile(name=@filename,dir=@dir)
fname = File.join(dir,name)
+
+ expf = File.expand_path(fname)
+ expd = File.expand_path(dir)
+ raise ArgumentError, _("%{name} (%{fname}) must be under %{dir}" % {
+ :name => name,
+ :fname => expf,
+ :dir => dir
+ }) unless expf.index(expd) == 0
+
if File.exist?(fname)
+ raise ArgumentError, _("%{name} is not a file" % {
+ :name => name
+ }) unless File.file?(fname)
factoids = File.readlines(fname)
return if factoids.empty?
firstline = factoids.shift
}
else
pattern.map! { |p| p.intern }
- raise ArgumentError, "fact must be the last field" unless pattern.last == :fact
+ 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
+ else
+ raise ArgumentError, _("%{name} (%{fname}) doesn't exist" % {
+ :name => name,
+ :fname => fname
+ })
end
end
:who => m.source.fullform,
:where => m.channel.to_s
)
- if @factoids.index(factoid)
- m.reply _("I already know that %{factoid}" % { :factoid => factoid })
+ if idx = @factoids.index(factoid)
+ m.reply _("I already know that %{factoid} [#%{idx}]" % {
+ :factoid => factoid,
+ :idx => idx
+ })
else
@factoids << factoid
@changed = true
m.okay
+ fact(m, :index => @factoids.length.to_s)
end
end
def forget(m, params)
- factoid = params[:stuff].to_s
- if @factoids.delete(factoid)
- @changed = true
- m.okay
+ if params[:index]
+ idx = params[:index].scan(/\d+/).first.to_i
+ total = @factoids.length
+ if idx <= 0 or idx > total
+ m.reply _("please select a fact number between 1 and %{total}" % { :total => total })
+ return
+ end
+ if factoid = @factoids.delete_at(idx-1)
+ m.reply _("I forgot that %{factoid}" % { :factoid => factoid })
+ @changed = true
+ else
+ m.reply _("I couldn't delete factoid %{idx}" % { :idx => idx })
+ end
else
- m.reply _("I didn't know that %{factoid}" % { :factoid => factoid })
+ 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
end
+ def long_fact(fact,index=nil,total=@factoids.length)
+ idx = index || @factoids.index(fact)+1
+ _("fact #%{idx} of %{total}: %{fact}" % {
+ :idx => idx,
+ :total => total,
+ :fact => fact.to_s(:meta => true)
+ })
+ end
+
def facts(m, params)
+ total = @factoids.length
if params[:words].empty?
- m.reply _("I know %{count} facts" % { :count => @factoids.length })
+ m.reply _("I know %{total} facts" % { :total => total })
else
rx = Regexp.new(params[:words].to_s, true)
known = @factoids.grep(rx)
+ reply = []
if known.empty?
- m.reply _("I know nothing about %{words}" % params)
+ reply << _("I know nothing about %{words}" % params)
else
- m.reply known.join(" | "), :split_at => /\s+\|\s+/
+ # TODO config
+ max_facts = 5
+ len = known.length
+ if len > max_facts
+ m.reply _("%{len} out of %{total} facts refer to %{words}, I'll only show %{max}" % {
+ :len => len,
+ :total => total,
+ :words => params[:words].to_s,
+ :max => max_facts
+ })
+ while known.length > max_facts
+ known.delete_one
+ end
+ end
+ known.each { |f|
+ reply << long_fact(f)
+ }
end
+ m.reply reply.join(" -- ")
end
end
def fact(m, params)
- known = nil
- if params[:words].empty?
- if @factoids.empty?
- m.reply _("I know nothing")
+ fact = nil
+ idx = 0
+ total = @factoids.length
+ if params[:index]
+ idx = params[:index].scan(/\d+/).first.to_i
+ if idx <= 0 or idx > total
+ m.reply _("please select a fact number between 1 and %{total}" % { :total => total })
return
end
- known = @factoids
+ fact = @factoids[idx-1]
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
+ 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
- fact = known.pick_one
- idx = @factoids.index(fact)+1
- m.reply _("fact %{idx}/%{total}: %{fact}" % {
- :idx => idx,
- :total => @factoids.length,
- :fact => fact
+ m.reply long_fact(fact, idx, total)
+ end
+
+ def edit_fact(m, params)
+ fact = nil
+ idx = 0
+ total = @factoids.length
+ idx = params[:index].scan(/\d+/).first.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]
+ begin
+ if params[:who]
+ who = params[:who].to_s.sub(/^me$/, m.source.fullform)
+ fact[:who] = who
+ @changed = true
+ end
+ if params[:when]
+ dstr = params[:when].to_s
+ begin
+ fact[:when] = Time.parse(dstr, "")
+ @changed = true
+ rescue
+ raise ArgumentError, _("not a date '%{dstr}'" % { :dstr => dstr })
+ end
+ end
+ if params[:where]
+ fact[:where] = params[:where].to_s
+ @changed = true
+ end
+ rescue Exception
+ m.reply _("couldn't change learn data for fact %{fact}: %{err}" % {
+ :fact => fact,
+ :err => $!
+ })
+ return
+ end
+ m.okay
+ end
+
+ def import(m, params)
+ fname = params[:filename].to_s
+ oldlen = @factoids.length
+ begin
+ read_factfile(fname)
+ rescue
+ m.reply _("failed to import facts from %{fname}: %{err}" % {
+ :fname => fname,
+ :err => $!
+ })
+ end
+ m.reply _("%{len} facts loaded from %{fname}" % {
+ :fname => fname,
+ :len => @factoids.length - oldlen
})
+ @changed = true
end
end
plugin = FactoidsPlugin.new
+
+plugin.default_auth('edit', false)
+plugin.default_auth('import', false)
+
plugin.map 'learn that *stuff'
-plugin.map 'forget that *stuff'
+plugin.map 'forget that *stuff', :auth_path => 'edit'
+plugin.map 'forget fact :index', :requirements => { :index => /^#?\d+$/ }, :auth_path => 'edit'
plugin.map 'facts [about *words]'
plugin.map 'fact [about *words]'
+plugin.map 'fact :index', :requirements => { :index => /^#?\d+$/ }
+
+plugin.map 'fact :index :learn from *who', :action => :edit_fact, :requirements => { :learn => /^((?:is|was)\s+)?learn(ed|t)$/, :index => /^#?\d+$/ }, :auth_path => 'edit'
+plugin.map 'fact :index :learn on *when', :action => :edit_fact, :requirements => { :learn => /^((?:is|was)\s+)?learn(ed|t)$/, :index => /^#?\d+$/ }, :auth_path => 'edit'
+plugin.map 'fact :index :learn in *where', :action => :edit_fact, :requirements => { :learn => /^((?:is|was)\s+)?learn(ed|t)$/, :index => /^#?\d+$/ }, :auth_path => 'edit'
+
+plugin.map 'facts import [from] *filename', :action => :import, :auth_path => 'import'
projects / user / henk / code / ruby / rbot.git / blobdiff