sergio/takana

1#!/usr/bin/env python3
2# drenar.py — el ORDEN DE MASTICADO: qué construir, en qué orden, para acercar una imagen a estar
3# completa. (P4 del plan docs/plan-catalogo-objetivo.md.)
4#
5# POR QUÉ. La cola de la granja era AUTORÍA: 55 ficheros en `tandas/` escritos a mano, y para la
6# deuda una lista que el hub medía con `saldar-deuda-static.sh` y pegaba en `campana-deuda.sh`. Pero
7# el grafo de estado ya sabe qué falta (`state`), qué lo traba (`blocked_by`) y a cuántos destraba
8# (`unblocks`). Este script convierte eso en la cola: DERIVADA, no escrita.
9#
10# ONDAS. No lista "lo que falta" en un montón: lo parte en ondas topológicas. Onda 1 = construible
11# YA (ninguna dep en deuda). Onda 2 = lo que la onda 1 destraba. Etc. Eso da tres cosas que una
12# lista plana no da: el orden correcto, la PROFUNDIDAD real de la cadena (cuántos pasos secuenciales
13# faltan, que es lo que manda en el reloj de pared), y qué se puede paralelizar sin pisarse.
14# Dentro de cada onda ordena por `unblocks` desc: primero lo que libera más.
15#
16# LA REGLA DE LA FUENTE ÚNICA. Un nodo `wanted` (sin receta) no tiene deps propias; se le miran las
17# SEMBRADAS (`seed-edges.json`, P3). En cuanto tiene receta, manda la receta — ya corregida por el
18# veredicto de harkaq. La semilla es andamio y se cae sola.
19#
20# NO LANZA NADA. Sólo imprime/escribe la cola; construir cuesta € y es decisión explícita
21# (`farm-up` + `campana-deuda.sh`). Ver el pie de la salida.
22#
23# Uso:  scripts/drenar.py --perfil escritorio-kde       # ondas de una imagen
24#       scripts/drenar.py                               # todo el corpus en deuda
25#       scripts/drenar.py --perfil cli --n 20           # sólo las 20 primeras construibles ya
26#       scripts/drenar.py --perfil escritorio-kde --tanda tandas/kde-drenaje.txt
27import json, sys
28from pathlib import Path
29
30ROOT = Path(__file__).resolve().parent.parent
31GRAFOS = ["docs/state/build-state.json", "docs/state/build-state-kde.json"]
32SEMILLA = ROOT / "docs/state/seed-edges.json"
33EN_DEUDA = ("debt", "never", "unhashable", "wanted")
34
35
36def cargar(perfil):
37    """Elige el grafo por la COLA que el perfil declara en `targets.toml`, no por cuál tiene más
38    nodos. La diferencia importa: la cola `incoming-kde` SOMBREA recetas canónicas, así que medir
39    `escritorio-mirada` contra el grafo KDE mide una imagen que nadie construye. Cada perfil se mide
40    contra el árbol de recetas del que realmente sale."""
41    sys.path.insert(0, str(ROOT / "scripts"))
42    import targets as targets_mod
43    cola = "corpus"
44    if perfil is not None:
45        try:
46            cola = targets_mod.load().get(perfil, {}).get("cola", "corpus")
47        except Exception:
48            pass
49    for f in GRAFOS:
50        p = ROOT / f
51        if not p.exists():
52            continue
53        d = json.loads(p.read_text())
54        if cola not in (d.get("by_queue") or {}):
55            continue
56        if perfil is None or any(perfil in (r.get("perfiles") or []) for r in d["nodes"].values()):
57            return d, f
58    return None, None
59
60
61def deps_de(nombre, nodo, semilla):
62    """La regla de la fuente única: con receta manda la receta; sin receta (nodo `wanted`) se usan
63    las aristas SEMBRADAS, que son hipótesis y se marcan como tales."""
64    if nodo["state"] != "wanted":
65        return nodo.get("deps") or [], "receta"
66    aristas = (semilla.get("nodos") or {}).get(nombre)
67    if isinstance(aristas, list):
68        return [a["dep"] for a in aristas if a.get("clase") in ("conocida", "frontera")], "semilla"
69    return [], "sin-datos"
70
71
72def ondas_de(perfil):
73    """(ondas, deuda, ambito, grafo) — el cálculo puro, sin imprimir. Reutilizado por --todos."""
74    doc, cual = cargar(perfil)
75    if doc is None:
76        return None, None, None, None
77    semilla = json.loads(SEMILLA.read_text()) if SEMILLA.exists() else {}
78    nodos = doc["nodes"]
79    ambito = {n: r for n, r in nodos.items()
80              if perfil is None or perfil in (r.get("perfiles") or [])}
81    deuda = {n: r for n, r in ambito.items() if r["state"] in EN_DEUDA}
82    restante, ondas = dict(deuda), []
83    while restante:
84        onda = [n for n, r in restante.items()
85                if not any(d in restante for d in deps_de(n, r, semilla)[0])]
86        if not onda:
87            ondas.append(sorted(restante)); break
88        onda.sort(key=lambda n: (-restante[n].get("unblocks", 0), restante[n].get("cls", ""), n))
89        ondas.append(onda)
90        for n in onda:
91            del restante[n]
92    return ondas, deuda, ambito, cual
93
94
95def todos_los_perfiles():
96    """Artefacto versionado con el drenaje de CADA imagen. El `git diff` entre dos corridas es el
97    avance real: qué salió de la deuda y qué onda se vació. Lo regenera el latido de la granja."""
98    sys.path.insert(0, str(ROOT / "scripts"))
99    import targets as targets_mod
100    doc = {"schema": "hammer-drenaje/1",
101           "nota": "DERIVADO del grafo de estado por scripts/drenar.py — no editar a mano.",
102           "perfiles": {}}
103    for p in sorted(targets_mod.load()):
104        ondas, deuda, ambito, cual = ondas_de(p)
105        if ondas is None:
106            continue
107        doc["perfiles"][p] = dict(
108            grafo=cual, nodos=len(ambito), en_deuda=len(deuda), ondas=len(ondas),
109            profundidad=len(ondas), construibles_ya=ondas[0] if ondas else [],
110            olas=[{"n": i + 1, "recetas": o} for i, o in enumerate(ondas)])
111        print(f"   {p:20} {len(deuda):3} en deuda · {len(ondas):2} ondas · "
112              f"onda 1 = {len(ondas[0]) if ondas else 0}")
113    salida = ROOT / "docs/state/drenaje.json"
114    salida.write_text(json.dumps(doc, indent=1, ensure_ascii=False) + "\n")
115    print(f"   → {salida.relative_to(ROOT)}")
116    return 0
117
118
119def main():
120    args = sys.argv[1:]
121    if "--todos" in args:
122        return todos_los_perfiles()
123    perfil = args[args.index("--perfil") + 1] if "--perfil" in args else None
124    tope = int(args[args.index("--n") + 1]) if "--n" in args else None
125    tanda = args[args.index("--tanda") + 1] if "--tanda" in args else None
126
127    doc, cual = cargar(perfil)
128    if doc is None:
129        print(f"no encontré un grafo con el perfil `{perfil}` "
130              f"(¿corriste scripts/build-state.py [--kde]?)", file=sys.stderr)
131        return 1
132    semilla = json.loads(SEMILLA.read_text()) if SEMILLA.exists() else {}
133    nodos = doc["nodes"]
134
135    ambito = {n: r for n, r in nodos.items()
136              if perfil is None or perfil in (r.get("perfiles") or [])}
137    deuda = {n: r for n, r in ambito.items() if r["state"] in EN_DEUDA}
138    etiqueta = f"perfil `{perfil}`" if perfil else "corpus entero"
139    print(f"== drenaje de {etiqueta} ({cual})")
140    print(f"   {len(ambito)} nodos en ámbito · {len(deuda)} en deuda "
141          f"({len(ambito)-len(deuda)} ya sellados)")
142    if not deuda:
143        print("   nada que drenar: la imagen está completa.")
144        return 0
145
146    # Ondas topológicas sobre el subgrafo EN DEUDA: una receta sólo espera por deps que también
147    # estén en deuda (las selladas ya están en el store y no bloquean).
148    restante = dict(deuda)
149    ondas, procedencia = [], {}
150    while restante:
151        onda = []
152        for n, r in restante.items():
153            ds, origen = deps_de(n, r, semilla)
154            procedencia[n] = origen
155            if not any(d in restante for d in ds):
156                onda.append(n)
157        if not onda:  # ciclo entre nodos en deuda: no debería pasar (topo_ok), pero no se oculta
158            print(f"\n   ⚠ CICLO entre {len(restante)} nodos en deuda: "
159                  f"{' '.join(sorted(restante)[:10])}")
160            ondas.append(sorted(restante))
161            break
162        onda.sort(key=lambda n: (-restante[n].get("unblocks", 0), restante[n].get("cls", ""), n))
163        ondas.append(onda)
164        for n in onda:
165            del restante[n]
166
167    print(f"\n== {len(ondas)} ondas (profundidad real de la cadena; el reloj de pared manda acá)")
168    for i, onda in enumerate(ondas, 1):
169        muestra = onda if tope is None or i > 1 else onda[:tope]
170        cab = f"   onda {i}: {len(onda):3} construibles"
171        if i == 1:
172            cab += "  ← ARRANCÁ POR ACÁ (nada las bloquea)"
173        print(cab)
174        for n in muestra[:25]:
175            r = deuda[n]
176            u = r.get("unblocks", 0)
177            marca = f" [{procedencia[n]}]" if procedencia[n] != "receta" else ""
178            print(f"        {n:28} {r['state']:10} destraba {u:3}  {r.get('cls','')}{marca}")
179        if len(muestra) > 25:
180            print(f"        … y {len(muestra)-25} más")
181
182    primera = ondas[0][:tope] if tope else ondas[0]
183    print(f"\n== cola para el worker ({len(primera)} recetas de la onda 1)")
184    print("   " + " ".join(primera))
185    if tanda:
186        p = ROOT / tanda
187        p.parent.mkdir(parents=True, exist_ok=True)
188        p.write_text(f"# DERIVADA de {cual} por scripts/drenar.py — no editar a mano.\n"
189                     f"# {etiqueta}: onda 1 de {len(ondas)}, ordenada por `unblocks` desc.\n"
190                     + "\n".join(primera) + "\n")
191        print(f"   → {tanda}")
192    print("\n   construir NO es gratis (€): esto no lanza nada. En el worker:")
193    print(f"     scripts/farm/campana-deuda.sh {' '.join(primera[:6])}…")
194    return 0
195
196
197if __name__ == "__main__":
198    sys.exit(main())