from pyinfra import host
from pyinfra.facts.server import LinuxDistribution
from pyinfra.operations import apt, files, server
import package_lists

is_pi = host.get_fact(LinuxDistribution)['name'] in ['Debian', 'Raspbian GNU/Linux']


def kitty():
    vers = '0.26.5'  # see https://github.com/kovidgoyal/kitty/releases
    home = '/home/drewp'
    local = f"{home}/.local/kitty"
    dl = f'/tmp/kitty-{vers}-x86_64.txz'
    files.download(src=f"https://github.com/kovidgoyal/kitty/releases/download/v{vers}/kitty-{vers}-x86_64.txz", dest=dl)
    files.directory(local)
    server.shell([
        f"mkdir -p {local}",  # https://github.com/Fizzadar/pyinfra/issues/777
        f"aunpack --extract-to={local} {dl}",
    ])
    files.link(target="{local}/bin/kitty", path="{home}/bin/kitty")


def pnpm():
    server.shell([
        # https://github.com/pnpm/pnpm/releases
        # but also https://pnpm.io/installation#compatibility
        "npm install -g pnpm@6.35.1",
    ])


def proper_locate():
    apt.packages(packages='mlocate', present=False)
    if not is_pi and host.name not in ['prime', 'pipe']:
        apt.packages(packages='plocate')


def proper_man():
    if host.name in ['pipe', 'prime'] or is_pi:
        apt.packages(packages=['mandb'], present=False)


def pkg_keys():
    # apt.key(keyserver='keyserver.ubuntu.com', keyid='04EE7237B7D453EC')
    # apt.key(keyserver='keyserver.ubuntu.com', keyid='648ACFD622F3D138')
    # apt.key(keyserver='keyserver.ubuntu.com', keyid='8B48AD6246925553')
    # apt.key(keyserver='keyserver.ubuntu.com', keyid='F24AEA9FB05498B7')
    apt.key(src='https://dl.google.com/linux/linux_signing_key.pub')
    apt.key(src='https://ftp-master.debian.org/keys/archive-key-8-security.asc')
    apt.key(src='https://ftp-master.debian.org/keys/archive-key-8.asc')
    apt.key(src='https://ftp-master.debian.org/keys/archive-key-9-security.asc')
    apt.key(src='https://packages.microsoft.com/keys/microsoft.asc')


pkg_keys()

apt.packages(packages=package_lists.setup)

if not is_pi:
    if host.name != 'pipe':
        apt.packages(packages=['reptyr'])
    kitty()
    pnpm()
else:
    apt.packages(packages=package_lists.pi_setup)

proper_locate()
proper_man()

apt.packages(packages=package_lists.general)
apt.packages(packages=package_lists.debug)

if host.name == "bang":
    apt.packages(packages=package_lists.for_bang)

if host.name == "pipe":
    apt.packages(packages=package_lists.for_pipe)

if host.name == "prime":
    apt.packages(packages=package_lists.for_prime)

if host.name == 'plus':
    apt.packages(packages=package_lists.laptop)

desktop_env = host.name in ['dash', 'slash', 'plus']
if desktop_env:
    apt.packages(packages=package_lists.xorg + package_lists.desktop)

if not is_pi:
    apt.packages(packages=package_lists.non_pi)