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
|
require 'set'
class Stack
def initialize
@hash = {}
end
def [](key)
@hash[key] = [] unless @hash[key]
@hash[key]
end
def contains?(key)
@hash.has_key?(key)
end
def clear(key)
@hash.delete(key)
end
end
class GentooFixupPlugin < Plugin
def initialize
super
@stack = Stack.new
@action = nil
@old_m = nil
end
# Name is overloaded
def whois(m)
nick = m.whois[:nick].downcase
# need to see if the whois reply was invoked by this plugin
return unless @stack.contains?(nick)
#msg = "Real bot host:" + m.parse_channel_list.inspect
channels = Set.new
channels.merge(m.whois[:channels].map{|c|c[0].downcase}) if m.whois.include?(:channels)
#msg = "Real bot channels:" + m.whois.channels
if false
elsif @action == 'ACTUAL'
reply_actual_channels(@old_m, channels)
elsif @action == 'AWOL'
reply_awol_channels(@old_m, channels)
elsif @action == 'FIX'
apply_fixup(@old_m, channels)
end
@action = nil
@old_m = nil
@stack.clear(nick)
end
def want_channels
return Set.new(@bot.config['irc.join_channels'].compact.map{|c|c.downcase})
end
def awol_channels(actual_channels)
return want_channels - actual_channels
end
def apply_fixup(m, actual_channels)
missing_channels = awol_channels(actual_channels).to_a.sort
s = missing_channels.join(', ')
#@bot.say 'robbat2|na', 'Fixing '+s
#@bot.say 'robbat2', 'Fixing '+s
m.reply 'Fixing '+s
missing_channels.each do |chan|
@bot.part(chan, "Fixing bot channels")
@bot.join(chan)
end
end
def do_lookup(m, params)
nick = @bot.config['irc.nick']
nick.downcase!
@stack[nick] << m.replyto
@bot.whois(nick)
end
def fixup(m, params)
@action = 'FIX'
@old_m = m
do_lookup(m, params)
end
def say_want_channels(m, params)
c = want_channels.to_a.sort
m.reply "Target channels (#{c.size}): "+c.join(', ')
end
def say_actual_channels(m, params)
@action = 'ACTUAL'
@old_m = m
do_lookup(m, params)
end
def reply_actual_channels(m, channels)
m.reply "Actual channels (#{channels.size}): " + channels.to_a.sort.join(', ')
end
def say_awol_channels(m, params)
@action = 'AWOL'
@old_m = m
do_lookup(m, params)
end
def reply_awol_channels(m, channels)
missing_channels = awol_channels(channels).to_a.sort
m.reply "AWOL channels (#{missing_channels.size}): " + missing_channels.to_a.sort.join(', ')
end
end
plugin = GentooFixupPlugin.new
plugin.map 'fixupjoin',
:action => 'fixup',
:auth_path => 'move'
plugin.map 'actualchannels',
:action => 'say_actual_channels',
:auth_path => 'move'
plugin.map 'wantchannels',
:action => 'say_want_channels',
:auth_path => 'move'
plugin.map 'awolchannels',
:action => 'say_awol_channels',
:auth_path => 'move'
# vim: et sts=2 ts=2 sw=2:
|
GitWeb