Commit a094b5d8 authored by programmerlexi's avatar programmerlexi Committed by Masum Reza
Browse files

nixos/limine: carefully remove files instead of nuking them

parent 2c7659b1

nixos/modules/system/boot/loader/limine/limine-install.py

+16 −13
Original line number Diff line number Diff line
@@ -74,6 +74,7 @@ def is_encrypted(device: str) -> bool: 
def is_fs_type_supported(fs_type: str) -> bool:

    return fs_type.startswith('vfat')



paths = {}



def get_copied_path_uri(path: str, target: str) -> str:

    result = ''
@@ -85,6 +86,8 @@ def get_copied_path_uri(path: str, target: str) -> str: 


    if not os.path.exists(dest_path):

        copy_file(path, dest_path)

    else:

        paths[dest_path] = True



    path_with_prefix = os.path.join('/limine', target, dest_file)

    result = f'boot():{path_with_prefix}'
@@ -206,6 +209,8 @@ def copy_file(from_path: str, to_path: str): 
    shutil.copyfile(from_path, to_path + ".tmp")

    os.rename(to_path + ".tmp", to_path)



    paths[to_path] = True



def option_from_config(name: str, config_path: List[str], conversion: Callable[[str], str] | None = None) -> str:

    if config(*config_path):

        return f'{name}: {conversion(config(*config_path)) if conversion else config(*config_path)}\n'
@@ -246,12 +251,10 @@ def main(): 


    if not os.path.exists(limine_dir):

        os.makedirs(limine_dir)



    if os.path.exists(os.path.join(limine_dir, 'kernels')):

        print(f'nuking {os.path.join(limine_dir, "kernels")}')

        shutil.rmtree(os.path.join(limine_dir, 'kernels'))



    os.makedirs(os.path.join(limine_dir, "kernels"))

    else:

        for dir, dirs, files in os.walk(limine_dir, topdown=True):

            for file in files:

                paths[os.path.join(dir, file)] = False



    profiles = [('system', get_gens())]
@@ -271,13 +274,6 @@ def main(): 
        default_entry: 2

    ''')



    if os.path.exists(os.path.join(limine_dir, 'wallpapers')):

        print(f'nuking {os.path.join(limine_dir, "wallpapers")}')

        shutil.rmtree(os.path.join(limine_dir, 'wallpapers'))



    if len(config('style', 'wallpapers')) > 0:

        os.makedirs(os.path.join(limine_dir, 'wallpapers'))



    for wallpaper in config('style', 'wallpapers'):

        config_file += f'''wallpaper: {get_copied_path_uri(wallpaper, 'wallpapers')}\n'''
@@ -319,6 +315,8 @@ def main(): 
        file.truncate()

        file.write(config_file.strip())



    paths[config_file_path] = True



    for dest_path, source_path in config('additionalFiles').items():

        dest_path = os.path.join(limine_dir, dest_path)
@@ -410,4 +408,9 @@ def main(): 
                'Failed to deploy BIOS stage 1 Limine bootloader!\n' +

                'You might want to try enabling the `boot.loader.limine.forceMbr` option.')



    print("removing unused boot files...")

    for path in paths:

        if not paths[path]:

            os.remove(path)



main()