spb/maclustr-dispatch

1"""CLI `mld` — MacLustr Dispatch (tourne sur la passerelle M1M32)."""2import argparse3import json4import sys5from . import config, deploy, manifest, nodes, picker, registry678def cmd_discover(a):9    nodes.discover()101112def cmd_scan(a):13    nodes.scan(aliases=a.nodes or None)141516def cmd_nodes(a):17    res = nodes.load_scan() or nodes.scan(verbose=False)18    nodes.print_scan(res)192021def cmd_apps(a):22    ms = manifest.load_all()23    r = registry.load()["apps"]24    print("%-18s %-8s %-6s %-30s %-8s %s" % ("app", "nœud", "port", "domaine", "épingle", "runtimes"))25    for name, m in ms.items():26        cur = r.get(name, {}).get("node") or "—"27        print("%-18s %-8s %-6s %-30s %-8s %s" % (name, cur, m.get("port") or "", m.get("domain") or "", m["placement"].get("pin") or "", ",".join(m["requires"]["runtimes"])))282930def cmd_show(a):31    print(json.dumps(manifest.load(a.app), indent=2, ensure_ascii=False))323334def cmd_pick(a):35    m = manifest.load(a.app) if a.app else {"app": "(ad hoc)", "requires": {"runtimes": a.runtime or ["pm2"], "ram_gb": a.ram or 1, "ports": []}, "ram_mb_observed": int((a.ram or 1) * 1024), "size_mb": 0, "placement": {"pin": None, "prefer": None, "avoid": [], "reason": ""}}36    scan = nodes.load_scan() if not a.rescan else nodes.scan(verbose=False)37    rows = picker.rank(m, scan, exclude=a.exclude or ())38    if a.json:39        print(json.dumps(rows, indent=2))40        return41    print("classement pour %s :" % m["app"])42    for r in rows:43        if r["ok"]:44            print("  %-8s score %.3f" % (r["node"], r["score"]))45        elif scan.get(r["node"], {}).get("online") and scan[r["node"]].get("role") == "worker":46            print("  %-8s —  %s" % (r["node"], r["why"]))474849def cmd_plan(a):50    ms = manifest.load_all()51    if a.apps:52        ms = {k: v for k, v in ms.items() if k in a.apps}53    scan = nodes.scan(verbose=False) if a.rescan else (nodes.load_scan() or nodes.scan(verbose=False))54    rows = picker.plan(ms, scan, exclude=a.exclude or ())55    if a.json:56        print(json.dumps(rows, indent=2, ensure_ascii=False))57        return58    print("%-19s %-8s %-8s %-6s %7s  %s" % ("app", "actuel", "proposé", "score", "taille", "note"))59    counts = {}60    moving_mb = 061    for r in rows:62        counts[r["node"]] = counts.get(r["node"], 0) + 163        mv = r["node"] and r["current"] and r["node"] != r["current"]64        if mv:65            moving_mb += r["size_mb"]66        note = ("épinglé — " + r["reason"]) if r["pin"] else (("alt: " + ", ".join("%s %.2f" % x for x in r["alternatives"])) if r["alternatives"] else "")67        if not r["node"]:68            note = "AUCUN NŒUD : " + " ; ".join("%s (%s)" % b for b in r["blockers"])69        flag = "→" if mv else ("=" if r["node"] else "!")70        print("%-19s %-8s %-8s %-6s %6dM %s %s" % (r["app"], r["current"] or "—", r["node"] or "—", ("%.3f" % r["score"]) if r["score"] is not None else "", r["size_mb"], flag, note))71    print("\nrépartition : " + ", ".join("%s×%s" % (k, v) for k, v in sorted(counts.items(), key=lambda x: -x[1]) if k))72    print("données à déplacer : %.1f Go (→ = migration, = = redéploiement en place)" % (moving_mb / 1024.0))737475def cmd_fetch(a):76    deploy.fetch(a.app, a.from_node, final=a.final)777879def cmd_deploy(a):80    if a.node:81        node = a.node82    else:83        scan = nodes.load_scan() or nodes.scan(verbose=False)84        rows = picker.rank(manifest.load(a.app), scan)85        if not rows or not rows[0]["ok"]:86            raise SystemExit("aucun nœud éligible : " + "; ".join("%s: %s" % (r["node"], r["why"]) for r in rows[:6]))87        node = rows[0]["node"]88        print("nœud choisi : %s (score %.3f)" % (node, rows[0]["score"]))89    try:90        deploy.deploy(a.app, node, reinstall=a.reinstall, skip_sync=a.skip_sync)91    except deploy.DeployError as e:92        raise SystemExit("ÉCHEC : %s" % e)939495def cmd_move(a):96    to = a.to97    if not to:98        scan = nodes.load_scan() or nodes.scan(verbose=False)99        rows = picker.rank(manifest.load(a.app), scan, exclude=[registry.node_of(a.app)] if a.away else ())100        if not rows or not rows[0]["ok"]:101            raise SystemExit("aucun nœud éligible : " + "; ".join("%s: %s" % (r["node"], r["why"]) for r in rows[:6]))102        to = rows[0]["node"]103        print("nœud choisi : %s (score %.3f)" % (to, rows[0]["score"]))104    try:105        deploy.move(a.app, to, from_alias=a.from_node, keep_source=a.keep_source, reinstall=a.reinstall)106    except deploy.DeployError as e:107        raise SystemExit("ÉCHEC : %s" % e)108109110def cmd_stop(a):111    deploy.stop(a.app, a.node or registry.node_of(a.app))112    registry.set_app(a.app, status="stopped")113114115def cmd_start(a):116    node = a.node or registry.node_of(a.app)117    deploy.start(a.app, node)118    ok, det = deploy.health(a.app, node)119    registry.set_app(a.app, node=node, status="online" if ok else "unhealthy", health=det)120    print("santé :", det)121122123def cmd_restart(a):124    cmd_stop(a)125    cmd_start(a)126127128def cmd_retire(a):129    deploy.retire(a.app, a.node or registry.node_of(a.app), remove_dir=not a.keep_dir)130    if not a.keep_registry:131        registry.remove_app(a.app, note=a.note or "")132133134def cmd_status(a):135    if a.live:136        for row in deploy.live_status(a.app):137            print("%-18s %-8s %-30s %-3s %s" % row)138    else:139        print("%-18s %-8s %-6s %-30s %-10s %s" % ("app", "nœud", "port", "domaine", "état", "déployée"))140        for row in deploy.status(a.app):141            print("%-18s %-8s %-6s %-30s %-10s %s" % tuple("" if x is None else x for x in row))142143144def cmd_health(a):145    ok, det = deploy.health(a.app, a.node or registry.node_of(a.app))146    print("OK" if ok else "KO", det)147    sys.exit(0 if ok else 1)148149150def cmd_registry(a):151    if a.push:152        res = registry.push_to_subscribers()153        bad = [x for x in res if not x[1]]154        sys.exit(1 if bad else 0)155    print(json.dumps(registry.load(), indent=2, ensure_ascii=False))156157158def cmd_subscribers(a):159    subs = registry.load_subscribers()160    if a.add:161        kind, target, path = a.add162        if kind not in ("app", "node"):163            raise SystemExit("usage : mld subscribers --add app|node <nom> <chemin>")164        subs = [s for s in subs if not (s.get(kind) == target and s.get("path") == path)]165        subs.append({kind: target, "path": path})166        registry.save_subscribers(subs)167    if a.remove:168        kind, target = a.remove169        subs = [s for s in subs if s.get(kind) != target]170        registry.save_subscribers(subs)171    r = registry.load()172    print("%-22s %-8s %-15s %s" % ("abonné", "nœud", "ip", "chemin"))173    for s in subs:174        try:175            label, node, ip, path = registry.resolve_subscriber(s, r)176            print("%-22s %-8s %-15s %s" % (label, node, ip, path))177        except ValueError as e:178            print("%-22s %-8s %-15s %s" % (s.get("app") or s.get("node"), "—", "—", "(%s)" % e))179180181def cmd_import(a):182    from . import importer183    for p in importer.import_node(a.node, apps_filter=a.apps or None):184        print("brouillon :", p)185186187def cmd_prepare(a):188    from . import prepare189    prepare.prepare(a.node, runtimes=a.runtime or None, ka_helpers=a.ka_helpers)190191192def cmd_bootstrap(a):193    from . import prepare194    prepare.bootstrap(a.node)195196197def cmd_logs(a):198    node = a.node or registry.node_of(a.app)199    from . import ssh200    ip = nodes.ip_of(node)201    names = manifest.pm2_names(manifest.load(a.app))202    rc, out, err = ssh.run(ip, "pm2 logs %s --nostream --lines %d 2>/dev/null" % (" ".join(names), a.lines), timeout=60)203    print(out)204205206def main(argv=None):207    p = argparse.ArgumentParser(prog="mld", description="MacLustr Dispatch — déploie les apps depuis la passerelle %s vers le nœud le plus efficient." % config.GATEWAY)208    sp = p.add_subparsers(dest="cmd", required=True)209    s = sp.add_parser("discover", help="balaye le LAN, met en cache les IP des nœuds"); s.set_defaults(f=cmd_discover)210    s = sp.add_parser("scan", help="sonde ressources + runtimes de chaque nœud"); s.add_argument("nodes", nargs="*"); s.set_defaults(f=cmd_scan)211    s = sp.add_parser("nodes", help="dernier scan"); s.set_defaults(f=cmd_nodes)212    s = sp.add_parser("apps", help="manifestes + emplacement courant"); s.set_defaults(f=cmd_apps)213    s = sp.add_parser("show", help="affiche un manifeste"); s.add_argument("app"); s.set_defaults(f=cmd_show)214    s = sp.add_parser("pick", help="classe les nœuds pour une app (ou un besoin ad hoc)"); s.add_argument("app", nargs="?"); s.add_argument("--ram", type=float); s.add_argument("--runtime", action="append"); s.add_argument("--exclude", action="append"); s.add_argument("--rescan", action="store_true"); s.add_argument("--json", action="store_true"); s.set_defaults(f=cmd_pick)215    s = sp.add_parser("plan", help="plan de placement pour toutes les apps (ou une liste)"); s.add_argument("apps", nargs="*"); s.add_argument("--exclude", action="append"); s.add_argument("--rescan", action="store_true"); s.add_argument("--json", action="store_true"); s.set_defaults(f=cmd_plan)216    s = sp.add_parser("fetch", help="copie l'app d'un nœud vers le staging"); s.add_argument("app"); s.add_argument("--from", dest="from_node", required=True); s.add_argument("--final", action="store_true"); s.set_defaults(f=cmd_fetch)217    s = sp.add_parser("deploy", help="staging -> nœud (choisi automatiquement sans --node)"); s.add_argument("app"); s.add_argument("--node"); s.add_argument("--reinstall", action="store_true"); s.add_argument("--skip-sync", action="store_true"); s.set_defaults(f=cmd_deploy)218    s = sp.add_parser("move", help="migre une app (fetch, stop, deploy, retire)"); s.add_argument("app"); s.add_argument("--to"); s.add_argument("--from", dest="from_node"); s.add_argument("--away", action="store_true", help="exclut le nœud actuel"); s.add_argument("--keep-source", action="store_true"); s.add_argument("--reinstall", action="store_true"); s.set_defaults(f=cmd_move)219    s = sp.add_parser("stop"); s.add_argument("app"); s.add_argument("--node"); s.set_defaults(f=cmd_stop)220    s = sp.add_parser("start"); s.add_argument("app"); s.add_argument("--node"); s.set_defaults(f=cmd_start)221    s = sp.add_parser("restart"); s.add_argument("app"); s.add_argument("--node"); s.set_defaults(f=cmd_restart)222    s = sp.add_parser("retire", help="arrête + efface la copie d'un nœud"); s.add_argument("app"); s.add_argument("--node"); s.add_argument("--keep-dir", action="store_true"); s.add_argument("--keep-registry", action="store_true"); s.add_argument("--note"); s.set_defaults(f=cmd_retire)223    s = sp.add_parser("status"); s.add_argument("app", nargs="?"); s.add_argument("--live", action="store_true"); s.set_defaults(f=cmd_status)224    s = sp.add_parser("health"); s.add_argument("app"); s.add_argument("--node"); s.set_defaults(f=cmd_health)225    s = sp.add_parser("logs"); s.add_argument("app"); s.add_argument("--node"); s.add_argument("--lines", type=int, default=40); s.set_defaults(f=cmd_logs)226    s = sp.add_parser("registry", help="registre JSON complet (--push : rediffuse aux abonnés)"); s.add_argument("--push", action="store_true"); s.set_defaults(f=cmd_registry)227    s = sp.add_parser("subscribers", help="abonnés du registre (copie poussée à chaque sauvegarde)"); s.add_argument("--add", nargs=3, metavar=("app|node", "NOM", "CHEMIN")); s.add_argument("--remove", nargs=2, metavar=("app|node", "NOM")); s.set_defaults(f=cmd_subscribers)228    s = sp.add_parser("import", help="brouillons de manifestes depuis les PM2 d'un nœud"); s.add_argument("node"); s.add_argument("apps", nargs="*"); s.set_defaults(f=cmd_import)229    s = sp.add_parser("prepare", help="installe runtimes/ngrok/pm2-startup sur un nœud"); s.add_argument("node"); s.add_argument("--runtime", action="append"); s.add_argument("--ka-helpers", action="store_true"); s.set_defaults(f=cmd_prepare)230    s = sp.add_parser("bootstrap", help="Xcode CLT + Homebrew sur un nœud vierge (long)"); s.add_argument("node"); s.set_defaults(f=cmd_bootstrap)231    a = p.parse_args(argv)232    a.f(a)233234235if __name__ == "__main__":236    main()237