feat: ✨ implement cronjob and db logic for sending emails (#25)

* feat: ✨ implement cronjob and db logic for sending emails

* fix: 🐛 email database schema

* fix: 🐛 update mimetext for Mailbox class

---------

Co-authored-by: Alexander Liu <[email protected]>

package.json

@@ -27,7 +27,7 @@
     "googleapis": "128.0.0",
     "lucia": "2.7.4",
     "lucide-svelte": "0.292.0",
-    "mimetext": "3.0.16",
+    "mimetext": "3.0.21",
     "postgres": "3.4.3",
     "zod": "3.22.4"
   },

services/send-email.ts

@@ -0,0 +1,30 @@
+import { asc, eq } from "drizzle-orm";
+import { drizzle } from "drizzle-orm/postgres-js";
+import { google } from "googleapis";
+import postgres from "postgres";
+
+import { email } from "../src/lib/db/schema";
+
+async function main() {
+  const { DATABASE_URL, GOOGLE_OAUTH_CLIENT_ID, GOOGLE_OAUTH_CLIENT_SECRET, MAILER_REFRESH_TOKEN } =
+    process.env;
+  const db = drizzle(postgres(DATABASE_URL, { max: 1, ssl: { rejectUnauthorized: false } }), {
+    schema: { email },
+  });
+  const auth = new google.auth.OAuth2(GOOGLE_OAUTH_CLIENT_ID, GOOGLE_OAUTH_CLIENT_SECRET);
+  auth.setCredentials({ refresh_token: MAILER_REFRESH_TOKEN });
+  const gmail = google.gmail({ version: "v1", auth });
+  const messages = await db
+    .select({ id: email.id, raw: email.raw })
+    .from(email)
+    .orderBy(asc(email.submittedAt))
+    .limit(15);
+  for (const { id, raw } of messages) {
+    await gmail.users.messages.send({ userId: "me", requestBody: { raw } });
+    await db.delete(email).where(eq(email.id, id));
+    await new Promise((_) => setTimeout(_, 1000));
+  }
+  process.exit(0);
+}
+
+main().then();

src/lib/db/schema.ts

@@ -1,5 +1,5 @@
 import { relations } from "drizzle-orm";
-import { pgSchema, pgEnum, text, timestamp, bigint, varchar } from "drizzle-orm/pg-core";
+import { pgSchema, pgEnum, text, timestamp, bigint, varchar, uuid } from "drizzle-orm/pg-core";
 
 export const schema = pgSchema("dev");
 
@@ -98,3 +98,9 @@ export const contactRelations = relations(contact, ({ one }) => ({
     references: [user.id],
   }),
 }));
+
+export const email = schema.table("email", {
+  id: uuid("id").primaryKey().defaultRandom(),
+  submittedAt: timestamp("submitted_at").defaultNow(),
+  raw: text("raw").notNull(),
+});

src/routes/api/email/send/+server.ts

@@ -1,11 +1,11 @@
 import { error, json } from "@sveltejs/kit";
-import { createMimeMessage } from "mimetext";
+import { createMimeMessage, Mailbox } from "mimetext";
 
 import type { RequestHandler } from "./$types";
 
-import { gmail } from "$lib/server/gmail";
+import { email } from "$lib/db/schema";
+import { drizzle } from "$lib/server/drizzle";
 import { auth } from "$lib/server/lucia";
-import { sleep } from "$lib/util/sleep";
 
 type AttachmentBase = {
   filename: string;
@@ -43,7 +43,7 @@ export const POST: RequestHandler = async (event) => {
     ({ attachments, htmlBody, plaintextBody, recipient, replyTo, subject }) => {
       const msg = createMimeMessage();
       msg.setSender({ name: "ICS Student Council", addr: "[email protected]" });
-      msg.setHeader("Reply-To", { addr: replyTo });
+      msg.setHeader("Reply-To", new Mailbox(replyTo));
       msg.setRecipient(recipient);
       msg.setSubject(subject);
       msg.addMessage({ contentType: "text/plain", data: plaintextBody });
@@ -61,9 +61,5 @@ export const POST: RequestHandler = async (event) => {
       return { raw: msg.asEncoded() };
     },
   );
-  for (const requestBody of messages) {
-    await gmail.users.messages.send({ userId: "me", requestBody });
-    await sleep(1000);
-  }
-  return json({});
+  return json(await drizzle.insert(email).values(messages).returning({ id: email.id }));
 };

sst.config.ts

@@ -1,6 +1,6 @@
 import {} from "aws-cdk-lib";
 import type { SSTConfig } from "sst";
-import { SvelteKitSite } from "sst/constructs";
+import { Cron, SvelteKitSite } from "sst/constructs";
 
 export default {
   config(_input) {
@@ -25,5 +25,13 @@ export default {
     if (app.stage !== "prod") {
       app.setDefaultRemovalPolicy("destroy");
     }
+    if (app.stage === "prod") {
+      app.stack(function SendEmail({ stack }) {
+        new Cron(stack, "sendEmail", {
+          schedule: "rate(10 minutes)",
+          job: "services/send-email.main",
+        });
+      });
+    }
   },
 } satisfies SSTConfig;