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.
292 lines
8.6 KiB
292 lines
8.6 KiB
from datetime import timedelta as td
|
|
import json
|
|
import re
|
|
from urllib.parse import quote, urlencode
|
|
|
|
from django import forms
|
|
from django.forms import URLField
|
|
from django.conf import settings
|
|
from django.core.exceptions import ValidationError
|
|
from hc.front.validators import (
|
|
CronExpressionValidator,
|
|
TimezoneValidator,
|
|
WebhookValidator,
|
|
)
|
|
import requests
|
|
|
|
|
|
class HeadersField(forms.Field):
|
|
message = """Use "Header-Name: value" pairs, one per line."""
|
|
|
|
def to_python(self, value):
|
|
if not value:
|
|
return {}
|
|
|
|
headers = {}
|
|
for line in value.split("\n"):
|
|
if not line.strip():
|
|
continue
|
|
|
|
if ":" not in line:
|
|
raise ValidationError(self.message)
|
|
|
|
n, v = line.split(":", maxsplit=1)
|
|
n, v = n.strip(), v.strip()
|
|
if not n or not v:
|
|
raise ValidationError(message=self.message)
|
|
|
|
headers[n] = v
|
|
|
|
return headers
|
|
|
|
def validate(self, value):
|
|
super().validate(value)
|
|
for k, v in value.items():
|
|
if len(k) > 1000 or len(v) > 1000:
|
|
raise ValidationError("Value too long")
|
|
|
|
|
|
class NameTagsForm(forms.Form):
|
|
name = forms.CharField(max_length=100, required=False)
|
|
tags = forms.CharField(max_length=500, required=False)
|
|
desc = forms.CharField(required=False)
|
|
|
|
def clean_tags(self):
|
|
result = []
|
|
|
|
for part in self.cleaned_data["tags"].split(" "):
|
|
part = part.strip()
|
|
if part != "":
|
|
result.append(part)
|
|
|
|
return " ".join(result)
|
|
|
|
|
|
class FilteringRulesForm(forms.Form):
|
|
filter_by_subject = forms.ChoiceField(choices=(("no", "no"), ("yes", "yes")))
|
|
subject = forms.CharField(required=False, max_length=200)
|
|
subject_fail = forms.CharField(required=False, max_length=200)
|
|
methods = forms.ChoiceField(required=False, choices=(("", "Any"), ("POST", "POST")))
|
|
manual_resume = forms.BooleanField(required=False)
|
|
|
|
def clean_subject(self):
|
|
if self.cleaned_data["filter_by_subject"] == "yes":
|
|
return self.cleaned_data["subject"]
|
|
|
|
return ""
|
|
|
|
def clean_subject_fail(self):
|
|
if self.cleaned_data["filter_by_subject"] == "yes":
|
|
return self.cleaned_data["subject_fail"]
|
|
|
|
return ""
|
|
|
|
|
|
class TimeoutForm(forms.Form):
|
|
timeout = forms.IntegerField(min_value=60, max_value=2592000)
|
|
grace = forms.IntegerField(min_value=60, max_value=2592000)
|
|
|
|
def clean_timeout(self):
|
|
return td(seconds=self.cleaned_data["timeout"])
|
|
|
|
def clean_grace(self):
|
|
return td(seconds=self.cleaned_data["grace"])
|
|
|
|
|
|
class CronForm(forms.Form):
|
|
schedule = forms.CharField(max_length=100, validators=[CronExpressionValidator()])
|
|
tz = forms.CharField(max_length=36, validators=[TimezoneValidator()])
|
|
grace = forms.IntegerField(min_value=1, max_value=43200)
|
|
|
|
|
|
class AddOpsgenieForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
region = forms.ChoiceField(initial="us", choices=(("us", "US"), ("eu", "EU")))
|
|
key = forms.CharField(max_length=40)
|
|
|
|
|
|
PRIO_CHOICES = [
|
|
("-2", "Lowest Priority"),
|
|
("-1", "Low Priority"),
|
|
("0", "Normal Priority"),
|
|
("1", "High Priority"),
|
|
("2", "Emergency Priority"),
|
|
]
|
|
|
|
|
|
class AddPushoverForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
pushover_user_key = forms.CharField()
|
|
prio = forms.ChoiceField(initial="0", choices=PRIO_CHOICES)
|
|
prio_up = forms.ChoiceField(initial="0", choices=PRIO_CHOICES)
|
|
|
|
def get_value(self):
|
|
key = self.cleaned_data["pushover_user_key"]
|
|
prio = self.cleaned_data["prio"]
|
|
prio_up = self.cleaned_data["prio_up"]
|
|
return "%s|%s|%s" % (key, prio, prio_up)
|
|
|
|
|
|
class AddEmailForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
value = forms.EmailField(max_length=100)
|
|
down = forms.BooleanField(required=False, initial=True)
|
|
up = forms.BooleanField(required=False, initial=True)
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
|
|
down = self.cleaned_data.get("down")
|
|
up = self.cleaned_data.get("up")
|
|
|
|
if not down and not up:
|
|
self.add_error("down", "Please select at least one.")
|
|
|
|
|
|
class AddUrlForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
value = forms.URLField(max_length=1000, validators=[WebhookValidator()])
|
|
|
|
|
|
METHODS = ("GET", "POST", "PUT")
|
|
|
|
|
|
class WebhookForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
name = forms.CharField(max_length=100, required=False)
|
|
|
|
method_down = forms.ChoiceField(initial="GET", choices=zip(METHODS, METHODS))
|
|
body_down = forms.CharField(max_length=1000, required=False)
|
|
headers_down = HeadersField(required=False)
|
|
url_down = URLField(
|
|
max_length=1000, required=False, validators=[WebhookValidator()]
|
|
)
|
|
|
|
method_up = forms.ChoiceField(initial="GET", choices=zip(METHODS, METHODS))
|
|
body_up = forms.CharField(max_length=1000, required=False)
|
|
headers_up = HeadersField(required=False)
|
|
url_up = forms.URLField(
|
|
max_length=1000, required=False, validators=[WebhookValidator()]
|
|
)
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
|
|
url_down = self.cleaned_data.get("url_down")
|
|
url_up = self.cleaned_data.get("url_up")
|
|
|
|
if not url_down and not url_up:
|
|
if not self.has_error("url_down"):
|
|
self.add_error("url_down", "Enter a valid URL.")
|
|
|
|
def get_value(self):
|
|
return json.dumps(dict(self.cleaned_data), sort_keys=True)
|
|
|
|
|
|
class AddShellForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
|
|
cmd_down = forms.CharField(max_length=1000, required=False)
|
|
cmd_up = forms.CharField(max_length=1000, required=False)
|
|
|
|
def get_value(self):
|
|
return json.dumps(dict(self.cleaned_data), sort_keys=True)
|
|
|
|
|
|
class PhoneNumberForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
label = forms.CharField(max_length=100, required=False)
|
|
phone = forms.CharField()
|
|
|
|
def clean_phone(self):
|
|
v = self.cleaned_data["phone"]
|
|
|
|
stripped = v.encode("ascii", "ignore").decode("ascii")
|
|
stripped = stripped.replace(" ", "").replace("-", "")
|
|
if not re.match(r"^\+\d{5,15}$", stripped):
|
|
raise forms.ValidationError("Invalid phone number format.")
|
|
|
|
return stripped
|
|
|
|
def get_json(self):
|
|
return json.dumps({"value": self.cleaned_data["phone"]})
|
|
|
|
|
|
class PhoneUpDownForm(PhoneNumberForm):
|
|
up = forms.BooleanField(required=False, initial=True)
|
|
down = forms.BooleanField(required=False, initial=True)
|
|
|
|
def get_json(self):
|
|
return json.dumps(
|
|
{
|
|
"value": self.cleaned_data["phone"],
|
|
"up": self.cleaned_data["up"],
|
|
"down": self.cleaned_data["down"],
|
|
}
|
|
)
|
|
|
|
|
|
class ChannelNameForm(forms.Form):
|
|
name = forms.CharField(max_length=100, required=False)
|
|
|
|
|
|
class AddMatrixForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
alias = forms.CharField(max_length=100)
|
|
|
|
def clean_alias(self):
|
|
v = self.cleaned_data["alias"]
|
|
|
|
# validate it by trying to join
|
|
url = settings.MATRIX_HOMESERVER
|
|
url += "/_matrix/client/r0/join/%s?" % quote(v)
|
|
url += urlencode({"access_token": settings.MATRIX_ACCESS_TOKEN})
|
|
r = requests.post(url, {})
|
|
if r.status_code == 429:
|
|
raise forms.ValidationError(
|
|
"Matrix server returned status code 429 (Too Many Requests), "
|
|
"please try again later."
|
|
)
|
|
|
|
doc = r.json()
|
|
if "error" in doc:
|
|
raise forms.ValidationError("Response from Matrix: %s" % doc["error"])
|
|
|
|
self.cleaned_data["room_id"] = doc["room_id"]
|
|
|
|
return v
|
|
|
|
|
|
class AddAppriseForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
url = forms.CharField(max_length=512)
|
|
|
|
|
|
class AddPdForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
value = forms.CharField(max_length=32)
|
|
|
|
|
|
ZULIP_TARGETS = (("stream", "Stream"), ("private", "Private"))
|
|
|
|
|
|
class AddZulipForm(forms.Form):
|
|
error_css_class = "has-error"
|
|
bot_email = forms.EmailField(max_length=100)
|
|
api_key = forms.CharField(max_length=50)
|
|
site = forms.URLField(max_length=100, validators=[WebhookValidator()])
|
|
mtype = forms.ChoiceField(choices=ZULIP_TARGETS)
|
|
to = forms.CharField(max_length=100)
|
|
|
|
def get_value(self):
|
|
return json.dumps(dict(self.cleaned_data), sort_keys=True)
|
|
|
|
|
|
class AddTrelloForm(forms.Form):
|
|
token = forms.RegexField(regex=r"^[0-9a-fA-F]{64}$")
|
|
board_name = forms.CharField(max_length=100)
|
|
list_name = forms.CharField(max_length=100)
|
|
list_id = forms.RegexField(regex=r"^[0-9a-fA-F]{16,32}$")
|
|
|
|
def get_value(self):
|
|
return json.dumps(dict(self.cleaned_data), sort_keys=True)
|