From eee6a74ea1547b5c0e8757c64cace42fc9aea9bf Mon Sep 17 00:00:00 2001 From: Giuseppe Bilotta Date: Wed, 2 Aug 2006 20:37:02 +0000 Subject: Initial factorization of botconfig into kernel functionality and a coremodule --- lib/rbot/config.rb | 321 +++++++++++++----------------------------------- lib/rbot/core/config.rb | 252 +++++++++++++++++++++++++++++++++++++ lib/rbot/core/core.rb | 60 +-------- lib/rbot/ircbot.rb | 11 +- lib/rbot/keywords.rb | 10 +- lib/rbot/plugins.rb | 2 +- 6 files changed, 357 insertions(+), 299 deletions(-) create mode 100644 lib/rbot/core/config.rb (limited to 'lib') diff --git a/lib/rbot/config.rb b/lib/rbot/config.rb index f91cfa70..e8cea284 100644 --- a/lib/rbot/config.rb +++ b/lib/rbot/config.rb @@ -1,7 +1,8 @@ +require 'singleton' + module Irc require 'yaml' - require 'rbot/messagemapper' unless YAML.respond_to?(:load_file) def YAML.load_file( filepath ) @@ -21,8 +22,11 @@ module Irc attr_reader :key attr_reader :wizard attr_reader :requires_restart + attr_reader :requires_rescan attr_reader :order + attr_reader :manager def initialize(key, params) + @manager = BotConfig::configmanager # Keys must be in the form 'module.name'. # They will be internally passed around as symbols, # but we accept them both in string and symbol form. @@ -31,7 +35,7 @@ module Irc end @order = @@order @@order += 1 - @key = key.intern + @key = key.to_sym if params.has_key? :default @default = params[:default] else @@ -43,6 +47,7 @@ module Irc @validate = params[:validate] @wizard = params[:wizard] @requires_restart = params[:requires_restart] + @requires_rescan = params[:requires_rescan] end def default if @default.instance_of?(Proc) @@ -52,16 +57,16 @@ module Irc end end def get - return BotConfig.config[@key] if BotConfig.config.has_key?(@key) + return @manager.config[@key] if @manager.config.has_key?(@key) return @default end alias :value :get def set(value, on_change = true) - BotConfig.config[@key] = value - @on_change.call(BotConfig.bot, value) if on_change && @on_change + @manager.config[@key] = value + @on_change.call(@manager.bot, value) if on_change && @on_change end def unset - BotConfig.config.delete(@key) + @manager.config.delete(@key) end # set string will raise ArgumentErrors on failed parse/validate @@ -73,7 +78,7 @@ module Irc raise ArgumentError, "invalid value: #{string}" end end - + # override this. the default will work for strings only def parse(string) string @@ -99,6 +104,7 @@ module Irc class BotConfigStringValue < BotConfigValue end + class BotConfigBooleanValue < BotConfigValue def parse(string) return true if string == "true" @@ -106,18 +112,21 @@ module Irc raise ArgumentError, "#{string} does not match either 'true' or 'false'" end end + class BotConfigIntegerValue < BotConfigValue def parse(string) raise ArgumentError, "not an integer: #{string}" unless string =~ /^-?\d+$/ string.to_i end end + class BotConfigFloatValue < BotConfigValue def parse(string) raise ArgumentError, "not a float #{string}" unless string =~ /^-?[\d.]+$/ string.to_f end end + class BotConfigArrayValue < BotConfigValue def parse(string) string.split(/,\s+/) @@ -135,6 +144,7 @@ module Irc set(curval - [val]) end end + class BotConfigEnumValue < BotConfigValue def initialize(key, params) super @@ -142,7 +152,7 @@ module Irc end def values if @values.instance_of?(Proc) - return @values.call(BotConfig.bot) + return @values.call(@manager.bot) else return @values end @@ -159,26 +169,51 @@ module Irc end # container for bot configuration - class BotConfig - # Array of registered BotConfigValues for defaults, types and help - @@items = Hash.new - def BotConfig.items - @@items + class BotConfigManagerClass + + include Singleton + + attr_reader :bot + attr_reader :items + attr_reader :config + + def initialize + bot_associate(nil,true) end - # Hash containing key => value pairs for lookup and serialisation - @@config = Hash.new(false) - def BotConfig.config - @@config + + def reset_config + @items = Hash.new + @config = Hash.new(false) end - def BotConfig.bot - @@bot + + # Associate with bot _bot_ + def bot_associate(bot, reset=false) + reset_config if reset + @bot = bot + return unless @bot + + if(File.exist?("#{@bot.botclass}/conf.yaml")) + begin + newconfig = YAML::load_file("#{@bot.botclass}/conf.yaml") + newconfig.each { |key, val| + @config[key.to_sym] = val + } + return + rescue + error "failed to read conf.yaml: #{$!}" + end + end + # if we got here, we need to run the first-run wizard + BotConfigWizard.new(@bot).run + # save newly created config + save end - - def BotConfig.register(item) + + def register(item) unless item.kind_of?(BotConfigValue) raise ArgumentError,"item must be a BotConfigValue" end - @@items[item.key] = item + @items[item.key] = item end # currently we store values in a hash but this could be changed in the @@ -186,17 +221,17 @@ module Irc # components that register their config keys and setup defaults are # supported via [] def [](key) - return @@items[key].value if @@items.has_key?(key) - return @@items[key.intern].value if @@items.has_key?(key.intern) + # return @items[key].value if @items.has_key?(key) + return @items[key.to_sym].value if @items.has_key?(key.to_sym) # try to still support unregistered lookups # but warn about them - if @@config.has_key?(key) - warning "Unregistered lookup #{key.inspect}" - return @@config[key] - end - if @@config.has_key?(key.intern) - warning "Unregistered lookup #{key.intern.inspect}" - return @@config[key.intern] + # if @config.has_key?(key) + # warning "Unregistered lookup #{key.inspect}" + # return @config[key] + # end + if @config.has_key?(key.to_sym) + warning "Unregistered lookup #{key.to_sym.inspect}" + return @config[key.to_sym] end return false end @@ -204,237 +239,54 @@ module Irc # TODO should I implement this via BotConfigValue or leave it direct? # def []=(key, value) # end - + # pass everything else through to the hash def method_missing(method, *args, &block) - return @@config.send(method, *args, &block) - end - - def handle_list(m, params) - modules = [] - if params[:module] - @@items.each_key do |key| - mod, name = key.to_s.split('.') - next unless mod == params[:module] - modules.push key unless modules.include?(name) - end - if modules.empty? - m.reply "no such module #{params[:module]}" - else - m.reply modules.join(", ") - end - else - @@items.each_key do |key| - name = key.to_s.split('.').first - modules.push name unless modules.include?(name) - end - m.reply "modules: " + modules.join(", ") - end - end - - def handle_get(m, params) - key = params[:key].to_s.intern - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - return - end - value = @@items[key].to_s - m.reply "#{key}: #{value}" - end - - def handle_desc(m, params) - key = params[:key].to_s.intern - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - end - puts @@items[key].inspect - m.reply "#{key}: #{@@items[key].desc}" - end - - def handle_unset(m, params) - key = params[:key].to_s.intern - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - end - @@items[key].unset - handle_get(m, params) - m.reply "this config change will take effect on the next restart" if @@items[key].requires_restart - end - - def handle_set(m, params) - key = params[:key].to_s.intern - value = params[:value].join(" ") - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - return - end - begin - @@items[key].set_string(value) - rescue ArgumentError => e - m.reply "failed to set #{key}: #{e.message}" - return - end - if @@items[key].requires_restart - m.reply "this config change will take effect on the next restart" - else - m.okay - end - end - - def handle_add(m, params) - key = params[:key].to_s.intern - value = params[:value] - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - return - end - unless @@items[key].class <= BotConfigArrayValue - m.reply "config key #{key} is not an array" - return - end - begin - @@items[key].add(value) - rescue ArgumentError => e - m.reply "failed to add #{value} to #{key}: #{e.message}" - return - end - handle_get(m,{:key => key}) - m.reply "this config change will take effect on the next restart" if @@items[key].requires_restart - end - - def handle_rm(m, params) - key = params[:key].to_s.intern - value = params[:value] - unless @@items.has_key?(key) - m.reply "no such config key #{key}" - return - end - unless @@items[key].class <= BotConfigArrayValue - m.reply "config key #{key} is not an array" - return - end - begin - @@items[key].rm(value) - rescue ArgumentError => e - m.reply "failed to remove #{value} from #{key}: #{e.message}" - return - end - handle_get(m,{:key => key}) - m.reply "this config change will take effect on the next restart" if @@items[key].requires_restart - end - - def handle_help(m, params) - topic = params[:topic] - case topic - when false - m.reply "config module - bot configuration. usage: list, desc, get, set, unset, add, rm" - when "list" - m.reply "config list => list configuration modules, config list => list configuration keys for module " - when "get" - m.reply "config get => get configuration value for key " - when "unset" - m.reply "reset key to the default" - when "set" - m.reply "config set => set configuration value for key to " - when "desc" - m.reply "config desc => describe what key configures" - when "add" - m.reply "config add to => add value to key if is an array" - when "rm" - m.reply "config rm from => remove value from key if is an array" - else - m.reply "no help for config #{topic}" - end - end - def usage(m,params) - m.reply "incorrect usage, try '#{@@bot.nick}: help config'" - end - - # bot:: parent bot class - # create a new config hash from #{botclass}/conf.rbot - # TODO make this into a core module to guide a BotCOnfigManagerClass - # singleton instance from IRC - # - def initialize(bot) - @@bot = bot - - # respond to config messages, to provide runtime configuration - # management - # messages will be: - # get - # set - # unset - # desc - # and for arrays: - # add - # remove - @handler = MessageMapper.new(self) - @handler.map 'config', 'config list :module', :action => 'handle_list', - :defaults => {:module => false} - @handler.map 'config', 'config get :key', :action => 'handle_get' - @handler.map 'config', 'config desc :key', :action => 'handle_desc' - @handler.map 'config', 'config describe :key', :action => 'handle_desc' - @handler.map 'config', 'config set :key *value', :action => 'handle_set' - @handler.map 'config', 'config add :value to :key', :action => 'handle_add' - @handler.map 'config', 'config rm :value from :key', :action => 'handle_rm' - @handler.map 'config', 'config del :value from :key', :action => 'handle_rm' - @handler.map 'config', 'config delete :value from :key', :action => 'handle_rm' - @handler.map 'config', 'config unset :key', :action => 'handle_unset' - @handler.map 'config', 'config reset :key', :action => 'handle_unset' - @handler.map 'config', 'config help :topic', :action => 'handle_help', - :defaults => {:topic => false} - @handler.map 'config', 'help config :topic', :action => 'handle_help', - :defaults => {:topic => false} - - if(File.exist?("#{@@bot.botclass}/conf.yaml")) - begin - newconfig = YAML::load_file("#{@@bot.botclass}/conf.yaml") - newconfig.each { |key, val| - @@config[key.intern] = val - } - return - rescue - error "failed to read conf.yaml: #{$!}" - end - end - # if we got here, we need to run the first-run wizard - BotConfigWizard.new(@@bot).run - # save newly created config - save + return @config.send(method, *args, &block) end # write current configuration to #{botclass}/conf.yaml def save begin debug "Writing new conf.yaml ..." - File.open("#{@@bot.botclass}/conf.yaml.new", "w") do |file| + File.open("#{@bot.botclass}/conf.yaml.new", "w") do |file| savehash = {} - @@config.each { |key, val| + @config.each { |key, val| savehash[key.to_s] = val } file.puts savehash.to_yaml end debug "Officializing conf.yaml ..." - File.rename("#{@@bot.botclass}/conf.yaml.new", - "#{@@bot.botclass}/conf.yaml") + File.rename("#{@bot.botclass}/conf.yaml.new", + "#{@bot.botclass}/conf.yaml") rescue => e error "failed to write configuration file conf.yaml! #{$!}" error "#{e.class}: #{e}" error e.backtrace.join("\n") end end + end - def privmsg(m) - @handler.handle(m) + module BotConfig + # Returns the only BotConfigManagerClass + # + def BotConfig.configmanager + return BotConfigManagerClass.instance + end + + # Register a config value + def BotConfig.register(item) + BotConfig.configmanager.register(item) end end class BotConfigWizard def initialize(bot) @bot = bot - @questions = BotConfig.items.values.find_all {|i| i.wizard } + @manager = BotConfig::configmanager + @questions = @manager.items.values.find_all {|i| i.wizard } end - + def run() puts "First time rbot configuration wizard" puts "====================================" @@ -464,4 +316,5 @@ module Irc end end end + end diff --git a/lib/rbot/core/config.rb b/lib/rbot/core/config.rb new file mode 100644 index 00000000..47bed108 --- /dev/null +++ b/lib/rbot/core/config.rb @@ -0,0 +1,252 @@ +#-- vim:sw=2:et +#++ + + +class ConfigModule < CoreBotModule + + def handle_list(m, params) + modules = [] + if params[:module] + @bot.config.items.each_key do |key| + mod, name = key.to_s.split('.') + next unless mod == params[:module] + modules.push key unless modules.include?(name) + end + if modules.empty? + m.reply "no such module #{params[:module]}" + else + m.reply modules.join(", ") + end + else + @bot.configitems.each_key do |key| + name = key.to_s.split('.').first + modules.push name unless modules.include?(name) + end + m.reply "modules: " + modules.join(", ") + end + end + + def handle_get(m, params) + key = params[:key].to_s.intern + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + return + end + value = @bot.config.items[key].to_s + m.reply "#{key}: #{value}" + end + + def handle_desc(m, params) + key = params[:key].to_s.intern + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + end + puts @bot.config.items[key].inspect + m.reply "#{key}: #{@bot.config.items[key].desc}" + end + + def handle_unset(m, params) + key = params[:key].to_s.intern + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + end + @bot.config.items[key].unset + handle_get(m, params) + m.reply "this config change will take effect on the next restart" if @bot.config.items[key].requires_restart + m.reply "this config change will take effect on the next rescan" if @bot.config.items[key].requires_rescan + end + + def handle_set(m, params) + key = params[:key].to_s.intern + value = params[:value].join(" ") + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + return + end + begin + @bot.config.items[key].set_string(value) + rescue ArgumentError => e + m.reply "failed to set #{key}: #{e.message}" + return + end + if @bot.config.items[key].requires_restart + m.reply "this config change will take effect on the next restart" + elsif @bot.config.items[key].requires_rescan + m.reply "this config change will take effect on the next rescan" + else + m.okay + end + end + + def handle_add(m, params) + key = params[:key].to_s.intern + value = params[:value] + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + return + end + unless @bot.config.items[key].class <= BotConfigArrayValue + m.reply "config key #{key} is not an array" + return + end + begin + @bot.config.items[key].add(value) + rescue ArgumentError => e + m.reply "failed to add #{value} to #{key}: #{e.message}" + return + end + handle_get(m,{:key => key}) + m.reply "this config change will take effect on the next restart" if @bot.config.items[key].requires_restart + m.reply "this config change will take effect on the next rescan" if @bot.config.items[key].requires_rescan + end + + def handle_rm(m, params) + key = params[:key].to_s.intern + value = params[:value] + unless @bot.config.items.has_key?(key) + m.reply "no such config key #{key}" + return + end + unless @bot.config.items[key].class <= BotConfigArrayValue + m.reply "config key #{key} is not an array" + return + end + begin + @bot.config.items[key].rm(value) + rescue ArgumentError => e + m.reply "failed to remove #{value} from #{key}: #{e.message}" + return + end + handle_get(m,{:key => key}) + m.reply "this config change will take effect on the next restart" if @bot.config.items[key].requires_restart + m.reply "this config change will take effect on the next rescan" if @bot.config.items[key].requires_rescan + end + + def bot_save(m, param) + @bot.save + m.okay + end + + def bot_rescan(m, param) + m.reply "saving ..." + @bot.save + m.reply "rescanning ..." + @bot.rescan + m.reply "done. #{@plugins.status(true)}" + end + + def bot_nick(m, param) + @bot.nickchg(param[:nick]) + end + + def bot_status(m, param) + m.reply @bot.status + end + + # TODO is this one of the methods that disappeared when the bot was moved + # from the single-file to the multi-file registry? + # + # def bot_reg_stat(m, param) + # m.reply @registry.stat.inspect + # end + + def bot_version(m, param) + m.reply "I'm a v. #{$version} rubybot, (c) Tom Gilbert - http://linuxbrit.co.uk/rbot/" + end + + def handle_help(m, params) + m.reply help(params[:topic]) + end + + def help(topic="") + case topic + when false + "config module - bot configuration. usage: list, desc, get, set, unset, add, rm" + when "list" + "config list => list configuration modules, config list => list configuration keys for module " + when "get" + "config get => get configuration value for key " + when "unset" + "reset key to the default" + when "set" + "config set => set configuration value for key to " + when "desc" + "config desc => describe what key configures" + when "add" + "config add to => add value to key if is an array" + when "rm" + "config rm from => remove value from key if is an array" + else + "no help for config #{topic}" + end + end + +end + +conf = ConfigModule.new + +conf.map 'config list :module', + :action => 'handle_list', + :defaults => {:module => false}, + :auth_path => 'show' +# TODO this one is presently a security risk, since the bot +# stores the master password in the config. Do we need auth levels +# on the BotConfig keys too? +conf.map 'config get :key', + :action => 'handle_get', + :auth_path => 'show' +conf.map 'config desc :key', + :action => 'handle_desc', + :auth_path => 'show' +conf.map 'config describe :key', + :action => 'handle_desc', + :auth_path => 'show' + +conf.map "save", + :action => 'bot_save' +conf.map "rescan", + :action => 'bot_rescan' +conf.map "nick :nick", + :action => 'bot_nick' +conf.map "status", + :action => 'bot_status', + :auth_path => 'show::status' +# TODO see above +# +# conf.map "registry stats", +# :action => 'bot_reg_stat', +# :auth_path => '!config::status' +conf.map "version", + :action => 'bot_version', + :auth_path => 'show::status' + +conf.map 'config set :key *value', + :action => 'handle_set', + :auth_path => 'edit' +conf.map 'config add :value to :key', + :action => 'handle_add', + :auth_path => 'edit' +conf.map 'config rm :value from :key', + :action => 'handle_rm', + :auth_path => 'edit' +conf.map 'config del :value from :key', + :action => 'handle_rm', + :auth_path => 'edit' +conf.map 'config delete :value from :key', + :action => 'handle_rm', + :auth_path => 'edit' +conf.map 'config unset :key', + :action => 'handle_unset', + :auth_path => 'edit' +conf.map 'config reset :key', + :action => 'handle_unset', + :auth_path => 'edit' + +conf.map 'config help :topic', + :action => 'handle_help', + :defaults => {:topic => false}, + :auth_path => '!help!' + +conf.default_auth('*', false) +conf.default_auth('show::status', true) + diff --git a/lib/rbot/core/core.rb b/lib/rbot/core/core.rb index 55da1a7d..cb5df226 100644 --- a/lib/rbot/core/core.rb +++ b/lib/rbot/core/core.rb @@ -2,7 +2,7 @@ #++ -class Core < CoreBotModule +class CoreModule < CoreBotModule def listen(m) return unless m.class <= PrivMessage @@ -41,15 +41,6 @@ class Core < CoreBotModule @bot.join 0 end - def bot_save(m, param) - @bot.save - m.okay - end - - def bot_nick(m, param) - @bot.nickchg(param[:nick]) - end - def bot_say(m, param) @bot.say param[:where], param[:what].join(" ") end @@ -66,14 +57,6 @@ class Core < CoreBotModule m.reply "pong" end - def bot_rescan(m, param) - m.reply "saving ..." - @bot.save - m.reply "rescanning ..." - @bot.rescan - m.reply "done. #{@plugins.status(true)}" - end - def bot_quiet(m, param) if param.has_key?(:where) @bot.set_quiet param[:where].sub(/^here$/, m.target) @@ -90,21 +73,6 @@ class Core < CoreBotModule end end - def bot_status(m, param) - m.reply @bot.status - end - - # TODO is this one of the methods that disappeared when the bot was moved - # from the single-file to the multi-file registry? - # - # def bot_reg_stat(m, param) - # m.reply @registry.stat.inspect - # end - - def bot_version(m, param) - m.reply "I'm a v. #{$version} rubybot, (c) Tom Gilbert - http://linuxbrit.co.uk/rbot/" - end - def bot_help(m, param) m.reply @bot.help(param[:topic].join(" ")) end @@ -160,7 +128,7 @@ class Core < CoreBotModule end end -core = Core.new +core = CoreModule.new core.map "quit *msg", :action => 'bot_quit', @@ -171,27 +139,6 @@ core.map "restart *msg", :defaults => { :msg => nil }, :auth_path => 'quit' -core.map "save", - :action => 'bot_save', - :auth_path => 'config' -core.map "rescan", - :action => 'bot_rescan', - :auth_path => 'config' -core.map "nick :nick", - :action => 'bot_nick', - :auth_path => 'config' -core.map "status", - :action => 'bot_status', - :auth_path => 'config::show' - # TODO see above - # - # core.map "registry stats", - # :action => 'bot_reg_stat', - # :auth_path => 'config::show' -core.map "version", - :action => 'bot_version', - :auth_path => 'config::show' - core.map "quiet", :action => 'bot_quiet', :auth_path => 'talk::set' @@ -235,8 +182,5 @@ core.map "help *topic", :default => { :topic => [""] }, :auth_path => '!help!' -# TODO the first line should probably go to the auth module? -# core.default_auth('*', false) -core.default_auth('config::show', true) diff --git a/lib/rbot/ircbot.rb b/lib/rbot/ircbot.rb index d96e0368..9664f653 100644 --- a/lib/rbot/ircbot.rb +++ b/lib/rbot/ircbot.rb @@ -268,7 +268,16 @@ class IrcBot @pong_timer = nil @last_ping = nil @startup_time = Time.new - @config = BotConfig.new(self) + + begin + @config = BotConfig.configmanager + @config.bot_associate(self) + rescue => e + fatal e.inspect + fatal e.backtrace.join("\n") + log_session_end + exit 2 + end if @config['core.run_as_daemon'] $daemonize = true diff --git a/lib/rbot/keywords.rb b/lib/rbot/keywords.rb index d65478fe..1604faf7 100644 --- a/lib/rbot/keywords.rb +++ b/lib/rbot/keywords.rb @@ -11,10 +11,10 @@ module Irc # 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 @@ -73,7 +73,7 @@ module Irc end end - # keywords class. + # keywords class. # # Handles all that stuff like "bot: foo is bar", "bot: foo?" # @@ -87,7 +87,7 @@ module Irc 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") - + # create a new Keywords instance, associated to bot +bot+ def initialize(bot) @bot = bot @@ -96,7 +96,7 @@ module Irc @keywords = DBTree.new bot, "keyword" scan - + # import old format keywords into DBHash if(File.exist?("#{@bot.botclass}/keywords.rbot")) log "auto importing old keywords.rbot" diff --git a/lib/rbot/plugins.rb b/lib/rbot/plugins.rb index 7e98b1b3..ec99fe30 100644 --- a/lib/rbot/plugins.rb +++ b/lib/rbot/plugins.rb @@ -176,7 +176,7 @@ module Plugins # return an identifier for this plugin, defaults to a list of the message # prefixes handled (used for error messages etc) def name - self.class.to_s.downcase.sub(/^#::/,"").sub(/(plugin)?$/,"") + self.class.to_s.downcase.sub(/^#::/,"").sub(/(plugin|module)?$/,"") end # just calls name -- cgit v1.2.3