deno/mail-relay/aws/mail.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

import { MailDeliverContext, MailDeliverHook } from "../mail.ts";

export class AwsMailMessageIdRewriteHook implements MailDeliverHook {
  readonly #lookup;

  constructor(lookup: (origin: string) => Promise<string | null>) {
    this.#lookup = lookup;
  }

  async callback(context: MailDeliverContext): Promise<void> {
    context.logger.info("Rewrite message ids...");
    const addresses = context.mail.simpleFindAllAddresses();
    context.logger.info(`Addresses found in mail: ${addresses.join(", ")}.`);
    for (const address of addresses) {
      const awsMessageId = await this.#lookup(address);
      if (awsMessageId != null && awsMessageId.length !== 0) {
        context.logger.info(`Rewrite ${address} to ${awsMessageId}.`);
        context.mail.raw = context.mail.raw.replaceAll(address, awsMessageId);
      }
    }
    context.logger.info("Done rewrite message ids.");
  }
}

export class AwsMailMessageIdSaveHook implements MailDeliverHook {
  readonly #record;

  constructor(record: (original: string, aws: string) => Promise<void>) {
    this.#record = record;
  }

  async callback(context: MailDeliverContext): Promise<void> {
    context.logger.info("Save aws message ids...");
    const messageId = context.mail
      .startSimpleParse(context.logger)
      .sections()
      .headers()
      .messageId();
    if (messageId == null) {
      context.logger.info(
        "Original mail does not have message id. Skip saving.",
      );
      return;
    }
    if (context.result.awsMessageId != null) {
      context.logger.info(
        `Saving ${messageId} => ${context.result.awsMessageId}.`,
      );
      await this.#record(messageId, context.result.awsMessageId);
    }
    context.logger.info("Done save message ids.");
  }
}