import os

from pyinfra import host
from pyinfra.facts.server import LinuxDistribution
from pyinfra.facts.files import FindFiles
from pyinfra.facts.server import Arch
from pyinfra.operations import apt, files, server, systemd

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

TZ = 'America/Los_Angeles'

server.hostname(hostname=host.name)

def timezone():
    files.link(path='/etc/localtime', target=f'/usr/share/zoneinfo/{TZ}')
    files.replace(path='/etc/timezone', text='.*', replace=TZ)
    
def apt_sources():
    if host.get_fact(Arch) == 'x86_64':
        server.shell(commands=['dpkg --add-architecture i386'])

    files.template(src='templates/sources.list.j2', dest='/etc/apt/sources.list.j2')
    if host.get_fact(FindFiles, '/etc/apt/sources.list.d/', quote_path=True):
        raise SystemExit(f"new files in {host.name} /etc/apt/sources.list.d/ - please remove")
    apt.packages(update=True,
                cache_time=86400,
                packages=['tzdata'],
                force=True,
                _env={
                    'TZ': TZ,
                    'LANG': 'en_US.UTF-8',
                    'DEBIAN_FRONTEND': 'noninteractive'
                })

def fstab():
    fstab_file = f'files/fstab/{host.name}'
    if os.path.exists(fstab_file):
        files.put(src=fstab_file, dest='/etc/fstab')

def pi_tmpfs():
    for line in [
            'tmpfs /var/log tmpfs defaults,noatime,mode=0755 0 0',
            'tmpfs /tmp tmpfs defaults,noatime 0 0',
    ]:
        files.line(path="/etc/fstab", line=line, replace=line)

    # stop SD card corruption (along with some mounts in fstab)
    apt.packages(packages=['dphys-swapfile'], present=False)


# don't try to get aufs-dkms on rpi-- https://github.com/docker/for-linux/issues/709
def podman_inecure_registry():
    files.template(src='templates/kube/podman_registries.conf.j2', dest='/etc/containers/registries.conf.d/bang.conf')


def no_sleep():
    server.shell(commands=['systemctl mask sleep.target suspend.target hibernate.target hybrid-sleep.target'])

def nfs_server():
    apt.packages(packages=['nfs-kernel-server'])
    files.template(src='templates/bang_exports.j2', dest='/etc/exports')

    # sudo zfs set sharenfs="rw=10.5.0.0/16" stor6

def smaller_journals():
    files.line(name='shorter systemctl log window, for disk space',
               path='/etc/systemd/journald.conf',
               line='MaxFileSec',
               replace="MaxFileSec=7day")

    for port in [80, 443]:
        files.template(src="templates/webforward.service.j2", dest=f"/etc/systemd/system/web_forward_{port}.service", port=port)
        systemd.service(service=f'web_forward_{port}', enabled=True, restarted=True)

def zfs_metrics():
    files.put(src='files/zfs_metrics/zfs_space_metrics.sh', dest='/opt/zfs_metrics/zfs_space_metrics.sh')
    files.put(src='files/zfs_metrics/zfs.mtail', dest='/opt/zfs_metrics/zfs.mtail')
    files.put(src='files/zfs_metrics/zfs_space_metrics.service',
               dest=f'/etc/systemd/system/zfs_space_metrics.service')
    systemd.service(service=f'zfs_space_metrics', enabled=True, restarted=True, daemon_reload=True)

timezone()
apt_sources()
fstab()

if not is_pi:
    files.line(path='/etc/update-manager/release-upgrades', line="^Prompt=", replace="Prompt=normal")

if is_pi and host.name != 'pipe':
    pi_tmpfs()
    files.template(src='templates/boot_config.txt.j2', dest='/boot/config.txt')

if not is_pi:    
    podman_inecure_registry()

if host.name in ['bang', 'pipe']:
    no_sleep()

if host.name == 'bang':
    nfs_server()
    zfs_metrics()

if host.name == 'prime':
    smaller_journals()