lib/rbot/core/config.rb

   1 #-- vim:sw=2:et
   2 #++
   3 #
   4 # :title: rbot config management from IRC
   5 #
   6 # Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
   7
   8 class ConfigModule < CoreBotModule
   9
  10   def version_string
  11     if $version_timestamp.to_i > 0
  12       ago = String.new ' ['
  13       ago << Utils.secs_to_string(Time.now.to_i - $version_timestamp.to_i)
  14       ago << ' ago]'
  15     else
  16       ago = ''
  17     end
  18     _("I'm a v. %{version}%{ago} rubybot%{copyright}%{url}") % {
  19       :version => $version,
  20       :ago => ago,
  21       :copyright => ", #{Irc::Bot::COPYRIGHT_NOTICE}",
  22       :url => " - #{Irc::Bot::SOURCE_URL}"
  23     }
  24   end
  25
  26   def save
  27     @bot.config.save
  28   end
  29
  30   def handle_list(m, params)
  31     modules = []
  32     if params[:module]
  33       @bot.config.items.each_key do |key|
  34         mod, name = key.to_s.split('.')
  35         next unless mod == params[:module]
  36         modules.push key unless modules.include?(name)
  37       end
  38       if modules.empty?
  39         m.reply _("no such module %{module}") % {:module => params[:module]}
  40       else
  41         m.reply modules.join(", ")
  42       end
  43     else
  44       @bot.config.items.each_key do |key|
  45         name = key.to_s.split('.').first
  46         modules.push name unless modules.include?(name)
  47       end
  48       m.reply "modules: " + modules.join(", ")
  49     end
  50   end
  51
  52   def handle_get(m, params)
  53     key = params[:key].to_s.intern
  54     unless @bot.config.items.has_key?(key)
  55       m.reply _("no such config key %{key}") % {:key => key}
  56       return
  57     end
  58     return if !@bot.auth.allow?(@bot.config.items[key].auth_path, m.source, m.replyto)
  59     value = @bot.config.items[key].to_s
  60     m.reply "#{key}: #{value}"
  61   end
  62
  63   def handle_desc(m, params)
  64     key = params[:key].to_s.intern
  65     unless @bot.config.items.has_key?(key)
  66       m.reply _("no such config key %{key}") % {:key => key}
  67     end
  68     m.reply "#{key}: #{@bot.config.items[key].desc}"
  69   end
  70
  71   def handle_search(m, params)
  72     rx = Regexp.new(params[:rx].to_s, true)
  73     cfs = []
  74     @bot.config.items.each do |k, v|
  75       cfs << v if k.to_s.match(rx) or (v.desc.match(rx) rescue false)
  76     end
  77     if cfs.empty?
  78       m.reply _("no config key found matching %{r}") % { :r => params[:rx].to_s}
  79     else
  80       m.reply _("possible keys: %{kl}") % { :kl => cfs.map { |c| c.key}.join(', ') }
  81       m.reply cfs.map { |c| [c.key, c.desc].join(': ') }.join("\n")
  82     end
  83   end
  84
  85   def handle_unset(m, params)
  86     key = params[:key].to_s.intern
  87     unless @bot.config.items.has_key?(key)
  88       m.reply _("no such config key %{key}") % {:key => key}
  89     end
  90     return if !@bot.auth.allow?(@bot.config.items[key].auth_path, m.source, m.replyto)
  91     @bot.config.items[key].unset
  92     handle_get(m, params)
  93     m.reply _("this config change will take effect on the next restart") if @bot.config.items[key].requires_restart
  94     m.reply _("this config change will take effect on the next rescan") if @bot.config.items[key].requires_rescan
  95   end
  96
  97   def handle_set(m, params)
  98     key = params[:key].to_s.intern
  99     value = params[:value].join(" ")
 100     unless @bot.config.items.has_key?(key)
 101       m.reply _("no such config key %{key}") % {:key => key} unless params[:silent]
 102       return false
 103     end
 104     return false if !@bot.auth.allow?(@bot.config.items[key].auth_path, m.source, m.replyto)
 105     begin
 106       @bot.config.items[key].set_string(value)
 107     rescue ArgumentError => e
 108       m.reply _("failed to set %{key}: %{error}") % {:key => key, :error => e.message} unless params[:silent]
 109       return false
 110     end
 111     if @bot.config.items[key].requires_restart
 112       m.reply _("this config change will take effect on the next restart") unless params[:silent]
 113       return :restart
 114     elsif @bot.config.items[key].requires_rescan
 115       m.reply _("this config change will take effect on the next rescan") unless params[:silent]
 116       return :rescan
 117     else
 118       m.okay unless params[:silent]
 119       return true
 120     end
 121   end
 122
 123   def handle_add(m, params)
 124     key = params[:key].to_s.intern
 125     value = params[:value]
 126     unless @bot.config.items.has_key?(key)
 127       m.reply _("no such config key %{key}") % {:key => key}
 128       return
 129     end
 130     unless @bot.config.items[key].kind_of?(Config::ArrayValue)
 131       m.reply _("config key %{key} is not an array") % {:key => key}
 132       return
 133     end
 134     return if !@bot.auth.allow?(@bot.config.items[key].auth_path, m.source, m.replyto)
 135     begin
 136       @bot.config.items[key].add(value)
 137     rescue ArgumentError => e
 138       m.reply _("failed to add %{value} to %{key}: %{error}") % {:value => value, :key => key, :error => e.message}
 139       return
 140     end
 141     handle_get(m,{:key => key})
 142     m.reply _("this config change will take effect on the next restart") if @bot.config.items[key].requires_restart
 143     m.reply _("this config change will take effect on the next rescan") if @bot.config.items[key].requires_rescan
 144   end
 145
 146   def handle_rm(m, params)
 147     key = params[:key].to_s.intern
 148     value = params[:value]
 149     unless @bot.config.items.has_key?(key)
 150       m.reply _("no such config key %{key}") % {:key => key}
 151       return
 152     end
 153     unless @bot.config.items[key].kind_of?(Config::ArrayValue)
 154       m.reply _("config key %{key} is not an array") % {:key => key}
 155       return
 156     end
 157     return if !@bot.auth.allow?(@bot.config.items[key].auth_path, m.source, m.replyto)
 158     begin
 159       @bot.config.items[key].rm(value)
 160     rescue ArgumentError => e
 161       m.reply _("failed to remove %{value} from %{key}: %{error}") % {:value => value, :key => key, :error => e.message}
 162       return
 163     end
 164     handle_get(m,{:key => key})
 165     m.reply _("this config change will take effect on the next restart") if @bot.config.items[key].requires_restart
 166     m.reply _("this config change will take effect on the next rescan") if @bot.config.items[key].requires_rescan
 167   end
 168
 169   def bot_save(m, param)
 170     @bot.save
 171     m.okay
 172   end
 173
 174   def bot_rescan(m, param)
 175     m.reply _("saving ...")
 176     @bot.save
 177     m.reply _("rescanning ...")
 178     @bot.rescan
 179     m.reply _("done. %{plugin_status}") % {:plugin_status => @bot.plugins.status(true)}
 180   end
 181
 182   def bot_nick(m, param)
 183     @bot.nickchg(param[:nick])
 184   end
 185
 186   def bot_status(m, param)
 187     m.reply @bot.status
 188   end
 189
 190   # TODO is this one of the methods that disappeared when the bot was moved
 191   # from the single-file to the multi-file registry?
 192   #
 193   #  def bot_reg_stat(m, param)
 194   #    m.reply @registry.stat.inspect
 195   #  end
 196
 197   def bot_version(m, param)
 198     m.reply version_string
 199   end
 200
 201   def ctcp_listen(m)
 202     who = m.private? ? "me" : m.target
 203     case m.ctcp.intern
 204     when :VERSION
 205       m.ctcp_reply version_string
 206     when :SOURCE
 207       m.ctcp_reply Irc::Bot::SOURCE_URL
 208     end
 209   end
 210
 211   def handle_help(m, params)
 212     m.reply help(params[:topic])
 213   end
 214
 215   def help(plugin, topic="")
 216     case plugin
 217     when "config"
 218       case topic
 219       when "list"
 220       _("config list => list configuration modules, config list <module> => list configuration keys for module <module>")
 221       when "get"
 222       _("config get <key> => get configuration value for key <key>")
 223       when "unset"
 224       _("reset key <key> to the default")
 225       when "set"
 226       _("config set <key> <value> => set configuration value for key <key> to <value>")
 227       when "desc"
 228       _("config desc <key> => describe what key <key> configures")
 229       when "add"
 230       _("config add <value> to <key> => add value <value> to key <key> if <key> is an array")
 231       when "rm"
 232       _("config rm <value> from <key> => remove value <value> from key <key> if <key> is an array")
 233       else
 234       _("config module - bot configuration. usage: list, desc, get, set, unset, add, rm")
 235       # else
 236       #   "no help for config #{topic}"
 237       end
 238     when "nick"
 239       _("nick <newnick> => change the bot nick to <newnick>, if possible")
 240     when "status"
 241       _("status => display some information on the bot's status")
 242     when "save"
 243       _("save => save current dynamic data and configuration")
 244     when "rescan"
 245       _("rescan => reload modules and static facts")
 246     when "version"
 247       _("version => describes software version")
 248     else
 249       _("config-related tasks: config, save, rescan, version, nick, status")
 250     end
 251   end
 252
 253 end
 254
 255 conf = ConfigModule.new
 256
 257 conf.map 'config list :module',
 258   :action => 'handle_list',
 259   :defaults => {:module => false},
 260   :auth_path => 'show'
 261 # TODO this one is presently a security risk, since the bot
 262 # stores the master password in the config. Do we need auth levels
 263 # on the Bot::Config keys too?
 264 conf.map 'config get :key',
 265   :action => 'handle_get',
 266   :auth_path => 'show'
 267 conf.map 'config desc :key',
 268   :action => 'handle_desc',
 269   :auth_path => 'show'
 270 conf.map 'config describe :key',
 271   :action => 'handle_desc',
 272   :auth_path => 'show::desc!'
 273 conf.map 'config search *rx',
 274   :action => 'handle_search',
 275   :auth_path => 'show'
 276
 277 conf.map "save",
 278   :action => 'bot_save'
 279 conf.map "rescan",
 280   :action => 'bot_rescan'
 281 conf.map "nick :nick",
 282   :action => 'bot_nick'
 283 conf.map "status",
 284   :action => 'bot_status',
 285   :auth_path => 'show::status'
 286 # TODO see above
 287 #
 288 # conf.map "registry stats",
 289 #   :action => 'bot_reg_stat',
 290 #   :auth_path => '!config::status'
 291 conf.map "version",
 292   :action => 'bot_version',
 293   :auth_path => 'show::status'
 294
 295 conf.map 'config set :key *value',
 296   :action => 'handle_set',
 297   :auth_path => 'edit'
 298 conf.map 'config add :value to :key',
 299   :action => 'handle_add',
 300   :auth_path => 'edit'
 301 conf.map 'config rm :value from :key',
 302   :action => 'handle_rm',
 303   :auth_path => 'edit'
 304 conf.map 'config del :value from :key',
 305   :action => 'handle_rm',
 306   :auth_path => 'edit'
 307 conf.map 'config delete :value from :key',
 308   :action => 'handle_rm',
 309   :auth_path => 'edit'
 310 conf.map 'config unset :key',
 311   :action => 'handle_unset',
 312   :auth_path => 'edit'
 313 conf.map 'config reset :key',
 314   :action => 'handle_unset',
 315   :auth_path => 'edit'
 316
 317 conf.map 'config help :topic',
 318   :action => 'handle_help',
 319   :defaults => {:topic => false},
 320   :auth_path => '!help!'
 321
 322 conf.default_auth('*', false)
 323 conf.default_auth('show', true)
 324 conf.default_auth('show::get', false)
 325 # TODO these shouldn't be set here, we need a way to let the default
 326 # permission be specified together with the ConfigValue
 327 conf.default_auth('key', true)
 328 conf.default_auth('key::auth::password', false)
 329