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.
62 lines
1.9 KiB
62 lines
1.9 KiB
from django.conf import settings
|
|
from django.core.signing import base64_hmac
|
|
from django.template.loader import render_to_string
|
|
from django.urls import reverse
|
|
|
|
WIDTHS = {"a": 7, "b": 7, "c": 6, "d": 7, "e": 6, "f": 4, "g": 7, "h": 7,
|
|
"i": 3, "j": 3, "k": 7, "l": 3, "m": 10, "n": 7, "o": 7, "p": 7,
|
|
"q": 7, "r": 4, "s": 6, "t": 5, "u": 7, "v": 7, "w": 9, "x": 6,
|
|
"y": 7, "z": 7, "0": 7, "1": 6, "2": 7, "3": 7, "4": 7, "5": 7,
|
|
"6": 7, "7": 7, "8": 7, "9": 7, "A": 8, "B": 7, "C": 8, "D": 8,
|
|
"E": 7, "F": 6, "G": 9, "H": 8, "I": 3, "J": 4, "K": 7, "L": 6,
|
|
"M": 10, "N": 8, "O": 9, "P": 6, "Q": 9, "R": 7, "S": 7, "T": 7,
|
|
"U": 8, "V": 8, "W": 11, "X": 7, "Y": 7, "Z": 7, "-": 4, "_": 6}
|
|
|
|
COLORS = {
|
|
"up": "#4c1",
|
|
"late": "#fe7d37",
|
|
"down": "#e05d44"
|
|
}
|
|
|
|
|
|
def get_width(s):
|
|
total = 0
|
|
for c in s:
|
|
total += WIDTHS.get(c, 7)
|
|
return total
|
|
|
|
|
|
def get_badge_svg(tag, status):
|
|
w1 = get_width(tag) + 10
|
|
w2 = get_width(status) + 10
|
|
ctx = {
|
|
"width": w1 + w2,
|
|
"tag_width": w1,
|
|
"status_width": w2,
|
|
"tag_center_x": w1 / 2,
|
|
"status_center_x": w1 + w2 / 2,
|
|
"tag": tag,
|
|
"status": status,
|
|
"color": COLORS[status]
|
|
}
|
|
|
|
return render_to_string("badge.svg", ctx)
|
|
|
|
|
|
def check_signature(username, tag, sig):
|
|
ours = base64_hmac(str(username), tag, settings.SECRET_KEY)
|
|
ours = ours[:8]
|
|
return ours == sig
|
|
|
|
|
|
def get_badge_url(username, tag, format="svg"):
|
|
sig = base64_hmac(str(username), tag, settings.SECRET_KEY)
|
|
|
|
if tag == "*":
|
|
view = "hc-badge-json-all" if format == "json" else "hc-badge-all"
|
|
url = reverse(view, args=[username, sig[:8]])
|
|
else:
|
|
view = "hc-badge-json" if format == "json" else "hc-badge"
|
|
url = reverse(view, args=[username, sig[:8], tag])
|
|
|
|
return settings.SITE_ROOT + url
|