aboutsummaryrefslogtreecommitdiffstats
path: root/deckit/modules/sbbuilder.py
blob: 13552a1be85e068f440eb2a6bbe20a2bd67689fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""sb-builder build state, read-only over ssh.

`sbb-remote stats` carries the dashboard: VM load, uptime, memory, disk, package
counts, orphans, slackrepo results and overlay pulls. `jobs` adds the recent
build list. Both run over one ssh connection and arrive as two JSON objects.
"""

import json
import subprocess

from ..module import Module


def _run(cmd, timeout=8):
    try:
        return subprocess.run(cmd, capture_output=True, text=True, timeout=timeout).stdout
    except (OSError, subprocess.TimeoutExpired):
        return ""


def make(cfg: dict) -> Module:
    host = cfg.get("sbbuilder", {}).get("host", "buildsystem")
    run = cfg.get("run", _run)

    def state():
        try:
            out = run(["ssh", "-o", "BatchMode=yes", "-o", "ConnectTimeout=4",
                       host, "sbb-remote stats; sbb-remote jobs"]).lstrip()
        except OSError:
            return {"offline": True}
        try:
            decoder = json.JSONDecoder()
            stats, end = decoder.raw_decode(out)
            jobs, _ = decoder.raw_decode(out[end:].lstrip())
        except (ValueError, TypeError):
            return {"offline": True}
        return {"offline": False, "stats": stats,
                "jobs": jobs.get("jobs", []) if isinstance(jobs, dict) else []}

    return Module(name="sbbuilder", title="Build VM", state=state)