mirror of
https://github.com/fosslinux/live-bootstrap.git
synced 2026-03-02 01:18:08 +01:00
81e3123262
There is nothing temporary about our "tmpdir" - its sole purpose is to contain the final product of the bootstrap process. Thus, removing it at the end of bootstrap amounts to doing the entire process for nothing. To remedy this, --tmpdir is renamed --target, keeping the -t short form, and defaulting to "target" instead of "tmp" to make its purpose clearer. The --preserve option is removed, as the target is now always preserved.
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# SPDX-FileCopyrightText: 2023 fosslinux <fosslinux@aussies.space>
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
"""
|
|
Contains a class that represents a target directory
|
|
"""
|
|
|
|
import enum
|
|
import getpass
|
|
import os
|
|
|
|
from lib.utils import mount, umount, create_disk, run_as_root
|
|
|
|
class TargetType(enum.Enum):
|
|
"""Different types of target dirs we can have"""
|
|
NONE = 0
|
|
TMPFS = 1
|
|
|
|
class Target:
|
|
"""
|
|
Represents a target directory
|
|
"""
|
|
|
|
_disks = {}
|
|
_disk_filesystems = {}
|
|
_mountpoints = {}
|
|
|
|
def __init__(self, path="target"):
|
|
self.path = os.path.abspath(path)
|
|
self._type = TargetType.NONE
|
|
|
|
if not os.path.exists(self.path):
|
|
os.mkdir(self.path)
|
|
|
|
def __del__(self):
|
|
for path in self._mountpoints:
|
|
print(f"Unmounting {path}")
|
|
umount(path)
|
|
|
|
for disk in self._disks.values():
|
|
print(f"Detaching {disk}")
|
|
run_as_root("losetup", "-d", disk)
|
|
|
|
def tmpfs(self, size="8G"):
|
|
"""Mount a tmpfs"""
|
|
print(f"Mounting tmpfs on {self.path}")
|
|
mount("tmpfs", self.path, "tmpfs", f"size={size}")
|
|
self._type = TargetType.TMPFS
|
|
|
|
# pylint: disable=too-many-arguments
|
|
def add_disk(self,
|
|
name,
|
|
size="16G",
|
|
filesystem="ext4",
|
|
tabletype="msdos",
|
|
bootable=False,
|
|
mkfs_args=None):
|
|
"""Add a disk"""
|
|
disk_path = os.path.join(self.path, f"{name}.img")
|
|
self._disks[name] = create_disk(disk_path,
|
|
tabletype,
|
|
filesystem,
|
|
size,
|
|
bootable,
|
|
mkfs_args)
|
|
self._disk_filesystems[name] = filesystem
|
|
# Allow executing user to access it
|
|
run_as_root("chown", getpass.getuser(), self._disks[name])
|
|
|
|
def mount_disk(self, name, mountpoint=None):
|
|
"""Mount the disk"""
|
|
if mountpoint is None:
|
|
mountpoint = f"{name}_mnt"
|
|
mountpoint = os.path.join(self.path, mountpoint)
|
|
os.mkdir(mountpoint)
|
|
mount(self._disks[name] + "p1", mountpoint, self._disk_filesystems[name])
|
|
# Allow executing user to access it
|
|
run_as_root("chown", getpass.getuser(), mountpoint)
|
|
self._mountpoints[name] = mountpoint
|
|
return mountpoint
|
|
|
|
def umount_disk(self, name):
|
|
"""Unmount a disk"""
|
|
umount(self._mountpoints[name])
|
|
del self._mountpoints[name]
|
|
|
|
def get_disk(self, name):
|
|
"""Get the path to a device of a disk"""
|
|
return self._disks.get(name)
|