Add setup-imapfilter

5 files changed, 145 insertions, 33 deletions

diff --git a/scripts/p7s/__init__.py b/scripts/p7s/__init__.py
index 17510190..8f6dea70 100644
--- a/scripts/p7s/__init__.py
+++ b/scripts/p7s/__init__.py
@@ -1,4 +1,4 @@
-import os, pathlib, shutil, subprocess, sys
+import os, pathlib, shutil, subprocess, sys, textwrap
 
 
 def nix_portable(command):
@@ -9,3 +9,13 @@ def nix_portable(command):
 
 
 BASHRC_D = pathlib.Path.home() / ".bashrc.d"
+
+
+def multiline_string(s):
+    return textwrap.dedent(s).lstrip()
+
+
+def assert_single(l):
+    assert len(l) == 1
+    return l[0]
+
diff --git a/scripts/p7s/mail/__init__.py b/scripts/p7s/mail/__init__.py
index 71675c70..e663d65c 100644
--- a/scripts/p7s/mail/__init__.py
+++ b/scripts/p7s/mail/__init__.py
@@ -1,14 +1,20 @@
 import pathlib
-import subprocess
-import textwrap
 
+from p7s import assert_single
+from p7s import multiline_string as _
 from p7s import bitwarden
+from p7s.mail import imapfilter
 from p7s.mail import mbsync
+from p7s import systemd
+
+
+MOM = "5c4d9e3b-121d-45f5-bab6-03b42d291326"
+GMAIL = "f9bba940-769d-430a-82f4-5da10990e8fd"
 
 
 def setup_mbsync():
-    gmail = bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", "f9bba940-769d-430a-82f4-5da10990e8fd")["login"]
-    yahoo = bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", "5c4d9e3b-121d-45f5-bab6-03b42d291326")["login"]
+    gmail = bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", GMAIL)["login"]
+    yahoo = bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", MOM)["login"]
     (pathlib.Path.home() / (".mbsyncrc")).write_text(
         mbsync.mbsync_gmail(gmail["username"], gmail["password"], "~/Mail") +
         "\n" +
@@ -18,31 +24,66 @@ def setup_mbsync():
     for username in [gmail["username"], yahoo["username"]]:
         (pathlib.Path.home() / "Mail" / username).mkdir(exist_ok=True, parents=True)
 
-    user_units = pathlib.Path.home() / ".config" / "systemd" / "user"
-    user_units.mkdir(exist_ok=True, parents=True)
-    (user_units / "mbsync.service").write_text(textwrap.dedent("""
-        [Unit]
-        Description=Mail synchronization
-
-        [Service]
-        Type=oneshot
-        ExecStart=/usr/bin/touch /home/alex/Mail/.lock ; /usr/bin/mbsync -qa ; /usr/bin/rm -f /home/alex/Mail/.lock
-        """).lstrip())
-
-    (user_units / "mbsync.timer").write_text(textwrap.dedent("""
-        [Unit]
-        Description=Mail synchronization
-
-        [Install]
-        WantedBy=timers.target
-
-        [Timer]
-        OnBootSec=1m
-        OnActiveSec=0s
-        OnUnitInactiveSec=30s
-        Unit=mbsync.service
-        """).lstrip())
-
-    subprocess.run(["systemctl", "--user", "daemon-reload"], check=True)
-    subprocess.run(["systemctl", "--user", "enable", "--now", "mbsync.timer"], check=True)
-    subprocess.run(["sudo", "loginctl", "enable-linger", "alex"], check=True)
+    systemd.create_user_unit("mbsync.service", _("""
+    [Unit]
+    Description=Mail synchronization
+
+    [Service]
+    Type=oneshot
+    ExecStart=/usr/bin/touch /home/alex/Mail/.lock ; /usr/bin/mbsync -qa ; /usr/bin/rm -f /home/alex/Mail/.lock
+    """))
+
+    systemd.create_user_unit("mbsync.timer", _("""
+    [Unit]
+    Description=Mail synchronization
+
+    [Install]
+    WantedBy=timers.target
+
+    [Timer]
+    OnBootSec=1m
+    OnActiveSec=0s
+    OnUnitInactiveSec=30s
+    Unit=mbsync.service
+    """))
+
+    systemd.reload()
+    systemd.enable_now("mbsync.timer")
+    systemd.enable_linger()
+
+
+def setup_imapfilter():
+    create_forward(
+        "mom_to_gmail",
+        _bitwarden_item_to_imapfilterserver("mom", bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", MOM)),
+        _bitwarden_item_to_imapfilterserver("gmail", bitwarden.get_item("https://vaultwarden.pdp7.net", "alex@corcoles.net", GMAIL)),
+    )
+
+
+def _bitwarden_item_to_imapfilterserver(name, i):
+    return imapfilter.ImapServer(
+        name=name,
+        server=assert_single([f for f in i["fields"] if f["name"] == "imap_server"])["value"],
+        username=i["login"]["username"],
+        password=i["login"]["password"],
+    )
+
+
+def create_forward(name, from_server, to_server):
+    config = pathlib.Path.home() / ".imapfilter" / f"{name}.lua"
+    config.write_text(
+        imapfilter.Sync(from_server=from_server, to_server=to_server).lua()
+    )
+
+    unit = f"imapfilter-{name}.service"
+    systemd.create_user_unit(unit, _(f"""
+    [Service]
+    ExecStart=/usr/bin/imapfilter -c {config} -v
+
+    [Install]
+    WantedBy=default.target
+    """))
+
+    systemd.reload()
+    systemd.enable_now(unit)
+    systemd.enable_linger()
diff --git a/scripts/p7s/mail/imapfilter.py b/scripts/p7s/mail/imapfilter.py
new file mode 100644
index 00000000..41cd519d
--- /dev/null
+++ b/scripts/p7s/mail/imapfilter.py
@@ -0,0 +1,39 @@
+from p7s import multiline_string as _
+
+
+class ImapServer:
+    def __init__(self, name, server, username, password):
+        self.name = name
+        self.server = server
+        self.username = username
+        self.password = password
+
+    def lua(self):
+        return _(f"""
+        {self.name} = IMAP {{
+            server = '{self.server}',
+            username = '{self.username}',
+            password = '{self.password}',
+            ssl = 'auto',
+        }}
+        """)
+
+
+class Sync:
+    def __init__(self, from_server: ImapServer, to_server: ImapServer):
+        self.from_server = from_server
+        self.to_server = to_server
+
+    def lua(self):
+        s = self.from_server.lua()
+        s += "\n"
+        s += self.to_server.lua()
+        s += "\n"
+        s += _(f"""
+        while true do
+            results = {self.from_server.name}.INBOX:is_unseen()
+            results:move_messages({self.to_server.name}.INBOX)
+            {self.from_server.name}.INBOX:enter_idle()
+        end
+        """)
+        return s
diff --git a/scripts/p7s/systemd.py b/scripts/p7s/systemd.py
new file mode 100644
index 00000000..3c2270f4
--- /dev/null
+++ b/scripts/p7s/systemd.py
@@ -0,0 +1,21 @@
+import getpass
+import pathlib
+import subprocess
+
+
+def create_user_unit(name, content):
+    user_units = pathlib.Path.home() / ".config" / "systemd" / "user"
+    user_units.mkdir(exist_ok=True, parents=True)
+    (user_units / name).write_text(content)
+
+
+def reload():
+    subprocess.run(["systemctl", "--user", "daemon-reload"], check=True)
+
+
+def enable_now(unit):
+    subprocess.run(["systemctl", "--user", "enable", "--now", unit], check=True)
+
+
+def enable_linger():
+    subprocess.run(["sudo", "loginctl", "enable-linger", getpass.getuser()], check=True)
diff --git a/scripts/pyproject.toml b/scripts/pyproject.toml
index 7a45f4e6..7803b64a 100644
--- a/scripts/pyproject.toml
+++ b/scripts/pyproject.toml
@@ -21,6 +21,7 @@ setup-emacs = 'p7s.emacs:setup_emacs'
 setup-mbsync = 'p7s.mail:setup_mbsync'
 setup-nextcloud = 'p7s.nextcloud:setup_nextcloud'
 setup-paperwm = 'p7s.paperwm:setup_paperwm'
+setup-imapfilter = 'p7s.mail:setup_imapfilter'
 setup-soju = 'p7s.soju:setup_soju'
 setup-ubpkg = 'p7s.ubpkg:setup_ubpkg'
 setup-x12 = 'p7s.x12:setup_x12'