Merge pull request 'API Improvements' (#4276) from api-cleanup into main

Reviewed-on: https://git.clan.lol/clan/clan-core/pulls/4276
This commit is contained in:
hsjobeki
2025-07-08 21:17:12 +00:00
6 changed files with 52 additions and 24 deletions

View File

@@ -18,7 +18,7 @@ class FlashOptions(TypedDict):
@API.register @API.register
def get_flash_options() -> FlashOptions: def get_machine_flash_options() -> FlashOptions:
"""Retrieve available languages and keymaps for flash configuration. """Retrieve available languages and keymaps for flash configuration.
Returns: Returns:
FlashOptions: A dictionary containing lists of available languages and keymaps. FlashOptions: A dictionary containing lists of available languages and keymaps.

View File

@@ -9,7 +9,7 @@ from clan_lib.machines.actions import list_machines
from clan_lib.machines.list import instantiate_inventory_to_machines from clan_lib.machines.list import instantiate_inventory_to_machines
from clan_lib.machines.machines import Machine from clan_lib.machines.machines import Machine
from clan_lib.machines.suggestions import validate_machine_names from clan_lib.machines.suggestions import validate_machine_names
from clan_lib.machines.update import run_machine_deploy from clan_lib.machines.update import run_machine_update
from clan_lib.nix import nix_config from clan_lib.nix import nix_config
from clan_lib.ssh.remote import Remote from clan_lib.ssh.remote import Remote
@@ -144,7 +144,7 @@ def update_command(args: argparse.Namespace) -> None:
tid=machine.name, tid=machine.name,
async_ctx=AsyncContext(prefix=machine.name), async_ctx=AsyncContext(prefix=machine.name),
), ),
run_machine_deploy, run_machine_update,
machine=machine, machine=machine,
target_host=target_host, target_host=target_host,
build_host=machine.build_host(), build_host=machine.build_host(),

View File

@@ -17,7 +17,7 @@ def example_function() -> None:
"""Example function for creating logs.""" """Example function for creating logs."""
def run_machine_deploy() -> None: def run_machine_update() -> None:
"""Function for deploying machines.""" """Function for deploying machines."""
@@ -41,7 +41,7 @@ def main() -> None:
for repo in repos: for repo in repos:
for machine in machines: for machine in machines:
log_manager.create_log_file( log_manager.create_log_file(
run_machine_deploy, run_machine_update,
f"deploy_{machine}", f"deploy_{machine}",
["clans", repo, "machines", machine], ["clans", repo, "machines", machine],
) )

View File

@@ -17,7 +17,7 @@ from clan_lib.log_manager import (
# Test functions for log creation # Test functions for log creation
def run_machine_deploy() -> None: def run_machine_update() -> None:
"""Test function for deploying machines.""" """Test function for deploying machines."""
@@ -194,13 +194,13 @@ class TestLogFileCreation:
for repo in repos: for repo in repos:
for machine in machines: for machine in machines:
log_file = configured_log_manager.create_log_file( log_file = configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
f"deploy_{machine}", f"deploy_{machine}",
["clans", repo, "machines", machine], ["clans", repo, "machines", machine],
) )
assert log_file.op_key == f"deploy_{machine}" assert log_file.op_key == f"deploy_{machine}"
assert log_file.func_name == "run_machine_deploy" assert log_file.func_name == "run_machine_update"
assert log_file.get_file_path().exists() assert log_file.get_file_path().exists()
# Check the group structure includes URL encoding for dynamic parts # Check the group structure includes URL encoding for dynamic parts
@@ -241,7 +241,7 @@ class TestFilterFunction:
"""Test that empty filter returns top-level groups.""" """Test that empty filter returns top-level groups."""
# Create some log files first # Create some log files first
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, "test_op", ["clans", "repo1", "machines", "machine1"] run_machine_update, "test_op", ["clans", "repo1", "machines", "machine1"]
) )
top_level = configured_log_manager.filter([]) top_level = configured_log_manager.filter([])
@@ -258,7 +258,7 @@ class TestFilterFunction:
for repo in repos: for repo in repos:
for machine in machines: for machine in machines:
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
f"deploy_{machine}", f"deploy_{machine}",
["clans", repo, "machines", machine], ["clans", repo, "machines", machine],
) )
@@ -281,7 +281,7 @@ class TestFilterFunction:
"""Test filtering with specific date.""" """Test filtering with specific date."""
# Create log file # Create log file
log_file = configured_log_manager.create_log_file( log_file = configured_log_manager.create_log_file(
run_machine_deploy, "test_op", ["clans", "repo1", "machines", "machine1"] run_machine_update, "test_op", ["clans", "repo1", "machines", "machine1"]
) )
# Filter with the specific date # Filter with the specific date
@@ -308,7 +308,7 @@ class TestGetLogFile:
"""Test getting log file by operation key.""" """Test getting log file by operation key."""
# Create log file # Create log file
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_wintux", "deploy_wintux",
["clans", "repo1", "machines", "wintux"], ["clans", "repo1", "machines", "wintux"],
) )
@@ -317,7 +317,7 @@ class TestGetLogFile:
found_log_file = configured_log_manager.get_log_file("deploy_wintux") found_log_file = configured_log_manager.get_log_file("deploy_wintux")
assert found_log_file is not None assert found_log_file is not None
assert found_log_file.op_key == "deploy_wintux" assert found_log_file.op_key == "deploy_wintux"
assert found_log_file.func_name == "run_machine_deploy" assert found_log_file.func_name == "run_machine_update"
def test_get_log_file_with_selector( def test_get_log_file_with_selector(
self, configured_log_manager: LogManager self, configured_log_manager: LogManager
@@ -325,12 +325,12 @@ class TestGetLogFile:
"""Test getting log file with specific selector like example_usage.py.""" """Test getting log file with specific selector like example_usage.py."""
# Create log files in different locations # Create log files in different locations
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_wintux", "deploy_wintux",
["clans", "repo1", "machines", "wintux"], ["clans", "repo1", "machines", "wintux"],
) )
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_wintux", "deploy_wintux",
["clans", "repo2", "machines", "wintux"], ["clans", "repo2", "machines", "wintux"],
) )
@@ -347,7 +347,7 @@ class TestGetLogFile:
"""Test getting log file with specific date.""" """Test getting log file with specific date."""
# Create log file # Create log file
log_file = configured_log_manager.create_log_file( log_file = configured_log_manager.create_log_file(
run_machine_deploy, "deploy_demo", ["clans", "repo1", "machines", "demo"] run_machine_update, "deploy_demo", ["clans", "repo1", "machines", "demo"]
) )
# Find it by op_key and date # Find it by op_key and date
@@ -384,10 +384,10 @@ class TestListLogDays:
"""Test listing log days when logs exist.""" """Test listing log days when logs exist."""
# Create log files # Create log files
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, "op1", ["clans", "repo1", "machines", "machine1"] run_machine_update, "op1", ["clans", "repo1", "machines", "machine1"]
) )
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, "op2", ["clans", "repo2", "machines", "machine2"] run_machine_update, "op2", ["clans", "repo2", "machines", "machine2"]
) )
days = configured_log_manager.list_log_days() days = configured_log_manager.list_log_days()
@@ -412,7 +412,7 @@ class TestApiCompatibility:
for repo in repos: for repo in repos:
for machine in machines: for machine in machines:
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
f"deploy_{machine}", f"deploy_{machine}",
["clans", repo, "machines", machine], ["clans", repo, "machines", machine],
) )
@@ -747,19 +747,19 @@ class TestLogFileSorting:
# This simulates the realistic scenario where the same operation runs on different machines # This simulates the realistic scenario where the same operation runs on different machines
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_operation", "deploy_operation",
["clans", "repo1", "machines", "machine1"], ["clans", "repo1", "machines", "machine1"],
) )
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_operation", "deploy_operation",
["clans", "repo1", "machines", "machine2"], ["clans", "repo1", "machines", "machine2"],
) )
configured_log_manager.create_log_file( configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_operation", "deploy_operation",
["clans", "repo2", "machines", "machine1"], ["clans", "repo2", "machines", "machine1"],
) )
@@ -825,7 +825,7 @@ class TestURLEncoding:
# Create log file with special characters # Create log file with special characters
log_file = configured_log_manager.create_log_file( log_file = configured_log_manager.create_log_file(
run_machine_deploy, run_machine_update,
"deploy_special", "deploy_special",
["clans", special_repo, "machines", special_machine], ["clans", special_repo, "machines", special_machine],
) )

View File

@@ -103,7 +103,7 @@ def upload_sources(machine: Machine, ssh: Remote) -> str:
@API.register @API.register
def run_machine_deploy( def run_machine_update(
machine: Machine, target_host: Remote, build_host: Remote | None machine: Machine, target_host: Remote, build_host: Remote | None
) -> None: ) -> None:
"""Update an existing machine using nixos-rebuild or darwin-rebuild. """Update an existing machine using nixos-rebuild or darwin-rebuild.

View File

@@ -118,6 +118,32 @@ def make_schema_name(func_name: str, part: str) -> str:
return f"{func_name}_{part}" return f"{func_name}_{part}"
def get_tag_key(tags: list[str]) -> tuple:
"""Convert list of tags to a tuple key for sorting."""
return tuple(tags)
def sort_openapi_paths_by_tag_tree(openapi: dict) -> None:
# Extract (tags, path, method, operation) tuples
operations = []
for path, methods in openapi["paths"].items():
for method, operation in methods.items():
tag_path = operation.get("tags", [])
operations.append((tag_path, path, method, operation))
# Sort by the tag hierarchy
operations.sort(key=lambda x: get_tag_key(x[0]))
# Rebuild sorted openapi["paths"]
sorted_paths: dict = {}
for _tag_path, path, method, operation in operations:
sorted_paths[path] = sorted_paths.get(path, {})
sorted_paths[path][method] = operation
openapi["paths"] = dict(sorted_paths) # Ensure it's a plain dict
def main() -> None: def main() -> None:
input_path = Path(os.environ["INPUT_PATH"]) input_path = Path(os.environ["INPUT_PATH"])
@@ -203,6 +229,8 @@ def main() -> None:
} }
} }
sort_openapi_paths_by_tag_tree(openapi)
# === Add global definitions from $defs === # === Add global definitions from $defs ===
for def_name, def_schema in defs.items(): for def_name, def_schema in defs.items():
fixed_schema = fix_nullables(deepcopy(def_schema)) fixed_schema = fix_nullables(deepcopy(def_schema))