119 lines
4.5 KiB
Python
119 lines
4.5 KiB
Python
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
from tempfile import NamedTemporaryFile
|
|
from typing import Optional
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from clan_cli.dirs import (
|
|
nixpkgs_source,
|
|
)
|
|
from clan_cli.errors import ClanError
|
|
from clan_cli.nix import nix_eval
|
|
|
|
|
|
def machine_schema(
|
|
flake_dir: Path,
|
|
config: dict,
|
|
clan_imports: Optional[list[str]] = None,
|
|
) -> dict:
|
|
# use nix eval to lib.evalModules .#nixosConfigurations.<machine_name>.options.clan
|
|
with NamedTemporaryFile(mode="w", dir=flake_dir) as clan_machine_settings_file:
|
|
env = os.environ.copy()
|
|
if clan_imports is not None:
|
|
config["clanImports"] = clan_imports
|
|
# dump config to file
|
|
json.dump(config, clan_machine_settings_file, indent=2)
|
|
clan_machine_settings_file.seek(0)
|
|
env["CLAN_MACHINE_SETTINGS_FILE"] = clan_machine_settings_file.name
|
|
# ensure that the requested clanImports exist
|
|
proc = subprocess.run(
|
|
nix_eval(
|
|
flags=[
|
|
"--impure",
|
|
"--show-trace",
|
|
"--expr",
|
|
f"""
|
|
let
|
|
b = builtins;
|
|
system = b.currentSystem;
|
|
flake = b.getFlake (toString {flake_dir});
|
|
clan-core = flake.inputs.clan-core;
|
|
config = b.fromJSON (b.readFile (b.getEnv "CLAN_MACHINE_SETTINGS_FILE"));
|
|
modules_not_found =
|
|
b.filter
|
|
(modName: ! clan-core.clanModules ? ${{modName}})
|
|
config.clanImports or [];
|
|
in
|
|
modules_not_found
|
|
""",
|
|
]
|
|
),
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=flake_dir,
|
|
env=env,
|
|
)
|
|
if proc.returncode != 0:
|
|
print(proc.stderr, file=sys.stderr)
|
|
raise ClanError(
|
|
f"Failed to check clanImports for existence:\n{proc.stderr}"
|
|
)
|
|
modules_not_found = json.loads(proc.stdout)
|
|
if len(modules_not_found) > 0:
|
|
raise HTTPException(
|
|
status_code=400,
|
|
detail={
|
|
"msg": "Some requested clan modules could not be found",
|
|
"modules_not_found": modules_not_found,
|
|
},
|
|
)
|
|
|
|
# get the schema
|
|
proc = subprocess.run(
|
|
nix_eval(
|
|
flags=[
|
|
"--impure",
|
|
"--show-trace",
|
|
"--expr",
|
|
f"""
|
|
let
|
|
system = builtins.currentSystem;
|
|
flake = builtins.getFlake (toString {flake_dir});
|
|
clan-core = flake.inputs.clan-core;
|
|
nixpkgsSrc = flake.inputs.nixpkgs or {nixpkgs_source()};
|
|
lib = import (nixpkgsSrc + /lib);
|
|
pkgs = import nixpkgsSrc {{ inherit system; }};
|
|
config = lib.importJSON (builtins.getEnv "CLAN_MACHINE_SETTINGS_FILE");
|
|
fakeMachine = pkgs.nixos {{
|
|
imports =
|
|
[
|
|
clan-core.nixosModules.clanCore
|
|
# potentially the config might affect submodule options,
|
|
# therefore we need to import it
|
|
config
|
|
]
|
|
# add all clan modules specified via clanImports
|
|
++ (map (name: clan-core.clanModules.${{name}}) config.clanImports or []);
|
|
}};
|
|
clanOptions = fakeMachine.options.clan;
|
|
jsonschemaLib = import {Path(__file__).parent / "jsonschema"} {{ inherit lib; }};
|
|
jsonschema = jsonschemaLib.parseOptions clanOptions;
|
|
in
|
|
jsonschema
|
|
""",
|
|
],
|
|
),
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=flake_dir,
|
|
env=env,
|
|
)
|
|
if proc.returncode != 0:
|
|
print(proc.stderr, file=sys.stderr)
|
|
raise ClanError(f"Failed to read schema:\n{proc.stderr}")
|
|
return json.loads(proc.stdout)
|