70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
"""authentik admin tasks"""
|
|
import re
|
|
from os import environ
|
|
|
|
from django.core.cache import cache
|
|
from django.core.validators import URLValidator
|
|
from packaging.version import parse
|
|
from prometheus_client import Info
|
|
from requests import RequestException, get
|
|
from structlog.stdlib import get_logger
|
|
|
|
from authentik import ENV_GIT_HASH_KEY, __version__
|
|
from authentik.events.models import Event, EventAction
|
|
from authentik.events.monitored_tasks import MonitoredTask, TaskResult, TaskResultStatus
|
|
from authentik.root.celery import CELERY_APP
|
|
|
|
LOGGER = get_logger()
|
|
VERSION_CACHE_KEY = "authentik_latest_version"
|
|
VERSION_CACHE_TIMEOUT = 8 * 60 * 60 # 8 hours
|
|
# Chop of the first ^ because we want to search the entire string
|
|
URL_FINDER = URLValidator.regex.pattern[1:]
|
|
PROM_INFO = Info("authentik_version", "Currently running authentik version")
|
|
|
|
|
|
def _set_prom_info():
|
|
"""Set prometheus info for version"""
|
|
PROM_INFO.info(
|
|
{
|
|
"version": __version__,
|
|
"latest": cache.get(VERSION_CACHE_KEY, ""),
|
|
"build_hash": environ.get(ENV_GIT_HASH_KEY, ""),
|
|
}
|
|
)
|
|
|
|
|
|
@CELERY_APP.task(bind=True, base=MonitoredTask)
|
|
def update_latest_version(self: MonitoredTask):
|
|
"""Update latest version info"""
|
|
try:
|
|
response = get("https://api.github.com/repos/goauthentik/authentik/releases/latest")
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
tag_name = data.get("tag_name")
|
|
upstream_version = tag_name.split("/")[1]
|
|
cache.set(VERSION_CACHE_KEY, upstream_version, VERSION_CACHE_TIMEOUT)
|
|
self.set_status(
|
|
TaskResult(TaskResultStatus.SUCCESSFUL, ["Successfully updated latest Version"])
|
|
)
|
|
_set_prom_info()
|
|
# Check if upstream version is newer than what we're running,
|
|
# and if no event exists yet, create one.
|
|
local_version = parse(__version__)
|
|
if local_version < parse(upstream_version):
|
|
# Event has already been created, don't create duplicate
|
|
if Event.objects.filter(
|
|
action=EventAction.UPDATE_AVAILABLE,
|
|
context__new_version=upstream_version,
|
|
).exists():
|
|
return
|
|
event_dict = {"new_version": upstream_version}
|
|
if match := re.search(URL_FINDER, data.get("body", "")):
|
|
event_dict["message"] = f"Changelog: {match.group()}"
|
|
Event.new(EventAction.UPDATE_AVAILABLE, **event_dict).save()
|
|
except (RequestException, IndexError) as exc:
|
|
cache.set(VERSION_CACHE_KEY, "0.0.0", VERSION_CACHE_TIMEOUT)
|
|
self.set_status(TaskResult(TaskResultStatus.ERROR).with_error(exc))
|
|
|
|
|
|
_set_prom_info()
|