diff system.py @ 1:b664f1027992

system.py port from ansible
author drewp@bigasterisk.com
date Thu, 04 Nov 2021 18:10:02 -0700
parents
children 7f7af7e2ba8d
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/system.py	Thu Nov 04 18:10:02 2021 -0700
@@ -0,0 +1,173 @@
+import os
+from pyinfra import host
+from pyinfra.operations import server, files, apt, ssh, systemd
+from pyinfra.facts.server import LinuxDistribution
+
+is_pi = host.get_fact(LinuxDistribution)['name'] == 'Debian'
+is_wifi_pi = host.name in ['frontdoor', 'living']
+
+TZ = 'America/Los_Angeles'
+
+#
+# system
+#
+
+server.hostname(hostname=host.name)
+files.link(path='/etc/localtime', target=f'/usr/share/zoneinfo/{TZ}')
+files.replace(path='/etc/timezone', match='.*', replace=TZ)
+apt.packages(update=True,
+             cache_time=86400,
+             packages=['tzdata'],
+             force=True,
+             env={
+                 'TZ': TZ,
+                 'LANG': 'en_US.UTF-8',
+                 'DEBIAN_FRONTEND': 'noninteractive'
+             })
+
+#
+# fstab
+#
+
+fstab_file = f'files/{host.name}_fstab'
+if os.path.exists(fstab_file):
+    files.put(src=fstab_file, dest='/etc/fstab')
+if is_pi:
+    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)
+
+#
+# pkgs
+#
+
+if not is_pi:
+    apt.key(keyserver='keyserver.ubuntu.com', keyid='8B48AD6246925553')
+
+if is_pi:
+    apt.packages(packages=['dirmngr', 'gnupg2', 'apt-utils'])
+    files.template(src='templates/pi_sources.list.j2', dest='/etc/apt/sources.list', rel='buster')
+
+    apt.key(src='https://ftp-master.debian.org/keys/archive-key-8.asc')
+    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-9-security.asc')
+
+    files.file(path='/etc/apt/sources.list.d/raspi.list', present=False)
+
+    if is_wifi_pi:
+        files.put(dest="/etc/network/interfaces.d/wlan0", src="files/pi_wlan0_powersave")
+        ssh.command(host.name, "iw wlan0 set power_save off")
+
+    # see https://www.raspberrypi.org/documentation/configuration/config-txt/memory.md#:~:text=txt-,gpu_mem,-Specifies
+    # to port to pyinfra
+    #- name: unused display; give ram to OS
+    #  lineinfile: dest=/boot/config.txt line="gpu_mem=16" regexp="^gpu_mem="
+    #  when: "'with_x11' not in group_names"
+
+    # for beacon
+    #enable_uart=1
+    #dtoverlay=pi3-miniuart-bt
+    #core_freq=250
+
+    # for tiny_screen
+    #to port to pyinfra
+    #- lineinfile: dest=/boot/config.txt line="dtparam=spi=on" regexp="^dtparam=spi="
+
+    # i hope this is deletable
+    # downgrade strictness so I can install from https://archive.raspberrypi.org/
+    # https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=907788
+    #- lineinfile: dest=/etc/ssl/openssl.cnf line="#CipherString = DEFAULT@SECLEVEL=2" regexp="CipherString ?="
+
+    # may be fixed in k3s, not sure
+    # raspbian defaults to `iptables -V` -> iptables v1.8.4 (nf_tables), which won't work with k3s
+    # - command: update-alternatives --set iptables /usr/sbin/iptables-legacy
+if not is_pi:
+    apt.key(src='https://dl.google.com/linux/linux_signing_key.pub')
+    apt.repo(src='deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main')
+
+    apt.key(src='https://packages.microsoft.com/keys/microsoft.asc')
+    apt.repo(src="deb [arch=amd64,arm64,armhf] http://packages.microsoft.com/repos/code stable main")
+
+    apt.ppa(src="ppa:savoury1/blender")
+
+    apt.key(keyserver='keyserver.ubuntu.com', keyid='F24AEA9FB05498B7')
+    apt.repo(src="deb [arch=amd64,i386] https://repo.steampowered.com/steam/ stable steam")
+
+if is_pi:
+    apt.key(src="https://download.docker.com/linux/raspbian/gpg")
+    apt.repo(src="deb [arch=armhf] https://download.docker.com/linux/raspbian stretch stable")
+    apt.repo(src='deb http://deb.debian.org/debian/ unstable main')  # maybe for WG
+
+# don't try to get aufs-dkms on rpi-- https://github.com/docker/for-linux/issues/709
+apt.packages(packages=['docker.io'], no_recommends=True)
+
+apt.packages(packages=[
+    'build-essential',
+    'python3-pip',
+    'python3-virtualenv',
+    'python3-docker',
+    'rsync',
+    'i2c-tools',
+    'sysstat',
+])
+
+#
+# ssh
+#
+
+systemd.service(
+    service='ssh',
+    running=True,
+    enabled=True,
+)
+
+files.line(path='/etc/ssh/ssh_config', line="HashKnownHosts", replace="HashKnownHosts no")
+
+if is_pi:
+    auth_keys = '/home/pi/.ssh/authorized_keys'
+    files.file(path=auth_keys, user='pi', group='pi', mode=600)
+    for pubkey in [
+            'ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBNlR7hereUHqw/RHQau0F7+vQZKAxduM+SD4R76FhC+4Zi078Pv04ZLe9qdM/NBlB/grLGhG58vaGmnWPpJ3QJs= drewp@plus',
+            'ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBOR+iV8Qm/rAfmq0epXYfnp5ZTfBl8eidFzw1GmyZ3fPUFAshWn839fQ5DPj9xDPtMy9kTtrB5bK1SnynFzDqzQ= drewp@bang',
+    ]:
+        files.line(path=auth_keys, line=pubkey, replace=pubkey)
+
+#
+# docker (delete this?)
+#
+
+files.put(src='files/docker_daemon.json', dest='/etc/docker/daemon.json')
+systemd.service(service='docker', running=True, enabled=True, restarted=True)
+
+if not is_pi:
+    files.line(path='/etc/update-manager/release-upgrades', line="^Prompt=", replace="Prompt=normal")
+
+    files.line(path='/etc/ssh/sshd_config', line="^UseDNS\b", replace="UseDNS no")
+    systemd.service(service='sshd', reloaded=True)
+
+#
+# special hosts
+#
+
+if host.name == "bang":
+    apt.packages(packages=[
+        'libzfs2linux',
+        'zfsutils-linux',
+        'zfs-zed',
+        'zfs-auto-snapshot',
+    ])
+
+# This is usable on pi where we don't care when they reboot:
+#- name: apt_upgrade
+#  apt: upgrade=full
+#- name: Check if a reboot is required
+#  register: file
+#  stat: path=/var/run/reboot-required get_md5=no
+#- name: Reboot the server
+#  command: /sbin/reboot
+#  when: file.stat.exists == true