From 0487284245703bc66001ec11d644050bafcd3ea0 Mon Sep 17 00:00:00 2001
From: dreamhunter2333 <dreamhunter2333@gmail.com>
Date: Thu, 24 Aug 2023 15:22:03 +0800
Subject: [PATCH] feat: add BLACK_LIST

---
 worker/src/email.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/worker/src/email.js b/worker/src/email.js
index 8a018ecb..51d53990 100644
--- a/worker/src/email.js
+++ b/worker/src/email.js
@@ -1,6 +1,11 @@
 const PostalMime = require("postal-mime");
 
 async function email(message, env, ctx) {
+    if (env.BLACK_LIST && env.BLACK_LIST.split(",").some(word => message.from.includes(word))) {
+        message.setReject("Missing from address");
+        console.log(`Reject message from ${message.from} to ${message.to}`);
+        return;
+    }
     if (!env.PREFIX || (message.to && message.to.startsWith(env.PREFIX))) {
         const reader = message.raw.getReader();
         const decoder = new TextDecoder("utf-8");
@@ -21,9 +26,11 @@ async function email(message, env, ctx) {
         ).bind(message.from, message.to, parsedEmail.subject || "", parsedEmail.html).run();
         if (!success) {
             message.setReject(`Failed save message to ${message.to}`);
+            console.log(`Failed save message from ${message.from} to ${message.to}`);
         }
     } else {
         message.setReject(`Unknown address ${message.to}`);
+        console.log(`Unknown address ${message.to}`);
     }
 }