prole/installer/ui/screens/cluster.py
chrisfu ce5ff83eb4 checkpoint: improve init scripts, installer flows, and kube context handling
- Fix kube context switching for k3s single-context kubeconfigs and k3d shorthand prefixes

- Update common init/status scripts (registry, kerberos, cnpg backup, service layer, common services)

- Add Gitea init script and installer ArgoCD screen

- Add Supabase realtime probe patching plus regression tests

- Extend installer core/UI test coverage
2026-03-14 20:07:54 -07:00

2144 lines
78 KiB
Python

"""Cluster lifecycle management (k3d / k3s / k8s)."""
import json
import os
import re
import subprocess
import sys
import threading
import time
import urllib.error
import urllib.request
from pathlib import Path
import tkinter as tk
from tkinter import ttk, messagebox, filedialog
from installer.screen import TerminalConsole
from installer import screen as ui
from installer import prole_conf
from installer.core.env import (
PROJECT_ROOT,
_default_opentofu_pipeline_url,
_deployment_mode_from_env,
_deployment_target_label,
_find_kubeconfig_file,
_k3d_prole_data_volume_args,
_kubectl_base_cmd_for_k3s as _kubectl_base_cmd_for_k3s_fn,
_looks_like_k8s_bearer_token,
_normalize_cluster_env,
_normalize_k3s_token,
_resolve_k3s_connection as _resolve_k3s_connection_fn,
_safe_str,
)
from installer.config import (
_encrypt_cfg_secret,
_merge_kubeconfig,
_write_k3s_kubeconfig,
)
from installer.core.actions import _reset_k3s_namespace
class ClusterScreenMixin:
"""Cluster lifecycle management (k3d / k3s / k8s)."""
def _render_init_cluster_page(self):
# Letterhead at top right
content_width = self.bg_canvas.winfo_width() or 975
right_margin = content_width - 48
ui.canvas_text(
self,
right_margin,
40,
"Prole",
fill="#6e6e73",
font=("SF Pro Text", 32, "bold"),
anchor="ne",
)
ui.canvas_text(
self,
right_margin,
85,
"Infrastructure Automated.",
fill="#6e6e73",
font=("SF Pro Text", 18),
anchor="ne",
)
# Tighten vertical spacing on this screen so the main status area (traffic lights +
# console output) has enough room in the fixed-size window.
self._render_title("Cluster Environment", y=110)
self._render_paragraph(
"Select a cluster environment and ensure the cluster is running and common services can be deployed.",
y=155,
)
# Cluster Selection (Radio Buttons)
x_label = 48
y = 210
self._canvas_items.append(
ui.canvas_text(
self,
x_label,
y,
"Select Environment:",
fill="black",
font=("SF Pro Text", 14, "bold"),
)
)
y += 28
cluster_options = [("dev", "Dev"), ("service", "Service"), ("prod", "Prod")]
# We need to trace cluster_env if not already traced
if not hasattr(self, "_cluster_env_trace"):
self._cluster_env_trace = self.cluster_env.trace_add(
"write", self._on_cluster_env_change
)
rb_x = x_label + 20
rb_gap = 150
for idx, (val, name) in enumerate(cluster_options):
rb = tk.Radiobutton(
self.bg_canvas,
text=name,
variable=self.cluster_env,
value=val,
bg="white",
fg="black",
activebackground="white",
selectcolor="white",
font=("SF Pro Text", 11),
)
rb_window = self.bg_canvas.create_window(
rb_x + idx * rb_gap, y, window=rb, anchor="nw"
)
self._canvas_items.append(rb_window)
self._overlay_widgets.append(rb)
selected_env_key = self._cluster_env_key()
y += 36
if selected_env_key == "dev":
# Combined Select / Add Cluster widget
self._canvas_items.append(
ui.canvas_text(
self,
x_label,
y,
"Cluster:",
fill="black",
font=("SF Pro Text", 12, "bold"),
)
)
y += 28
clusters = self._get_k3d_cluster_list()
combo = ttk.Combobox(
self.bg_canvas,
textvariable=self.selected_k3d_cluster,
values=clusters,
width=30,
)
if not self.selected_k3d_cluster.get() and clusters:
self.selected_k3d_cluster.set(clusters[0])
combo.bind(
"<<ComboboxSelected>>", lambda _evt: self._on_k3d_cluster_select()
)
combo_win = self.bg_canvas.create_window(
x_label + 20, y, window=combo, anchor="nw"
)
self._canvas_items.append(combo_win)
self._overlay_widgets.append(combo)
add_btn = tk.Button(
self.bg_canvas,
text="Add",
command=self._on_create_k3d_cluster,
bg="#F5F5DC",
fg="black",
activebackground="#E5E5D5",
activeforeground="black",
highlightbackground="#F5F5DC",
highlightthickness=0,
relief="flat",
font=("SF Pro Text", 10),
padx=10,
)
add_win = self.bg_canvas.create_window(
x_label + 290, y - 4, window=add_btn, anchor="nw"
)
self._canvas_items.append(add_win)
self._overlay_widgets.append(add_btn)
delete_btn = tk.Button(
self.bg_canvas,
text="Delete",
command=self._on_delete_k3d_cluster,
bg="#F5F5DC",
fg="#cc0000",
activebackground="#E5E5D5",
activeforeground="#cc0000",
highlightbackground="#F5F5DC",
highlightthickness=0,
relief="flat",
font=("SF Pro Text", 10),
padx=10,
)
delete_win = self.bg_canvas.create_window(
x_label + 350, y - 4, window=delete_btn, anchor="nw"
)
self._canvas_items.append(delete_win)
self._overlay_widgets.append(delete_btn)
y += 40
else:
# Service / Prod Section
# Context selection for both Service and Prod (managed via kubectx)
self._canvas_items.append(
ui.canvas_text(
self,
x_label,
y,
"Kubernetes Context:",
fill="black",
font=("SF Pro Text", 12, "bold"),
)
)
values = self._get_kubectx_list()
combo = ttk.Combobox(
self.bg_canvas,
textvariable=self.selected_kubectx,
values=values,
state="readonly",
width=40,
)
combo.bind("<<ComboboxSelected>>", self._on_kubectx_select)
if not self.selected_kubectx.get() and values:
# Default to the current context if it exists in the list; otherwise pick
# a sensible default without switching immediately.
current_ctx = (getattr(self, "_kubectx_applied", "") or "").strip()
if not current_ctx:
current_ctx = (self._current_kubectl_context() or "").strip()
if current_ctx:
self._kubectx_applied = current_ctx
if current_ctx and current_ctx in values:
self.selected_kubectx.set(current_ctx)
elif selected_env_key == "service" and "prole-k3s" in values:
self.selected_kubectx.set("prole-k3s")
else:
self.selected_kubectx.set(values[0])
combo_win = self.bg_canvas.create_window(
x_label + 190, y - 6, window=combo, anchor="nw"
)
self._canvas_items.append(combo_win)
self._overlay_widgets.append(combo)
# Keep references for UI tests / layout verification.
self._kubectx_combo = combo
self._kubectx_combo_canvas_window = combo_win
ui.canvas_text(
self,
x_label + 580,
y + 2,
"via kubectx",
fill="#6e6e73",
font=("SF Pro Text", 10),
)
apply_btn = tk.Button(
self.bg_canvas,
text="Apply",
command=self._on_kubectx_apply,
bg="#F5F5DC",
fg="black",
activebackground="#E5E5D5",
highlightbackground="#F5F5DC",
highlightthickness=0,
relief="flat",
font=("SF Pro Text", 10),
padx=10,
state="disabled",
)
apply_win = self.bg_canvas.create_window(
x_label + 670, y - 10, window=apply_btn, anchor="nw"
)
self._canvas_items.append(apply_win)
self._overlay_widgets.append(apply_btn)
# Keep references for UI tests / layout verification.
self._kubectx_apply_btn = apply_btn
self._kubectx_apply_btn_canvas_window = apply_win
self._update_kubectx_apply_button()
y += 34
if selected_env_key == "prod":
# Staging directory input
self._canvas_items.append(
ui.canvas_text(
self,
x_label,
y,
"Local Artifact Staging Directory:",
fill="black",
font=("SF Pro Text", 12, "bold"),
)
)
entry = tk.Entry(
self.bg_canvas,
textvariable=self.prod_artifacts_path,
width=50,
bg="white",
fg="black",
insertbackground="black",
highlightbackground="#CCCCCC",
highlightthickness=1,
relief="flat",
font=("SF Pro Text", 11),
)
entry_win = self.bg_canvas.create_window(
x_label + 260, y - 6, window=entry, anchor="nw"
)
self._canvas_items.append(entry_win)
self._overlay_widgets.append(entry)
browse_btn = tk.Button(
self.bg_canvas,
text="Browse...",
command=lambda: self.prod_artifacts_path.set(
filedialog.askdirectory() or self.prod_artifacts_path.get()
),
)
browse_win = self.bg_canvas.create_window(
x_label + 770, y - 8, window=browse_btn, anchor="nw"
)
self._canvas_items.append(browse_win)
self._overlay_widgets.append(browse_btn)
y += 34
ui.canvas_text(
self,
x_label + 20,
y,
"(Used by etc/deploy_pipeline.sh --mode gcp)",
fill="#6e6e73",
font=("SF Pro Text", 10),
)
y += 34
# Common Services Status Check (Notebook)
# Show if in Service/Prod or if a cluster is selected in Dev
show_notebook = False
if selected_env_key == "dev":
if self.selected_k3d_cluster.get():
show_notebook = True
else:
show_notebook = True
if show_notebook:
ui.canvas_text(
self,
x_label,
y,
"Common Core Services Namespace:",
fill="black",
font=("SF Pro Text", 12, "bold"),
)
ns_entry = tk.Entry(
self.bg_canvas,
textvariable=self.service_namespace,
width=20,
bg="white",
fg="black",
insertbackground="black",
highlightbackground="#CCCCCC",
highlightthickness=1,
relief="flat",
font=("SF Pro Text", 11),
)
ns_win = self.bg_canvas.create_window(
x_label + 250, y - 6, window=ns_entry, anchor="nw"
)
self._canvas_items.append(ns_win)
self._overlay_widgets.append(ns_entry)
# Keep references for UI tests / layout verification.
self._service_namespace_entry = ns_entry
self._service_namespace_entry_canvas_window = ns_win
save_btn = tk.Button(
self.bg_canvas,
text="Save",
command=self._validate_and_save_cluster_config,
bg="#F5F5DC",
fg="black",
activebackground="#E5E5D5",
highlightbackground="#F5F5DC",
highlightthickness=0,
relief="flat",
font=("SF Pro Text", 10),
padx=10,
)
save_win = self.bg_canvas.create_window(
x_label + 480, y - 10, window=save_btn, anchor="nw"
)
self._canvas_items.append(save_win)
self._overlay_widgets.append(save_btn)
y += 34
# Traffic light status indicators for each service component
self._service_traffic_lights = {}
components = [
"Registry",
"CertMgr",
"Garage",
"Kong",
"OpenBao",
"OpenTofu",
]
columns = 3
col_gap = 180
row_gap = 24
start_x = x_label + 20
for idx, comp in enumerate(components):
row = idx // columns
col = idx % columns
tl_x = start_x + (col * col_gap)
tl_y = y + (row * row_gap)
# Circle indicator (default grey)
indicator = self.bg_canvas.create_oval(
tl_x,
tl_y,
tl_x + 14,
tl_y + 14,
fill="#8e8e93",
outline="#8e8e93",
)
label = ui.canvas_text(
self,
tl_x + 20,
tl_y,
comp,
fill="black",
font=("SF Pro Text", 11),
)
self._canvas_items.extend([indicator, label])
self._service_traffic_lights[comp.lower()] = indicator
rows = (len(components) + columns - 1) // columns
y += (rows * row_gap)
# Repair button should remain reachable; place it adjacent to the status area
# instead of pushing it below the console.
btn_text = "Repair"
deploy_btn = tk.Button(
self.bg_canvas,
text=btn_text,
command=self._deploy_k3s_services,
bg="#F5F5DC",
fg="black",
activebackground="#E5E5D5",
highlightbackground="#F5F5DC",
highlightthickness=0,
relief="flat",
font=("SF Pro Text", 10),
padx=10,
pady=5,
)
self._k3s_deploy_btn = deploy_btn
notebook = ttk.Notebook(self.bg_canvas)
status_frame = tk.Frame(notebook, bg="white")
deploy_frame = tk.Frame(notebook, bg="white")
status_console = ui.TerminalConsole(status_frame)
deploy_console = ui.TerminalConsole(deploy_frame)
status_console.pack(fill="both", expand=True)
deploy_console.pack(fill="both", expand=True)
notebook.add(status_frame, text="Status")
notebook.add(deploy_frame, text="Repair Output")
self._k3s_service_notebook = notebook
self._k3s_service_status_console = status_console
self._k3s_service_deploy_console = deploy_console
self._k3s_service_status_tab = status_frame
self._k3s_service_deploy_tab = deploy_frame
console_width = 900
console_height = 260
deploy_win = self.bg_canvas.create_window(
x_label + console_width - 96,
y - 34,
window=deploy_btn,
anchor="nw",
)
self._canvas_items.append(deploy_win)
self._overlay_widgets.append(deploy_btn)
self._k3s_deploy_btn_canvas_window = deploy_win
console_window = self.bg_canvas.create_window(
x_label,
y,
window=notebook,
anchor="nw",
width=console_width,
height=console_height,
)
self._canvas_items.append(console_window)
self._k3s_service_notebook_canvas_window = console_window
self._overlay_widgets.extend(
[notebook, status_frame, deploy_frame, status_console, deploy_console]
)
# Always start on the Status tab with a clean console
try:
status_console.clear()
self._k3s_service_notebook.select(self._k3s_service_status_tab)
except Exception:
pass
y += console_height + 12
self._verify_k3s_services()
def _cluster_env_key(self, env_label: str | None = None) -> str:
"""Return normalized cluster env key ('dev', 'service', 'prod')."""
value = env_label
if value is None:
try:
value = self.cluster_env.get()
except Exception:
value = ""
try:
if hasattr(value, "get") and callable(value.get):
value = value.get()
except Exception:
pass
if not value:
try:
value = (
(self.prole_cfg_data.get("Initialize Cluster", {}) or {})
.get("ENVIRONMENT", "")
.strip()
)
except Exception:
value = ""
if not value:
try:
value = (
(self.prole_cfg_data.get("Global", {}) or {})
.get("CLUSTER_ENV", "")
.strip()
)
except Exception:
value = ""
key = _normalize_cluster_env(value)
if not key:
return "dev"
return key
def _get_service_namespace(self) -> str:
"""Resolve service namespace with UI/env/config fallbacks.
Safety rule: avoid defaulting to the Kubernetes `default` namespace unless we truly
cannot infer a better option.
"""
# 1) Explicit user/config/env choice wins.
try:
ns = (self.service_namespace.get() or "").strip()
except Exception:
ns = ""
if not ns:
try:
ns = (
(self.prole_cfg_data.get("Global", {}) or {})
.get("SERVICE_NAMESPACE", "")
.strip()
)
except Exception:
ns = ""
if not ns:
ns = (os.environ.get("SERVICE_NAMESPACE") or "").strip()
if ns:
return ns
# 2) If the cluster is reachable, infer the common services namespace.
inferred = ""
try:
inferred = (self._infer_common_services_namespace() or "").strip()
except Exception:
inferred = ""
if inferred:
return inferred
# 3) Stable defaults for known environments.
try:
# For local dev (k3d) and service (k3s), keep knoe-system stable across restarts.
if self._cluster_env_key() in ("dev", "service"):
return "knoe-system"
except Exception:
pass
# 4) Absolute last resort.
return "default"
def _infer_common_services_namespace(self) -> str:
"""Infer the namespace hosting common services.
If the cluster is reachable, try to locate an existing Docker Registry workload
(image containing `registry:2`). If found, treat that namespace as the common
services namespace.
This method is read-only.
"""
# Cache to avoid repeated `kubectl -A` probes during periodic refresh.
cache_ttl_s = 60
now = time.time()
cached_ns = getattr(self, "_common_services_ns_cache", "")
cached_at = getattr(self, "_common_services_ns_cache_at", 0.0)
if cached_ns and (now - float(cached_at or 0.0)) < cache_ttl_s:
return cached_ns
# Ensure kubeconfig is available when possible (read-only action).
try:
self._ensure_k3s_kubeconfig_merged()
except Exception:
pass
# Build a kubectl command suitable for the active mode.
try:
mode = self._deployment_mode()
except Exception:
mode = None
try:
base_cmd = self._kubectl_base_cmd(mode)
except Exception:
base_cmd = ["kubectl"]
# Respect an existing kubeconfig if present.
env = os.environ.copy()
try:
kc = _find_kubeconfig_file(env)
except Exception:
kc = _find_kubeconfig_file()
if kc:
env["KUBECONFIG"] = kc
def _probe(kind: str) -> list[str]:
cmd = base_cmd + ["get", kind, "-A", "-o", "json"]
res = subprocess.run(
cmd,
capture_output=True,
text=True,
env=env,
timeout=5,
)
if res.returncode != 0:
return []
try:
data = json.loads(res.stdout or "{}")
except Exception:
return []
out = []
for item in (data.get("items") or []):
try:
ns = (item.get("metadata") or {}).get("namespace") or ""
containers = (
(((item.get("spec") or {}).get("template") or {}).get("spec") or {}).get(
"containers"
)
or []
)
for c in containers:
img = (c.get("image") or "").strip()
if "registry:2" in img:
if ns:
out.append(ns)
break
except Exception:
continue
return out
namespaces = []
# Deployments are the most common shape; fall back to StatefulSets.
namespaces.extend(_probe("deploy"))
if not namespaces:
namespaces.extend(_probe("sts"))
# Prefer knoe-system if present, otherwise prefer a non-default namespace.
ns_choice = ""
if namespaces:
if "knoe-system" in namespaces:
ns_choice = "knoe-system"
else:
non_default = [n for n in namespaces if n and n != "default"]
ns_choice = (non_default[0] if non_default else namespaces[0]) or ""
if ns_choice:
setattr(self, "_common_services_ns_cache", ns_choice)
setattr(self, "_common_services_ns_cache_at", now)
return ns_choice
def _on_cluster_env_change(self, *args):
# Keep configuration environment selection explicit and stable.
try:
conf_dir = self._resolve_prole_conf_dir()
prole_conf.activate_environment(conf_dir, self.cluster_env.get())
except Exception:
pass
self._set_deploy_target_from_cluster_env()
# Trigger status check or refresh UI
env_key = self._cluster_env_key()
if env_key == "service":
try:
self._apply_k3s_defaults()
except Exception:
pass
# Prefer prole-k3s context if present, but require explicit Apply.
try:
values = self._get_kubectx_list()
except Exception:
values = []
if hasattr(self, "selected_kubectx") and "prole-k3s" in (values or []):
self.selected_kubectx.set("prole-k3s")
self._update_kubectx_apply_button()
elif env_key == "dev":
name = self.selected_k3d_cluster.get()
if name:
if hasattr(self, "selected_kubectx"):
self.selected_kubectx.set(f"k3d-{name}")
self._switch_kubectx(f"k3d-{name}")
self._verify_k3s_services()
self.show_page("init_cluster")
def _deployment_mode(self) -> str:
try:
env_val = self.cluster_env.get()
except Exception:
env_val = ""
return _deployment_mode_from_env(env_val)
def _deployment_pipeline_url(self, target_key: str) -> str:
env_url = (
os.environ.get("PROLE_OPENTOFU_URL") or os.environ.get("OPENTOFU_URL") or ""
).strip()
if env_url:
return env_url
section = None
if target_key == "service":
section = "Service Cluster (k3s)"
elif target_key == "prod":
section = "Prod Cluster (k8s)"
if section:
url = (
(self.prole_cfg_data.get(section, {}) or {})
.get("PIPELINE_URL", "")
.strip()
)
if url:
return url
return _default_opentofu_pipeline_url()
def _set_deploy_target_from_cluster_env(self):
if not hasattr(self, "deploy_target"):
return
if getattr(self, "_syncing_deploy_target", False):
return
self._syncing_deploy_target = True
try:
self.deploy_target.set(_deployment_target_label(self.cluster_env.get()))
except Exception:
pass
finally:
self._syncing_deploy_target = False
def _on_deploy_target_change(self, *args):
if getattr(self, "_syncing_deploy_target", False):
return
target = ""
try:
target = self.deploy_target.get()
except Exception:
target = ""
key = _normalize_cluster_env(target)
if key == "dev":
desired = "dev"
elif key == "service":
desired = "service"
elif key == "prod":
desired = "prod"
else:
return
self._syncing_deploy_target = True
try:
if self.cluster_env.get() != desired:
self.cluster_env.set(desired)
finally:
self._syncing_deploy_target = False
def _dashboard_kong_status(self, env: str | None = None) -> dict:
"""Check Kubernetes Dashboard (Kong) pod health and return status info."""
ns = "kubernetes-dashboard"
base_cmd = self._kubectl_base_cmd(env)
def _pod_healthy(pod_status: dict) -> bool:
phase = (pod_status.get("phase") or "").strip()
if phase not in ("Running", "Succeeded"):
return False
statuses = pod_status.get("containerStatuses") or []
if not statuses:
return phase == "Succeeded"
for cs in statuses:
if cs.get("ready") is True:
continue
state = cs.get("state") or {}
waiting = state.get("waiting") or {}
reason = (waiting.get("reason") or "").lower()
if reason:
return False
return False
return True
# Ensure namespace exists
rc_ns, out_ns = self._run_cmd_capture(
base_cmd + ["get", "ns", ns, "-o", "name"], timeout=6
)
if rc_ns != 0 or ns not in (out_ns or ""):
return {
"ok": False,
"msg": "Dashboard (Kong): Missing",
"fill": "#ff9f0a",
"reset_pods": [],
}
rc, out = self._run_cmd_capture(
base_cmd + ["get", "pods", "-n", ns, "-o", "json"], timeout=8
)
if rc != 0:
return {
"ok": False,
"msg": "Dashboard (Kong): Unable to query",
"fill": "#ff9f0a",
"reset_pods": [],
}
try:
payload = json.loads(out or "{}")
except Exception:
return {
"ok": False,
"msg": "Dashboard (Kong): Unable to parse status",
"fill": "#ff9f0a",
"reset_pods": [],
}
kong_pods = []
for item in payload.get("items", []):
name = (item.get("metadata", {}) or {}).get("name", "")
name_lc = name.lower()
if "kong" in name_lc and "dashboard" in name_lc:
kong_pods.append(item)
if not kong_pods:
return {
"ok": False,
"msg": "Dashboard (Kong): Missing",
"fill": "#ff9f0a",
"reset_pods": [],
}
unhealthy = []
for item in kong_pods:
name = (item.get("metadata", {}) or {}).get("name", "")
status = item.get("status") or {}
if not _pod_healthy(status):
unhealthy.append(name)
if unhealthy:
return {
"ok": False,
"msg": "Dashboard (Kong): Unhealthy (safe to reset)",
"fill": "#ff9f0a",
"reset_pods": unhealthy,
}
return {
"ok": True,
"msg": "Dashboard (Kong): Running",
"fill": "#34c759",
"reset_pods": [],
}
def _set_cluster_env_message(
self, text: str, color: str = "#34c759", clear_after_ms: int | None = None
):
"""Set a transient status message on the Cluster Environment screen."""
def _apply():
if not hasattr(self, "_cluster_env_message_label"):
return
if not self.bg_canvas.winfo_exists():
return
try:
self.bg_canvas.itemconfig(
self._cluster_env_message_label, text=text, fill=color
)
except Exception:
pass
self.safe_after(_apply)
if clear_after_ms is not None and clear_after_ms >= 0:
def _clear():
if not hasattr(self, "_cluster_env_message_label"):
return
if not self.bg_canvas.winfo_exists():
return
try:
self.bg_canvas.itemconfig(self._cluster_env_message_label, text="")
except Exception:
pass
self.safe_after(_clear, delay=clear_after_ms)
def _authority_context_missing(self) -> bool:
"""Detect missing authority Docker context when Kerberos is enabled."""
try:
if not self.kerberos_enabled.get():
return False
except Exception:
return False
candidates = []
env_home = (os.environ.get("PROLE_HOME") or "").strip()
if env_home:
candidates.append(Path(env_home))
candidates.append(PROJECT_ROOT)
for base in candidates:
try:
if (base / "authority").is_dir():
return False
if (base / "prole" / "authority").is_dir():
return False
except Exception:
continue
return True
def _maybe_run_repair_pipeline(self, status: dict, kong_status: dict | None):
"""Run repair pipeline when cluster is ready and anomalies are detected."""
if not status.get("cluster_ok"):
return
anomalies = []
if kong_status and not kong_status.get("ok", True):
anomalies.append("dashboard")
if self._authority_context_missing():
anomalies.append("authority")
if not anomalies:
return
now = time.time()
if self._repair_inflight:
return
if now - self._repair_last_run_at < self._repair_cooldown_s:
return
self._repair_last_run_at = now
self._repair_inflight = True
def worker():
try:
env_key = status.get("env") or self._cluster_env_key()
if env_key == "service":
ns = self._get_service_namespace()
else:
ns = (self.db_namespace.get() or "").strip() or "default"
env = self._script_env_for_namespace(ns)
if env_key == "service":
env["PROLE_MODE"] = "k3s"
elif env_key == "dev":
env["PROLE_MODE"] = "k3d"
self._set_cluster_env_message(
"Repair pipeline started", "#34c759", clear_after_ms=3000
)
self.controller.run_script(
"repair_pipeline.sh",
args=["-n", ns],
env=env,
)
self._set_cluster_env_message(
"Repair pipeline complete", "#34c759", clear_after_ms=3000
)
except Exception as e:
self._set_cluster_env_message(
f"Repair pipeline failed: {e}", "#ff3b30", clear_after_ms=6000
)
finally:
self._repair_inflight = False
threading.Thread(target=worker, daemon=True).start()
def _check_opentofu_health(self) -> bool:
"""Check OpenTofu readiness in the selected namespace."""
try:
env_key = self._cluster_env_key()
except Exception:
env_key = "dev"
try:
if env_key == "service":
ns = self._get_service_namespace()
else:
ns = (self.db_namespace.get() or "").strip() or "default"
except Exception:
ns = "default"
try:
base_cmd = self._kubectl_base_cmd(env_key)
res_dep = subprocess.run(
base_cmd
+ [
"-n",
ns,
"get",
"deploy",
"opentofu",
"-o",
"jsonpath={.status.readyReplicas}",
],
capture_output=True,
text=True,
timeout=3,
)
ready_dep = res_dep.returncode == 0 and (
res_dep.stdout or "0"
).strip() not in ("", "0")
if ready_dep:
return True
res_sts = subprocess.run(
base_cmd
+ [
"-n",
ns,
"get",
"statefulset",
"opentofu",
"-o",
"jsonpath={.status.readyReplicas}",
],
capture_output=True,
text=True,
timeout=3,
)
ready_sts = res_sts.returncode == 0 and (
res_sts.stdout or "0"
).strip() not in ("", "0")
return ready_sts
except Exception:
return False
def _check_k8s_cluster(self, env_label: str) -> tuple[bool, str]:
env_key = self._cluster_env_key(env_label)
label = (
"K3s Cluster"
if env_key == "service"
else "Prod Cluster" if env_key == "prod" else "Kubernetes Cluster"
)
try:
kubectl = subprocess.run(["which", "kubectl"], capture_output=True)
if kubectl.returncode != 0:
return False, f"{label}: kubectl not found"
if env_key == "service":
kubeconfig = _find_kubeconfig_file()
if not kubeconfig:
server, token = self._k3s_connection_info()
if not server:
return False, f"{label}: missing server URL or kubeconfig"
cmd = self._kubectl_base_cmd(env_key) + ["cluster-info"]
res = subprocess.run(cmd, capture_output=True, text=True, timeout=8)
ok = res.returncode == 0
except Exception:
ok = False
msg = f"{label}: Connected" if ok else f"{label}: Not reachable"
return ok, msg
def _get_k3d_cluster_list(self) -> list[str]:
"""Get list of k3d clusters."""
try:
res = subprocess.run(
["k3d", "cluster", "list", "--no-headers"],
capture_output=True,
text=True,
timeout=5,
)
if res.returncode == 0:
# Extract first column (cluster names)
return [
line.split()[0] for line in res.stdout.splitlines() if line.strip()
]
except Exception:
pass
return []
def _on_create_k3d_cluster(self):
name = self.selected_k3d_cluster.get().strip()
if not name:
messagebox.showerror("Error", "Cluster name cannot be empty")
return
# Persist the typed name so the refreshed page keeps it selected
self.k3d_cluster_name.set(name)
# If the cluster already exists just select it (triggers notebook display)
existing = self._get_k3d_cluster_list()
if name in existing:
self.selected_k3d_cluster.set(name)
self.show_page("init_cluster")
return
# Refresh the page so the notebook/console is visible before streaming output
self.selected_k3d_cluster.set(name)
self.show_page("init_cluster")
def do_create():
console = getattr(self, "_k3s_service_deploy_console", None)
if console:
console.clear()
console.write(
f"Creating k3d cluster '{name}' @ {time.strftime('%Y-%m-%d %H:%M:%S')}\n\n"
)
try:
if self._k3s_service_notebook and self._k3s_service_deploy_tab:
self._k3s_service_notebook.select(self._k3s_service_deploy_tab)
except Exception:
pass
try:
# Build the k3d create command
prole_data = str(self._resolve_env_dir("PROLE_DATA", "data"))
volume_args = _k3d_prole_data_volume_args(prole_data)
reg_args = []
if getattr(self, "registry_url", None):
if self.registry_url.startswith("localhost:5000"):
reg_args = ["--registry-create", "prole-registry:0.0.0.0:5000"]
else:
reg_args = ["--registry-use", self.registry_url]
cmd = (
["k3d", "cluster", "create", name, "-a", "2", "--wait"]
+ volume_args
+ reg_args
+ ["--timestamps"]
)
if console:
console.write(f"$ {' '.join(cmd)}\n\n")
proc = subprocess.Popen(
cmd,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
text=True,
cwd=PROJECT_ROOT,
)
for line in proc.stdout:
if console:
console.write(line)
proc.wait()
if proc.returncode != 0:
if console:
console.write(
f"\nk3d cluster create failed (exit {proc.returncode})\n"
)
return
if console:
console.write(f"\nCluster '{name}' created successfully.\n\n")
# Merge kubeconfig so kubectl can reach the new cluster
try:
subprocess.run(
[
"k3d",
"kubeconfig",
"merge",
name,
"--kubeconfig-switch-context",
],
capture_output=True,
text=True,
timeout=10,
)
if console:
console.write(f"Kubeconfig merged for cluster '{name}'.\n\n")
except Exception as kce:
if console:
console.write(f"Warning: failed to merge kubeconfig: {kce}\n")
# Refresh UI and then deploy common services check
self.safe_after(self._refresh_k3d_ui)
# Trigger common services check in the deploy console
self.safe_after(self._deploy_k3s_services)
except Exception as e:
if console:
console.write(f"\nError: {e}\n")
self.safe_after(
lambda: messagebox.showerror(
"Error", f"Failed to create k3d cluster: {e}"
)
)
threading.Thread(target=do_create, daemon=True).start()
def _refresh_k3d_ui(self):
clusters = self._get_k3d_cluster_list()
self.k3d_cluster_list.set(clusters)
name = self.k3d_cluster_name.get().strip()
if name in clusters:
self.selected_k3d_cluster.set(name)
self.show_page("init_cluster")
def _on_k3d_cluster_select(self, *args):
"""Called when a k3d cluster is selected from the dropdown."""
cluster_name = self.selected_k3d_cluster.get()
if cluster_name:
try:
# Merge kubeconfig and switch context to the selected cluster
subprocess.run(
["k3d", "kubeconfig", "merge", cluster_name, "--switch-context"],
capture_output=True,
text=True,
)
# Ensure kubectx also knows about it
self._switch_kubectx(f"k3d-{cluster_name}")
except Exception:
pass
self.show_page("init_cluster")
def _on_delete_k3d_cluster(self):
"""Delete the selected k3d cluster after user confirmation."""
name = self.selected_k3d_cluster.get().strip()
if not name:
messagebox.showerror("Error", "No cluster selected to delete.")
return
confirmed = messagebox.askyesno(
"Confirm Delete", f"Are you sure you want to delete cluster '{name}'?"
)
if not confirmed:
return
# Show the notebook/console before streaming output
self.show_page("init_cluster")
def do_delete():
console = getattr(self, "_k3s_service_deploy_console", None)
if console:
console.clear()
console.write(
f"Deleting k3d cluster '{name}' @ {time.strftime('%Y-%m-%d %H:%M:%S')}\n\n"
)
try:
if self._k3s_service_notebook and self._k3s_service_deploy_tab:
self._k3s_service_notebook.select(self._k3s_service_deploy_tab)
except Exception:
pass
try:
cmd = ["k3d", "cluster", "delete", name]
if console:
console.write(f"$ {' '.join(cmd)}\n\n")
proc = subprocess.Popen(
cmd,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
text=True,
cwd=PROJECT_ROOT,
)
for line in proc.stdout:
if console:
console.write(line)
proc.wait()
if proc.returncode != 0:
if console:
console.write(
f"\nk3d cluster delete failed (exit {proc.returncode})\n"
)
return
if console:
console.write(f"\nCluster '{name}' deleted successfully.\n")
# Clear selection and refresh UI
self.selected_k3d_cluster.set("")
self.k3d_cluster_name.set("")
self.safe_after(self._refresh_k3d_ui)
except Exception as e:
if console:
console.write(f"\nError: {e}\n")
self.safe_after(
lambda: messagebox.showerror(
"Error", f"Failed to delete k3d cluster: {e}"
)
)
threading.Thread(target=do_delete, daemon=True).start()
def _on_kubectx_select(self, *args):
"""Called when a generic kubernetes context is selected from the dropdown."""
self._update_kubectx_apply_button()
ctx = (self.selected_kubectx.get() or "").strip()
if ctx:
self._set_cluster_env_message(
f"Context change pending: {ctx} (click Apply)",
"#ff9500",
clear_after_ms=4000,
)
def _on_kubectx_apply(self):
ctx = (self.selected_kubectx.get() or "").strip()
if not ctx:
return
ok = self._switch_kubectx(ctx)
if ok:
self._kubectx_applied = ctx
self._update_kubectx_apply_button()
self._set_cluster_env_message(
f"Context applied: {ctx}", "#34c759", clear_after_ms=3000
)
self._verify_k3s_services()
self.show_page("init_cluster")
def _update_kubectx_apply_button(self):
btn = getattr(self, "_kubectx_apply_btn", None)
if not btn:
return
selected = ""
try:
if hasattr(self, "selected_kubectx") and self.selected_kubectx is not None:
selected = (self.selected_kubectx.get() or "").strip()
except Exception:
selected = ""
applied = (getattr(self, "_kubectx_applied", "") or "").strip()
state = "normal" if (selected and selected != applied) else "disabled"
try:
btn.config(state=state)
except Exception:
pass
def _switch_kubectx(self, context_name: str) -> bool:
"""Switch the current kubernetes context using kubectx or kubectl."""
if not context_name:
return False
# Determine which kubeconfig to modify.
# Use the same logic as the rest of the application.
kc = _find_kubeconfig_file()
env = os.environ.copy()
if kc:
env["KUBECONFIG"] = kc
try:
# Try kubectx first
res = subprocess.run(
["kubectx", context_name], capture_output=True, text=True, env=env
)
ok = res.returncode == 0
err = (res.stderr or "").strip()
if not ok:
# Fallback to kubectl config use-context
res2 = subprocess.run(
["kubectl", "config", "use-context", context_name],
capture_output=True,
text=True,
env=env
)
ok = res2.returncode == 0
if not ok:
err2 = (res2.stderr or "").strip()
err = err2 or err
# Update the UI dropdown to match (but don't trigger re-select)
if (
ok
and hasattr(self, "selected_kubectx")
and self.selected_kubectx.get() != context_name
):
try:
# Check if it's in the current values list
current_values = []
if hasattr(self, "kubectx_list") and self.kubectx_list is not None:
current_values = list(self.kubectx_list.get() or [])
if not current_values:
current_values = self._get_kubectx_list()
if context_name in (current_values or []):
self.selected_kubectx.set(context_name)
except Exception:
pass
if not ok:
try:
messagebox.showerror(
"Kubernetes Context",
f"Failed to switch to context '{context_name}'.\n\n{err or 'Unknown error'}",
)
except Exception:
pass
return ok
except Exception as e:
try:
messagebox.showerror(
"Kubernetes Context",
f"Failed to switch to context '{context_name}': {e}",
)
except Exception:
pass
return False
def _current_kubectl_context(self) -> str:
kc = _find_kubeconfig_file()
env = os.environ.copy()
if kc:
env["KUBECONFIG"] = kc
try:
res = subprocess.run(
["kubectl", "config", "current-context"],
capture_output=True,
text=True,
env=env,
)
if res.returncode == 0:
return (res.stdout or "").strip()
except Exception:
pass
return ""
def _get_kubectx_list(self) -> list[str]:
"""Get list of kubernetes contexts."""
kc = _find_kubeconfig_file()
env = os.environ.copy()
if kc:
env["KUBECONFIG"] = kc
try:
res = subprocess.run(["kubectx"], capture_output=True, text=True, env=env)
if res.returncode == 0:
return res.stdout.strip().split("\n")
# Fallback to kubectl
res = subprocess.run(
["kubectl", "config", "get-contexts", "-o", "name"],
capture_output=True,
text=True,
env=env
)
if res.returncode == 0:
return res.stdout.strip().split("\n")
except Exception:
pass
return ["default"]
def _validate_and_save_cluster_config(self) -> bool:
"""Verify the config then write the values to prole.cfg."""
env_key = self._cluster_env_key()
prev_service_ns = ""
try:
prev_service_ns = (
(self.prole_cfg_data.get("Global", {}) or {})
.get("SERVICE_NAMESPACE", "")
.strip()
)
except Exception:
prev_service_ns = ""
if env_key == "dev":
cluster_val = self.selected_k3d_cluster.get() or "knoe-dev-cluster"
if not cluster_val:
messagebox.showwarning(
"Validation", "Please select or create a k3d cluster."
)
return False
# Context switch (if context matches cluster name)
try:
subprocess.run(
["kubectl", "config", "use-context", f"k3d-{cluster_val}"],
capture_output=True,
text=True,
)
except Exception:
pass
elif env_key == "service":
# Service deployments can authenticate via an existing kubeconfig/context
# (preferred), or via a token-based generated kubeconfig.
kubeconfig = _find_kubeconfig_file()
has_token_flow = bool((self.k3s_server_url.get() or "").strip()) and bool(
(self.k3s_token.get() or "").strip()
)
if not kubeconfig and not has_token_flow:
messagebox.showwarning(
"Validation",
"Missing cluster credentials. Please select a working Kubernetes context (kubectx) "
"or provide K3s server URL + token.",
)
return False
cluster_val = self.cluster_env.get()
elif env_key == "prod":
path = self.prod_artifacts_path.get().strip()
if not path:
messagebox.showwarning(
"Validation", "Please specify an artifact staging directory."
)
return False
p = Path(path).expanduser()
if not p.exists():
try:
p.mkdir(parents=True, exist_ok=True)
except Exception as e:
messagebox.showerror(
"Error", f"Failed to create directory {path}: {e}"
)
return False
cluster_val = self.cluster_env.get()
else:
cluster_val = self.cluster_env.get()
# Capture cluster info for config.
# ENVIRONMENT here is used as an env/mode hint; keep it as the selected env key.
self.prole_cfg_data["Initialize Cluster"]["ENVIRONMENT"] = (
self.cluster_env.get() or ""
).strip()
self.prole_cfg_data["Initialize Cluster"]["K3S_SERVER_URL"] = (
self.k3s_server_url.get() or ""
).strip()
self.prole_cfg_data["Initialize Cluster"]["K3S_TOKEN"] = _encrypt_cfg_secret(
self.k3s_token.get() or ""
)
self.prole_cfg_data["Global"][
"SERVICE_NAMESPACE"
] = self._get_service_namespace()
new_service_ns = self.prole_cfg_data["Global"]["SERVICE_NAMESPACE"]
self.prole_cfg_data["Optional Features"]["SUPABASE_ENABLED"] = str(
self.supabase_enabled.get()
)
self.prole_cfg_data["Optional Features"]["GITOPS_ENABLED"] = str(
self.gitops_enabled.get()
)
self.prole_cfg_data["Optional Features"]["KERBEROS_ENABLED"] = str(
self.kerberos_enabled.get()
)
self.prole_cfg_data["Optional Features"]["AT_REST_ENCRYPTION_ENABLED"] = str(
self.at_rest_encryption_enabled.get()
)
if env_key == "prod":
self.prole_cfg_data["Prod Cluster (k8s)"][
"STAGING_DIRECTORY"
] = self.prod_artifacts_path.get()
# Save config
self._save_prole_cfg()
# Namespace change cleanup (prevents collisions when switching namespaces).
try:
prev = (prev_service_ns or "").strip()
new = (new_service_ns or "").strip()
if prev and new and prev != new:
do_cleanup = False
try:
do_cleanup = messagebox.askyesno(
"Namespace Changed",
(
f"Common Core Services namespace changed from '{prev}' to '{new}'.\n\n"
f"Reset the OLD namespace '{prev}' to avoid resource collisions?\n\n"
"This deletes workloads/services in that namespace (PVCs are preserved)."
),
)
except Exception:
do_cleanup = False
if do_cleanup:
def _cleanup_worker():
try:
self._set_cluster_env_message(
f"Resetting old namespace: {prev}",
"#ff9500",
clear_after_ms=3000,
)
try:
self._ensure_k3s_kubeconfig_merged()
except Exception:
pass
server, token = ("", "")
try:
server, token = self._k3s_connection_info()
except Exception:
server, token = ("", "")
controller = getattr(self, "controller", None)
project_root = getattr(controller, "project_root", None)
if project_root:
_reset_k3s_namespace(project_root, prev, server, token)
self._set_cluster_env_message(
f"Old namespace reset complete: {prev}",
"#34c759",
clear_after_ms=4000,
)
except Exception as e:
self._set_cluster_env_message(
f"Namespace reset failed: {e}",
"#ff3b30",
clear_after_ms=6000,
)
threading.Thread(target=_cleanup_worker, daemon=True).start()
except Exception:
pass
# On save, immediately run a services status refresh so the UI reflects the
# newly persisted namespace/cluster settings.
try:
self._verify_k3s_services()
except Exception:
pass
return True
def _cluster_status_snapshot(self) -> dict:
env = self._cluster_env_key()
cluster_ok = False
if env == "dev":
cluster_name = self.selected_k3d_cluster.get() or "knoe-dev-cluster"
try:
res = subprocess.run(
["k3d", "cluster", "list", "--no-headers"],
capture_output=True,
text=True,
)
cluster_ok = cluster_name in (res.stdout or "")
except Exception:
cluster_ok = False
cluster_msg = (
f"K3D Cluster ({cluster_name}): Running"
if cluster_ok
else f"K3D Cluster ({cluster_name}): Not found"
)
else:
cluster_ok, cluster_msg = self._check_k8s_cluster(env)
cluster_fill = "#34c759" if cluster_ok else "#ff9f0a"
common_ok = cluster_ok
common_msg = (
"Common Services: Ready to deploy"
if common_ok
else "Common Services: Not ready"
)
common_fill = "#34c759" if common_ok else "#ff9f0a"
return {
"env": env,
"cluster_ok": cluster_ok,
"cluster_msg": cluster_msg,
"cluster_fill": cluster_fill,
"common_ok": common_ok,
"common_msg": common_msg,
"common_fill": common_fill,
}
def _cluster_ready_for_navigation(self) -> bool:
status = self._cluster_status_snapshot()
if not status["cluster_ok"]:
try:
messagebox.showerror(
"Cluster",
"Cluster is not reachable. Please verify your cluster and try again.",
)
except Exception:
pass
return False
return True
def check_cluster_status_async(self):
def worker():
notice_active = False
self._set_cluster_env_message("Collect Status", "#34c759")
status = self._cluster_status_snapshot()
self._set_cluster_env_message("Verify State", "#34c759")
kong_status = None
if status.get("cluster_ok"):
kong_status = self._dashboard_kong_status(status.get("env"))
self._set_cluster_env_message("Observe", "#34c759")
# Observe-only: do not mutate the cluster automatically.
anomalies = []
try:
if kong_status and not kong_status.get("ok", True):
anomalies.append("dashboard")
except Exception:
pass
try:
if self._authority_context_missing():
anomalies.append("authority")
except Exception:
pass
if anomalies:
self._set_cluster_env_message(
"Notice: issues detected. Click Repair to reconcile.",
"#ff9f0a",
clear_after_ms=6000,
)
notice_active = True
def update_ui():
if hasattr(self, "k3d_status_label"):
self.bg_canvas.itemconfig(
self.k3d_status_label,
text=status["cluster_msg"],
fill=status["cluster_fill"],
)
if hasattr(self, "common_services_status_label"):
self.bg_canvas.itemconfig(
self.common_services_status_label,
text=status["common_msg"],
fill=status["common_fill"],
)
if hasattr(self, "_init_cluster_button"):
try:
self._init_cluster_button.configure(text="Save")
except Exception:
pass
self.root.after(0, update_ui)
# Clear message after a short delay if no notice is active
if not notice_active:
self._set_cluster_env_message("", "#34c759", clear_after_ms=2000)
threading.Thread(target=worker, daemon=True).start()
def _services_status_snapshot(self) -> dict:
opentofu_ok = self._check_opentofu_health()
opentofu_msg = "OpenTofu: Running" if opentofu_ok else "OpenTofu: Not reachable"
opentofu_fill = "#34c759" if opentofu_ok else "#ff9f0a"
return {
"opentofu_msg": opentofu_msg,
"opentofu_fill": opentofu_fill,
}
def check_services_status_async(self):
def worker():
status = self._services_status_snapshot()
def update_ui():
if hasattr(self, "_db_opentofu_status_label"):
self.bg_canvas.itemconfig(
self._db_opentofu_status_label,
text=status["opentofu_msg"],
fill=status["opentofu_fill"],
)
self.root.after(0, update_ui)
threading.Thread(target=worker, daemon=True).start()
def ensure_cluster_ready(self):
self._action_flags["init_cluster.start_cluster"] = True
# Implementation of cluster creation/startup
def worker():
cluster_env = self._cluster_env_key()
if cluster_env == "dev":
# 1. Start Docker if not running
if not self.controller.check_docker_running():
# Attempt to start Docker on macOS
subprocess.run(["open", "-a", "Docker"], capture_output=True)
# Wait for it to start
for _ in range(30):
time.sleep(2)
if self.controller.check_docker_running():
break
if not self.controller.check_docker_running():
self.root.after(
0,
lambda: messagebox.showerror(
"Docker",
"Could not start Docker. Please start it manually.",
),
)
return
# 2. Manage or verify cluster
cluster_name = "knoe-dev-cluster"
res = subprocess.run(
["k3d", "cluster", "list", "--no-headers"],
capture_output=True,
text=True,
)
res_stdout = res.stdout or ""
if self._reset_cluster:
subprocess.run(
["k3d", "cluster", "delete", cluster_name], capture_output=True
)
res_stdout = ""
self._reset_cluster = False
if cluster_name not in res_stdout:
# Create it
# Default args based on README.md
prole_data = str(self._resolve_env_dir("PROLE_DATA", "data"))
volume_args = _k3d_prole_data_volume_args(prole_data)
cmd = [
"k3d",
"cluster",
"create",
cluster_name,
"-a",
"2",
] + volume_args
reg_args = []
try:
if self.ensure_local_registry_available():
reg_args = ["--registry-use", "k3d-prole-registry:5000"]
except Exception:
reg_args = []
cmd += reg_args + ["--api-port", "0.0.0.0:6443"]
# Run in terminal or capture output? Let's use a console window later.
# For now, run it and update status.
subprocess.run(cmd, capture_output=True)
else:
# Start it if it's stopped
subprocess.run(
["k3d", "cluster", "start", cluster_name], capture_output=True
)
self.check_cluster_status_async()
self.root.after(
0,
lambda: messagebox.showinfo(
"Cluster", f"Cluster {cluster_name} is ready."
),
)
else:
if self._reset_cluster and cluster_env in ("service", "k3s"):
ns = (self._get_service_namespace() or "").strip() or "default"
server, token = self._k3s_connection_info()
_reset_k3s_namespace(
self.controller.project_root, ns, server, token
)
self._reset_cluster = False
ok, msg = self._check_k8s_cluster(cluster_env)
if not ok:
self.root.after(0, lambda: messagebox.showerror("Cluster", msg))
return
self.check_cluster_status_async()
self.root.after(0, lambda: messagebox.showinfo("Cluster", msg))
threading.Thread(target=worker, daemon=True).start()
def _verify_k3s_services(self):
if not self._k3s_service_status_console:
self._stop_k3s_service_status_updates()
return
try:
self._k3s_service_status_console.clear()
except Exception:
pass
try:
if self._k3s_service_notebook and self._k3s_service_status_tab:
self._k3s_service_notebook.select(self._k3s_service_status_tab)
except Exception:
pass
self._start_k3s_service_status_updates()
def _start_k3s_service_status_updates(self):
if not self._k3s_status_active:
self._k3s_status_active = True
if self._k3s_status_after_id:
try:
self.root.after_cancel(self._k3s_status_after_id)
except Exception:
pass
self._k3s_status_after_id = None
self._schedule_k3s_service_status_update(0)
def _stop_k3s_service_status_updates(self):
self._k3s_status_active = False
self._k3s_status_inflight = False
if self._k3s_status_after_id:
try:
self.root.after_cancel(self._k3s_status_after_id)
except Exception:
pass
self._k3s_status_after_id = None
def _schedule_k3s_service_status_update(self, delay_ms: int):
if not self._k3s_status_active:
return
if not self.root or not self.root.winfo_exists():
return
try:
self._k3s_status_after_id = self.root.after(
delay_ms, self._run_k3s_service_status_once
)
except Exception:
self._k3s_status_after_id = None
def _run_k3s_service_status_once(self):
if self._k3s_status_inflight:
return
if not self._k3s_service_status_console:
self._k3s_status_inflight = False
self.safe_after(lambda: self._schedule_k3s_service_status_update(30000))
return
self._k3s_status_inflight = True
console = self._k3s_service_status_console
console.clear()
console.write(
f"Service status refresh @ {time.strftime('%Y-%m-%d %H:%M:%S')}\n\n"
)
def worker():
collected_lines = []
try:
ns = self._get_service_namespace()
env = self._script_env_for_namespace(ns)
env["PROLE_MODE"] = self._deployment_mode()
if not env.get("KUBECONFIG"):
console.write(
"KUBECONFIG not generated. Check cluster credentials in prole.cfg.\n"
)
return
verify_args = ["-n", ns, "verify"]
def _on_line(line):
console.write(line)
collected_lines.append(line)
rc = self.controller.run_script(
"init_common_services.sh",
args=verify_args,
env=env,
on_line=_on_line,
)
console.write(f"\nVerification Exit status: {rc}\n")
# Parse per-component traffic light status from [OK]/[FAIL] lines
component_status = self._parse_service_status_lines(collected_lines)
all_green = rc == 0
def _update_traffic_lights():
self._all_services_green = all_green
# Ensure common services success flag is synced for navigation
self._common_services_success = all_green
lights = getattr(self, "_service_traffic_lights", {})
for comp_key, indicator_id in lights.items():
if comp_key in component_status:
color = (
"#34c759" if component_status[comp_key] else "#ff3b30"
)
else:
color = "#34c759" if all_green else "#8e8e93"
try:
self.bg_canvas.itemconfig(
indicator_id, fill=color, outline=color
)
except Exception:
pass
# Update footer to enable Next button if everything is green
self.update_footer()
self.safe_after(_update_traffic_lights)
finally:
self._k3s_status_inflight = False
self.safe_after(lambda: self._schedule_k3s_service_status_update(30000))
threading.Thread(target=worker, daemon=True).start()
def _parse_service_status_lines(self, lines: list[str]) -> dict[str, bool]:
"""Parse [OK]/[FAIL] lines from status_common_services.sh output.
Returns a dict mapping component keys (registry, certmgr,
garage, kong, openbao, opentofu) to True (healthy) or False (unhealthy).
"""
# Map resource names to traffic light component keys
name_map = {
"registry": "registry",
"opentofu": "opentofu",
"garage": "garage",
"openbao": "openbao",
"prole-svc-kong": "kong",
"cert-manager": "certmgr",
"cert-manager-cainjector": "certmgr",
"cert-manager-webhook": "certmgr",
}
# Start with all components healthy (True); any FAIL sets to False
result: dict[str, bool] = {}
import re as _re
for line in lines:
m = _re.match(r"\[(OK|FAIL)\]\s+(\S+)/(\S+)", line)
if not m:
continue
status_tag = m.group(1)
resource_name = m.group(3)
comp_key = name_map.get(resource_name)
if not comp_key:
continue
if comp_key not in result:
result[comp_key] = True
if status_tag == "FAIL":
result[comp_key] = False
return result
def _deploy_k3s_services(self):
"""Deploy common services (Registry/OpenTofu/Garage) to remote k3s cluster."""
def _deploy():
console = self._k3s_service_deploy_console
if console:
console.clear()
console.write(
f"Deploying common services @ {time.strftime('%Y-%m-%d %H:%M:%S')}\n\n"
)
try:
if self._k3s_service_notebook and self._k3s_service_deploy_tab:
self._k3s_service_notebook.select(self._k3s_service_deploy_tab)
except Exception:
pass
service_ns = self._get_service_namespace()
env = self._script_env_for_namespace(service_ns)
env["PROLE_MODE"] = self._deployment_mode()
log_path = self._common_services_log_path()
env["COMMON_SERVICES_INIT_LOG"] = str(log_path)
try:
log_fp = log_path.open("a", encoding="utf-8")
log_fp.write(
f"\n# Common services deploy @ {time.strftime('%Y-%m-%d %H:%M:%S')}\n"
)
log_fp.flush()
except Exception:
log_fp = None
if console:
console.write(f"Log file: {log_path}\n\n")
if not env.get("KUBECONFIG"):
if console:
console.write(
"KUBECONFIG not generated. Check cluster credentials in prole.cfg.\n"
)
if log_fp:
try:
log_fp.write(
"KUBECONFIG not generated. Check cluster credentials in prole.cfg.\n"
)
log_fp.close()
except Exception:
pass
self._verify_k3s_services()
return
try:
common_args = ["-n", env["NAMESPACE"]]
if self.kerberos_enabled.get():
common_args.append("-k")
common_args.append("update")
def _log_line(line: str):
if console:
console.write(line)
if log_fp:
try:
log_fp.write(line)
log_fp.flush()
except Exception:
pass
rc = self.controller.run_script(
"init_common_services.sh",
args=common_args,
env=env,
on_line=_log_line,
)
if console:
console.write(f"\nExit status: {rc}\n")
if log_fp:
try:
log_fp.write(f"\nExit status: {rc}\n")
except Exception:
pass
finally:
if log_fp:
try:
log_fp.close()
except Exception:
pass
self._verify_k3s_services()
threading.Thread(target=_deploy, daemon=True).start()
def _ensure_k3s_kubeconfig_merged(self):
"""Ensure a valid kubeconfig is available and merged into ~/.kube/config.
Prefers an existing Ansible-fetched kubeconfig (with client-certificate
auth) over generating a new token-based one.
"""
# 1. Check for existing cert-based kubeconfig first
existing = _find_kubeconfig_file()
if existing:
self._managed_kubeconfig = existing
_merge_kubeconfig(existing)
return
# 2. Fall back to token-based generation
server, token = self._k3s_connection_info()
if not server or not token:
return
kubeconfig_path = str(_write_k3s_kubeconfig(server, token))
self._managed_kubeconfig = kubeconfig_path
_merge_kubeconfig(kubeconfig_path)
def _k3s_connection_info(self) -> tuple[str, str]:
# Check UI fields first; fall back to the shared 3-source resolver.
ui_server = (_safe_str(self.k3s_server_url.get()) or "").strip()
ui_token = (_safe_str(self.k3s_token.get()) or "").strip()
if ui_token:
ui_token = self._resolve_secret_value(ui_token)
ui_token = _normalize_k3s_token(ui_token)
if ui_server and not ui_server.startswith("http"):
ui_server = f"https://{ui_server}"
# If UI fields are populated, use them directly.
if ui_server and ui_token:
return ui_server, ui_token
# Otherwise delegate to the unified resolver (env → cfg → ansible).
resolved_server, resolved_token = _resolve_k3s_connection_fn(
project_root=PROJECT_ROOT,
)
return ui_server or resolved_server, ui_token or resolved_token
def _k3s_connection_raw(self) -> tuple[str, str]:
"""Return (server_url, token) without attempting secret resolution."""
server = (
_safe_str(self.k3s_server_url.get())
or os.environ.get("PROLE_K3S_SERVER")
or os.environ.get("K3S_SERVER_URL")
or ""
).strip()
token = (
_safe_str(self.k3s_token.get())
or os.environ.get("PROLE_K3S_TOKEN")
or os.environ.get("K3S_TOKEN")
or ""
).strip()
if server and not server.startswith("http"):
server = f"https://{server}"
return server, token