-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.ts
96 lines (89 loc) · 2.08 KB
/
index.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
94
95
96
import * as cloudfront from 'aws-cdk-lib/aws-cloudfront'
import type { Construct } from 'constructs'
/** Properties for creating a CloudFrontFunction */
export interface CloudFrontFunctionProps {
/**
* A name to identify the function.
*
* @default - generated from the `id`
*/
readonly functionName?: string
/**
* A comment to describe the function.
*
* @default - same as `functionName`
*/
readonly comment?: string
}
export interface ReplaceInUriCloudFrontFunctionProps {
/** What to replace */
pattern: RegExp | string
/** What to place */
replacement: string
}
export class CloudFrontFunction extends cloudfront.Function {
static forSpa(
scope: Construct,
id: string,
props?: CloudFrontFunctionProps,
): CloudFrontFunction {
return new CloudFrontFunction(scope, id, {
...props,
code: cloudfront.FunctionCode.fromInline(`
function handler(event) {
var request = event.request;
if (!request.uri.includes('.')) {
request.uri = '/index.html';
}
return request;
}
`),
})
}
/**
* Replace some part of request uri
*
* @example
*
* ```ts
* // Replaces `/api/some-route` to `/some-route`
* CloudFrontFunction.replaceInUri(scope, id, {
* pattern: /^\/api/,
* replacement: '',
* })
* ```
*
* @example
*
* ```ts
* // Replaces `/api/aaa/some-route` to `/api/some-route`
* CloudFrontFunction.replaceInUri(scope, id, {
* pattern: '^/api/aaa,
* replacement: '/api',
* })
* ```
*/
static replaceInUri(
scope: Construct,
id: string,
props: ReplaceInUriCloudFrontFunctionProps,
): CloudFrontFunction {
const escapedPattern = (
typeof props.pattern === 'string'
? new RegExp(props.pattern)
: props.pattern
).toString()
return new CloudFrontFunction(scope, id, {
...props,
code: cloudfront.FunctionCode.fromInline(`
function handler(event) {
var request = event.request;
request.uri = request.uri.replace(${escapedPattern}, ${JSON.stringify(
props.replacement,
)});
return request;
}
`),
})
}
}