summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/markov.rb
blob: 7408a43ecabc285f519f3798d36a8e10076c2738 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
class MarkovPlugin < Plugin
  def initialize
    super
    @registry.set_default([])
    @registry['enabled'] = false unless @registry.has_key?('enabled')
    @lastline = false
  end

  def generate_string(word1, word2)
    # limit to max of 50 words
    output = word1 + " " + word2

    # try to avoid :nonword in the first iteration
    wordlist = @registry["#{word1} #{word2}"]
    wordlist.delete(:nonword)
    if not wordlist.empty?
      word3 = wordlist[rand(wordlist.length)]
      output = output + " " + word3
      word1, word2 = word2, word3
    end

    49.times do
      wordlist = @registry["#{word1} #{word2}"]
      break if wordlist.empty?
      word3 = wordlist[rand(wordlist.length)]
      break if word3 == :nonword
      output = output + " " + word3
      word1, word2 = word2, word3
    end
    return output
  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 <foo> <bar>' => riff on a word pair (if possible)"
  end

  def clean_str(s)
    str = s.dup
    str.gsub!(/^\S+[:,;]/, "")
    str.gsub!(/\s{2,}/, ' ') # fix for two or more spaces
    return str.strip
  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
  end

  def status(m,params)
    enabled = @registry['enabled']
    if (enabled)
      m.reply "markov is currently enabled, #{probability?}% chance of chipping in"
    else
      m.reply "markov is currently disabled"
    end
  end

  def ignore?(user=nil)
    @registry['ignore_users'].each do |mask|
      return true if Irc.netmaskmatch mask, user
    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']
        s.delete user
        @registry['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)
          m.reply "#{user} already in list"
        else
          @registry['ignore_users'] = @registry['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(", ")}"
    else
      m.reply "have markov ignore the input from a hostmask.  usage: markov ignore add <mask>; markov ignore remove <mask>; markov ignore list"
    end
  end

  def enable(m, params)
    @registry['enabled'] = true
    m.okay
  end

  def probability(m, params)
    @registry['probability'] = params[:probability].to_i
    m.okay
  end

  def disable(m, params)
    @registry['enabled'] = false
    m.okay
  end

  def should_talk
    return false unless @registry['enabled']
    prob = probability?
    return true if prob > rand(100)
    return false
  end

  def delay
    1 + rand(5)
  end

  def random_markov(m, message)
    return unless should_talk

    word1, word2 = message.split(/\s+/)
    line = generate_string(word1, word2)
    return unless line
    return if line == message
    @bot.timer.add_once(delay, m) {|m|
      m.reply line
    }
  end

  def chat(m, params)
    line = generate_string(params[:seed1], params[:seed2])
    if line != "#{params[:seed1]} #{params[:seed2]}"
      m.reply line 
    else
      m.reply "I can't :("
    end
  end

  def rand_chat(m, params)
    # pick a random pair from the db and go from there
    word1, word2 = :nonword, :nonword
    output = Array.new
    50.times do
      wordlist = @registry["#{word1} #{word2}"]
      break if wordlist.empty?
      word3 = wordlist[rand(wordlist.length)]
      break if word3 == :nonword
      output << word3
      word1, word2 = word2, word3
    end
    if output.length > 1
      m.reply output.join(" ")
    else
      m.reply "I can't :("
    end
  end
  
  def listen(m)
    return unless m.kind_of?(PrivMessage) && m.public?
    return if m.address?
    return if ignore? m.source

    # in channel message, the kind we are interested in
    message = clean_str m.message

    if m.action?
      message = "#{m.sourcenick} #{message}"
    end
    
    wordlist = message.split(/\s+/)
    return unless wordlist.length >= 2
    @lastline = message
    word1, word2 = :nonword, :nonword
    wordlist.each do |word3|
      @registry["#{word1} #{word2}"] = @registry["#{word1} #{word2}"].push(word3)
      word1, word2 = word2, word3
    end
    @registry["#{word1} #{word2}"] = @registry["#{word1} #{word2}"].push(:nonword)

    return if m.replied?
    random_markov(m, message)
  end
end
plugin = MarkovPlugin.new
plugin.map 'markov ignore :action :option', :action => "ignore"
plugin.map 'markov ignore :action', :action => "ignore"
plugin.map 'markov ignore', :action => "ignore"
plugin.map 'markov enable', :action => "enable"
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",
           :requirements => {:probability => /^\d+%?$/}