You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

414 lines
13 KiB

from django.conf import settings
from django.template.loader import render_to_string
from django.utils import timezone
import json
import requests
from six.moves.urllib.parse import quote
from hc.accounts.models import Profile
from hc.lib import emails
def tmpl(template_name, **ctx):
template_path = "integrations/%s" % template_name
# \xa0 is non-breaking space. It causes SMS messages to use UCS2 encoding
# and cost twice the money.
return render_to_string(template_path, ctx).strip().replace(u"\xa0", " ")
class Transport(object):
def __init__(self, channel):
self.channel = channel
def notify(self, check):
""" Send notification about current status of the check.
This method returns None on success, and error message
on error.
"""
raise NotImplementedError()
def is_noop(self, check):
""" Return True if transport will ignore check's current status.
This method is overriden in Webhook subclass where the user can
configure webhook urls for "up" and "down" events, and both are
optional.
"""
return False
def checks(self):
return self.channel.user.check_set.order_by("created")
class Email(Transport):
def notify(self, check, bounce_url):
if not self.channel.email_verified:
return "Email not verified"
headers = {"X-Bounce-Url": bounce_url}
try:
# Look up the sorting preference for this email address
p = Profile.objects.get(user__email=self.channel.value)
sort = p.sort
except Profile.DoesNotExist:
# Default sort order is by check's creation time
sort = "created"
ctx = {
"check": check,
"checks": self.checks(),
"sort": sort,
"now": timezone.now(),
"unsub_link": self.channel.get_unsub_link()
}
emails.alert(self.channel.value, ctx, headers)
def is_noop(self, check):
return not self.channel.email_verified
class HttpTransport(Transport):
@classmethod
def _request(cls, method, url, **kwargs):
try:
options = dict(kwargs)
options["timeout"] = 5
if "headers" not in options:
options["headers"] = {}
if "User-Agent" not in options["headers"]:
options["headers"]["User-Agent"] = "healthchecks.io"
r = requests.request(method, url, **options)
if r.status_code not in (200, 201, 204):
return "Received status code %d" % r.status_code
except requests.exceptions.Timeout:
# Well, we tried
return "Connection timed out"
except requests.exceptions.ConnectionError:
return "Connection failed"
@classmethod
def get(cls, url, **kwargs):
# Make 3 attempts--
for x in range(0, 3):
error = cls._request("get", url, **kwargs)
if error is None:
break
return error
@classmethod
def post(cls, url, **kwargs):
# Make 3 attempts--
for x in range(0, 3):
error = cls._request("post", url, **kwargs)
if error is None:
break
return error
@classmethod
def put(cls, url, **kwargs):
# Make 3 attempts--
for x in range(0, 3):
error = cls._request("put", url, **kwargs)
if error is None:
break
return error
class Webhook(HttpTransport):
def prepare(self, template, check, urlencode=False):
""" Replace variables with actual values.
There should be no bad translations if users use $ symbol in
check's name or tags, because $ gets urlencoded to %24
"""
def safe(s):
return quote(s) if urlencode else s
result = template
if "$CODE" in result:
result = result.replace("$CODE", str(check.code))
if "$STATUS" in result:
result = result.replace("$STATUS", check.status)
if "$NOW" in result:
s = timezone.now().replace(microsecond=0).isoformat()
result = result.replace("$NOW", safe(s))
if "$NAME" in result:
result = result.replace("$NAME", safe(check.name))
if "$TAG" in result:
for i, tag in enumerate(check.tags_list()):
placeholder = "$TAG%d" % (i + 1)
result = result.replace(placeholder, safe(tag))
return result
def is_noop(self, check):
if check.status == "down" and not self.channel.url_down:
return True
if check.status == "up" and not self.channel.url_up:
return True
return False
def notify(self, check):
url = self.channel.url_down
if check.status == "up":
url = self.channel.url_up
assert url
url = self.prepare(url, check, urlencode=True)
headers = {}
for key, value in self.channel.headers.items():
headers[key] = self.prepare(value, check)
if self.channel.post_data:
payload = self.prepare(self.channel.post_data, check)
return self.post(url, data=payload.encode("utf-8"), headers=headers)
else:
return self.get(url, headers=headers)
class Slack(HttpTransport):
def notify(self, check):
text = tmpl("slack_message.json", check=check)
payload = json.loads(text)
return self.post(self.channel.slack_webhook_url, json=payload)
class HipChat(HttpTransport):
def notify(self, check):
text = tmpl("hipchat_message.json", check=check)
payload = json.loads(text)
self.channel.refresh_hipchat_access_token()
return self.post(self.channel.hipchat_webhook_url, json=payload)
class OpsGenie(HttpTransport):
def notify(self, check):
payload = {
"apiKey": self.channel.value,
"alias": str(check.code),
"source": "healthchecks.io"
}
if check.status == "down":
payload["tags"] = ",".join(check.tags_list())
payload["message"] = tmpl("opsgenie_message.html", check=check)
payload["note"] = tmpl("opsgenie_note.html", check=check)
url = "https://api.opsgenie.com/v1/json/alert"
if check.status == "up":
url += "/close"
return self.post(url, json=payload)
class PagerDuty(HttpTransport):
URL = "https://events.pagerduty.com/generic/2010-04-15/create_event.json"
def notify(self, check):
description = tmpl("pd_description.html", check=check)
payload = {
"vendor": settings.PD_VENDOR_KEY,
"service_key": self.channel.pd_service_key,
"incident_key": str(check.code),
"event_type": "trigger" if check.status == "down" else "resolve",
"description": description,
"client": settings.SITE_NAME,
"client_url": settings.SITE_ROOT
}
return self.post(self.URL, json=payload)
class PagerTree(HttpTransport):
def notify(self, check):
url = self.channel.value
headers = {
"Conent-Type": "application/json"
}
payload = {
"incident_key": str(check.code),
"event_type": "trigger" if check.status == "down" else "resolve",
"title": tmpl("pagertree_title.html", check=check),
"description": tmpl("pagertree_description.html", check=check),
"client": settings.SITE_NAME,
"client_url": settings.SITE_ROOT,
"tags": ",".join(check.tags_list())
}
return self.post(url, json=payload, headers=headers)
class Pushbullet(HttpTransport):
def notify(self, check):
text = tmpl("pushbullet_message.html", check=check)
url = "https://api.pushbullet.com/v2/pushes"
headers = {
"Access-Token": self.channel.value,
"Conent-Type": "application/json"
}
payload = {
"type": "note",
"title": "healthchecks.io",
"body": text
}
return self.post(url, json=payload, headers=headers)
class Pushover(HttpTransport):
URL = "https://api.pushover.net/1/messages.json"
def notify(self, check):
others = self.checks().filter(status="down").exclude(code=check.code)
ctx = {
"check": check,
"down_checks": others,
}
text = tmpl("pushover_message.html", **ctx)
title = tmpl("pushover_title.html", **ctx)
user_key, prio = self.channel.value.split("|")
payload = {
"token": settings.PUSHOVER_API_TOKEN,
"user": user_key,
"message": text,
"title": title,
"html": 1,
"priority": int(prio),
}
# Emergency notification
if prio == "2":
payload["retry"] = settings.PUSHOVER_EMERGENCY_RETRY_DELAY
payload["expire"] = settings.PUSHOVER_EMERGENCY_EXPIRATION
return self.post(self.URL, data=payload)
class VictorOps(HttpTransport):
def notify(self, check):
description = tmpl("victorops_description.html", check=check)
mtype = "CRITICAL" if check.status == "down" else "RECOVERY"
payload = {
"entity_id": str(check.code),
"message_type": mtype,
"entity_display_name": check.name_then_code(),
"state_message": description,
"monitoring_tool": "healthchecks.io",
}
return self.post(self.channel.value, json=payload)
class Discord(HttpTransport):
def notify(self, check):
text = tmpl("slack_message.json", check=check)
payload = json.loads(text)
url = self.channel.discord_webhook_url + "/slack"
return self.post(url, json=payload)
class Telegram(HttpTransport):
SM = "https://api.telegram.org/bot%s/sendMessage" % settings.TELEGRAM_TOKEN
@classmethod
def send(cls, chat_id, text):
return cls.post(cls.SM, json={
"chat_id": chat_id,
"text": text,
"parse_mode": "html"
})
def notify(self, check):
text = tmpl("telegram_message.html", check=check)
return self.send(self.channel.telegram_id, text)
class Sms(HttpTransport):
URL = 'https://api.twilio.com/2010-04-01/Accounts/%s/Messages.json'
def is_noop(self, check):
return check.status != "down"
def notify(self, check):
profile = Profile.objects.for_user(self.channel.user)
if not profile.authorize_sms():
return "Monthly SMS limit exceeded"
url = self.URL % settings.TWILIO_ACCOUNT
auth = (settings.TWILIO_ACCOUNT, settings.TWILIO_AUTH)
text = tmpl("sms_message.html", check=check,
site_name=settings.SITE_NAME)
data = {
'From': settings.TWILIO_FROM,
'To': self.channel.value,
'Body': text,
}
return self.post(url, data=data, auth=auth)
class Zendesk(HttpTransport):
TMPL = "https://%s.zendesk.com/api/v2/requests.json"
def get_payload(self, check):
return {
"request": {
"subject": tmpl("zendesk_title.html", check=check),
"type": "incident",
"comment": {
"body": tmpl("zendesk_description.html", check=check)
}
}
}
def notify_down(self, check):
headers = {"Authorization": "Bearer %s" % self.channel.zendesk_token}
url = self.TMPL % self.channel.zendesk_subdomain
return self.post(url, headers=headers, json=self.get_payload(check))
def notify_up(self, check):
# Get the list of requests made by us, in newest-to-oldest order
url = self.TMPL % self.channel.zendesk_subdomain
url += "?sort_by=created_at&sort_order=desc"
headers = {"Authorization": "Bearer %s" % self.channel.zendesk_token}
r = requests.get(url, headers=headers, timeout=10)
if r.status_code != 200:
return "Received status code %d" % r.status_code
# Update the first request that has check.code in its description
doc = r.json()
if "requests" in doc:
for obj in doc["requests"]:
if str(check.code) in obj["description"]:
payload = self.get_payload(check)
return self.put(obj["url"], headers=headers, json=payload)
return "Could not find a ticket to update"
def notify(self, check):
if check.status == "down":
return self.notify_down(check)
if check.status == "up":
return self.notify_up(check)