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:
@@ -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.
|
||||||
|
|||||||
@@ -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(),
|
||||||
|
|||||||
@@ -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],
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -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],
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -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.
|
||||||
|
|||||||
@@ -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))
|
||||||
|
|||||||
Reference in New Issue
Block a user