2023-07-26 06:00:07 +00:00
|
|
|
import os
|
2023-08-26 08:32:04 +00:00
|
|
|
import tempfile
|
2023-07-26 06:00:07 +00:00
|
|
|
|
2023-09-14 13:22:13 +00:00
|
|
|
from .dirs import nixpkgs_flake, nixpkgs_source, unfree_nixpkgs
|
2023-09-13 14:01:03 +00:00
|
|
|
|
|
|
|
|
2023-09-15 14:22:05 +00:00
|
|
|
def nix_command(flags: list[str]) -> list[str]:
|
|
|
|
return ["nix", "--experimental-features", "nix-command flakes"] + flags
|
|
|
|
|
|
|
|
|
2023-09-14 13:22:13 +00:00
|
|
|
def nix_build(
|
|
|
|
flags: list[str],
|
2023-09-13 14:40:26 +00:00
|
|
|
) -> list[str]:
|
2023-09-15 14:22:05 +00:00
|
|
|
return (
|
|
|
|
nix_command(
|
|
|
|
[
|
|
|
|
"build",
|
|
|
|
"--no-link",
|
|
|
|
"--print-out-paths",
|
|
|
|
"--extra-experimental-features",
|
|
|
|
"nix-command flakes",
|
|
|
|
]
|
|
|
|
)
|
|
|
|
+ flags
|
|
|
|
)
|
2023-08-26 09:23:15 +00:00
|
|
|
|
2023-07-26 06:00:07 +00:00
|
|
|
|
2023-08-26 08:32:04 +00:00
|
|
|
def nix_eval(flags: list[str]) -> list[str]:
|
2023-09-15 14:22:05 +00:00
|
|
|
default_flags = nix_command(
|
|
|
|
[
|
|
|
|
"eval",
|
|
|
|
"--show-trace",
|
|
|
|
"--json",
|
|
|
|
]
|
|
|
|
)
|
2023-09-15 11:44:04 +00:00
|
|
|
if os.environ.get("IN_NIX_SANDBOX"):
|
|
|
|
with tempfile.TemporaryDirectory() as nix_store:
|
|
|
|
return (
|
|
|
|
default_flags
|
|
|
|
+ [
|
|
|
|
"--override-input",
|
|
|
|
"nixpkgs",
|
|
|
|
str(nixpkgs_source()),
|
|
|
|
# --store is required to prevent this error:
|
|
|
|
# error: cannot unlink '/nix/store/6xg259477c90a229xwmb53pdfkn6ig3g-default-builder.sh': Operation not permitted
|
|
|
|
"--store",
|
|
|
|
nix_store,
|
|
|
|
]
|
|
|
|
+ flags
|
|
|
|
)
|
|
|
|
return default_flags + flags
|
2023-08-26 08:32:04 +00:00
|
|
|
|
|
|
|
|
2023-07-26 06:00:07 +00:00
|
|
|
def nix_shell(packages: list[str], cmd: list[str]) -> list[str]:
|
2023-08-26 09:23:15 +00:00
|
|
|
# we cannot use nix-shell inside the nix sandbox
|
|
|
|
# in our tests we just make sure we have all the packages
|
|
|
|
if os.environ.get("IN_NIX_SANDBOX"):
|
|
|
|
return cmd
|
|
|
|
wrapped_packages = [f"nixpkgs#{p}" for p in packages]
|
|
|
|
return (
|
2023-09-15 14:22:05 +00:00
|
|
|
nix_command(
|
|
|
|
[
|
|
|
|
"shell",
|
|
|
|
"--inputs-from",
|
|
|
|
f"{str(nixpkgs_flake())}",
|
|
|
|
]
|
|
|
|
)
|
2023-08-26 09:23:15 +00:00
|
|
|
+ wrapped_packages
|
|
|
|
+ ["-c"]
|
|
|
|
+ cmd
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def unfree_nix_shell(packages: list[str], cmd: list[str]) -> list[str]:
|
|
|
|
if os.environ.get("IN_NIX_SANDBOX"):
|
2023-07-26 06:00:07 +00:00
|
|
|
return cmd
|
2023-08-26 09:23:15 +00:00
|
|
|
return (
|
2023-09-15 14:22:05 +00:00
|
|
|
nix_command(
|
|
|
|
[
|
|
|
|
"shell",
|
|
|
|
"-f",
|
|
|
|
str(unfree_nixpkgs()),
|
|
|
|
]
|
|
|
|
)
|
2023-08-26 09:23:15 +00:00
|
|
|
+ packages
|
|
|
|
+ ["-c"]
|
|
|
|
+ cmd
|
|
|
|
)
|