aws-sdk-js-codemod/dist/transforms/v2-to-v3/apis/replaceS3GetSignedUrlApi.js

Collection of codemod scripts that help update AWS SDK for JavaScript APIs

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.replaceS3GetSignedUrlApi = void 0;
const getClientApiCallExpression_1 = require("./getClientApiCallExpression");
const getCommandName_1 = require("./getCommandName");
// Updates `s3.getSignedUrl()` API with `await getSignedUrl(s3, command)` API.
const replaceS3GetSignedUrlApi = (j, source, clientIdentifiers) => {
    for (const clientId of clientIdentifiers) {
        for (const getSignedUrlApiName of ["getSignedUrl", "getSignedUrlPromise"]) {
            source
                .find(j.CallExpression, (0, getClientApiCallExpression_1.getClientApiCallExpression)(clientId, getSignedUrlApiName))
                .replaceWith((callExpression) => {
                const args = callExpression.node.arguments;
                if (args[0].type !== "Literal" && args[0].type !== "StringLiteral") {
                    return callExpression;
                }
                if (typeof args[0].value !== "string") {
                    return callExpression;
                }
                const apiName = args[0].value;
                const params = args[1];
                const options = j.objectExpression([]);
                if (params.type === "ObjectExpression") {
                    // Check if params has property 'Expires' and add it to options.
                    for (const property of params.properties) {
                        if (property.type !== "Property" && property.type !== "ObjectProperty") {
                            continue;
                        }
                        if (property.key.type === "Identifier") {
                            if (property.key.name === "Expires") {
                                // Add 'expiresIn' property to options.
                                options.properties.push(j.objectProperty.from({
                                    key: j.identifier("expiresIn"),
                                    value: property.value,
                                    shorthand: true,
                                }));
                                // Remove 'Expires' property from params.
                                params.properties = params.properties.filter((property) => {
                                    if (property.type !== "Property" && property.type !== "ObjectProperty") {
                                        return true;
                                    }
                                    if (property.key.type !== "Identifier")
                                        return true;
                                    return property.key.name !== "Expires";
                                });
                            }
                        }
                    }
                }
                else {
                    // Add comment that expiredIn need to explicitly set.
                    options.properties.push(j.objectProperty.from({
                        key: j.identifier("expiresIn"),
                        value: j.literal.from({
                            value: "/* add value from 'Expires' from v2 call if present, else remove */",
                        }),
                        shorthand: true,
                    }));
                }
                const getSignedUrlArgs = [
                    clientId,
                    j.newExpression(j.identifier((0, getCommandName_1.getCommandName)(apiName)), [params]),
                ];
                if (options.properties.length > 0) {
                    getSignedUrlArgs.push(options);
                }
                const outputCallExpression = j.callExpression.from({
                    callee: j.identifier("getSignedUrl"),
                    arguments: getSignedUrlArgs,
                });
                return getSignedUrlApiName.endsWith("Promise")
                    ? outputCallExpression
                    : j.awaitExpression.from({
                        argument: outputCallExpression,
                    });
            });
        }
    }
};
exports.replaceS3GetSignedUrlApi = replaceS3GetSignedUrlApi;