FilesExpand file tree

 master

/

lambda.cfc

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

186 lines (161 loc) · 7.38 KB

 master

/

lambda.cfc

Copy path

Top

File metadata and controls

• Code
• Blame

186 lines (161 loc) · 7.38 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

/**

* lambda.cfc

* Copyright 2018-2019 Matthew Clemente, Brian Klaas

* Licensed under MIT

*/

component {

/**

* @classPath a Lucee specific option; provide the path to a directory containing the AWS Java SDK class files for Lambda.

*/

public any function init(

required string accessKey,

required string secretKey,

string defaultArn = '',

string classPath = ''

) {

variables.serverVersion = server.keyExists('lucee') ? 'Lucee' : 'ColdFusion';

if (variables.serverVersion != 'Lucee' && len(arguments.classPath)) {

throw('Sorry, the option to provide a class path to the AWS SDK jars is only available with Lucee CFML.');

}

variables.accessKey = accessKey;

variables.secretKey = secretKey;

variables.defaultArn = defaultArn;

variables.classPath = arguments.classPath;

return this;

}

// For reference - full Java API reference for AWS SDK: https://sdk.amazonaws.com/java/api/latest/index.html

// Lambda: https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/services/lambda/package-summary.html

/**

* @hint Calls the Lambda function specified by the default ARN, passing in the payload. You must set the default ARN in order to use this.

* @payload Can be passed in as JSON, an array, or a struct. Structs and arrays will be converted to JSON, as required by Lambda

*/

public string function invoke(any payload = {}) {

if (!len(variables.defaultArn))

throw(

message = 'Default Lambda ARN missing',

detail = 'Unable to call #getFunctionCalledName()#() without a default Lambda ARN. You can set the default ARN on init() or manually, via setDefaultArn().'

);

return invokeLambda(arn = variables.defaultArn, payload = payload);

}

/**

* @hint Calls the Lambda function specified by the ARN, passing in the payload

* @payload Can be passed in as JSON, an array, or a struct. Structs and arrays will be converted to JSON, as required by Lambda

*/

public string function invokeFunction(required string arn, any payload = {}) {

return invokeLambda(arn = arn, payload = payload);

}

/**

* @hint Handles the actual interaction with Lambda for the public methods.

*/

private string function invokeLambda(required string arn, required any payload) {

var lambda = lambdaClient(arn = arn);

var invokeRequest = createObjectHelper('software.amazon.awssdk.services.lambda.model.InvokeRequest').builder();

invokeRequest.functionName(arn);

var jsonPayload = parsePayload(payload);

if (jsonPayload.len()) {

// https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/core/SdkBytes.html

var SdkBytes = createObjectHelper('software.amazon.awssdk.core.SdkBytes').fromUtf8String(jsonPayload);

invokeRequest.payload(SdkBytes);

}

var test = lambda.build();

var response = test.invoke(invokeRequest.build());

return decodeResponse(response);

}

private any function lambdaClient(required string arn) {

// https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/auth/credentials/AwsBasicCredentials.html

var awsCredentials = createObjectHelper('software.amazon.awssdk.auth.credentials.AwsBasicCredentials').create(

variables.accessKey,

variables.secretKey

);

// https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/auth/credentials/StaticCredentialsProvider.html

var awsStaticCredentialsProvider = createObjectHelper(

'software.amazon.awssdk.auth.credentials.StaticCredentialsProvider'

).create(awsCredentials);

return buildFromArn(arn, awsStaticCredentialsProvider);

}

/**

* @hint Takes an arn and combines it with the credentials to return the Lambda client

*/

private any function buildFromArn(arn, awsStaticCredentialsProvider) {

var arnComponents = parseArn(arn);

// https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/regions/Region.html

var awsRegion = createObjectHelper('software.amazon.awssdk.regions.Region').of(arnComponents.region);

return createObjectHelper('software.amazon.awssdk.services.lambda.LambdaClient')

.builder()

.credentialsProvider(awsStaticCredentialsProvider)

.region(awsRegion);

}

/**

* @hint The payload returned from a Lambda function invocation in the Java SDK is always a Java binary stream. As such, it needs to be decoded into a string of characters.

* @response must be the object returned by the Lambda object's invoke() method

*/

private string function decodeResponse(required any response) {

var charset = createObject('java', 'java.nio.charset.Charset').forName('UTF-8');

var charsetDecoder = charset.newDecoder();

// https://sdk.amazonaws.com/java/api/latest/software/amazon/awssdk/services/lambda/model/InvokeResponse.html

return response.payload().asUtf8String();

}

/**

* @hint ensures the payload passed to Lambda is JSON

*/

private string function parsePayload(required any payload) {

if (isStruct(payload) || isArray(payload)) return serializeJSON(payload);

else if (isJSON(payload)) return payload;

else return '';

}

/**

* @hint Parses an Amazon Resource Name (ARN) and returns its component parts as an object.

* This follows the general format of ARNs outlined by Amazon (http://docs.aws.amazon.com/general/latest/gr/aws-arns-and-namespaces.html), but does not fully account for all possible formats

* Derived from https://gist.github.com/gene1wood/5299969edc4ef21d8efcfea52158dd40

*/

private struct function parseArn(required string arn) {

var elements = arn.listToArray(':', true);

var result = {

'original': arn,

'arn': elements[1],

'partition': elements[2],

'service': elements[3],

'region': elements[4],

'account': elements[5]

};

if (elements.len() >= 7) {

result['resourcetype'] = elements[6];

result['resource'] = elements[7];

} else if (!elements[6].find('/')) {

result['resource'] = elements[6];

result['resourcetype'] = '';

} else {

result['resourcetype'] = elements[6].listFirst('/');

result['resource'] = elements[6].listRest('/');

}

return result;

}

/**

* @hint Enables us to pass a specific class path when using Lucee

*/

private any function createObjectHelper(required string classname) {

if (hasClassPath()) {

return createObject(

'java',

classname,

directoryList(

expandPath(variables.classPath),

true,

'path',

'',

'',

'file'

)

);

} else {

if (variables.serverVersion == 'ColdFusion') {

return createObject('java', classname);

} else {

return createObject('java', classname);

}

}

}

private boolean function hasClassPath() {

return len(variables.classPath);

}

}