DGNum/stateless-uptime-kuma
14
0
Fork 1
Code Issues 3 Pull requests Projects Releases Packages Wiki Activity

Compare commits

base: DGNum:c1863033bccc50f1b0d7209fc26eb89672894163
Branches Tags
DGNum:master
..
compare: DGNum:a6904ba37034e36ceae3dc3efa6e376ff20070a4
Branches Tags
DGNum:master

No commits in common. "c1863033bccc50f1b0d7209fc26eb89672894163" and "a6904ba37034e36ceae3dc3efa6e376ff20070a4" have entirely different histories.
c1863033bc ... a6904ba370

6 changed files with 29 additions and 147 deletions
Show stats Expand all files Collapse all files

69
lib/default.nix
View file

@ -1,69 +0,0 @@
{ lib }:
rec {
mkFqdn = _: cfg: cfg.networking.fqdn;
fromHive =
{
builder,
nodes,
excludes ? [ ],
}:
lib.mkMerge (
builtins.map (node: builder node nodes.${node}) (
lib.subtractLists excludes (builtins.attrNames nodes)
)
);
pingProbesFromHive =
{
mkHost,
nodes,
prefix ? "Ping ",
excludes ? [ ],
tags ? [ ],
}:
fromHive {
builder = (
node: module: {
monitors = {
${prefix + node} = {
type = "ping";
inherit tags;
hostname = mkHost node module.config;
};
};
}
);
inherit nodes excludes;
};
httpProbesFromConfig =
{
config,
excludes ? [ ],
prefix ? "",
type ? "keyword",
tags ? [ ],
}:
let
filter = k: v: !builtins.elem k excludes && v.globalRedirect == null;
in
{
monitors = lib.mapAttrs' (
vhostName: vhost:
let
hasSSL = vhost.onlySSL || vhost.enableSSL || vhost.addSSL || vhost.forceSSL;
serverName = if vhost.serverName != null then vhost.serverName else vhostName;
in
{
name = prefix + serverName;
value = {
inherit type;
inherit tags;
url = "http${lib.optionalString hasSSL "s"}://${serverName}";
method = "get";
};
}
) (lib.filterAttrs filter config.services.nginx.virtualHosts);
};
}

16
nixos/module.nix
View file

@ -22,31 +22,29 @@ in
Extra arguments to use for executing `stateless-uptime-kuma`.
'';
};
lib = lib.mkOption { type = lib.types.raw; };
probesConfig = {
monitors = lib.mkOption {
type = with lib.types; attrsOf probesFormat.type;
default = { };
inherit (probesFormat) type;
default = [ ];
};
tags = lib.mkOption {
type = with lib.types; attrsOf probesFormat.type;
default = { };
inherit (probesFormat) type;
default = [ ];
};
notifications = lib.mkOption {
type = with lib.types; attrsOf probesFormat.type;
default = { };
inherit (probesFormat) type;
default = [ ];
};
};
};
config.statelessUptimeKuma = {
lib = import ../lib { inherit lib; };
build = {
json = probesFormat.generate "probes.json" cfg.probesConfig;
script = pkgs.writeShellApplication {
name = "deploy-uptime-kuma-probes";
runtimeInputs = [ pkgs.statelessUptimeKuma ];
text = ''
stateless-uptime-kuma apply-json -f ${cfg.build.json} ${builtins.concatStringsSep " " cfg.extraFlags}
stateless-uptime-kuma apply-json -f ${cfg.build.json} ${cfg.extraFlags}
'';
};
};

21
stateless_uptime_kuma/cli.py
View file

@ -35,34 +35,19 @@ def cli():
help="Scrape keywords for http probe",
default=False,
)
@click.option(
"--keywords-fallback/--no-keywords-fallback",
help="When scrapping keywords, fallback to http if no keywords found",
default=True,
)
@click.option(
"--no-autocreate-tags",
"-t",
is_flag=True,
help="Don't automatically create tags if not in tags section of input",
default=False,
)
def apply_json(file, scrape_http_keywords, no_autocreate_tags, keywords_fallback):
def apply_json(file, scrape_http_keywords):
"""
Apply json probes
"""
logger.debug(
f"Flags value:\n - scrape_http_keywords: {scrape_http_keywords}\n - no_autocreate_tags: {no_autocreate_tags}\n - keywords_fallback: {keywords_fallback}"
)
with UptimeKumaApi("http://localhost:3001") as api:
api.login("admin", "123456789a")
logging.debug("Reading json")
data = json.load(file)
logging.debug("Parsing json")
tree = from_dict(api, data, not no_autocreate_tags)
tree = from_dict(api, data)
if scrape_http_keywords:
hydrate_http_probes(tree)
logging.debug("Sync probes")
api.login("admin", "123456789a")
Manager(api, tree).process()

25
stateless_uptime_kuma/hydratation.py
View file

@ -8,10 +8,10 @@ from uptime_kuma_api import MonitorType
logger = logging.getLogger(__name__)
def hydrate_http_probes(tree, excludes=[], fallback_to_http = True):
def hydrate_http_probes(tree, excludes=[]):
for probe in tree.get("monitors", []):
if "type" not in probe.kwargs:
logger.error(f"Fatal: probe {probe.name} must have a 'type' parameter")
logger.error("Fatal: probes must have a 'type' parameter")
sys.exit(1)
if (
probe.kwargs["type"] == MonitorType.KEYWORD
@ -28,20 +28,9 @@ def hydrate_http_probes(tree, excludes=[], fallback_to_http = True):
method = probe.kwargs["method"]
headers = probe.kwargs.get("headers")
body = probe.kwargs.get("body")
req = requests.request(
method, url, headers=headers, data=body, allow_redirects=True
)
req.encoding = req.apparent_encoding
if req.status_code not in probe.get_status_codes():
logger.error(f"{probe.name} is not returning the right status code")
m = re.search("(?s)<title[^>]*>(.*?)</title>", req.text)
content = requests.request(method, url, headers=headers, data=body).text
m = re.search("<title>(.*?)</title>", content)
if m is None:
logger.warn(
f"Didn't find keywords for probe {probe.name}, falling back on classic HTTP probe"
)
logging.debug(req.text)
if fallback_to_http:
probe.kwargs["type"] = "http"
else:
probe.kwargs["keyword"] = m.group(1).strip()
logger.debug(f"New keyword: {m.group(1).strip()}")
logger.info(f"Didn't find keywords for probe {probe.name}, skipping")
probe.kwargs["keyword"] = m.group(1)
logger.debug(f"New keyword: {m.group(1)}")

26
stateless_uptime_kuma/tree_gen.py
View file

@ -17,33 +17,31 @@ def die_tag_format_error():
sys.exit(1)
def from_dict(api, tree, autocreate_tags=True):
def from_dict(api, tree):
notif = tree.get("notifications", [])
indexed_notifications = {
name: Notification(api, name, **kwargs) for name, kwargs in notif.items()
}
indexed_notifications = {n["name"]: Notification(api, **n) for n in notif}
tags = tree.get("tags", [])
indexed_tags = {name: Tag(api, name, **kwargs) for name, kwargs in tags.items()}
indexed_tags = {t["name"]: Tag(api, **t) for t in tags}
monitors = tree.get("monitors", [])
indexed_monitors = {}
for monitor_name, monitor_kwargs in monitors.items():
for m in monitors:
associated_tags = []
for tag in monitor_kwargs.get("tags", []):
for tag in m.get("tags", []):
if not isinstance(tag, dict) or "name" not in tag:
die_tag_format_error()
try:
if autocreate_tags and tag["name"] not in indexed_tags:
indexed_tags[tag["name"]] = Tag(api, name=tag["name"])
associated_tags.append((indexed_tags[tag["name"]], tag.get("value")))
except IndexError:
die_tag_format_error()
monitor_kwargs["tags"] = associated_tags
m["tags"] = associated_tags
associated_notifications = [
indexed_notifications[notif]
for notif in monitor_kwargs.get("notifications", [])
indexed_notifications[notif] for notif in m.get("notifications", [])
]
monitor_kwargs["notifications"] = associated_notifications
indexed_monitors[monitor_name] = Monitor(api, monitor_name, **monitor_kwargs)
m["notifications"] = associated_notifications
if "name" not in m:
logger.error("Fatal: All monitors must have a name")
sys.exit(1)
indexed_monitors[m["name"]] = Monitor(api, **m)
return {
"monitors": indexed_monitors.values(),
"tags": indexed_tags.values(),

19
stateless_uptime_kuma/uptime_kuma.py
View file

@ -3,7 +3,6 @@ Classes to make the needed operations to reach the specified state.
"""
import logging
import sys
logger = logging.getLogger(__name__)
@ -86,24 +85,6 @@ class Monitor(Item):
self.notifications = notifications
self.saved = False
def get_status_codes(self):
if "accepted_statuscodes" not in self.kwargs:
return set(range(200, 300))
accepted_statuscodes = set()
for codes in self.kwargs["accepted_statuscodes"]:
if "-" in codes:
c = codes.split("-")
if len(c) != 2:
logger.error(
f"Fatal: status codes for {self} must be in the format 'number' or 'number-number'"
)
sys.exit(1)
a, b = int(c[0]), int(c[1])
accepted_statuscodes.update(range(a, b))
else:
accepted_statuscodes.add(int(codes))
return accepted_statuscodes
def save(self):
if self.saved:
return