# type of keyword (e.g. "is" or "are")
attr_reader :type
-
+
# type:: type of keyword (e.g "is" or "are")
# values:: array of values
- #
+ #
# create a keyword of type +type+ with values +values+
def initialize(type, values)
@type = type.downcase
end
end
-# keywords class.
+# keywords class.
#
# Handles all that stuff like "bot: foo is bar", "bot: foo?"
#
BotConfig.register BotConfigBooleanValue.new('keyword.address',
:default => true,
:desc => "Should the bot require that keyword lookups are addressed to it? If not, the bot will attempt to lookup foo if someone says 'foo?' in channel")
-
+ BotConfig.register BotConfigIntegerValue.new('keyword.search_results',
+ :default => 3,
+ :desc => "How many search results to display at a time")
+
# create a new KeywordPlugin instance, associated to bot +bot+
def initialize
super
upgrade_data
scan
-
+
# import old format keywords into DBHash
if(File.exist?("#{@bot.botclass}/keywords.rbot"))
log "auto importing old keywords.rbot"
File.rename("#{@bot.botclass}/keywords.rbot", "#{@bot.botclass}/keywords.rbot.old")
end
end
-
+
# drop static keywords and reload them from files, picking up any new
# keyword files that have been added
def rescan
next unless f =~ /\.db$/
log "upgrading keyword db #{f} (rbot 0.9.5 or prior) database format"
newname = f.gsub(/\.db$/, ".kdb")
- old = BDB::Hash.open f, nil,
+ old = BDB::Hash.open f, nil,
"r+", 0600
- new = BDB::CIBtree.open(newname, nil,
+ new = BDB::CIBtree.open(newname, nil,
BDB::CREATE | BDB::EXCL,
0600)
old.each {|k,v|
new.close
File.delete(f)
}
-
+
# then scan for current DBTree files, and load them
Dir["#{@bot.botclass}/keywords/*"].each {|f|
next unless f =~ /\.kdb$/
debug "keywords module: loading DBTree file #{f}, key #{key}"
@statickeywords[key] = hsh
}
-
+
# then scan for non DB files, and convert/import them and delete
Dir["#{@bot.botclass}/keywords/*"].each {|f|
next if f =~ /\.kdb$/
def upgrade_data
if File.exist?("#{@bot.botclass}/keywords.db")
log "upgrading old keywords (rbot 0.9.5 or prior) database format"
- old = BDB::Hash.open "#{@bot.botclass}/keywords.db", nil,
+ old = BDB::Hash.open "#{@bot.botclass}/keywords.db", nil,
"r+", 0600
old.each {|k,v|
@keywords[k] = v
@keywords.flush
File.rename("#{@bot.botclass}/keywords.db", "#{@bot.botclass}/keywords.db.old")
end
-
+
if File.exist?("#{@bot.botclass}/keyword.db")
log "upgrading old keywords (rbot 0.9.9 or prior) database format"
- old = BDB::CIBtree.open "#{@bot.botclass}/keyword.db", nil,
+ old = BDB::CIBtree.open "#{@bot.botclass}/keyword.db", nil,
"r+", 0600
old.each {|k,v|
@keywords[k] = v
end
end
end
-
+
# lookup keyword +key+, return it or nil
def [](key)
return nil if key.nil?
# m:: PrivMessage containing message info
# key:: key being queried
# quiet:: optional, if false, complain if +key+ is not found
- #
+ #
# handle a message asking about a keyword
def keyword_lookup(m, key, quiet = false)
return if key.nil?
m.reply "sorry, I don't know about \"#{key}\"" unless quiet
return
end
-
+
response = kw.to_s
response.gsub!(/<who>/, m.sourcenick)
-
+
if(response =~ /^<reply>\s*(.*)/)
m.reply $1
elsif(response =~ /^<action>\s*(.*)/)
end
end
-
+
# handle a message which alters a keyword
# like "foo is bar" or "foo is also qux"
def keyword_command(m, lhs, mhs, rhs, quiet = false)
debug "got keyword command #{lhs}, #{mhs}, #{rhs}"
-
+ return if lhs.strip.empty?
+
+ overwrite = false
+ overwrite = true if(lhs.gsub!(/^no,\s*/, ""))
+ also = false
also = true if(rhs.gsub!(/^also\s+/, ""))
-
+
values = rhs.split(/\s+\|\s+/)
lhs = Keyword.unescape lhs
-
- if(also && has_key?(lhs))
+
+ if(overwrite || also || !has_key?(lhs))
+ if(also && has_key?(lhs))
+ kw = self[lhs]
+ kw << values
+ @keywords[lhs] = kw.dump
+ else
+ @keywords[lhs] = Keyword.new(mhs, values).dump
+ end
+ m.okay if !quiet
+ elsif(has_key?(lhs))
kw = self[lhs]
- kw << values
- @keywords[lhs] = kw.dump
- else
- @keywords[lhs] = Keyword.new(mhs, values).dump
+ m.reply "but #{lhs} #{kw.type} #{kw.desc}" if kw && !quiet
end
-
- @bot.okay m.target if !quiet
end
# return help string for Keywords with option topic +topic+
def help(plugin, topic = '')
- case topic
- when 'lookup'
- 'keyword [lookup] <keyword> => look up the definition for a keyword; writing "lookup" is optional'
- when 'set'
- 'keyword set <keyword> is/are <definition> => define a keyword, definition can contain "|" to separate multiple randomly chosen replies'
- when 'forget'
- 'keyword forget <keyword> => forget a keyword'
- when 'tell'
- 'keyword tell <nick> about <keyword> => tell somebody about a keyword'
- when 'search'
- 'keyword search [--all] [--full] <pattern> => search keywords for <pattern>, which can be a regular expression. If --all is set, search static keywords too, if --full is set, search definitions too.'
- when 'listen'
- 'when the config option "keyword.listen" is set to false, rbot will try to extract keyword definitions from regular channel messages'
- when 'address'
- 'when the config option "keyword.address" is set to true, rbot will try to answer channel questions of the form "<keyword>?"'
- when '<reply>'
- '<reply> => normal response is "<keyword> is <definition>", but if <definition> begins with <reply>, the response will be "<definition>"'
- when '<action>'
- '<action> => makes keyword respond with "/me <definition>"'
- when '<who>'
- '<who> => replaced with questioner in reply'
- when '<topic>'
- '<topic> => respond by setting the topic to the rest of the definition'
+ case plugin
+ when /keyword/
+ case topic
+ when 'lookup'
+ 'keyword [lookup] <keyword> => look up the definition for a keyword; writing "lookup" is optional'
+ when 'set'
+ 'keyword set <keyword> is/are <definition> => define a keyword, definition can contain "|" to separate multiple randomly chosen replies'
+ when 'forget'
+ 'keyword forget <keyword> => forget a keyword'
+ when 'tell'
+ 'keyword tell <nick> about <keyword> => tell somebody about a keyword'
+ when 'search'
+ 'keyword search [--all] [--full] <pattern> => search keywords for <pattern>, which can be a regular expression. If --all is set, search static keywords too, if --full is set, search definitions too.'
+ when 'listen'
+ 'when the config option "keyword.listen" is set to false, rbot will try to extract keyword definitions from regular channel messages'
+ when 'address'
+ 'when the config option "keyword.address" is set to true, rbot will try to answer channel questions of the form "<keyword>?"'
+ when '<reply>'
+ '<reply> => normal response is "<keyword> is <definition>", but if <definition> begins with <reply>, the response will be "<definition>"'
+ when '<action>'
+ '<action> => makes keyword respond with "/me <definition>"'
+ when '<who>'
+ '<who> => replaced with questioner in reply'
+ when '<topic>'
+ '<topic> => respond by setting the topic to the rest of the definition'
+ else
+ 'keyword module (fact learning and regurgitation) topics: lookup, set, forget, tell, search, listen, address, <reply>, <action>, <who>, <topic>'
+ end
+ when "forget"
+ 'forget <keyword> => forget a keyword'
+ when "tell"
+ 'tell <nick> about <keyword> => tell somebody about a keyword'
+ when "learn"
+ 'learn that <keyword> is/are <definition> => define a keyword, definition can contain "|" to separate multiple randomly chosen replies'
else
'keyword module (fact learning and regurgitation) topics: lookup, set, forget, tell, search, listen, address, <reply>, <action>, <who>, <topic>'
end
m.reply @bot.lang.get("dunno_about_X") % key
return
end
-
+ if target == @bot.nick
+ m.reply "very funny, trying to make me tell something to myself"
+ return
+ end
+
response = kw.to_s
response.gsub!(/<who>/, m.sourcenick)
if(response =~ /^<reply>\s*(.*)/)
end
# search for keywords, optionally also the definition and the static keywords
- def keyword_search(m, key, full = false, all = false)
+ def keyword_search(m, key, full = false, all = false, from = 1)
begin
if key =~ /^\/(.+)\/$/
re = Regexp.new($1, Regexp::IGNORECASE)
else
re = Regexp.new(Regexp.escape(key), Regexp::IGNORECASE)
end
-
+
matches = Array.new
@keywords.each {|k,v|
kw = Keyword.restore(v)
}
}
end
-
+
if matches.length == 1
rkw = matches[0]
m.reply "#{rkw[0]} #{rkw[1].type} #{rkw[1].desc}"
elsif matches.length > 0
- i = 0
+ if from > matches.length
+ m.reply "#{matches.length} found, can't tell you about #{from}"
+ return
+ end
+ i = 1
matches.each {|rkw|
- m.reply "[#{i+1}/#{matches.length}] #{rkw[0]} #{rkw[1].type} #{rkw[1].desc}"
+ m.reply "[#{i}/#{matches.length}] #{rkw[0]} #{rkw[1].type} #{rkw[1].desc}" if i >= from
i += 1
- break if i == 4
+ break if i == from+@bot.config['keyword.search_results']
}
else
m.reply "no keywords match #{key}"
# privmsg handler
def privmsg(m)
- case m.params
- when /^set\s+(.+?)\s+(is|are)\s+(.+)$/
- keyword_command(m, $1, $2, $3) if @bot.auth.allow?('keycmd', m.source, m.replyto)
- when /^forget\s+(.+)$/
- keyword_forget(m, $1) if @bot.auth.allow?('keycmd', m.source, m.replyto)
- when /^lookup\s+(.+)$/
- keyword_lookup(m, $1) if @bot.auth.allow?('keyword', m.source, m.replyto)
- when /^stats\s*$/
- keyword_stats(m) if @bot.auth.allow?('keyword', m.source, m.replyto)
- when /^search\s+(.+)$/
- key = $1
- full = key.sub!('--full ', '')
- all = key.sub!('--all ', '')
- keyword_search(m, key, full, all) if @bot.auth.allow?('keyword', m.source, m.replyto)
- when /^tell\s+(\S+)\s+about\s+(.+)$/
- keyword_tell(m, $1, $2) if @bot.auth.allow?('keyword', m.source, m.replyto)
- else
- keyword_lookup(m, m.params) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ case m.plugin
+ when "keyword"
+ case m.params
+ when /^set\s+(.+?)\s+(is|are)\s+(.+)$/
+ keyword_command(m, $1, $2, $3) if @bot.auth.allow?('keycmd', m.source, m.replyto)
+ when /^forget\s+(.+)$/
+ keyword_forget(m, $1) if @bot.auth.allow?('keycmd', m.source, m.replyto)
+ when /^lookup\s+(.+)$/
+ keyword_lookup(m, $1) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ when /^stats\s*$/
+ keyword_stats(m) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ when /^search\s+(.+)$/
+ key = $1
+ full = key.sub!('--full ', '')
+ all = key.sub!('--all ', '')
+ if key.sub!(/--from (\d+) /, '')
+ from = $1.to_i
+ else
+ from = 1
+ end
+ from = 1 unless from > 0
+ keyword_search(m, key, full, all, from) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ when /^tell\s+(\S+)\s+about\s+(.+)$/
+ keyword_tell(m, $1, $2) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ else
+ keyword_lookup(m, m.params) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ end
+ when "forget"
+ keyword_forget(m, m.params) if @bot.auth.allow?('keycmd', m.source, m.replyto)
+ when "tell"
+ if m.params =~ /(\S+)\s+about\s+(.+)$/
+ keyword_tell(m, $1, $2) if @bot.auth.allow?('keyword', m.source, m.replyto)
+ else
+ m.reply "wrong 'tell' syntax"
+ end
+ when "learn"
+ if m.params =~ /^that\s+(.+?)\s+(is|are)\s+(.+)$/
+ keyword_command(m, $1, $2, $3) if @bot.auth.allow?('keycmd', m.source, m.replyto)
+ else
+ m.reply "wrong 'learn' syntax"
+ end
end
end
- def listen(m)
- return if m.address?
+ def unreplied(m)
+ # return if m.address?
# in channel message, not to me
# TODO option to do if(m.message =~ /^(.*)$/, ie try any line as a
# keyword lookup.
plugin = Keywords.new
plugin.register 'keyword'
+plugin.register 'forget'
+plugin.register 'tell'
+plugin.register 'learn'
+
projects / user / henk / code / ruby / rbot.git / blobdiff