Add key-value pair to existing var declarator in import equals (#379)

Author: trivikr

src/transforms/v2-to-v3/modules/addV3ClientNamedImportEquals.ts

@@ -2,6 +2,7 @@ import { Collection, JSCodeshift } from "jscodeshift";
 
 import { getV3ClientDefaultLocalName } from "../utils";
 import { getImportEqualsDeclaration } from "./getImportEqualsDeclaration";
+import { getV3ClientRequireProperty } from "./getV3ClientRequireProperty";
 import { V3ClientModulesOptions, V3ClientRequirePropertyOptions } from "./types";
 
 export const addV3ClientNamedImportEquals = (
@@ -15,20 +16,26 @@ export const addV3ClientNamedImportEquals = (
   }: V3ClientModulesOptions & V3ClientRequirePropertyOptions
 ) => {
   const v3ClientDefaultLocalName = getV3ClientDefaultLocalName(v2ClientLocalName);
+  const namedImportObjectProperty = getV3ClientRequireProperty(j, { keyName, valueName });
+
+  const existingVarDeclarator = source.find(j.VariableDeclarator, {
+    type: "VariableDeclarator",
+    init: { type: "Identifier", name: v3ClientDefaultLocalName },
+  });
+
+  if (existingVarDeclarator.size()) {
+    existingVarDeclarator.get(0).node.id.properties.push(namedImportObjectProperty);
+    return;
+  }
+
   const existingImportEquals = source.find(
     j.TSImportEqualsDeclaration,
     getImportEqualsDeclaration(v3ClientPackageName)
   );
 
   const varDeclaration = j.variableDeclaration("const", [
     j.variableDeclarator(
-      j.objectPattern([
-        j.objectProperty.from({
-          key: j.identifier(keyName),
-          value: j.identifier(valueName),
-          shorthand: true,
-        }),
-      ]),
+      j.objectPattern([namedImportObjectProperty]),
       j.identifier(v3ClientDefaultLocalName)
     ),
   ]);