pep-mklive/pylibraries/pepperpy/hooks.py

309 lines
12 KiB
Python

"""
* Author: "PeppermintOS Team(peppermintosteam@proton.me)
*
* License: SPDX-License-Identifier: GPL-3.0-or-later
*
* This is the hook module use dto run for the normal hooks,
* this will run afte rthe ISO main build begins.
"""
import os
import platform
def apt_install_packages():
"""
The place where installing apt packaged is preffered during the
hook run ran ther then the general build
"""
os.system('apt update')
os.system('apt install --yes peppermint-wallpapers')
def remove_packages_from_the_build():
"""
Uninstall unwanted packages from the build
"""
packages = [
'konsole',
'malcontent',
'systemsettings',
'nvidia*',
'imagemagick-6.q16'
]
for package in packages:
os.system(f'apt --purge --yes autoremove {package}')
def remove_desktop_files():
"""
Use this def to remove any .desktop files you donot need
"""
os.system('rm /usr/share/applications/btop.desktop')
os.system('rm /usr/share/applications/librewolf.desktop')
def install_librewolf_browser():
"""
Installs LibreWolf browser for Kumo by setting up sources and
installing the package.
"""
path_to_file = '/opt/pepconf/'
build_id_list = [
path_to_file + 'deb.64xfc',
path_to_file + 'dev.64xfc',
path_to_file + 'deb.armxfc',
path_to_file + 'dev.armxfc',
path_to_file + 'deb.64gfb',
path_to_file + 'dev.64gfb',
path_to_file + 'deb.64opb',
path_to_file + 'dev.64opb'
]
distro_cmd = (
"distro=$(if echo ' una bookworm vanessa focal jammy bullseye vera uma ' | "
"grep -q ' $(lsb_release -sc) '; then lsb_release -sc; else echo bookworm; fi); "
"echo $distro"
)
for build_id in build_id_list:
if os.path.exists(build_id):
distro = os.popen(distro_cmd).read().strip()
content = (
f"Types: deb\n"
f"URIs: https://deb.librewolf.net\n"
f"Suites: {distro}\n"
f"Components: main\n"
f"Architectures: amd64\n"
f"Signed-By: /usr/share/keyrings/librewolf.gpg\n"
)
file_path = '/etc/apt/sources.list.d/librewolf.sources'
os.system('apt update && '
'apt install -y wget gnupg lsb-release apt-transport-https ca-certificates')
os.system('wget -O- https://deb.librewolf.net/keyring.gpg | '
'gpg --dearmor -o /usr/share/keyrings/librewolf.gpg')
try:
with open(file_path, 'w', encoding='utf-8') as f:
f.write(content)
os.chmod(file_path, 0o644)
except PermissionError:
print(f"Permission denied. Please run this script with "
f"appropriate permissions to write to {file_path}.")
return
os.system('apt update')
os.system('apt install librewolf -y')
print("Browser installed!")
else:
print("No builds were met not installed")
def setup_plymouth_theme_grub():
"""
Set the plymouth boot theme and desktop theme, as well as grub
"""
os.system('plymouth-set-default-theme -R joy')
os.system('rm -f /etc/alternatives/desktop-theme')
os.system('rm -f /etc/alternatives/desktop-grub')
theme_source = "/usr/share/desktop-base/joy-theme"
theme_destination = "/etc/alternatives/desktop-theme"
theme_command = f"ln -s {theme_source} {theme_destination}"
os.system(theme_command)
grub_source = (
"/usr/share/desktop-base/active-theme/grub/"
"pep-grub-16x9.png"
)
grub_destination = "/etc/alternatives/desktop-grub"
grub_command = f"ln -s {grub_source} {grub_destination}"
os.system(grub_command)
def hblock_setup():
"""
Install hblock to the system
"""
hb_url = "https://raw.githubusercontent.com/hectorm/hblock/master/hblock"
hb_output = "/tmp/hblock"
hb_command = f"curl {hb_url} --output {hb_output}"
os.system(hb_command)
file_path = "/tmp/hblock"
expected_hash = "4031d86cd04fd7c6cb1b7e9acb1ffdbe9a3f84f693bfb287c68e1f1fa2c14c3b"
sh_command = f'[ "$(sha256sum {file_path})" == "{expected_hash}" ]'
os.system(sh_command)
os.system('mv /tmp/hblock /usr/local/bin/hblock')
os.system('chown 0:0 /usr/local/bin/hblock')
os.system('chmod 755 /usr/local/bin/hblock')
os.system('hblock -S none -D none exit 0')
## this function may not be necessary because the bug is already fixed
## in live-build.
def manage_raspi_firmware():
"""
Manage when the raspi firmware, is removed.
"""
if os.path.exists("/usr/share/peppermint/pep_id"):
os.system("apt remove raspi-firmware -y")
os.system("apt purge raspi-firmware -y")
os.system("apt autoremove -y")
else:
print("not there")
def manage_symlinks():
"""
Symlink management
"""
# Workaround for using snapd on debian. Thank You - @stevesveryown.
os.system('[ ! -e /etc/skel/.local/share ] && mkdir -p /etc/skel/.local/share')
os.system('ln -s snap /usr/bin/snap-store')
# These are for the 3 installer scripts - icons.sh, themes.sh & walls.sh
# In 0610- we set the permissions to 777 on the DIRECTORY, not the contents.
os.system('ln -s icons /usr/share/pepicons')
os.system('ln -s themes /usr/share/pepthemes')
os.system('ln -s backgrounds /usr/share/pepwallpaper')
def set_vi_vim():
"""
Setting the better version of vim
"""
# A better version of vi and vim than vim-tiny
os.system('rm /etc/alternatives/vi')
os.system('ln -s ../../etc/alternatives/vim /usr/bin/vim')
os.system('ln -s ../../usr/bin/vim.tiny /etc/alternatives/vim')
os.system('ln -s ../../usr/bin/busybox /etc/alternatives/vi')
def setting_permissions():
"""
This allows write access to the Welcome Screen database to users
in the "cdrom" group. IF this a system wide file, one user can
disable or turn it off for *ALL* accounts. If this is meant to be
on a per user basis, put the .db in ${HOME}/.config . Or rewrite
welcome.py to write to the .db WHICH user requested to disable
this feature. Set Permissions on desktop files
"""
# Desktop links
os.system('chmod 755 /usr/share/applications/Welcome.desktop')
os.system('chmod 755 /usr/share/applications/plank.desktop')
os.system('chmod 755 /usr/share/applications/kumo.desktop')
os.system('chmod 755 /usr/share/applications/calamares-install-peppermint.desktop')
# Executables
os.system('chmod 755 /usr/local/bin/xDaily')
os.system('chmod 755 /usr/local/bin/kumo')
os.system('chmod 755 /usr/local/bin/welcome')
os.system('chmod 755 /usr/bin/install-peppermint')
os.system('chmod 755 /opt/pepconf/*')
# Set permissions on the Python libs
os.system('chmod 755 -R /usr/lib/python3/dist-packages/tendo')
os.system('chmod 755 -R /usr/lib/python3/dist-packages/tendo-0.3.0.dist-info')
os.system('chmod 755 -R /usr/lib/python3/dist-packages/ttkbootstrap')
os.system('chmod 755 -R /usr/lib/python3/dist-packages/ttkbootstrap-1.10.1.dist-info')
os.system('chmod 755 -R /usr/lib/python3/dist-packages/ttkcreator')
# Set permissions on the lines theme
os.system('chmod 755 -R /usr/share/desktop-base/lines-theme')
# Set Grub Themes Permission
os.system('chmod 755 -R /boot/grub/themes')
# Set the Calamares Permissions]
os.system('chmod 755 -R /etc/calamares')
### After installation, additional groups to add new users to.
os.system('grep -B99 "#EXTRA_GROUPS=" /etc/adduser.conf' +
'> /etc/adduser.conf.new')
os.system('grep "#EXTRA_GROUPS=" /etc/adduser.conf' +
' | cut -c2-' +
'>> /etc/adduser.conf.new'
)
os.system('grep -B3 "#ADD_EXTRA_GROUPS=" /etc/adduser.conf >> /etc/adduser.conf.new')
os.system('grep "#ADD_EXTRA_GROUPS=" /etc/adduser.conf | cut -c2- >> /etc/adduser.conf.new')
os.system('grep -B3 "#NAME_REGEX=" /etc/adduser.conf >> /etc/adduser.conf.new')
# These are for the 3 installer scripts - icons.sh, themes.sh & walls.sh
# In 0600- we created symlinks to these directories. This makes them writable.
os.system('chmod 755 -R /usr/share/icons')
os.system('chmod 755 -R /usr/share/themes')
os.system('chmod 755 -R /usr/share/backgrounds')
os.system('chmod 755 -R /usr/share/pixmaps')
def final_cleanup():
"""
This section is the "lint-trap" to remove files and/or
directories not associated with or required by PepOS.
Followed by "\" , add files to be removed, one per line.
"""
files_to_remove = [ "/usr/bin/install-debian",
"/usr/share/applications/min.desktop",
"/usr/share/applications/calamares-install-debian.desktop"
]
for file_path in files_to_remove:
if os.path.exists(file_path):
os.system(f"rm -rf {file_path}")
print(f" Removed: {file_path}")
else:
print(f" The path {file_path} was not found and couldn't be removed.")
### This might be better in an OS-tweaks hook script.
# Lowers the footprint in RAM by 200 MB at the small expense of added size to the ISO.
os.system('update-icon-caches /usr/share/icons/*')
### Setting --apt-recommends and --apt-suggests defaults to '0'
# Create and write to the file
os.system('echo "# Changing these values to \\"1\\" may quickly fill up a small partition" > /etc/apt/apt.conf.d/99No-Recommends')
# Append additional content to the file
os.system('echo "APT::Install-Recommends \\"0\\";\nAPT::Install-Suggests \\"0\\";" >> /etc/apt/apt.conf.d/99No-Recommends')
### Fixes the "Not installing grub for UEFI Secure Boot" in all versions , after the default was changed.
#sed s/keyutils/"keyutils --install-recommends"/ /usr/sbin/bootloader-config > /tmp/bootloader-config
os.system('echo -e "$(grep -A1 -B20 "Installing grub-efi (uefi)..." /usr/sbin/bootloader-config) --install-recommends" > /tmp/bootloader-config')
os.system('echo -e "$(grep -A2 "else" /usr/sbin/bootloader-config) --install-recommends\nfi" >> /tmp/bootloader-config')
os.system('chmod +x /tmp/bootloader-config && mv /tmp/bootloader-config /usr/sbin/bootloader-confi')
def rename_kernel():
"""
These commands are typically used in the context of updating the
initial ramdisk image and the kernel image for a live Linux system.
The steps can be summarized as follows:
Removal of the current boot files: initrd.img and vmlinuz are
removed to clear out the old versions.
Copying the updated versions: New versions of these files,
which match the patterns initrd.img-* and vmlinuz-*, are copied
to the standard names initrd.img and vmlinuz.
"""
os.system('rm live/initrd.img')
os.system('rm live/vmlinuz')
os.system('cp live/initrd.img-* live/initrd.img')
os.system('cp live/vmlinuz-* live/vmlinuz')
def start():
""" This will begin the execution process of the hooks
First see what arch is being ran and run the correct
functions
"""
architecture = platform.architecture()[0]
if '64bit' in architecture:
apt_install_packages()
remove_packages_from_the_build()
install_librewolf_browser()
remove_desktop_files()
setup_plymouth_theme_grub()
hblock_setup()
manage_raspi_firmware()
manage_symlinks()
set_vi_vim()
setting_permissions()
final_cleanup()
rename_kernel()
else:
apt_install_packages()
remove_desktop_files()
remove_packages_from_the_build()
setup_plymouth_theme_grub()
hblock_setup()
manage_raspi_firmware()
manage_symlinks()
set_vi_vim()
setting_permissions()
final_cleanup()
rename_kernel()
start()