Skip to content

Commit

Permalink
add body parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
teijosol committed Dec 10, 2024
1 parent 094aa7a commit d7c6e93
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 1 deletion.
2 changes: 1 addition & 1 deletion package.json
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{
"name": "@digitraffic/common",
"version": "2024.11.29-1",
"version": "2024.12.10-1",
"description": "",
"type": "module",
"repository": {
Expand Down
15 changes: 15 additions & 0 deletions src/aws/infra/api/integration.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,19 +14,23 @@ const VELOCITY_ALL_PARAMS = `#foreach($paramName in $params.keySet())
#set($tmp = $paramMap.put($paramName, $params[$paramName]))
#end`;

const VELOCITY_PASS_BODY = `#set($tmp = $paramMap.put('body', $util.escapeJavaScript($input.json('$'))))`;

export class DigitrafficIntegration<T extends string> {
readonly lambda: IFunction;
readonly mediaType: MediaType;
readonly parameters: ApiParameter[] = [];
readonly sunset?: string;

_passAllQueryParameters: boolean;
_passBody: boolean;

constructor(lambda: IFunction, mediaType: MediaType = MediaType.TEXT_PLAIN, sunset?: string) {
this.lambda = lambda;
this.mediaType = mediaType;
this.sunset = sunset;
this._passAllQueryParameters = false;
this._passBody = false;
}

passAllQueryParameters(): this {
Expand All @@ -38,6 +42,16 @@ export class DigitrafficIntegration<T extends string> {
return this;
}

/**
* Body is passed as an escaped string, so broken input won't break anything. You should
* parse and validate the input in the lambda.
*/
passBody(): this {
this._passBody = true;

return this;
}

addPathParameter(...names: T[]): this {
names.forEach((name) => this.parameters.push({ type: "path", name }));

Expand Down Expand Up @@ -139,6 +153,7 @@ export class DigitrafficIntegration<T extends string> {
#set($params = $input.params().get("querystring"))
${parameterAssignments.join("\n")}
${this._passAllQueryParameters ? VELOCITY_ALL_PARAMS : ""}
${this._passBody ? VELOCITY_PASS_BODY : ""}
{
#foreach($paramName in $paramMap.keySet())
#if( $paramName[0] != '_')
Expand Down

0 comments on commit d7c6e93

Please sign in to comment.