80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
import argparse
|
|
import json
|
|
import os
|
|
import pathlib
|
|
import subprocess
|
|
import sys
|
|
|
|
selected_shell_file = pathlib.Path(".direnv/selected-shell")
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="Select a devshell")
|
|
parser.add_argument("shell", help="the name of the devshell to select", nargs="?")
|
|
parser.add_argument("--list", action="store_true", help="list available devshells")
|
|
parser.add_argument(
|
|
"--show", action="store_true", help="show the currently selected devshell"
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def get_current_shell() -> str | None:
|
|
if selected_shell_file.exists():
|
|
with open(selected_shell_file) as f:
|
|
return f.read().strip()
|
|
return None
|
|
|
|
|
|
def print_current_shell() -> None:
|
|
current_shell = get_current_shell()
|
|
if current_shell:
|
|
print(f"Currently selected devshell: {current_shell}")
|
|
else:
|
|
print("No devshell selected")
|
|
|
|
|
|
def print_devshells() -> None:
|
|
project_root = os.environ.get("PRJ_ROOT")
|
|
flake_show = subprocess.run(
|
|
[
|
|
"nix",
|
|
"eval",
|
|
"--json",
|
|
"--no-write-lock-file",
|
|
"--apply",
|
|
"shells: builtins.mapAttrs (name: _shell: name) shells",
|
|
f"{project_root}#devShells.x86_64-linux",
|
|
],
|
|
stdout=subprocess.PIPE,
|
|
)
|
|
names = json.loads(flake_show.stdout.decode())
|
|
print("Available devshells:\n")
|
|
print("\n".join(names))
|
|
|
|
|
|
def select_shell(shell: str) -> None:
|
|
if shell == get_current_shell():
|
|
print(f"{shell} devshell already selected. No changes made.")
|
|
sys.exit(0)
|
|
else:
|
|
with open(selected_shell_file, "w") as f:
|
|
f.write(shell)
|
|
print(f"{shell} devshell selected")
|
|
|
|
|
|
def main() -> None:
|
|
args = parse_args()
|
|
if args.show:
|
|
print_current_shell()
|
|
elif args.list:
|
|
print_devshells()
|
|
elif args.shell:
|
|
select_shell(args.shell)
|
|
else:
|
|
print_current_shell()
|
|
print("Use --help for more information")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|