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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// spellchecker: words sesv2 amazonses
import { SendEmailCommand, SESv2Client } from "@aws-sdk/client-sesv2";
import log from "../log.ts";
import { DbService } from "../db.ts";
import {
Mail,
MailDeliverContext,
MailDeliverHook,
SyncMailDeliverer,
} from "../mail.ts";
import { AwsContext } from "./context.ts";
declare module "../mail.ts" {
interface MailDeliverResult {
awsMessageId?: string;
}
}
export class AwsMailMessageIdRewriteHook implements MailDeliverHook {
readonly #db;
constructor(db: DbService) {
this.#db = db;
}
async callback(context: MailDeliverContext): Promise<void> {
log.info("Rewrite message ids...");
const addresses = context.mail.simpleFindAllAddresses();
log.info(`Addresses found in mail: ${addresses.join(", ")}.`);
for (const address of addresses) {
const awsMessageId = await this.#db.messageIdToAws(address);
if (awsMessageId != null && awsMessageId.length !== 0) {
log.info(`Rewrite ${address} to ${awsMessageId}.`);
context.mail.raw = context.mail.raw.replaceAll(address, awsMessageId);
}
}
log.info("Done rewrite message ids.");
}
}
export class AwsMailMessageIdSaveHook implements MailDeliverHook {
readonly #db;
constructor(db: DbService) {
this.#db = db;
}
async callback(context: MailDeliverContext): Promise<void> {
log.info("Save aws message ids...");
const messageId = context.mail.startSimpleParse().sections().headers()
.messageId();
if (messageId == null) {
log.info("Original mail does not have message id. Skip saving.");
return;
}
if (context.result.awsMessageId != null) {
log.info(`Saving ${messageId} => ${context.result.awsMessageId}.`);
await this.#db.addMessageIdMap({
message_id: messageId,
aws_message_id: context.result.awsMessageId,
});
}
log.info("Done save message ids.");
}
}
export class AwsMailDeliverer extends SyncMailDeliverer {
readonly name = "aws";
readonly #aws;
readonly #ses;
constructor(aws: AwsContext) {
super();
this.#aws = aws;
this.#ses = new SESv2Client(aws);
}
protected override async doDeliver(
mail: Mail,
context: MailDeliverContext,
): Promise<void> {
log.info("Begin to call aws send-email api...");
try {
const sendCommand = new SendEmailCommand({
Content: {
Raw: { Data: mail.toUtf8Bytes() },
},
});
const res = await this.#ses.send(sendCommand);
if (res.MessageId == null) {
log.warn("Aws send-email returns no message id.");
} else {
context.result.awsMessageId =
`${res.MessageId}@${this.#aws.region}.amazonses.com`;
}
context.result.recipients.set("*", {
kind: "done",
message:
`Successfully called aws send-email, message id ${context.result.awsMessageId}.`,
});
} catch (cause) {
context.result.recipients.set("*", {
kind: "fail",
message: "An error was thrown when calling aws send-email." + cause,
cause,
});
}
}
}
|
