henrik/dotfiles
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
ff7282ee404337c3299b4eb80b079c94efb317b0
dotfiles/endos.py
T
Henrik Bakken ff7282ee40 endos.py tweaks
2025-11-14 10:40:03 +01:00

500 lines
16 KiB
Python
Executable File
Raw Blame History

#!/usr/bin/env -S uv run --script
# /// script
# requires-python = ">=3.13"
# dependencies = [
# "defopt",
# ]
# ///
# TODO
# fix btop colors
# pre-luks remote ssh
# add
# - sudo ufw allow 22/tcp comment "ssh"
# - sudo ufw default allow FORWARD
# - sudo ufw allow 2222
# - ... etc
# - sudo enable ufw
from difflib import unified_diff
from os.path import lexists
from pathlib import Path
from shutil import rmtree
from subprocess import run
from urllib.request import urlretrieve
HOSTNAME = open("/etc/hostname").read().strip()
HOME_TGT = Path("~").expanduser()
ROOT_TGT = Path("/")
DOTFILES = HOME_TGT / "dotfiles"
HOME_SRC = DOTFILES / "HOME"
ROOT_SRC = DOTFILES / "ROOT"
CFG_TGT = HOME_TGT / ".config"
CFG_SRC = DOTFILES / "CONFIG"
CUSTOM_SRC = DOTFILES / f"custom-{HOSTNAME}"
installmap = dict(
fonts=(
"noto-fonts-emoji",
"ttf-hack",
"font-manager",
),
zsh=(
"zsh",
"zsh-autosuggestions",
"zsh-syntax-highlighting",
"zsh-vi-mode",
"atuin",
"fzf",
"git-delta",
"starship",
"zoxide",
"bat",
),
tmux=("tmux", "urlscan"),
editors=(
"neovim",
"helix",
"ripgrep",
"npm", # required for nvim plugins
"go", # required for vim-hexokinase build
),
utils=("uv", "ncdu", "unzip", "jq", "bluetui"),
gittools=(
"tig",
"diff-so-fancy",
"git-secret",
"git-delta",
"git-lfs",
"lazygit",
"bat",
),
readers=(
"sioyek",
"zathura",
"zathura-pdf-mupdf",
"zathura-djvu",
"zathura-ps",
"typst",
"okular",
"calligra",
"libreoffice-fresh",
),
mediaviewers=(
# video
"vlc",
"mpv",
"protobuf",
"yt-dlp",
"v4l-utils",
# photos
"qimgv",
# music
"quodlibet",
"gst-plugins-good", # required deb
# audio
"pamixer",
"noisetorch",
),
filebrowsers=(
"yazi",
"ranger",
"pcmanfm",
"thunar",
"gparted",
"sshfs",
"file-roller",
"7zip", # explore zip files
"zoxide", # quick search
"eza", # pretty alternative to `ls`
"glow", # markdown renderer
),
netbrowsers=(
"qutebrowser",
"firefox",
"google-chrome",
"microsoft-edge-stable-bin",
"w3m",
"python-adblock", # for qutebrowser ad blocker
"python-tldextract",
"bitwarden-cli", # for qutebrowser autofill
),
chat=("discordo-git", "gurk"),
emailcalrss=(
"khard", # contacts
"khal", # calendar
"aerc", # email
"newsboat", # rss reader
"vdirsyncer", # sync calendar+contacts
"pandoc", # md2html for aerc
"pass", # password manager for aerc and newsboat
"w3m", # terminal browser for aerc and newsboat
"urlscan", # url finder for newsboat
"python-aiohttp-oauthlib", # for google vdirsyncer
),
monitors=(
"htop", # hardware
"btop", # hardware
"nvtop", # gpu
"lazyjournal", # journald
"isd", # systemd
"bandwhich", # network
"sysstat",
),
apps=("keepassxc", "bitwarden", "qalculate-gtk", "vesktop"),
sway=(
# visuals
"wlsunset", # eye saver
"wdisplays", # ui for display settings
"flashfocus", # quick flash when changing app in focus
# clipboard
"wl-clip-persist", # keep clipboard after close
# div
"unipicker", # unicode symbol selector
"blueman", # bluetooth
"wev", # debugging of ui
),
remotedata=("rclone", "dropbox", "minio-client"),
screensharing=("wireplumber",),
nvidia=("cuda", "nvidia-settings"),
coolercontrol=("coolercontrol",),
docker=(
"docker",
"dry-bin", # docker tui
),
k8s=(
"docker-buildx", # advanced build
"qemu-user-static-binfmt", # build arm64
"qemu-user-static", # build arm64
"k9s", # kubernetes tui
"wireguard-tools", # includes wg-quick
"sops", # secret mgmt
"flux-bin",
"open-iscsi", # required by longhorn
"kubectl-cnpg",
),
)
def compare(src: Path, tgt: Path) -> str:
if src.is_dir():
diff = ""
for subsrc in src.glob("**/*"):
subpath = subsrc.relative_to(src) # NOTE: dirty code!
subtgt = tgt / subpath
if subtgt.exists() and lexists(subtgt):
diff += compare(subsrc, subtgt)
return diff
else:
srcdata = open(src).readlines() if src.exists() else []
tgtdata = open(tgt).readlines() if tgt.exists() else []
return "".join(unified_diff(srcdata, tgtdata, str(src), str(tgt)))
def helper_maybe_copy(
src_folder: Path, tgt_folder: Path, sub: str, overwrite: bool, symlink: bool
) -> bool:
src = src_folder.expanduser() / sub
tgt = tgt_folder.expanduser() / sub
assert src.exists()
tgt.parent.mkdir(exist_ok=True)
if lexists(tgt):
if not overwrite:
if diff := compare(src, tgt):
print("DIFF:\n" + diff)
if input("overwrite? (y/n) ") == "n":
return False
if tgt.is_file() or tgt.is_symlink():
tgt.unlink()
else:
rmtree(str(tgt))
if symlink:
tgt.symlink_to(src)
else:
run(["cp", str(src), str(tgt)])
return True
def helper_clone_foldercontents(
src_parent: Path,
tgt_parent: Path,
folder: str,
overwrite: bool,
symlink: bool = True,
) -> None:
src_folder = src_parent.expanduser() / folder
tgt_folder = tgt_parent.expanduser() / folder
tgt_folder.mkdir(exist_ok=True)
assert src_folder.is_dir()
assert tgt_folder.is_dir() or tgt_folder.is_symlink()
for src in src_folder.iterdir():
name = src.name
if name.endswith(".secret"): # git-secret item
continue
helper_maybe_copy(src_folder, tgt_folder, name, overwrite, symlink=symlink)
def helper_check_if_installed(pkg: str) -> bool:
return run(["pacman", "-Q", pkg], capture_output=True).returncode == 0
def helper_uninstall(*pkgs: str) -> None:
for pkg in pkgs:
run(["yay", "-Rns", pkg], capture_output=True)
def helper_install(*pkgs: str, reinstall: bool) -> None:
for pkg in pkgs:
if reinstall or not helper_check_if_installed(pkg):
assert run(["yay", "-S", pkg]).returncode == 0
def install_fonts(reinstall: bool) -> None:
tgt = Path("~/.local/share/fonts/Akkurat-Mono.otf").expanduser()
tgt.parent.mkdir(exist_ok=True)
urlretrieve(
"https://raw.githubusercontent.com/SUNET/static_sunet_se/refs/heads/master/fonts/Akkurat-Mono.otf",
tgt,
)
helper_install(*installmap["fonts"], reinstall=reinstall)
def install_zsh(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["zsh"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "zsh", overwrite)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "atuin", overwrite)
run("chsh -s /usr/bin/zsh".split())
helper_maybe_copy(CFG_SRC, CFG_TGT, "starship.toml", overwrite, symlink=True)
helper_maybe_copy(HOME_SRC, HOME_TGT, ".zshenv", overwrite, symlink=True)
def install_tmux(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["tmux"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "tmux", overwrite)
tpmpath = CFG_TGT / "tmux/plugins/tpm"
if overwrite or not lexists(tpmpath):
if lexists(tpmpath):
if tpmpath.is_symlink():
tpmpath.unlink()
else:
rmtree(str(tpmpath))
run(["git", "clone", "https://github.com/tmux-plugins/tpm", tpmpath])
def install_editors(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["editors"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "nvim", overwrite)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "helix", overwrite)
def install_gittools(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["gittools"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "tig", overwrite)
helper_maybe_copy(HOME_SRC, HOME_TGT, ".gitconfig", overwrite, symlink=True)
def install_readers(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["readers"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "sioyek", overwrite)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "zathura", overwrite)
def install_mediaviewers(reinstall: bool) -> None:
helper_install(*installmap["mediaviewers"], reinstall=reinstall)
def install_filebrowsers(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["filebrowsers"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "ranger", overwrite)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "yazi", overwrite)
for plugin in [
"yazi-rs/plugins:chmod",
"yazi-rs/plugins:git",
"yazi-rs/plugins:mount",
"yazi-rs/plugins:piper",
"yazi-rs/plugins:smart-enter",
"yazi-rs/plugins:smart-filter",
"yazi-rs/plugins:toggle-pane",
"boydaihungst/file-extra-metadata",
]:
run(f"ya pkg add {plugin}".split())
tgt = ".local/share/applications/userapp-file-roller.desktop"
helper_maybe_copy(HOME_SRC, HOME_TGT, tgt, overwrite, symlink=True)
def install_netbrowsers(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["netbrowsers"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "qutebrowser", overwrite)
helper_maybe_copy(HOME_SRC, HOME_TGT, ".w3m/keymap", overwrite, symlink=True)
(HOME_TGT / "Downloads").mkdir(exist_ok=True)
def install_chat(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["chat"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "discordo", overwrite)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "gurk", overwrite)
def install_monitors(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["monitors"], reinstall=reinstall)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "btop", overwrite)
def install_emailcalrss(overwrite: bool, reinstall: bool) -> None:
helper_install(*installmap["emailcalrss"], reinstall=reinstall)
for tgt in ["vdirsyncer", "khard", "khal", "aerc", "newsboat"]:
helper_clone_foldercontents(CFG_SRC, CFG_TGT, tgt, overwrite)
tgt = ".local/share/applications/userapp-khalimport.desktop"
helper_maybe_copy(HOME_SRC, HOME_TGT, tgt, overwrite, symlink=True)
run("systemctl enable --user --now vdirsyncer.timer".split())
run(f"chmod 600 {CFG_TGT / 'aerc/accounts.conf'}".split())
(HOME_TGT / ".cache/newsboat").mkdir(exist_ok=True)
(HOME_TGT / "Calendars").mkdir(exist_ok=True)
(HOME_TGT / "Contacts").mkdir(exist_ok=True)
def install_sway(overwrite: bool, reinstall: bool) -> None:
# sudo
helper_uninstall("autotiling", "cliphist")
helper_install(*installmap["sway"], reinstall=reinstall)
for sub in [
"etc/systemd/logind.conf.d/suspend.conf",
"etc/systemd/sleep.conf.d/hibernate.conf",
"etc/greetd/sway.cfg",
"lib/systemd/system-sleep/iptsd.sh",
]:
src = CUSTOM_SRC / "ROOT" / sub
if not src.exists():
continue
tgt = ROOT_TGT / sub
run(["sudo", "mkdir", "-p", str(tgt.parent)])
if tgt.exists():
if diff := compare(src, tgt):
print("DIFF:\n" + diff)
if input("overwrite? (y/n) ") == "n":
continue
run(["sudo", "rm", str(tgt)])
run(["sudo", "cp", str(src), str(tgt)])
run("sudo systemctl enable --now bluetooth".split())
# user
run("systemctl --user enable --now flashfocus".split())
if (tgt := CFG_TGT / "waybar/config").exists():
tgt.unlink()
for sub in ["sway", "waybar", "gtk-3.0", "mako", "fuzzel", "nwg-drawer"]:
helper_clone_foldercontents(CFG_SRC, CFG_TGT, sub, overwrite)
helper_clone_foldercontents(CUSTOM_SRC / "CONFIG", CFG_TGT, "sway", overwrite)
helper_clone_foldercontents(CUSTOM_SRC / "CONFIG", CFG_TGT, "waybar", overwrite)
helper_maybe_copy(CFG_SRC, CFG_TGT, "mimeapps.list", overwrite, symlink=True)
helper_clone_foldercontents(
HOME_SRC,
HOME_TGT,
".local/share/applications",
overwrite,
)
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "foot", overwrite)
def configure_pytools(overwrite: bool) -> None:
for sub in [".ipython/profile_default", ".jupyter"]:
tgt = HOME_TGT / sub
run(["mkdir", "-p", str(tgt)])
helper_clone_foldercontents(HOME_SRC, HOME_TGT, sub, overwrite)
def install_k8sreqs(overwrite: bool, reinstall: bool) -> None:
# sudo
helper_install(*installmap["k8s"], reinstall=reinstall)
run("sudo systemctl enable --now docker.service".split())
sub = "etc/modules-load.d/br_netfilter.conf"
run(["sudo", "cp", str(ROOT_SRC / sub), str(ROOT_TGT / sub)])
run("sudo systemctl enable --now iscsid".split())
# user
helper_clone_foldercontents(CFG_SRC, CFG_TGT, "k9s", overwrite)
def installer(
*,
overwrite: bool = False,
reinstall: bool = False,
) -> None:
install_fonts(reinstall)
print("installed fonts")
install_zsh(overwrite, reinstall)
print("installed zsh")
install_tmux(overwrite, reinstall)
print("installed tmux")
install_editors(overwrite, reinstall)
print("installed editors")
helper_install(*installmap["utils"], reinstall=reinstall)
print("installed utils")
install_gittools(overwrite, reinstall)
print("installed gittools")
install_readers(overwrite, reinstall)
print("installed readers")
install_mediaviewers(reinstall)
print("installed mediaviewers")
install_filebrowsers(overwrite, reinstall)
print("installed filebrowsers")
install_netbrowsers(overwrite, reinstall)
print("installed netbrowsers")
install_emailcalrss(overwrite, reinstall)
print("installed emailcalrss")
install_chat(overwrite, reinstall)
print("installed chat")
install_monitors(overwrite, reinstall)
print("installed monitors")
helper_install(*installmap["apps"], reinstall=reinstall)
print("installed apps")
install_sway(overwrite, reinstall)
print("installed sway")
helper_install(*installmap["remotedata"], reinstall=reinstall)
print("installed remotedata")
helper_install(*installmap["screensharing"], reinstall=reinstall)
print("installed screensharing")
configure_pytools(overwrite=overwrite)
print("configured python tools")
if HOSTNAME in ["mothership", "droneship"]:
helper_install(*installmap["nvidia"], reinstall=reinstall)
print("installed nvidia")
if HOSTNAME in ["mothership", "droneship"]:
install_k8sreqs(overwrite, reinstall)
print("installed k8s requirements")
if HOSTNAME in ["mothership", "droneship"]:
helper_install(*installmap["coolercontrol"], reinstall=reinstall)
run("sudo systemctl enable --now coolercontrold.service".split())
print("installed coolercontrol")
if HOSTNAME in ["mothership"]:
helper_install(*installmap["docker"], reinstall=reinstall)
helper_clone_foldercontents(
CFG_SRC, CFG_TGT, "homeassistant", overwrite, symlink=False
)
print("installed docker + home assistant")
print("""
MANUAL NEXT STEPS:
# vdirsyncer
- set up with google calendar using
https://vdirsyncer.pimutils.org/en/stable/config.html#google
- run `vdirsyncer discover` then `vdiscover sync`
# firefox
- allow windowed fullscreen by setting full-screen-api.ignore-widgets to
true in about:config
# coolercontrol
- set coolercontrold log level to WARN: `sudo systemctl edit
coolercontrold.service`
# k8s
- configure `/etc/hosts`
- run `hjarl-system/k8s.py` scripts
# docker / home assistant
- docker with non-root daemon
`sudo groupadd docker && sudo usermod -aG docker $USER`
- run home assistant and music assistant scripts to start docker services
""")
if __name__ == "__main__":
import defopt
defopt.run(installer)
Reference in New Issue View Git Blame Copy Permalink