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
|
# encoding: UTF-8
require 'timeout'
require 'log4r'
module Mauve
class Person < Struct.new(:username, :password, :holiday_url, :urgent, :normal, :low, :email, :xmpp, :sms)
attr_reader :notification_thresholds, :last_pop3_login
def initialize(*args)
@notification_thresholds = nil
@suppressed = false
#
# TODO fix up web login so pop3 can be used as a proxy.
#
@last_pop3_login = {:from => nil, :at => nil}
super(*args)
end
def logger ; @logger ||= Log4r::Logger.new self.class.to_s ; end
def suppressed?
@suppressed
end
def notification_thresholds
#
# By default send 10 thresholds in a minute maximum
#
@notification_thresholds ||= { }
end
# This class implements an instance_eval context to execute the blocks
# for running a notification block for each person.
#
class NotificationCaller
def initialize(person, alert, other_alerts, base_conditions={})
@person = person
@alert = alert
@other_alerts = other_alerts
@base_conditions = base_conditions
end
def logger ; @logger ||= Log4r::Logger.new self.class.to_s ; end
#
# This method makes sure things like
#
# xmpp
#
# works
#
def method_missing(name, *args)
#
# Work out the destination
#
if args.first.is_a?(String)
destination = args.pop
else
destination = @person.__send__(name)
end
if args.first.is_a?(Array)
conditions = @base_conditions.merge(args[0])
else
conditions = @base_conditions
end
notification_method = Configuration.current.notification_methods[name.to_s]
raise NoMethodError.new("#{name} not defined as a notification method") unless notification_method
# Methods are expected to return true or false so the user can chain
# them together with || as fallbacks. So we have to catch exceptions
# and turn them into false.
#
res = notification_method.send_alert(destination, @alert, @other_alerts, conditions)
#
# Log the result
note = "#{@alert.update_type.capitalize} #{name} notification to #{@person.username} (#{destination}) " + (res ? "succeeded" : "failed" )
logger.info note+" about #{@alert}."
h = History.new(:alerts => [@alert], :type => "notification", :event => note)
logger.error "Unable to save history due to #{h.errors.inspect}" if !h.save
return res
end
end
#
#
# Sends the alert
#
def send_alert(level, alert)
now = Time.now
was_suppressed = self.suppressed?
@suppressed = self.notification_thresholds.any? do |period, previous_alert_times|
#
# Choose the second one as the first.
#
first = previous_alert_times[1]
last = previous_alert_times[-1]
first.is_a?(Time) and (
(now - first) < period or
(was_suppressed and (now - last) < period)
)
end
logger.info "Starting to send notifications again for #{username}." if was_suppressed and not self.suppressed?
#
# We only suppress notifications if we were suppressed before we started,
# and are still suppressed.
#
if (was_suppressed and self.suppressed?) or self.is_on_holiday?
note = "#{alert.update_type.capitalize} notification to #{self.username} suppressed"
logger.info note + " about #{alert}."
History.create(:alerts => [alert], :type => "notification", :event => note)
return true
end
# FIXME current_alerts is very slow. So much so it slows everything
# down. A lot.
result = NotificationCaller.new(
self,
alert,
[],
# current_alerts,
{:is_suppressed => @suppressed,
:was_suppressed => was_suppressed, }
).instance_eval(&__send__(level))
if result
#
# Remember that we've sent an alert
#
self.notification_thresholds.each do |period, previous_alert_times|
#
# Hmm.. not sure how to make this thread-safe.
#
self.notification_thresholds[period].push Time.now
self.notification_thresholds[period].shift
end
return true
end
return false
end
#
# Returns the subset of current alerts that are relevant to this Person.
#
# This is currently very CPU intensive, and slows things down a lot. So
# I've commented it out when sending notifications.
#
def current_alerts
Alert.all_raised.select do |alert|
my_last_update = AlertChanged.first(:person => username, :alert_id => alert.id)
my_last_update && my_last_update.update_type != "cleared"
end
end
# Whether the person is on holiday or not.
#
# @return [Boolean] True if person on holiday, false otherwise.
def is_on_holiday? ()
return false if holiday_url.nil? or holiday_url.empty?
return CalendarInterface.is_user_on_holiday?(holiday_url, username)
end
end
end
|
