pyrox.dev
1#! @python@/bin/python
2import argparse
3import shutil
4import os
5import errno
6import subprocess
7import glob
8import tempfile
9import errno
10import warnings
11
12def copy_if_not_exists(source, dest):
13 if not os.path.exists(dest):
14 shutil.copyfile(source, dest)
15
16system_dir = lambda generation: "/nix/var/nix/profiles/system-%d-link" % (generation)
17
18def write_entry(generation, kernel, initrd):
19 entry_file = "@efiSysMountPoint@/loader/entries/nixos-generation-%d.conf" % (generation)
20 generation_dir = os.readlink(system_dir(generation))
21 tmp_path = "%s.tmp" % (entry_file)
22 kernel_params = "systemConfig=%s init=%s/init " % (generation_dir, generation_dir)
23 with open("%s/kernel-params" % (generation_dir)) as params_file:
24 kernel_params = kernel_params + params_file.read()
25 with open(tmp_path, 'w') as f:
26 print >> f, "title NixOS"
27 print >> f, "version Generation %d" % (generation)
28 if machine_id is not None: print >> f, "machine-id %s" % (machine_id)
29 print >> f, "linux %s" % (kernel)
30 print >> f, "initrd %s" % (initrd)
31 print >> f, "options %s" % (kernel_params)
32 os.rename(tmp_path, entry_file)
33
34def write_loader_conf(generation):
35 with open("@efiSysMountPoint@/loader/loader.conf.tmp", 'w') as f:
36 if "@timeout@" != "":
37 print >> f, "timeout @timeout@"
38 print >> f, "default nixos-generation-%d" % (generation)
39 os.rename("@efiSysMountPoint@/loader/loader.conf.tmp", "@efiSysMountPoint@/loader/loader.conf")
40
41def copy_from_profile(generation, name, dry_run=False):
42 store_file_path = os.readlink("%s/%s" % (system_dir(generation), name))
43 suffix = os.path.basename(store_file_path)
44 store_dir = os.path.basename(os.path.dirname(store_file_path))
45 efi_file_path = "/efi/nixos/%s-%s.efi" % (store_dir, suffix)
46 if not dry_run:
47 copy_if_not_exists(store_file_path, "@efiSysMountPoint@%s" % (efi_file_path))
48 return efi_file_path
49
50def add_entry(generation):
51 efi_kernel_path = copy_from_profile(generation, "kernel")
52 efi_initrd_path = copy_from_profile(generation, "initrd")
53 write_entry(generation, efi_kernel_path, efi_initrd_path)
54
55def mkdir_p(path):
56 try:
57 os.makedirs(path)
58 except OSError as e:
59 if e.errno != errno.EEXIST or not os.path.isdir(path):
60 raise
61
62def get_generations(profile):
63 gen_list = subprocess.check_output([
64 "@nix@/bin/nix-env",
65 "--list-generations",
66 "-p",
67 "/nix/var/nix/profiles/%s" % (profile),
68 "--option", "build-users-group", ""
69 ])
70 gen_lines = gen_list.split('\n')
71 gen_lines.pop()
72 return [ int(line.split()[0]) for line in gen_lines ]
73
74def remove_old_entries(gens):
75 slice_start = len("@efiSysMountPoint@/loader/entries/nixos-generation-")
76 slice_end = -1 * len(".conf")
77 known_paths = []
78 for gen in gens:
79 known_paths.append(copy_from_profile(gen, "kernel", True))
80 known_paths.append(copy_from_profile(gen, "initrd", True))
81 for path in glob.iglob("@efiSysMountPoint@/loader/entries/nixos-generation-[1-9]*.conf"):
82 try:
83 gen = int(path[slice_start:slice_end])
84 if not gen in gens:
85 os.unlink(path)
86 except ValueError:
87 pass
88 for path in glob.iglob("@efiSysMountPoint@/efi/nixos/*"):
89 if not path in known_paths:
90 os.unlink(path)
91
92parser = argparse.ArgumentParser(description='Update NixOS-related systemd-boot files')
93parser.add_argument('default_config', metavar='DEFAULT-CONFIG', help='The default NixOS config to boot')
94args = parser.parse_args()
95
96if os.getenv("NIXOS_INSTALL_GRUB") == "1":
97 warnings.warn("NIXOS_INSTALL_GRUB env var deprecated, use NIXOS_INSTALL_BOOTLOADER", DeprecationWarning)
98 os.environ["NIXOS_INSTALL_BOOTLOADER"] = "1"
99
100if os.getenv("NIXOS_INSTALL_BOOTLOADER") == "1":
101 if "@canTouchEfiVariables@" == "1":
102 subprocess.check_call(["@systemd@/bin/bootctl", "--path=@efiSysMountPoint@", "install"])
103 else:
104 subprocess.check_call(["@systemd@/bin/bootctl", "--path=@efiSysMountPoint@", "--no-variables", "install"])
105
106mkdir_p("@efiSysMountPoint@/efi/nixos")
107mkdir_p("@efiSysMountPoint@/loader/entries")
108try:
109 with open("/etc/machine-id") as machine_file:
110 machine_id = machine_file.readlines()[0]
111except IOError as e:
112 if e.errno != errno.ENOENT:
113 raise
114 machine_id = None
115
116gens = get_generations("system")
117remove_old_entries(gens)
118for gen in gens:
119 add_entry(gen)
120 if os.readlink(system_dir(gen)) == args.default_config:
121 write_loader_conf(gen)