From 57eb219269effac2f86da34c425df19b36309b5e Mon Sep 17 00:00:00 2001
From: Jeffrey Serio <hyperreal@moonshadow.dev>
Date: Sat, 9 Nov 2024 14:38:57 -0600
Subject: [PATCH] Use age

---
 sync_to_remotes.py | 52 ++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 46 insertions(+), 6 deletions(-)

diff --git a/sync_to_remotes.py b/sync_to_remotes.py
index c4f6a8f..1273cbc 100755
--- a/sync_to_remotes.py
+++ b/sync_to_remotes.py
@@ -4,7 +4,7 @@
 # ]
 # ///
 
-
+import os
 import socket
 import subprocess
 from pathlib import Path
@@ -12,8 +12,9 @@ from pathlib import Path
 import resend
 
 
-def send_email(program: str, log: str):
+def send_email(program: str, log: str, **kwargs: BaseException):
     resend.api_key = Path("/usr/local/etc/resend_api_key.txt").read_text().strip("\n")
+    err_msg = kwargs.get("err_msg", None)
 
     match log:
         case "ok":
@@ -21,7 +22,7 @@ def send_email(program: str, log: str):
             msg = f"{program} on {socket.getfqdn()} ran successfully!"
         case "err":
             subj = f"[{socket.getfqdn()}] {program} Error ❌"
-            msg = f"There was an error running {program} on {socket.getfqdn()}. Please investigate."
+            msg = f"There was an error running {program} on {socket.getfqdn()}: {err_msg}.\n\nPlease investigate."
         case _:
             subj = ""
             msg = ""
@@ -50,8 +51,41 @@ def run_kbackup():
         print(f"kbackup ran successfully")
     except subprocess.CalledProcessError as e:
         print(f"Error running kbackup: {e}")
-        send_email("sync_to_remotes", "err")
+        send_email("sync_to_remotes", "err", err_msg=e)
         exit(1)
+    except KeyboardInterrupt as e:
+        print(f"Run of kbackup interrupted: {e}")
+        send_email("sync_to_remotes", "err", err_msg=e)
+        exit(130)
+
+
+def encrypt_tar():
+    tar_dir = Path("/home/jas/sync_to_remotes")
+    for tarfile in tar_dir.iterdir():
+        if tarfile.name.endswith("tar"):
+            age_cmd = [
+                "age",
+                "-e",
+                "-i",
+                "/home/jas/documents/age-key.txt",
+                "-o",
+                f"{tarfile}.age",
+                tarfile,
+            ]
+
+            print(f"Encrypting {tarfile}...")
+            try:
+                subprocess.run(age_cmd, check=True)
+                print(f"Successfully encrypted {tarfile}")
+                os.remove(tarfile)
+            except subprocess.CalledProcessError as e:
+                print(f"Error during archive encryption of {tarfile}: {e}")
+                send_email("sync_to_remotes", "err", err_msg=e)
+                exit(1)
+            except KeyboardInterrupt as e:
+                print(f"Archive encryption of {tarfile} interrupted: {e}")
+                send_email("sync_to_remotes", "err", err_msg=e)
+                exit(130)
 
 
 def sync_to_remotes(src: str, dest: str):
@@ -62,17 +96,23 @@ def sync_to_remotes(src: str, dest: str):
         print(f"Successful sync to {dest}")
     except subprocess.CalledProcessError as e:
         print(f"Error during sync to {dest}: {e}")
-        send_email("sync_to_remotes", "err")
+        send_email("sync_to_remotes", "err", err_msg=e)
         exit(1)
+    except KeyboardInterrupt as e:
+        print(f"Sync to {dest} interrupted: {e}")
+        send_email("sync_to_remotes", "err", err_msg=e)
+        exit(130)
 
 
 if __name__ == "__main__":
     remotes = [
-        ("/home/jas/sync_to_remotes/", "wasabi:/desktop-backups-rep/"),
+        ("/home/jas/sync_to_remotes/", "wasabi:/desktop-backups-moonshadow/"),
     ]
 
     run_kbackup()
 
+    encrypt_tar()
+
     for remote in remotes:
         sync_to_remotes(remote[0], remote[1])