"""
* 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()