-
Notifications
You must be signed in to change notification settings - Fork 0
/
scheduler.ts
93 lines (77 loc) · 2.54 KB
/
scheduler.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
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
import { CloudWatchEvents, Lambda } from "aws-sdk";
const cw = new CloudWatchEvents();
const lambda = new Lambda();
export const createCloudWatchEventSchedule = async (
discogsUsername: string,
shipsFrom: string,
destinationEmail: string
) => {
var params = {
Name: `MarketMonitor-${discogsUsername}`,
ScheduleExpression: "rate(12 hours)",
State: "ENABLED",
Description: `Discogs Market Monitor Event Rule for ${discogsUsername}`,
};
const { RuleArn } = await cw.putRule(params).promise();
const targetParams = {
Rule: `MarketMonitor-${discogsUsername}`,
Targets: [
{
Arn: process.env.MARKET_MONITOR_LAMBDA_FUNCTION_ARN || "",
Id: "RunMarketMonitorLambda",
Input: `{"destinationEmail":"${destinationEmail}", "username":"${discogsUsername}", "shipsFrom":"${shipsFrom}"}`,
},
],
};
await cw.putTargets(targetParams).promise();
await addLambdaPermission(discogsUsername, RuleArn);
};
const addLambdaPermission = async (
discogsUsername: string,
sourceArn?: string
) => {
const addPermissionParams = {
Action: "lambda:InvokeFunction",
FunctionName: process.env.MARKET_MONITOR_LAMBDA_FUNCTION_ARN || "",
Principal: "events.amazonaws.com",
StatementId: `MarketMonitor-${discogsUsername}`,
SourceArn: sourceArn,
};
try {
const { Policy } = await lambda
.getPolicy({ FunctionName: addPermissionParams.FunctionName })
.promise();
const policyDocument = JSON.parse(Policy || "");
const permissionExists = policyDocument.Statement.some(
(statement: { Sid?: string }) =>
statement.Sid === addPermissionParams.StatementId
);
if (!permissionExists) {
await lambda.addPermission(addPermissionParams).promise();
}
} catch (err: any) {
if (err.code === "ResourceNotFoundException") {
await lambda.addPermission(addPermissionParams).promise();
} else {
throw err;
}
}
};
export const deleteCloudWatchEventSchedule = async (
discogsUsername: string
) => {
const removeTargetsParams = {
Rule: `MarketMonitor-${discogsUsername}`,
Ids: ["RunMarketMonitorLambda"],
};
await cw.removeTargets(removeTargetsParams).promise();
const deleteRuleParams = {
Name: `MarketMonitor-${discogsUsername}`,
};
await cw.deleteRule(deleteRuleParams).promise();
const removePermissionParams = {
FunctionName: process.env.MARKET_MONITOR_LAMBDA_FUNCTION_ARN || "",
StatementId: `MarketMonitor-${discogsUsername}`,
};
await lambda.removePermission(removePermissionParams).promise();
};