initial commit

backend/src/services/dynamodb.ts

@@ -0,0 +1,60 @@
+import { DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { DynamoDBDocumentClient, GetCommand, PutCommand, DeleteCommand } from '@aws-sdk/lib-dynamodb';
+import { config } from '../config.js';
+import { normalizeEmail } from '../utils/email.js';
+
+export interface EmailRule {
+  email_address: string;
+  ooo_active?: boolean;
+  ooo_message?: string;
+  ooo_content_type?: string;
+  forwards?: string[];
+}
+
+export interface BlockList {
+  email_address: string;
+  blocked_patterns: string[];
+}
+
+export class DynamoRulesService {
+  private doc = DynamoDBDocumentClient.from(new DynamoDBClient({ region: config.awsRegion }), {
+    marshallOptions: { removeUndefinedValues: true },
+  });
+
+  async getRules(email: string): Promise<EmailRule> {
+    const email_address = normalizeEmail(email);
+    const resp = await this.doc.send(new GetCommand({ TableName: config.rulesTable, Key: { email_address } }));
+    return (resp.Item as EmailRule) ?? { email_address, ooo_active: false, ooo_message: '', ooo_content_type: 'text/plain', forwards: [] };
+  }
+
+  async putRules(rule: EmailRule): Promise<EmailRule> {
+    const item: EmailRule = {
+      email_address: normalizeEmail(rule.email_address),
+      ooo_active: !!rule.ooo_active,
+      ooo_message: rule.ooo_message ?? '',
+      ooo_content_type: rule.ooo_content_type ?? 'text/plain',
+      forwards: (rule.forwards ?? []).map(normalizeEmail).filter(Boolean),
+    };
+    await this.doc.send(new PutCommand({ TableName: config.rulesTable, Item: item }));
+    return item;
+  }
+
+  async getBlocklist(email: string): Promise<BlockList> {
+    const email_address = normalizeEmail(email);
+    const resp = await this.doc.send(new GetCommand({ TableName: config.blockedTable, Key: { email_address } }));
+    return (resp.Item as BlockList) ?? { email_address, blocked_patterns: [] };
+  }
+
+  async putBlocklist(email: string, patterns: string[]): Promise<BlockList> {
+    const item: BlockList = {
+      email_address: normalizeEmail(email),
+      blocked_patterns: patterns.map((p) => p.trim().toLowerCase()).filter(Boolean),
+    };
+    if (item.blocked_patterns.length === 0) {
+      await this.doc.send(new DeleteCommand({ TableName: config.blockedTable, Key: { email_address: item.email_address } }));
+      return item;
+    }
+    await this.doc.send(new PutCommand({ TableName: config.blockedTable, Item: item }));
+    return item;
+  }
+}