From f7af23ec3f6b0c31028327b08240bd5ed95b8a98 Mon Sep 17 00:00:00 2001
From: kaylee <kaylee@32bit.cafe>
Date: Fri, 2 Feb 2024 08:55:12 +0000
Subject: [PATCH] added check function

---
 uvgotmail.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 mode change 100644 => 100755 uvgotmail.py

diff --git a/uvgotmail.py b/uvgotmail.py
old mode 100644
new mode 100755
index e69de29..3fbbc7e
--- a/uvgotmail.py
+++ b/uvgotmail.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+import argparse
+import sys
+import os
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--daemon", help="run in daemon mode (put this at end of .bashrc)", action="store_true")
+    parser.add_argument("--config", help="specify config file in muttrc format")
+    parser.add_argument("--check", help="run in check mode (put this in PS1)", action="store_true")
+    parser.add_argument("--no-count", help="don't display how many unread messages there are", action="store_false")
+    parser.add_argument("--mailbox-char", help="specify a character instead of the mailbox emoji",default="📬")
+    args = parser.parse_args()
+
+    global mailchar, countmail;
+    mailchar = args.mailbox_char
+    countmail = args.no_count;
+
+    if(args.daemon == args.check):
+        parser.print_help()
+        sys.exit(0)
+    if(args.daemon):
+        daemon()
+    else:
+        check()
+
+def check():
+    filename = '/tmp/uvgotmail/'+os.getlogin()+'/unread'
+    if os.path.exists(filename):
+        with open(filename) as f:
+            unread = int(f.read());
+        if unread == 0:
+            sys.exit(0)
+        pre = ""
+        if(unread > 1 and countmail):
+            pre = "("+str(unread)+") "
+        print(pre+mailchar,end=' ')
+
+main();