X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Fmarkov.rb;h=bd7a673efbf1cdf0076e23cb446e016af3ab8e03;hb=24bb60775741d3731400f1e430ef6bf3a2e1b933;hp=7408a43ecabc285f519f3798d36a8e10076c2738;hpb=683c2328496a60529310a10cb46582d800d71888;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/markov.rb b/data/rbot/plugins/markov.rb index 7408a43e..bd7a673e 100644 --- a/data/rbot/plugins/markov.rb +++ b/data/rbot/plugins/markov.rb @@ -1,8 +1,37 @@ +#-- vim:sw=2:et +#++ +# +# :title: Markov plugin +# +# Author:: Tom Gilbert +# Copyright:: (C) 2005 Tom Gilbert +# +# Contribute to chat with random phrases built from word sequences learned +# by listening to chat + class MarkovPlugin < Plugin + Config.register Config::BooleanValue.new('markov.enabled', + :default => false, + :desc => "Enable and disable the plugin") + Config.register Config::IntegerValue.new('markov.probability', + :default => 25, + :validate => Proc.new { |v| (0..100).include? v }, + :desc => "Percentage chance of markov plugin chipping in") + Config.register Config::ArrayValue.new('markov.ignore_users', + :default => [], + :desc => "Hostmasks of users to be ignored") + def initialize super @registry.set_default([]) - @registry['enabled'] = false unless @registry.has_key?('enabled') + if @registry.has_key?('enabled') + @bot.config['markov.enabled'] = @registry['enabled'] + @registry.delete('enabled') + end + if @registry.has_key?('probability') + @bot.config['markov.probability'] = @registry['probability'] + @registry.delete('probability') + end @lastline = false end @@ -31,7 +60,7 @@ class MarkovPlugin < Plugin end def help(plugin, topic="") - "markov plugin: listens to chat to build a markov chain, with which it can (perhaps) attempt to (inanely) contribute to 'discussion'. Sort of.. Will get a *lot* better after listening to a lot of chat. usage: 'markov' to attempt to say something relevant to the last line of chat, if it can. other options to markov: 'ignore' => ignore a hostmask (accept no input), 'status' => show current status, 'probability' => set the % chance of rbot responding to input, 'chat' => try and say something intelligent, 'chat about ' => riff on a word pair (if possible)" + "markov plugin: listens to chat to build a markov chain, with which it can (perhaps) attempt to (inanely) contribute to 'discussion'. Sort of.. Will get a *lot* better after listening to a lot of chat. usage: 'markov' to attempt to say something relevant to the last line of chat, if it can. other options to markov: 'ignore' => ignore a hostmask (accept no input), 'status' => show current status, 'probability []' => set the % chance of rbot responding to input, or display the current probability, 'chat' => try and say something intelligent, 'chat about ' => riff on a word pair (if possible)" end def clean_str(s) @@ -42,16 +71,11 @@ class MarkovPlugin < Plugin end def probability? - prob = @registry['probability'] - prob = 25 if prob.kind_of? Array; - prob = 0 if prob < 0 - prob = 100 if prob > 100 - return prob + return @bot.config['markov.probability'] end def status(m,params) - enabled = @registry['enabled'] - if (enabled) + if @bot.config['markov.enabled'] m.reply "markov is currently enabled, #{probability?}% chance of chipping in" else m.reply "markov is currently disabled" @@ -59,63 +83,65 @@ class MarkovPlugin < Plugin end def ignore?(user=nil) - @registry['ignore_users'].each do |mask| - return true if Irc.netmaskmatch mask, user + return false unless user + @bot.config['markov.ignore_users'].each do |mask| + return true if user.matches?(mask) end return false end def ignore(m, params) - if @registry['ignore_users'].nil? - @registry['ignore_users'] = [] - end action = params[:action] user = params[:option] case action when 'remove': - if @registry['ignore_users'].include? user - s = @registry['ignore_users'] + if @bot.config['markov.ignore_users'].include? user + s = @bot.config['markov.ignore_users'] s.delete user - @registry['ignore_users'] = s + @bot.config['ignore_users'] = s m.reply "#{user} removed" else m.reply "not found in list" end when 'add': if user - if @registry['ignore_users'].include?(user) + if @bot.config['markov.ignore_users'].include?(user) m.reply "#{user} already in list" else - @registry['ignore_users'] = @registry['ignore_users'].push user + @bot.config['markov.ignore_users'] = @bot.config['markov.ignore_users'].push user m.reply "#{user} added to markov ignore list" end else m.reply "give the name of a person to ignore" end when 'list': - m.reply "I'm ignoring #{@registry['ignore_users'].join(", ")}" + m.reply "I'm ignoring #{@bot.config['markov.ignore_users'].join(", ")}" else m.reply "have markov ignore the input from a hostmask. usage: markov ignore add ; markov ignore remove ; markov ignore list" end end def enable(m, params) - @registry['enabled'] = true + @bot.config['markov.enabled'] = true m.okay end def probability(m, params) - @registry['probability'] = params[:probability].to_i - m.okay + if params[:probability] + @bot.config['probability'] = params[:probability].to_i + m.okay + else + m.reply _("markov has a %{prob}% chance of chipping in") % { :prob => probability? } + end end def disable(m, params) - @registry['enabled'] = false + @bot.config['markov.enabled'] = false m.okay end def should_talk - return false unless @registry['enabled'] + return false unless @bot.config['markov.enabled'] prob = probability? return true if prob > rand(100) return false @@ -132,7 +158,7 @@ class MarkovPlugin < Plugin line = generate_string(word1, word2) return unless line return if line == message - @bot.timer.add_once(delay, m) {|m| + @bot.timer.add_once(delay) { m.reply line } end @@ -191,6 +217,7 @@ class MarkovPlugin < Plugin random_markov(m, message) end end + plugin = MarkovPlugin.new plugin.map 'markov ignore :action :option', :action => "ignore" plugin.map 'markov ignore :action', :action => "ignore" @@ -200,5 +227,5 @@ plugin.map 'markov disable', :action => "disable" plugin.map 'markov status', :action => "status" plugin.map 'chat about :seed1 :seed2', :action => "chat" plugin.map 'chat', :action => "rand_chat" -plugin.map 'markov probability :probability', :action => "probability", +plugin.map 'markov probability [:probability]', :action => "probability", :requirements => {:probability => /^\d+%?$/}