Add code-server support to auplc-installer env selection and image builds.

Wire auplc-code-cpu/gpu into catalog, Makefile all, overlay tagging, and the
basic preset; rename the TUI to env selection with back navigation; and show
build/pull progress during standalone img and dev quick flows.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: KerwinTsaiii

auplc_installer/catalog.py

@@ -45,6 +45,8 @@ class Course:
 COURSE_CATALOG: tuple[Course, ...] = (
     Course("cpu", "auplc-default", False, "base-cpu", "Base CPU (Python Base Environment)"),
     Course("gpu", "auplc-base", True, "base-rocm", "Base GPU (GPU Base Environment)"),
+    Course("code-cpu", "auplc-code-cpu", False, "code-cpu", "Code Server CPU Environment"),
+    Course("code-gpu", "auplc-code-gpu", True, "code-gpu", "Code Server GPU Environment"),
     Course("Course-CV", "auplc-cv", True, "cv", "Computer Vision Course"),
     Course("Course-DL", "auplc-dl", True, "dl", "Deep Learning Course"),
     Course("Course-LLM", "auplc-llm", True, "llm", "Large Language Model Course"),
@@ -54,7 +56,7 @@ class Course:
 COURSE_KEYS_ALL: tuple[str, ...] = tuple(c.key for c in COURSE_CATALOG)
 COURSE_BY_KEY: dict[str, Course] = {c.key: c for c in COURSE_CATALOG}
 
-COURSE_PRESET_BASIC: tuple[str, ...] = ("cpu", "gpu")
+COURSE_PRESET_BASIC: tuple[str, ...] = ("cpu", "gpu", "code-cpu", "code-gpu")
 COURSE_PRESET_ALL: tuple[str, ...] = COURSE_KEYS_ALL
 
 
@@ -78,12 +80,39 @@ class Course:
 
 
 BASE_TEAM_MAPPING: dict[str, list[str]] = {
-    "cpu": ["cpu"],
-    "gpu": ["Course-CV", "Course-DL", "Course-LLM", "Course-PhySim"],
-    "official": ["cpu", "gpu", "Course-CV", "Course-DL", "Course-LLM", "Course-PhySim"],
+    "cpu": ["cpu", "code-cpu"],
+    "gpu": ["code-gpu", "Course-CV", "Course-DL", "Course-LLM", "Course-PhySim"],
+    "official": [
+        "cpu",
+        "gpu",
+        "code-cpu",
+        "code-gpu",
+        "Course-CV",
+        "Course-DL",
+        "Course-LLM",
+        "Course-PhySim",
+    ],
     "AUP": ["Course-CV", "Course-DL", "Course-LLM", "Course-PhySim"],
-    "native-users": ["Course-CV", "Course-DL", "Course-LLM", "Course-PhySim", "cpu", "gpu"],
-    "github-users": ["cpu", "gpu", "Course-CV", "Course-DL", "Course-LLM", "Course-PhySim"],
+    "native-users": [
+        "code-cpu",
+        "code-gpu",
+        "Course-CV",
+        "Course-DL",
+        "Course-LLM",
+        "Course-PhySim",
+        "cpu",
+        "gpu",
+    ],
+    "github-users": [
+        "cpu",
+        "gpu",
+        "code-cpu",
+        "code-gpu",
+        "Course-CV",
+        "Course-DL",
+        "Course-LLM",
+        "Course-PhySim",
+    ],
 }
 
 

auplc_installer/cli.py

@@ -46,6 +46,7 @@
     ensure_sudo_session,
     log,
     log_error,
+    log_success,
     set_verbose,
     start_sudo_keepalive,
 )
@@ -99,7 +100,7 @@
   rt remove             Remove JupyterHub runtime
 
   img build [target...] Build custom images (default: all, or hub + selected courses)
-                        Targets: all, hub, base-cpu, base-rocm, cv, dl, llm, physim
+                        Targets: all, hub, base-cpu, base-rocm, code, code-cpu, code-gpu, cv, dl, llm, physim
   img pull              Pull external images for offline use
 
   detect-gpu            Show detected GPU configuration
@@ -146,7 +147,7 @@
                     unselected courses are hidden in the spawn UI. Empty (the
                     default) keeps the historical "all courses" behaviour.
                       all     - every course
-                      basic   - cpu base + gpu base only
+                      basic   - cpu/gpu base + code-server (code-cpu, code-gpu)
                       none    - Hub only, no courses
                       <list>  - comma-separated keys, e.g. cpu,gpu,Course-CV
                     Env: AUPLC_COURSES
@@ -176,7 +177,7 @@
     ./auplc-installer install --gpu=strix-halo
     ./auplc-installer install --gpu=auto --dry-run
     ./auplc-installer install --gpu=phx --docker=0     # legacy flags
-    ./auplc-installer install --courses=basic          # cpu + gpu only
+    ./auplc-installer install --courses=basic          # base + code-server envs
     ./auplc-installer install --courses=cpu,gpu,Course-CV
     ./auplc-installer img build base-rocm --gpu=strix
     ./auplc-installer install --mirror=mirror.example.com
@@ -565,17 +566,19 @@ def cmd_dev_quick(state: InstallerState) -> None:
     log("Dev quick cycle: build hub → restart pod")
     log("===========================================")
     detect_and_configure_gpu(state.gpu, gpu_type_override=state.gpu_type)
-    local_image_build(
-        ["hub"],
-        cfg=state.gpu,
-        courses=state.courses,
-        mirror_prefix=state.mirror_prefix,
-        mirror_pip=state.mirror_pip,
-        mirror_npm=state.mirror_npm,
-        use_docker=state.use_docker,
-        k3s_images_dir=state.k3s_images_dir,
-    )
-    dev_quick_rollout()
+    with stage("Building Hub image", idx=1, total=2):
+        local_image_build(
+            ["hub"],
+            cfg=state.gpu,
+            courses=state.courses,
+            mirror_prefix=state.mirror_prefix,
+            mirror_pip=state.mirror_pip,
+            mirror_npm=state.mirror_npm,
+            use_docker=state.use_docker,
+            k3s_images_dir=state.k3s_images_dir,
+        )
+    with stage("Restarting Hub pod", idx=2, total=2):
+        dev_quick_rollout()
 
 
 def cmd_dev_deploy(state: InstallerState) -> None:
@@ -686,25 +689,29 @@ def cmd_rt_reinstall(state: InstallerState) -> None:
 
 def cmd_img_build(state: InstallerState, targets: Sequence[str]) -> None:
     detect_and_configure_gpu(state.gpu, gpu_type_override=state.gpu_type)
-    local_image_build(
-        targets,
-        cfg=state.gpu,
-        courses=state.courses,
-        mirror_prefix=state.mirror_prefix,
-        mirror_pip=state.mirror_pip,
-        mirror_npm=state.mirror_npm,
-        use_docker=state.use_docker,
-        k3s_images_dir=state.k3s_images_dir,
-    )
+    with stage("Building custom images", idx=1, total=1):
+        local_image_build(
+            targets,
+            cfg=state.gpu,
+            courses=state.courses,
+            mirror_prefix=state.mirror_prefix,
+            mirror_pip=state.mirror_pip,
+            mirror_npm=state.mirror_npm,
+            use_docker=state.use_docker,
+            k3s_images_dir=state.k3s_images_dir,
+        )
+    log_success("Custom images built successfully.")
 
 
 def cmd_img_pull(state: InstallerState) -> None:
-    pull_external_images(
-        skip_build_only=False,
-        use_docker=state.use_docker,
-        k3s_images_dir=state.k3s_images_dir,
-        mirror_prefix=state.mirror_prefix,
-    )
+    with stage("Pulling external images", idx=1, total=1):
+        pull_external_images(
+            skip_build_only=False,
+            use_docker=state.use_docker,
+            k3s_images_dir=state.k3s_images_dir,
+            mirror_prefix=state.mirror_prefix,
+        )
+    log_success("External images pulled successfully.")
 
 
 # ---------------------------------------------------------------------------
@@ -871,7 +878,7 @@ def _dispatch(
         else:
             log_error(
                 "Usage: auplc-installer img {build [target...]|pull}. "
-                "Targets: all, hub, base-cpu, base-rocm, cv, dl, llm, physim"
+                "Targets: all, hub, base-cpu, base-rocm, code, code-cpu, code-gpu, cv, dl, llm, physim"
             )
             sys.exit(1)
         return

auplc_installer/overlay.py

@@ -29,6 +29,7 @@
 # so YAML emission is deterministic.
 _RESOURCE_IMAGE_BASE: dict[str, str] = {
     "gpu": "auplc-base",
+    "code-gpu": "auplc-code-gpu",
     "Course-CV": "auplc-cv",
     "Course-DL": "auplc-dl",
     "Course-LLM": "auplc-llm",
@@ -62,7 +63,7 @@ def emit_overlay(
     if not homogeneous_target:
         targets = " ".join(s.gpu_target for s in cfg.skus)
         buf.write(f"# Mixed gfx targets: {targets}\n")
-    buf.write(f"# Course selection : {courses.description()}\n")
+    buf.write(f"# Env selection : {courses.description()}\n")
     buf.write("# Regenerated on install/upgrade.\n")
     buf.write("custom:\n")
 
@@ -188,7 +189,7 @@ def generate_values_overlay(
 # header by :func:`emit_overlay`. Used by upgrade flows so that a bare
 # ``rt upgrade`` (no ``--courses=`` flag) preserves whatever the user
 # originally installed with instead of silently expanding to "all".
-_COURSE_HEADER_RE = re.compile(r"^# Course selection\s*:\s*(.+?)\s*$")
+_COURSE_HEADER_RE = re.compile(r"^# (?:Env selection|Course selection)\s*:\s*(.+?)\s*$")
 
 
 def try_load_courses_from_overlay(overlay_path: Path) -> CourseSelection | None:

auplc_installer/summary.py

@@ -68,7 +68,7 @@ def format_configuration_summary(state: InstallerState, *, image_source_label: s
     lines.append(f"  Registry mirror  : {state.mirror_prefix or '(none)'}")
     lines.append(f"  PyPI mirror      : {state.mirror_pip or '(default)'}")
     lines.append(f"  npm mirror       : {state.mirror_npm or '(default)'}")
-    lines.append(f"  Courses          : {state.courses.description()}")
+    lines.append(f"  Environments     : {state.courses.description()}")
     return "\n".join(lines)
 
 
@@ -104,5 +104,5 @@ def row(key: str, value: str, *, accent: bool = False, faint: bool = False) -> s
         lines.append(row("npm mirror", state.mirror_npm))
     else:
         lines.append(row("npm mirror", "(default)", faint=True))
-    lines.append(row("Courses", state.courses.description(), accent=True))
+    lines.append(row("Environments", state.courses.description(), accent=True))
     return "\n".join(lines)