From 0a5754541bb01e96021ca7ee74f1256a8ee68bc4 Mon Sep 17 00:00:00 2001
From: tdback <tyler@tdback.net>
Date: Sat, 21 Dec 2024 15:32:13 -0500
Subject: initial commit to self-hosted git

---
 modules/scripts/motd/default.nix     | 99 ++++++++++++++++++++++++++++++++++++
 modules/scripts/pushover/default.nix | 47 +++++++++++++++++
 2 files changed, 146 insertions(+)
 create mode 100644 modules/scripts/motd/default.nix
 create mode 100644 modules/scripts/pushover/default.nix

(limited to 'modules/scripts')

diff --git a/modules/scripts/motd/default.nix b/modules/scripts/motd/default.nix
new file mode 100644
index 0000000..d59f787
--- /dev/null
+++ b/modules/scripts/motd/default.nix
@@ -0,0 +1,99 @@
+{ config, lib, pkgs, ... }:
+let
+  motd =
+    pkgs.writeShellScriptBin "motd" ''
+      #!/usr/bin/env bash
+      RED="\e[31m"
+      GREEN="\e[32m"
+      YELLOW="\e[33m"
+      BOLD="\e[1m"
+      ENDCOLOR="\e[0m"
+
+      case "$(date +'%H')" in
+        [0-9]|1[0-1])
+          TIME="morning"
+          ;;
+        1[2-7])
+          TIME="afternoon"
+          ;;
+        *)
+          TIME="evening"
+          ;;
+      esac
+
+      UPTIME=$(cat /proc/uptime | cut -f1 -d.)
+      UPDAYS=$((UPTIME/60/60/24))
+      UPHOURS=$((UPTIME/60/60%24))
+      UPMINS=$((UPTIME/60%60))
+      UPSECS=$((UPTIME%60))
+
+      MEMORY=$(free -m | awk 'NR == 2 { printf "%s/%sMB (%.2f%%)\n", $3, $2, ($3 * 100) / $2 }')
+
+      SERVICES=$(systemctl list-units | grep -P 'podman-|${lib.strings.concatStringsSep "|" config.motd.servicesToCheck}')
+
+      printf "\n"
+      printf "''${BOLD}Good $TIME $(whoami), welcome to $(hostname)!$ENDCOLOR\n"
+      printf "\n"
+      ${lib.strings.concatStrings (lib.lists.forEach config.motd.networkInterfaces (x:
+        "printf \"$BOLD  * %-20s$ENDCOLOR %s\\n\" \"IPv4 ${x}\" \"$(ip -4 addr show ${x} | grep -oP '(?<=inet\\s)\\d+(\\.\\d+){3}')\"\n"
+      ))}
+      printf "$BOLD  * %-20s$ENDCOLOR %s\n" "Release" "$(awk -F= '/PRETTY_NAME/ { print $2 }' /etc/os-release | tr -d '"')"
+      printf "$BOLD  * %-20s$ENDCOLOR %s\n" "Kernel" "$(uname -rs)"
+      printf "\n"
+      printf "$BOLD  * %-20s$ENDCOLOR %s\n" "CPU Usage" "$(awk '{ print $1 ", " $2 ", " $3 }' /proc/loadavg) (1, 5, 15 min)"
+      printf "$BOLD  * %-20s$ENDCOLOR %s\n" "Memory" "$MEMORY"
+      printf "$BOLD  * %-20s$ENDCOLOR %s\n" "System Uptime" "$UPDAYS days $UPHOURS hours $UPMINS minutes $UPSECS seconds"
+      printf "\n"
+
+      [ -z "$SERVICES" ] && exit
+
+      printf "''${BOLD}Service status:$ENDCOLOR\n"
+      while IFS= read -r line; do
+        if [[ ! $line =~ ".service" ]] || [[ $line =~ ".mount" ]]; then
+          continue
+        fi
+        if echo "$line" | grep -q 'failed'; then
+          name=$(echo "$line" | awk '{ print $1 }' | sed 's/podman-//g')
+          printf "$RED• $ENDCOLOR%-50s $RED[failed]$ENDCOLOR\n" "$name"
+        elif echo "$line" | grep -q 'running'; then
+          name=$(echo "$line" | awk '{ print $1 }' | sed 's/podman-//g')
+          printf "$GREEN• $ENDCOLOR%-50s $GREEN[active]$ENDCOLOR\n" "$name"
+        elif echo "$line" | grep -q 'exited'; then
+          name=$(echo "$line" | awk '{ print $1 }' | sed 's/podman-//g')
+          printf "$YELLOW• $ENDCOLOR%-50s $YELLOW[exited]$ENDCOLOR\n" "$name"
+        else
+          echo "service status unknown"
+        fi
+      done <<< "$SERVICES"
+      printf "\n"
+    '';
+in
+{
+  options.motd = {
+    networkInterfaces = lib.mkOption {
+      description = "Network interfaces to monitor.";
+      type = lib.types.listOf lib.types.str;
+      default = [ ];
+    };
+
+    servicesToCheck = lib.mkOption {
+      description = "Services to validate alongside podman containers.";
+      type = lib.types.listOf lib.types.str;
+      default = [ ];
+    };
+  };
+
+  config = {
+    environment.systemPackages = [
+      motd
+    ];
+
+    programs.bash.loginShellInit = ''
+      [ -z "$PS1" ] && return
+
+      if command -v motd &> /dev/null; then
+        motd
+      fi
+    '';
+  };
+}
diff --git a/modules/scripts/pushover/default.nix b/modules/scripts/pushover/default.nix
new file mode 100644
index 0000000..f20a5be
--- /dev/null
+++ b/modules/scripts/pushover/default.nix
@@ -0,0 +1,47 @@
+{ inputs, config, pkgs, ... }:
+let
+  pushover =
+    pkgs.writeShellScriptBin "pushover" ''
+      #!/bin/sh
+
+      die() { echo "$0: $*" >&2; exit 111; }
+
+      APP=$(cat ${config.age.secrets.pushoverAppToken.path})
+      USER=$(cat ${config.age.secrets.pushoverUserToken.path})
+
+      while getopts ":t:" args; do
+        case "$args" in
+          t)
+            TITLE="$OPTARG"
+            ;;
+          :)
+            die "missing option argument for -$OPTARG"
+            ;;
+          *)
+            die "invalid option -$OPTARG"
+            ;;
+        esac
+      done
+      shift $((OPTIND - 1))
+
+      MESSAGE="$*"
+      if [ -z "$MESSAGE" ] || [ "$MESSAGE" = " " ]; then
+        MESSAGE="No errors to report."
+      fi
+
+      /run/current-system/sw/bin/curl -s \
+        --form-string "token=$APP" \
+        --form-string "user=$USER" \
+        --form-string "title=$TITLE" \
+        --form-string "message=$MESSAGE" \
+        https://api.pushover.net/1/messages.json
+    '';
+in
+{
+  age.secrets = {
+    pushoverAppToken.file = "${inputs.self}/secrets/pushoverAppToken.age";
+    pushoverUserToken.file = "${inputs.self}/secrets/pushoverUserToken.age";
+  };
+
+  environment.systemPackages = [ pushover ];
+}
-- 
cgit v1.2.3