@trackunit/iris-app 1.4.48 → 1.4.49

package/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 1.4.49 (2025-08-14)
+
+This was a version bump only for iris-app to align it with other projects, there were no code changes.
+
 ## 1.4.48 (2025-08-14)
 
 ### 🧱 Updated Dependencies

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trackunit/iris-app",
-  "version": "1.4.48",
+  "version": "1.4.49",
   "license": "SEE LICENSE IN LICENSE.txt",
   "main": "src/index.js",
   "generators": "./generators.json",

package/src/generators/extend/generator.d.ts

@@ -1,5 +1,10 @@
 import { Tree } from "@nx/devkit";
 import { IrisAppExtensionGeneratorSchema } from "./schema";
+/**
+ * Ensures that jest.preset.js exists and includes the required @trackunit/react-test-setup preset.
+ * If the file doesn't exist, creates it. If it exists but doesn't include the preset, updates it.
+ */
+export declare function ensureJestPresetConfiguration(tree: Tree, templateOptions?: object): void;
 /**
  * Iris App Extension Generator.
  *

package/src/generators/extend/generator.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.IrisAppExtensionGenerator = void 0;
+exports.ensureJestPresetConfiguration = ensureJestPresetConfiguration;
 const tslib_1 = require("tslib");
 const devkit_1 = require("@nx/devkit");
 const eslint_1 = require("@nx/eslint");
@@ -13,6 +14,7 @@ const fs = tslib_1.__importStar(require("fs"));
 const path = tslib_1.__importStar(require("path"));
 const string_ts_1 = require("string-ts");
 const ts_morph_1 = require("ts-morph");
+const astUtils_1 = require("../../utils/ast/astUtils");
 const dependencies_1 = require("./dependencies");
 function normalizeOptions(tree, options) {
     const name = (0, devkit_1.names)(options.name).fileName;
@@ -39,6 +41,35 @@ function normalizeOptions(tree, options) {
         dependencyDefinitionFile,
     };
 }
+/**
+ * Ensures that jest.preset.js exists and includes the required @trackunit/react-test-setup preset.
+ * If the file doesn't exist, creates it. If it exists but doesn't include the preset, updates it.
+ */
+function ensureJestPresetConfiguration(tree, templateOptions) {
+    const jestPresetPath = "jest.preset.js";
+    if (!tree.exists(jestPresetPath)) {
+        // File doesn't exist, generate it using the template
+        (0, devkit_1.generateFiles)(tree, path.join(__dirname, "files", "preset"), "/", templateOptions || {});
+        return;
+    }
+    // File exists, use AST utilities for reliable manipulation
+    let currentContent = tree.read(jestPresetPath, "utf-8");
+    if (!currentContent) {
+        throw new Error(`Could not read ${jestPresetPath}`);
+    }
+    // Extract variable name from existing import if it exists
+    let presetVariableName = "irisPreset"; // default name
+    const existingImportMatch = currentContent.match(/const\s+(\w+)\s*=\s*require\(["']@trackunit\/react-test-setup\/preset["']\)/);
+    if (existingImportMatch && existingImportMatch[1]) {
+        presetVariableName = existingImportMatch[1];
+    }
+    // Add the require statement if it doesn't exist
+    currentContent = (0, astUtils_1.addRequireStatement)(currentContent, presetVariableName, "@trackunit/react-test-setup/preset");
+    // Add the spread assignment to the module.exports object
+    currentContent = (0, astUtils_1.addSpreadAssignment)(currentContent, presetVariableName);
+    // Write the updated content back to the tree
+    tree.write(jestPresetPath, currentContent);
+}
 /**
  * Iris App Extension Generator.
  *
@@ -85,7 +116,9 @@ const IrisAppExtensionGenerator = async function (tree, options) {
     (0, devkit_1.generateFiles)(tree, path.join(__dirname, "files", irisAppExtensionType), normalizedOptions.projectRoot, templateOptions);
     if (!iris_app_api_1.nonUiExtensionTypes.includes(irisAppExtensionType)) {
         (0, devkit_1.generateFiles)(tree, path.join(__dirname, "files", "react-app"), normalizedOptions.projectRoot, templateOptions);
-        (0, devkit_1.generateFiles)(tree, path.join(__dirname, "files", "preset"), "/", templateOptions);
+        if (normalizedOptions.extendJestPreset) {
+            ensureJestPresetConfiguration(tree, templateOptions);
+        }
     }
     const rootTsConfigPath = (0, js_1.getRootTsConfigPathInTree)(tree);
     if (!rootTsConfigPath) {

package/src/generators/extend/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../libs/iris-app-sdk/iris-app/src/generators/extend/generator.ts"],"names":[],"mappings":";;;;AAAA,uCAWoB;AACpB,uCAAoC;AACpC,+BAAmD;AACnD,+EAA0E;AAC1E,qCAA6C;AAE7C,0DAAwF;AACxF,uCAAkC;AAClC,+CAAyB;AACzB,mDAA6B;AAC7B,yCAAsC;AACtC,uCAAoE;AACpE,iDAAmE;AAanE,SAAS,gBAAgB,CAAC,IAAU,EAAE,OAAwC;IAC5E,MAAM,IAAI,GAAG,IAAA,cAAK,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;IAC1C,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,IAAA,cAAK,EAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IACnG,MAAM,WAAW,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC;IACxE,MAAM,gBAAgB,GAAG,IAAA,qBAAS,EAAC,WAAW,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,WAAW,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,EAAE,CAAC;IAC9D,MAAM,UAAU,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,MAAM,CAAC;IACjE,MAAM,wBAAwB,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,eAAe,CAAC;IAExF,MAAM,QAAQ,GAAG,IAAA,2BAAW,EAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,QAAQ,IAAI,WAAW,EAAE,CAAC;IAEjD,OAAO;QACL,GAAG,OAAO;QACV,WAAW;QACX,WAAW;QACX,gBAAgB;QAChB,gBAAgB;QAChB,UAAU;QACV,UAAU;QACV,wBAAwB;KACzB,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACI,MAAM,yBAAyB,GAAG,KAAK,WAAW,IAAU,EAAE,OAAwC;IAC3G,MAAM,eAAe,GAAG,MAAM,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAE3D,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IAElE,MAAM,eAAe,GAAG;QACtB,GAAG,iBAAiB;QACpB,GAAG,IAAA,cAAK,EAAC,iBAAiB,CAAC,IAAI,CAAC;QAChC,cAAc,EAAE,IAAA,uBAAc,EAAC,iBAAiB,CAAC,WAAW,CAAC;QAC7D,QAAQ,EAAE,EAAE;KACb,CAAC;IACF,MAAM,oBAAoB,GAAG,iBAAiB,CAAC,IAAI,CAAC;IAEpD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACjF,MAAM,IAAI,KAAK,CACb,2BAA2B,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IACzD,iBAAiB,CAAC,GACpB,yCAAyC,CAC1C,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,uBAAuB,CAAC;IACzG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CACb,kDAAkD,YAAY,4EAA4E,CAC3I,CAAC;IACJ,CAAC;IAED,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,CAAC;QACxC,IAAA,qCAA4B,EAC1B,IAAI,EACJ,EAAE,EACF;YACE,GAAG,8BAAe;YAClB,GAAG,+BAAgB;SACpB,CACF,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAW;QACrB,IAAI,EAAE,iBAAiB,CAAC,WAAW;QACnC,KAAK,EAAE,MAAM;QACb,UAAU,EAAE,KAAK;QACjB,MAAM,EAAE,IAAI;QACZ,cAAc,EAAE,MAAM;QACtB,MAAM,EAAE,eAAM,CAAC,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,iBAAiB,CAAC,WAAW;QACxC,YAAY,EAAE,KAAK;QACnB,eAAe,EAAE,IAAI;KACtB,CAAC;IAEF,MAAM,IAAA,wBAAgB,EAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAErC,IAAA,sBAAa,EACX,IAAI,EACJ,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,oBAAoB,CAAC,EACnD,iBAAiB,CAAC,WAAW,EAC7B,eAAe,CAChB,CAAC;IAEF,IAAI,CAAC,kCAAmB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACxD,IAAA,sBAAa,EAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,WAAW,CAAC,EAAE,iBAAiB,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;QAChH,IAAA,sBAAa,EAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,GAAG,EAAE,eAAe,CAAC,CAAC;IACrF,CAAC;IAED,MAAM,gBAAgB,GAAG,IAAA,8BAAyB,EAAC,IAAI,CAAC,CAAC;IACzD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IACD,IAAA,mBAAU,EAAC,IAAI,EAAE,gBAAgB,EAAE,YAAY,CAAC,EAAE;QAChD,2HAA2H;QAC3H,IAAI,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAC5D,YAAY,CAAC,eAAe,GAAG;gBAC7B,GAAG,YAAY,CAAC,eAAe;gBAC/B,KAAK,EAAE,CAAC,MAAM,CAAC;gBACf,eAAe,EAAE,IAAI;aACtB,CAAC;QACJ,CAAC;QAED,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,iBAAiB,CAAC,UAAU,CAAC,GAAG;YACjE,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,uBAAuB,CAAC;SAClE,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,mEAAmE;IACnE,IAAI,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC5D,IAAA,mBAAU,EAAC,IAAI,EAAE,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,cAAc,CAAC,EAAE,WAAW,CAAC,EAAE;YAC/F,WAAW,CAAC,OAAO,CAAC,KAAK,GAAG;gBAC1B,QAAQ,EAAE,aAAa;gBACvB,OAAO,EAAE;oBACP,WAAW,EAAE,GAAG,iBAAiB,CAAC,WAAW,QAAQ;oBACrD,OAAO,EAAE,IAAI;iBACd;gBACD,cAAc,EAAE;oBACd,YAAY,EAAE;wBACZ,OAAO,EAAE,aAAa;qBACvB;iBACF;aACF,CAAC;YACF,WAAW,CAAC,OAAO,CAAC,KAAK,GAAG;gBAC1B,QAAQ,EAAE,qBAAqB;gBAC/B,OAAO,EAAE,CAAC,sBAAsB,CAAC;gBACjC,oBAAoB,EAAE,YAAY;gBAClC,OAAO,EAAE;oBACP,QAAQ,EAAE,MAAM;oBAChB,UAAU,EAAE,QAAQ,iBAAiB,CAAC,WAAW,EAAE;oBACnD,MAAM,EAAE,CAAC,KAAK,CAAC;oBACf,MAAM,EAAE,IAAI;oBACZ,UAAU,EAAE,IAAI;oBAChB,QAAQ,EAAE,EAAE;oBACZ,IAAI,EAAE,GAAG,iBAAiB,CAAC,UAAU,WAAW;oBAChD,QAAQ,EAAE,QAAQ,iBAAiB,CAAC,gBAAgB,oBAAoB;oBACxE,cAAc,EAAE;wBACd,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE;4BACZ,KAAK,EAAE,KAAK;yBACb;qBACF;oBACD,cAAc,EAAE;wBACd,WAAW,EAAE,EAAE;wBACf,UAAU,EAAE;4BACV,cAAc,EAAE;gCACd,SAAS,EAAE,KAAK;gCAChB,YAAY,EAAE;oCACZ,KAAK,EAAE,KAAK;iCACb;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;YACF,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iGAAiG;IACjG,MAAM,OAAO,GAAG,IAAI,kBAAO,CAAC;QAC1B,gBAAgB,EAAE,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,oBAAoB;KACnG,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAE5D,QAAQ,CAAC,oBAAoB,CAAC;QAC5B,eAAe,EAAE,iBAAiB,CAAC,UAAU;QAC7C,aAAa,EAAE,iBAAiB,CAAC,gBAAgB;KAClD,CAAC,CAAC;IAEH,kDAAkD;IAClD,MAAM,mBAAmB,GAAG,QAAQ,CAAC,6BAA6B,EAAE,CAAC;IAErE,2DAA2D;IAC3D,MAAM,aAAa,GAAG,mBAAmB,CAAC,gBAAgB,EAAE,IAAI,mBAAmB,CAAC;IAEpF,4CAA4C;IAC5C,MAAM,WAAW,GAAG,aAAa,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC;IAEvD,IAAI,CAAC,8BAAmB,CAAC,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD,MAAM,qBAAqB,GAAG,WAAW,CAAC,cAAc,EAAE,EAAE,aAAa,CAAC,qBAAU,CAAC,uBAAuB,CAAC,CAAC;IAE9G,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,+CAA+C,YAAY,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,iEAAiE;IACjE,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;IAElF,IAAI,kBAAkB,CAAC,OAAO,EAAE,KAAK,qBAAU,CAAC,kBAAkB,EAAE,CAAC;QACnE,MAAM,eAAe,GAAG,kBAAkB,CAAC,0BAA0B,CAAC,qBAAU,CAAC,sBAAsB,CAAC,CAAC;QACzG,eAAe,CAAC,UAAU,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAEtB,IAAI,CAAC,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC7D,IAAI,CAAC,MAAM,CAAC,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IACnF,CAAC;IAED,IAAA,mBAAU,EAAC,IAAI,EAAE,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,oBAAoB,CAAC,EAAE,YAAY,CAAC,EAAE;QACtG,oHAAoH;QACpH,YAAY,CAAC,OAAO,GAAG,CAAC,GAAG,YAAY,CAAC,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,CAAC,CAAC;QAC/F,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,MAAM,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC;IACxB,IAAA,4BAAmB,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAChC,OAAO,GAAG,EAAE;QACV,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,oFAAoF,CAAC,CAAC;IACpG,CAAC,CAAC;AACJ,CAAC,CAAC;AAhMW,QAAA,yBAAyB,6BAgMpC;AACF,MAAM,aAAa,GAAG,KAAK,EAAE,IAAU,EAAE,OAAwC,EAAE,EAAE;IACnF,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,EAAE;aACf,WAAW,CAAC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aACtE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAEtC,MAAM,QAAQ,GAAkB,EAAE,CAAC;QAEnC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YACvB,MAAM,eAAe,GAAG,EAAE;iBACvB,WAAW,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;iBAC1F,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC;YAEtD,IAAI,eAAe,EAAE,CAAC;gBACpB,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,EAAE;qBAClB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;qBAC9F,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBACtC,UAAU,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;oBAC7B,MAAM,qBAAqB,GAAG,EAAE;yBAC7B,WAAW,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,IAAI,SAAS,CAAC,IAAI,EAAE,EAAE;wBACnF,aAAa,EAAE,IAAI;qBACpB,CAAC;yBACD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC;oBAEtD,IAAI,qBAAqB,EAAE,CAAC;wBAC1B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,yBAAyB,CAAC;QACzD,CAAC;aAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;QACpF,CAAC;aAAM,CAAC;YACN,MAAM,YAAY,GAAG;gBACnB,IAAI,EAAE,KAAK;gBACX,OAAO,EAAE,2DAA2D;gBACpE,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;aACpC,CAAC;YACF,MAAM,IAAA,iBAAM,EAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,MAAc,EAAE,EAAE;gBACjD,8CAA8C;gBAC9C,2DAA2D;gBAC3D,OAAO,CAAC,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,kBAAe,iCAAyB,CAAC","sourcesContent":["import {\n  addDependenciesToPackageJson,\n  formatFiles,\n  generateFiles,\n  getWorkspaceLayout,\n  installPackagesTask,\n  joinPathFragments,\n  names,\n  offsetFromRoot,\n  Tree,\n  updateJson,\n} from \"@nx/devkit\";\nimport { Linter } from \"@nx/eslint\";\nimport { getRootTsConfigPathInTree } from \"@nx/js\";\nimport { getNpmScope } from \"@nx/js/src/utils/package-json/get-npm-scope\";\nimport { libraryGenerator } from \"@nx/react\";\nimport { Schema } from \"@nx/react/src/generators/library/schema\";\nimport { nonUiExtensionTypes, serverSideExtensionTypes } from \"@trackunit/iris-app-api\";\nimport { prompt } from \"enquirer\";\nimport * as fs from \"fs\";\nimport * as path from \"path\";\nimport { camelCase } from \"string-ts\";\nimport { Project, SyntaxKind, VariableDeclaration } from \"ts-morph\";\nimport { spaDependencies, testDependencies } from \"./dependencies\";\nimport { IrisAppExtensionGeneratorSchema } from \"./schema\";\n\ninterface NormalizedSchema extends IrisAppExtensionGeneratorSchema {\n  projectName: string;\n  projectRoot: string;\n  projectDirectory: string;\n  projectCamelcase: string;\n  importPath: string;\n  sourceRoot: string;\n  dependencyDefinitionFile: string;\n}\n\nfunction normalizeOptions(tree: Tree, options: IrisAppExtensionGeneratorSchema): NormalizedSchema {\n  const name = names(options.name).fileName;\n  const projectDirectory = options.directory ? `${names(options.directory).fileName}/${name}` : name;\n  const projectName = projectDirectory.replace(new RegExp(\"/\", \"g\"), \"-\");\n  const projectCamelcase = camelCase(projectName);\n  const wsLayout = getWorkspaceLayout(tree);\n  const projectRoot = `${wsLayout.libsDir}/${projectDirectory}`;\n  const sourceRoot = `${wsLayout.libsDir}/${projectDirectory}/src`;\n  const dependencyDefinitionFile = `${wsLayout.libsDir}/${projectDirectory}/package.json`;\n\n  const npmScope = getNpmScope(tree);\n  if (!npmScope) {\n    throw new Error(\"Could not find npm scope in package.json\");\n  }\n  const importPath = `@${npmScope}/${projectName}`;\n\n  return {\n    ...options,\n    projectName,\n    projectRoot,\n    projectDirectory,\n    projectCamelcase,\n    importPath,\n    sourceRoot,\n    dependencyDefinitionFile,\n  };\n}\n\n/**\n * Iris App Extension Generator.\n *\n * @param tree the nx tree\n * @param options the generator options\n * @returns { Promise<void> } void\n */\nexport const IrisAppExtensionGenerator = async function (tree: Tree, options: IrisAppExtensionGeneratorSchema) {\n  const adjustedOptions = await updateAppName(tree, options);\n\n  const normalizedOptions = normalizeOptions(tree, adjustedOptions);\n\n  const templateOptions = {\n    ...normalizedOptions,\n    ...names(normalizedOptions.name),\n    offsetFromRoot: offsetFromRoot(normalizedOptions.projectRoot),\n    template: \"\",\n  };\n  const irisAppExtensionType = normalizedOptions.type;\n\n  if (!tree.exists(`${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}`)) {\n    throw new Error(\n      `The app does not exist: ${getWorkspaceLayout(tree).appsDir}/${\n        normalizedOptions.app\n      } make sure you enter the right app name`\n    );\n  }\n\n  const manifestPath = `${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}/iris-app-manifest.ts`;\n  if (!tree.exists(manifestPath)) {\n    throw new Error(\n      `The app does not have an iris-app-manifest.ts: ${manifestPath} make sure you enter the right app name, which has an iris-app-manifest.ts`\n    );\n  }\n\n  if (normalizedOptions.updatePackageJson) {\n    addDependenciesToPackageJson(\n      tree,\n      {},\n      {\n        ...spaDependencies,\n        ...testDependencies,\n      }\n    );\n  }\n\n  const schema: Schema = {\n    name: normalizedOptions.projectName,\n    style: \"none\",\n    skipFormat: false,\n    strict: true,\n    unitTestRunner: \"jest\",\n    linter: Linter.EsLint,\n    routing: false,\n    directory: normalizedOptions.projectRoot,\n    skipTsConfig: false,\n    skipPackageJson: true,\n  };\n\n  await libraryGenerator(tree, schema);\n\n  generateFiles(\n    tree,\n    path.join(__dirname, \"files\", irisAppExtensionType),\n    normalizedOptions.projectRoot,\n    templateOptions\n  );\n\n  if (!nonUiExtensionTypes.includes(irisAppExtensionType)) {\n    generateFiles(tree, path.join(__dirname, \"files\", \"react-app\"), normalizedOptions.projectRoot, templateOptions);\n    generateFiles(tree, path.join(__dirname, \"files\", \"preset\"), \"/\", templateOptions);\n  }\n\n  const rootTsConfigPath = getRootTsConfigPathInTree(tree);\n  if (!rootTsConfigPath) {\n    throw new Error(\"Could not find root tsconfig.json\");\n  }\n  updateJson(tree, rootTsConfigPath, tsConfigBase => {\n    // For server side extension types, we need to add deno types to the tsconfig and make Typescript accept Deno import syntax\n    if (serverSideExtensionTypes.includes(irisAppExtensionType)) {\n      tsConfigBase.compilerOptions = {\n        ...tsConfigBase.compilerOptions,\n        types: [\"node\"],\n        esModuleInterop: true,\n      };\n    }\n\n    tsConfigBase.compilerOptions.paths[normalizedOptions.importPath] = [\n      path.join(normalizedOptions.projectRoot, \"extension-manifest.ts\"),\n    ];\n    return tsConfigBase;\n  });\n\n  // Add working serve target to the project.json for DENO extensions\n  if (serverSideExtensionTypes.includes(irisAppExtensionType)) {\n    updateJson(tree, joinPathFragments(normalizedOptions.projectRoot, \"project.json\"), projectJson => {\n      projectJson.targets.serve = {\n        executor: \"@nx/js:node\",\n        options: {\n          buildTarget: `${normalizedOptions.projectName}:build`,\n          inspect: true,\n        },\n        configurations: {\n          breakOnStart: {\n            inspect: \"inspect-brk\",\n          },\n        },\n      };\n      projectJson.targets.build = {\n        executor: \"@nx/esbuild:esbuild\",\n        outputs: [\"{options.outputPath}\"],\n        defaultConfiguration: \"production\",\n        options: {\n          platform: \"node\",\n          outputPath: `dist/${normalizedOptions.projectRoot}`,\n          format: [\"cjs\"],\n          bundle: true,\n          thirdParty: true,\n          external: [],\n          main: `${normalizedOptions.sourceRoot}/index.ts`,\n          tsConfig: `libs/${normalizedOptions.projectDirectory}/tsconfig.lib.json`,\n          esbuildOptions: {\n            sourcemap: true,\n            outExtension: {\n              \".js\": \".js\",\n            },\n          },\n          configurations: {\n            development: {},\n            production: {\n              esbuildOptions: {\n                sourcemap: false,\n                outExtension: {\n                  \".js\": \".js\",\n                },\n              },\n            },\n          },\n        },\n      };\n      return projectJson;\n    });\n  }\n\n  //TODO: Change the following to update the manifest in the virtual Tree - not on the file system.\n  const project = new Project({\n    tsConfigFilePath: `${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}/tsconfig.app.json`,\n  });\n\n  const manifest = project.getSourceFileOrThrow(manifestPath);\n\n  manifest.addImportDeclaration({\n    moduleSpecifier: normalizedOptions.importPath,\n    defaultImport: normalizedOptions.projectCamelcase,\n  });\n\n  // Get the default export symbol from the manifest\n  const defaultExportSymbol = manifest.getDefaultExportSymbolOrThrow();\n\n  // Get the aliased symbol if the default export is an alias\n  const aliasedSymbol = defaultExportSymbol.getAliasedSymbol() || defaultExportSymbol;\n\n  // Get the declaration of the aliased symbol\n  const declaration = aliasedSymbol.getDeclarations()[0];\n\n  if (!VariableDeclaration.isVariableDeclaration(declaration)) {\n    throw new Error(\"Expected variable declaration\");\n  }\n  const irisAppManifestObject = declaration.getInitializer()?.asKindOrThrow(SyntaxKind.ObjectLiteralExpression);\n\n  if (!irisAppManifestObject) {\n    throw new Error(`irisAppManifest not exported from the file: ${manifestPath}`);\n  }\n  // Find the extensions property within the irisAppManifest object\n  const extensionsProperty = irisAppManifestObject.getPropertyOrThrow(\"extensions\");\n\n  if (extensionsProperty.getKind() === SyntaxKind.PropertyAssignment) {\n    const extensionsArray = extensionsProperty.getFirstChildByKindOrThrow(SyntaxKind.ArrayLiteralExpression);\n    extensionsArray.addElement(`${normalizedOptions.projectCamelcase}`);\n  }\n\n  await manifest.save();\n\n  if (!serverSideExtensionTypes.includes(irisAppExtensionType)) {\n    tree.delete(joinPathFragments(normalizedOptions.projectRoot, \"src\", \"index.ts\"));\n  }\n\n  updateJson(tree, joinPathFragments(normalizedOptions.projectRoot, \"tsconfig.spec.json\"), tsConfigSpec => {\n    // extension-manifest.ts is a build time file, so we only need to include it in the spec file for type/lint checking\n    tsConfigSpec.include = [...tsConfigSpec.include, \"**/jest.setup.ts*\", \"extension-manifest.ts\"];\n    return tsConfigSpec;\n  });\n\n  await formatFiles(tree);\n  installPackagesTask(tree, true);\n  return () => {\n    // eslint-disable-next-line no-console\n    console.log(\"🥳 Successfully created an app extensions and it is already added to the Iris App!\");\n  };\n};\nconst updateAppName = async (tree: Tree, options: IrisAppExtensionGeneratorSchema) => {\n  if (!options.app) {\n    const folders = fs\n      .readdirSync(getWorkspaceLayout(tree).appsDir, { withFileTypes: true })\n      .filter(item => item.isDirectory());\n\n    const tileApps: Array<string> = [];\n\n    folders.forEach(folder => {\n      const irisAppManifest = fs\n        .readdirSync(`${getWorkspaceLayout(tree).appsDir}/${folder.name}`, { withFileTypes: true })\n        .find(item => item.name === \"iris-app-manifest.ts\");\n\n      if (irisAppManifest) {\n        tileApps.push(folder.name);\n      } else {\n        const subFolders = fs\n          .readdirSync(path.join(getWorkspaceLayout(tree).appsDir, folder.name), { withFileTypes: true })\n          .filter(item => item.isDirectory());\n        subFolders.forEach(subFolder => {\n          const subDirIrisAppManifest = fs\n            .readdirSync(`${getWorkspaceLayout(tree).appsDir}/${folder.name}/${subFolder.name}`, {\n              withFileTypes: true,\n            })\n            .find(item => item.name === \"iris-app-manifest.ts\");\n\n          if (subDirIrisAppManifest) {\n            tileApps.push(path.join(folder.name, subFolder.name));\n          }\n        });\n      }\n    });\n\n    if (tileApps.length === 1) {\n      options.app = tileApps[0] || \"<Missing iris app name>\";\n    } else if (tileApps.length === 0) {\n      throw new Error(\"Please add an iris-app using: nx g @trackunit/iris-app:create \");\n    } else {\n      const promptObject = {\n        name: \"app\",\n        message: \"Select the iris app that this extension should belong to:\",\n        type: \"autocomplete\",\n        choices: tileApps.map(item => item),\n      };\n      await prompt(promptObject).then((values: object) => {\n        // @ts-ignore - suppressImplicitAnyIndexErrors\n        // eslint-disable-next-line @typescript-eslint/dot-notation\n        options.app = values[\"app\"];\n      });\n    }\n  }\n  return options;\n};\n\nexport default IrisAppExtensionGenerator;\n"]}
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../libs/iris-app-sdk/iris-app/src/generators/extend/generator.ts"],"names":[],"mappings":";;;AAqEA,sEAgCC;;AArGD,uCAWoB;AACpB,uCAAoC;AACpC,+BAAmD;AACnD,+EAA0E;AAC1E,qCAA6C;AAE7C,0DAAwF;AACxF,uCAAkC;AAClC,+CAAyB;AACzB,mDAA6B;AAC7B,yCAAsC;AACtC,uCAAoE;AACpE,uDAAoF;AACpF,iDAAmE;AAanE,SAAS,gBAAgB,CAAC,IAAU,EAAE,OAAwC;IAC5E,MAAM,IAAI,GAAG,IAAA,cAAK,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;IAC1C,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,IAAA,cAAK,EAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IACnG,MAAM,WAAW,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC;IACxE,MAAM,gBAAgB,GAAG,IAAA,qBAAS,EAAC,WAAW,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,WAAW,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,EAAE,CAAC;IAC9D,MAAM,UAAU,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,MAAM,CAAC;IACjE,MAAM,wBAAwB,GAAG,GAAG,QAAQ,CAAC,OAAO,IAAI,gBAAgB,eAAe,CAAC;IAExF,MAAM,QAAQ,GAAG,IAAA,2BAAW,EAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,QAAQ,IAAI,WAAW,EAAE,CAAC;IAEjD,OAAO;QACL,GAAG,OAAO;QACV,WAAW;QACX,WAAW;QACX,gBAAgB;QAChB,gBAAgB;QAChB,UAAU;QACV,UAAU;QACV,wBAAwB;KACzB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,6BAA6B,CAAC,IAAU,EAAE,eAAwB;IAChF,MAAM,cAAc,GAAG,gBAAgB,CAAC;IAExC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,qDAAqD;QACrD,IAAA,sBAAa,EAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,GAAG,EAAE,eAAe,IAAI,EAAE,CAAC,CAAC;QACzF,OAAO;IACT,CAAC;IAED,2DAA2D;IAC3D,IAAI,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACxD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CAAC,kBAAkB,cAAc,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,0DAA0D;IAC1D,IAAI,kBAAkB,GAAG,YAAY,CAAC,CAAC,eAAe;IACtD,MAAM,mBAAmB,GAAG,cAAc,CAAC,KAAK,CAC9C,6EAA6E,CAC9E,CAAC;IACF,IAAI,mBAAmB,IAAI,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;QAClD,kBAAkB,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,gDAAgD;IAChD,cAAc,GAAG,IAAA,8BAAmB,EAAC,cAAc,EAAE,kBAAkB,EAAE,oCAAoC,CAAC,CAAC;IAE/G,yDAAyD;IACzD,cAAc,GAAG,IAAA,8BAAmB,EAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;IAEzE,6CAA6C;IAC7C,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;AAC7C,CAAC;AAED;;;;;;GAMG;AACI,MAAM,yBAAyB,GAAG,KAAK,WAAW,IAAU,EAAE,OAAwC;IAC3G,MAAM,eAAe,GAAG,MAAM,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAE3D,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IAElE,MAAM,eAAe,GAAG;QACtB,GAAG,iBAAiB;QACpB,GAAG,IAAA,cAAK,EAAC,iBAAiB,CAAC,IAAI,CAAC;QAChC,cAAc,EAAE,IAAA,uBAAc,EAAC,iBAAiB,CAAC,WAAW,CAAC;QAC7D,QAAQ,EAAE,EAAE;KACb,CAAC;IACF,MAAM,oBAAoB,GAAG,iBAAiB,CAAC,IAAI,CAAC;IAEpD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACjF,MAAM,IAAI,KAAK,CACb,2BAA2B,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IACzD,iBAAiB,CAAC,GACpB,yCAAyC,CAC1C,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,uBAAuB,CAAC;IACzG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CACb,kDAAkD,YAAY,4EAA4E,CAC3I,CAAC;IACJ,CAAC;IAED,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,CAAC;QACxC,IAAA,qCAA4B,EAC1B,IAAI,EACJ,EAAE,EACF;YACE,GAAG,8BAAe;YAClB,GAAG,+BAAgB;SACpB,CACF,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAW;QACrB,IAAI,EAAE,iBAAiB,CAAC,WAAW;QACnC,KAAK,EAAE,MAAM;QACb,UAAU,EAAE,KAAK;QACjB,MAAM,EAAE,IAAI;QACZ,cAAc,EAAE,MAAM;QACtB,MAAM,EAAE,eAAM,CAAC,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,iBAAiB,CAAC,WAAW;QACxC,YAAY,EAAE,KAAK;QACnB,eAAe,EAAE,IAAI;KACtB,CAAC;IAEF,MAAM,IAAA,wBAAgB,EAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAErC,IAAA,sBAAa,EACX,IAAI,EACJ,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,oBAAoB,CAAC,EACnD,iBAAiB,CAAC,WAAW,EAC7B,eAAe,CAChB,CAAC;IAEF,IAAI,CAAC,kCAAmB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACxD,IAAA,sBAAa,EAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,WAAW,CAAC,EAAE,iBAAiB,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;QAChH,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;YACvC,6BAA6B,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,MAAM,gBAAgB,GAAG,IAAA,8BAAyB,EAAC,IAAI,CAAC,CAAC;IACzD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IACD,IAAA,mBAAU,EAAC,IAAI,EAAE,gBAAgB,EAAE,YAAY,CAAC,EAAE;QAChD,2HAA2H;QAC3H,IAAI,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAC5D,YAAY,CAAC,eAAe,GAAG;gBAC7B,GAAG,YAAY,CAAC,eAAe;gBAC/B,KAAK,EAAE,CAAC,MAAM,CAAC;gBACf,eAAe,EAAE,IAAI;aACtB,CAAC;QACJ,CAAC;QAED,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,iBAAiB,CAAC,UAAU,CAAC,GAAG;YACjE,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,uBAAuB,CAAC;SAClE,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,mEAAmE;IACnE,IAAI,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC5D,IAAA,mBAAU,EAAC,IAAI,EAAE,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,cAAc,CAAC,EAAE,WAAW,CAAC,EAAE;YAC/F,WAAW,CAAC,OAAO,CAAC,KAAK,GAAG;gBAC1B,QAAQ,EAAE,aAAa;gBACvB,OAAO,EAAE;oBACP,WAAW,EAAE,GAAG,iBAAiB,CAAC,WAAW,QAAQ;oBACrD,OAAO,EAAE,IAAI;iBACd;gBACD,cAAc,EAAE;oBACd,YAAY,EAAE;wBACZ,OAAO,EAAE,aAAa;qBACvB;iBACF;aACF,CAAC;YACF,WAAW,CAAC,OAAO,CAAC,KAAK,GAAG;gBAC1B,QAAQ,EAAE,qBAAqB;gBAC/B,OAAO,EAAE,CAAC,sBAAsB,CAAC;gBACjC,oBAAoB,EAAE,YAAY;gBAClC,OAAO,EAAE;oBACP,QAAQ,EAAE,MAAM;oBAChB,UAAU,EAAE,QAAQ,iBAAiB,CAAC,WAAW,EAAE;oBACnD,MAAM,EAAE,CAAC,KAAK,CAAC;oBACf,MAAM,EAAE,IAAI;oBACZ,UAAU,EAAE,IAAI;oBAChB,QAAQ,EAAE,EAAE;oBACZ,IAAI,EAAE,GAAG,iBAAiB,CAAC,UAAU,WAAW;oBAChD,QAAQ,EAAE,QAAQ,iBAAiB,CAAC,gBAAgB,oBAAoB;oBACxE,cAAc,EAAE;wBACd,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE;4BACZ,KAAK,EAAE,KAAK;yBACb;qBACF;oBACD,cAAc,EAAE;wBACd,WAAW,EAAE,EAAE;wBACf,UAAU,EAAE;4BACV,cAAc,EAAE;gCACd,SAAS,EAAE,KAAK;gCAChB,YAAY,EAAE;oCACZ,KAAK,EAAE,KAAK;iCACb;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;YACF,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iGAAiG;IACjG,MAAM,OAAO,GAAG,IAAI,kBAAO,CAAC;QAC1B,gBAAgB,EAAE,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,iBAAiB,CAAC,GAAG,oBAAoB;KACnG,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAE5D,QAAQ,CAAC,oBAAoB,CAAC;QAC5B,eAAe,EAAE,iBAAiB,CAAC,UAAU;QAC7C,aAAa,EAAE,iBAAiB,CAAC,gBAAgB;KAClD,CAAC,CAAC;IAEH,kDAAkD;IAClD,MAAM,mBAAmB,GAAG,QAAQ,CAAC,6BAA6B,EAAE,CAAC;IAErE,2DAA2D;IAC3D,MAAM,aAAa,GAAG,mBAAmB,CAAC,gBAAgB,EAAE,IAAI,mBAAmB,CAAC;IAEpF,4CAA4C;IAC5C,MAAM,WAAW,GAAG,aAAa,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC;IAEvD,IAAI,CAAC,8BAAmB,CAAC,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD,MAAM,qBAAqB,GAAG,WAAW,CAAC,cAAc,EAAE,EAAE,aAAa,CAAC,qBAAU,CAAC,uBAAuB,CAAC,CAAC;IAE9G,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,+CAA+C,YAAY,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,iEAAiE;IACjE,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;IAElF,IAAI,kBAAkB,CAAC,OAAO,EAAE,KAAK,qBAAU,CAAC,kBAAkB,EAAE,CAAC;QACnE,MAAM,eAAe,GAAG,kBAAkB,CAAC,0BAA0B,CAAC,qBAAU,CAAC,sBAAsB,CAAC,CAAC;QACzG,eAAe,CAAC,UAAU,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAEtB,IAAI,CAAC,uCAAwB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC7D,IAAI,CAAC,MAAM,CAAC,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IACnF,CAAC;IAED,IAAA,mBAAU,EAAC,IAAI,EAAE,IAAA,0BAAiB,EAAC,iBAAiB,CAAC,WAAW,EAAE,oBAAoB,CAAC,EAAE,YAAY,CAAC,EAAE;QACtG,oHAAoH;QACpH,YAAY,CAAC,OAAO,GAAG,CAAC,GAAG,YAAY,CAAC,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,CAAC,CAAC;QAC/F,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,MAAM,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC;IACxB,IAAA,4BAAmB,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAChC,OAAO,GAAG,EAAE;QACV,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,oFAAoF,CAAC,CAAC;IACpG,CAAC,CAAC;AACJ,CAAC,CAAC;AAlMW,QAAA,yBAAyB,6BAkMpC;AACF,MAAM,aAAa,GAAG,KAAK,EAAE,IAAU,EAAE,OAAwC,EAAE,EAAE;IACnF,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,EAAE;aACf,WAAW,CAAC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aACtE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAEtC,MAAM,QAAQ,GAAkB,EAAE,CAAC;QAEnC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YACvB,MAAM,eAAe,GAAG,EAAE;iBACvB,WAAW,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;iBAC1F,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC;YAEtD,IAAI,eAAe,EAAE,CAAC;gBACpB,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,EAAE;qBAClB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;qBAC9F,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBACtC,UAAU,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;oBAC7B,MAAM,qBAAqB,GAAG,EAAE;yBAC7B,WAAW,CAAC,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,IAAI,SAAS,CAAC,IAAI,EAAE,EAAE;wBACnF,aAAa,EAAE,IAAI;qBACpB,CAAC;yBACD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC;oBAEtD,IAAI,qBAAqB,EAAE,CAAC;wBAC1B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,yBAAyB,CAAC;QACzD,CAAC;aAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;QACpF,CAAC;aAAM,CAAC;YACN,MAAM,YAAY,GAAG;gBACnB,IAAI,EAAE,KAAK;gBACX,OAAO,EAAE,2DAA2D;gBACpE,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;aACpC,CAAC;YACF,MAAM,IAAA,iBAAM,EAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,MAAc,EAAE,EAAE;gBACjD,8CAA8C;gBAC9C,2DAA2D;gBAC3D,OAAO,CAAC,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,kBAAe,iCAAyB,CAAC","sourcesContent":["import {\n  addDependenciesToPackageJson,\n  formatFiles,\n  generateFiles,\n  getWorkspaceLayout,\n  installPackagesTask,\n  joinPathFragments,\n  names,\n  offsetFromRoot,\n  Tree,\n  updateJson,\n} from \"@nx/devkit\";\nimport { Linter } from \"@nx/eslint\";\nimport { getRootTsConfigPathInTree } from \"@nx/js\";\nimport { getNpmScope } from \"@nx/js/src/utils/package-json/get-npm-scope\";\nimport { libraryGenerator } from \"@nx/react\";\nimport { Schema } from \"@nx/react/src/generators/library/schema\";\nimport { nonUiExtensionTypes, serverSideExtensionTypes } from \"@trackunit/iris-app-api\";\nimport { prompt } from \"enquirer\";\nimport * as fs from \"fs\";\nimport * as path from \"path\";\nimport { camelCase } from \"string-ts\";\nimport { Project, SyntaxKind, VariableDeclaration } from \"ts-morph\";\nimport { addRequireStatement, addSpreadAssignment } from \"../../utils/ast/astUtils\";\nimport { spaDependencies, testDependencies } from \"./dependencies\";\nimport { IrisAppExtensionGeneratorSchema } from \"./schema\";\n\ninterface NormalizedSchema extends IrisAppExtensionGeneratorSchema {\n  projectName: string;\n  projectRoot: string;\n  projectDirectory: string;\n  projectCamelcase: string;\n  importPath: string;\n  sourceRoot: string;\n  dependencyDefinitionFile: string;\n}\n\nfunction normalizeOptions(tree: Tree, options: IrisAppExtensionGeneratorSchema): NormalizedSchema {\n  const name = names(options.name).fileName;\n  const projectDirectory = options.directory ? `${names(options.directory).fileName}/${name}` : name;\n  const projectName = projectDirectory.replace(new RegExp(\"/\", \"g\"), \"-\");\n  const projectCamelcase = camelCase(projectName);\n  const wsLayout = getWorkspaceLayout(tree);\n  const projectRoot = `${wsLayout.libsDir}/${projectDirectory}`;\n  const sourceRoot = `${wsLayout.libsDir}/${projectDirectory}/src`;\n  const dependencyDefinitionFile = `${wsLayout.libsDir}/${projectDirectory}/package.json`;\n\n  const npmScope = getNpmScope(tree);\n  if (!npmScope) {\n    throw new Error(\"Could not find npm scope in package.json\");\n  }\n  const importPath = `@${npmScope}/${projectName}`;\n\n  return {\n    ...options,\n    projectName,\n    projectRoot,\n    projectDirectory,\n    projectCamelcase,\n    importPath,\n    sourceRoot,\n    dependencyDefinitionFile,\n  };\n}\n\n/**\n * Ensures that jest.preset.js exists and includes the required @trackunit/react-test-setup preset.\n * If the file doesn't exist, creates it. If it exists but doesn't include the preset, updates it.\n */\nexport function ensureJestPresetConfiguration(tree: Tree, templateOptions?: object): void {\n  const jestPresetPath = \"jest.preset.js\";\n\n  if (!tree.exists(jestPresetPath)) {\n    // File doesn't exist, generate it using the template\n    generateFiles(tree, path.join(__dirname, \"files\", \"preset\"), \"/\", templateOptions || {});\n    return;\n  }\n\n  // File exists, use AST utilities for reliable manipulation\n  let currentContent = tree.read(jestPresetPath, \"utf-8\");\n  if (!currentContent) {\n    throw new Error(`Could not read ${jestPresetPath}`);\n  }\n\n  // Extract variable name from existing import if it exists\n  let presetVariableName = \"irisPreset\"; // default name\n  const existingImportMatch = currentContent.match(\n    /const\\s+(\\w+)\\s*=\\s*require\\([\"']@trackunit\\/react-test-setup\\/preset[\"']\\)/\n  );\n  if (existingImportMatch && existingImportMatch[1]) {\n    presetVariableName = existingImportMatch[1];\n  }\n\n  // Add the require statement if it doesn't exist\n  currentContent = addRequireStatement(currentContent, presetVariableName, \"@trackunit/react-test-setup/preset\");\n\n  // Add the spread assignment to the module.exports object\n  currentContent = addSpreadAssignment(currentContent, presetVariableName);\n\n  // Write the updated content back to the tree\n  tree.write(jestPresetPath, currentContent);\n}\n\n/**\n * Iris App Extension Generator.\n *\n * @param tree the nx tree\n * @param options the generator options\n * @returns { Promise<void> } void\n */\nexport const IrisAppExtensionGenerator = async function (tree: Tree, options: IrisAppExtensionGeneratorSchema) {\n  const adjustedOptions = await updateAppName(tree, options);\n\n  const normalizedOptions = normalizeOptions(tree, adjustedOptions);\n\n  const templateOptions = {\n    ...normalizedOptions,\n    ...names(normalizedOptions.name),\n    offsetFromRoot: offsetFromRoot(normalizedOptions.projectRoot),\n    template: \"\",\n  };\n  const irisAppExtensionType = normalizedOptions.type;\n\n  if (!tree.exists(`${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}`)) {\n    throw new Error(\n      `The app does not exist: ${getWorkspaceLayout(tree).appsDir}/${\n        normalizedOptions.app\n      } make sure you enter the right app name`\n    );\n  }\n\n  const manifestPath = `${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}/iris-app-manifest.ts`;\n  if (!tree.exists(manifestPath)) {\n    throw new Error(\n      `The app does not have an iris-app-manifest.ts: ${manifestPath} make sure you enter the right app name, which has an iris-app-manifest.ts`\n    );\n  }\n\n  if (normalizedOptions.updatePackageJson) {\n    addDependenciesToPackageJson(\n      tree,\n      {},\n      {\n        ...spaDependencies,\n        ...testDependencies,\n      }\n    );\n  }\n\n  const schema: Schema = {\n    name: normalizedOptions.projectName,\n    style: \"none\",\n    skipFormat: false,\n    strict: true,\n    unitTestRunner: \"jest\",\n    linter: Linter.EsLint,\n    routing: false,\n    directory: normalizedOptions.projectRoot,\n    skipTsConfig: false,\n    skipPackageJson: true,\n  };\n\n  await libraryGenerator(tree, schema);\n\n  generateFiles(\n    tree,\n    path.join(__dirname, \"files\", irisAppExtensionType),\n    normalizedOptions.projectRoot,\n    templateOptions\n  );\n\n  if (!nonUiExtensionTypes.includes(irisAppExtensionType)) {\n    generateFiles(tree, path.join(__dirname, \"files\", \"react-app\"), normalizedOptions.projectRoot, templateOptions);\n    if (normalizedOptions.extendJestPreset) {\n      ensureJestPresetConfiguration(tree, templateOptions);\n    }\n  }\n\n  const rootTsConfigPath = getRootTsConfigPathInTree(tree);\n  if (!rootTsConfigPath) {\n    throw new Error(\"Could not find root tsconfig.json\");\n  }\n  updateJson(tree, rootTsConfigPath, tsConfigBase => {\n    // For server side extension types, we need to add deno types to the tsconfig and make Typescript accept Deno import syntax\n    if (serverSideExtensionTypes.includes(irisAppExtensionType)) {\n      tsConfigBase.compilerOptions = {\n        ...tsConfigBase.compilerOptions,\n        types: [\"node\"],\n        esModuleInterop: true,\n      };\n    }\n\n    tsConfigBase.compilerOptions.paths[normalizedOptions.importPath] = [\n      path.join(normalizedOptions.projectRoot, \"extension-manifest.ts\"),\n    ];\n    return tsConfigBase;\n  });\n\n  // Add working serve target to the project.json for DENO extensions\n  if (serverSideExtensionTypes.includes(irisAppExtensionType)) {\n    updateJson(tree, joinPathFragments(normalizedOptions.projectRoot, \"project.json\"), projectJson => {\n      projectJson.targets.serve = {\n        executor: \"@nx/js:node\",\n        options: {\n          buildTarget: `${normalizedOptions.projectName}:build`,\n          inspect: true,\n        },\n        configurations: {\n          breakOnStart: {\n            inspect: \"inspect-brk\",\n          },\n        },\n      };\n      projectJson.targets.build = {\n        executor: \"@nx/esbuild:esbuild\",\n        outputs: [\"{options.outputPath}\"],\n        defaultConfiguration: \"production\",\n        options: {\n          platform: \"node\",\n          outputPath: `dist/${normalizedOptions.projectRoot}`,\n          format: [\"cjs\"],\n          bundle: true,\n          thirdParty: true,\n          external: [],\n          main: `${normalizedOptions.sourceRoot}/index.ts`,\n          tsConfig: `libs/${normalizedOptions.projectDirectory}/tsconfig.lib.json`,\n          esbuildOptions: {\n            sourcemap: true,\n            outExtension: {\n              \".js\": \".js\",\n            },\n          },\n          configurations: {\n            development: {},\n            production: {\n              esbuildOptions: {\n                sourcemap: false,\n                outExtension: {\n                  \".js\": \".js\",\n                },\n              },\n            },\n          },\n        },\n      };\n      return projectJson;\n    });\n  }\n\n  //TODO: Change the following to update the manifest in the virtual Tree - not on the file system.\n  const project = new Project({\n    tsConfigFilePath: `${getWorkspaceLayout(tree).appsDir}/${normalizedOptions.app}/tsconfig.app.json`,\n  });\n\n  const manifest = project.getSourceFileOrThrow(manifestPath);\n\n  manifest.addImportDeclaration({\n    moduleSpecifier: normalizedOptions.importPath,\n    defaultImport: normalizedOptions.projectCamelcase,\n  });\n\n  // Get the default export symbol from the manifest\n  const defaultExportSymbol = manifest.getDefaultExportSymbolOrThrow();\n\n  // Get the aliased symbol if the default export is an alias\n  const aliasedSymbol = defaultExportSymbol.getAliasedSymbol() || defaultExportSymbol;\n\n  // Get the declaration of the aliased symbol\n  const declaration = aliasedSymbol.getDeclarations()[0];\n\n  if (!VariableDeclaration.isVariableDeclaration(declaration)) {\n    throw new Error(\"Expected variable declaration\");\n  }\n  const irisAppManifestObject = declaration.getInitializer()?.asKindOrThrow(SyntaxKind.ObjectLiteralExpression);\n\n  if (!irisAppManifestObject) {\n    throw new Error(`irisAppManifest not exported from the file: ${manifestPath}`);\n  }\n  // Find the extensions property within the irisAppManifest object\n  const extensionsProperty = irisAppManifestObject.getPropertyOrThrow(\"extensions\");\n\n  if (extensionsProperty.getKind() === SyntaxKind.PropertyAssignment) {\n    const extensionsArray = extensionsProperty.getFirstChildByKindOrThrow(SyntaxKind.ArrayLiteralExpression);\n    extensionsArray.addElement(`${normalizedOptions.projectCamelcase}`);\n  }\n\n  await manifest.save();\n\n  if (!serverSideExtensionTypes.includes(irisAppExtensionType)) {\n    tree.delete(joinPathFragments(normalizedOptions.projectRoot, \"src\", \"index.ts\"));\n  }\n\n  updateJson(tree, joinPathFragments(normalizedOptions.projectRoot, \"tsconfig.spec.json\"), tsConfigSpec => {\n    // extension-manifest.ts is a build time file, so we only need to include it in the spec file for type/lint checking\n    tsConfigSpec.include = [...tsConfigSpec.include, \"**/jest.setup.ts*\", \"extension-manifest.ts\"];\n    return tsConfigSpec;\n  });\n\n  await formatFiles(tree);\n  installPackagesTask(tree, true);\n  return () => {\n    // eslint-disable-next-line no-console\n    console.log(\"🥳 Successfully created an app extensions and it is already added to the Iris App!\");\n  };\n};\nconst updateAppName = async (tree: Tree, options: IrisAppExtensionGeneratorSchema) => {\n  if (!options.app) {\n    const folders = fs\n      .readdirSync(getWorkspaceLayout(tree).appsDir, { withFileTypes: true })\n      .filter(item => item.isDirectory());\n\n    const tileApps: Array<string> = [];\n\n    folders.forEach(folder => {\n      const irisAppManifest = fs\n        .readdirSync(`${getWorkspaceLayout(tree).appsDir}/${folder.name}`, { withFileTypes: true })\n        .find(item => item.name === \"iris-app-manifest.ts\");\n\n      if (irisAppManifest) {\n        tileApps.push(folder.name);\n      } else {\n        const subFolders = fs\n          .readdirSync(path.join(getWorkspaceLayout(tree).appsDir, folder.name), { withFileTypes: true })\n          .filter(item => item.isDirectory());\n        subFolders.forEach(subFolder => {\n          const subDirIrisAppManifest = fs\n            .readdirSync(`${getWorkspaceLayout(tree).appsDir}/${folder.name}/${subFolder.name}`, {\n              withFileTypes: true,\n            })\n            .find(item => item.name === \"iris-app-manifest.ts\");\n\n          if (subDirIrisAppManifest) {\n            tileApps.push(path.join(folder.name, subFolder.name));\n          }\n        });\n      }\n    });\n\n    if (tileApps.length === 1) {\n      options.app = tileApps[0] || \"<Missing iris app name>\";\n    } else if (tileApps.length === 0) {\n      throw new Error(\"Please add an iris-app using: nx g @trackunit/iris-app:create \");\n    } else {\n      const promptObject = {\n        name: \"app\",\n        message: \"Select the iris app that this extension should belong to:\",\n        type: \"autocomplete\",\n        choices: tileApps.map(item => item),\n      };\n      await prompt(promptObject).then((values: object) => {\n        // @ts-ignore - suppressImplicitAnyIndexErrors\n        // eslint-disable-next-line @typescript-eslint/dot-notation\n        options.app = values[\"app\"];\n      });\n    }\n  }\n  return options;\n};\n\nexport default IrisAppExtensionGenerator;\n"]}

package/src/generators/extend/schema.d.ts

@@ -16,4 +16,5 @@ export interface IrisAppExtensionGeneratorSchema {
     | "LIFECYCLE_EXTENSION"
     | "SERVERLESS_FUNCTION_EXTENSION"
     | "AI_AGENT_EXTENSION";
+  extendJestPreset?: boolean;
 }

package/src/generators/extend/schema.json

@@ -59,6 +59,11 @@
           "SERVERLESS_FUNCTION_EXTENSION"
         ]
       }
+    },
+    "extendJestPreset": {
+      "type": "boolean",
+      "description": "If the root jest.preset.js should be extended with the Iris app preset",
+      "default": true
     }
   },
   "required": ["name", "directory", "type"]

package/src/utils/ast/astUtils.d.ts

@@ -15,3 +15,18 @@ export declare function removePropertyWithName(fileContent: string, ...propertyN
  * @param {string | string[]} propertyValue - the value of the property to add
  */
 export declare function addPropertyWithName(fileContent: string, propertyName: string, propertyValue: string | Array<string> | Record<string, string>): string;
+/**
+ * This function adds a spread assignment to an object literal.
+ *
+ * @param {string} fileContent - the source code as text
+ * @param {string} variableName - the name of the variable to spread
+ */
+export declare function addSpreadAssignment(fileContent: string, variableName: string): string;
+/**
+ * This function adds a require statement to a file after existing require statements.
+ *
+ * @param {string} fileContent - the source code as text
+ * @param {string} variableName - the name of the variable to assign to
+ * @param {string} modulePath - the module path to require
+ */
+export declare function addRequireStatement(fileContent: string, variableName: string, modulePath: string): string;

package/src/utils/ast/astUtils.js

@@ -2,6 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.removePropertyWithName = removePropertyWithName;
 exports.addPropertyWithName = addPropertyWithName;
+exports.addSpreadAssignment = addSpreadAssignment;
+exports.addRequireStatement = addRequireStatement;
 const tslib_1 = require("tslib");
 const shared_utils_1 = require("@trackunit/shared-utils");
 const tsquery_1 = require("@phenomnomnominal/tsquery");
@@ -68,6 +70,77 @@ function addPropertyWithName(fileContent, propertyName, propertyValue) {
     });
     return ts.createPrinter().printFile(updatedSourceFile);
 }
+/**
+ * This function adds a spread assignment to an object literal.
+ *
+ * @param {string} fileContent - the source code as text
+ * @param {string} variableName - the name of the variable to spread
+ */
+function addSpreadAssignment(fileContent, variableName) {
+    const sourceFile = tsquery_1.tsquery.ast(fileContent);
+    // First check if the spread already exists
+    const existingSpreads = (0, tsquery_1.tsquery)(sourceFile, "SpreadAssignment");
+    const hasExistingSpread = existingSpreads.some(spread => {
+        if (ts.isSpreadAssignment(spread)) {
+            return spread.expression.getText() === variableName;
+        }
+        return false;
+    });
+    if (hasExistingSpread) {
+        return fileContent; // Already has the spread, no changes needed
+    }
+    const updatedSourceFile = tsquery_1.tsquery.map(sourceFile, "ObjectLiteralExpression", node => {
+        if (ts.isObjectLiteralExpression(node)) {
+            // Add the spread assignment to the object literal
+            const spreadAssignment = ts.factory.createSpreadAssignment(ts.factory.createIdentifier(variableName));
+            const updatedProperties = [...node.properties, spreadAssignment];
+            return ts.factory.createObjectLiteralExpression(updatedProperties, true);
+        }
+        return node;
+    });
+    return ts.createPrinter().printFile(updatedSourceFile);
+}
+/**
+ * This function adds a require statement to a file after existing require statements.
+ *
+ * @param {string} fileContent - the source code as text
+ * @param {string} variableName - the name of the variable to assign to
+ * @param {string} modulePath - the module path to require
+ */
+function addRequireStatement(fileContent, variableName, modulePath) {
+    const sourceFile = tsquery_1.tsquery.ast(fileContent);
+    // Check if the require statement already exists
+    const existingRequires = (0, tsquery_1.tsquery)(sourceFile, "CallExpression");
+    const hasExistingRequire = existingRequires.some(call => {
+        if (ts.isCallExpression(call) &&
+            ts.isIdentifier(call.expression) &&
+            call.expression.text === "require" &&
+            call.arguments.length > 0) {
+            const firstArg = call.arguments[0];
+            if (firstArg && ts.isStringLiteral(firstArg)) {
+                return firstArg.text === modulePath;
+            }
+        }
+        return false;
+    });
+    if (hasExistingRequire) {
+        return fileContent; // Already has the require, no changes needed
+    }
+    // Find all variable statements with require calls
+    const requireStatements = (0, tsquery_1.tsquery)(sourceFile, "VariableStatement")
+        .filter(stmt => stmt.getText().includes("require("));
+    const requireStatement = `const ${variableName} = require("${modulePath}").default;`;
+    if (requireStatements.length > 0) {
+        // Add after the last require statement
+        const lastRequire = requireStatements[requireStatements.length - 1];
+        if (lastRequire) {
+            const lastRequireEnd = lastRequire.getEnd();
+            return fileContent.slice(0, lastRequireEnd) + '\n' + requireStatement + fileContent.slice(lastRequireEnd);
+        }
+    }
+    // Add at the beginning if no require statements found
+    return requireStatement + '\n' + fileContent;
+}
 function createObjectLiteralElementLike(obj) {
     const properties = [];
     for (const [key, value] of (0, shared_utils_1.objectEntries)(obj)) {

package/src/utils/ast/astUtils.js.map

@@ -1 +1 @@
-{"version":3,"file":"astUtils.js","sourceRoot":"","sources":["../../../../../../../libs/iris-app-sdk/iris-app/src/utils/ast/astUtils.ts"],"names":[],"mappings":";;AAWA,wDAiBC;AAUD,kDAwCC;;AA9ED,0DAAwD;AACxD,uDAAoD;AACpD,uDAAiC;AAEjC;;;;;;GAMG;AACH,SAAgB,sBAAsB,CAAC,WAAmB,EAAE,GAAG,aAA4B;IACzF,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,iBAAiB,GAAG,iBAAO,CAAC,GAAG,CAAC,UAAU,EAAE,yBAAyB,EAAE,IAAI,CAAC,EAAE;QAClF,IAAI,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,gCAAgC;YAChC,MAAM,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE;gBAC1D,OAAO,CAAC,CACN,EAAE,CAAC,oBAAoB,CAAC,QAAQ,CAAC;oBACjC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAC9B,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;AACzD,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,mBAAmB,CACjC,WAAmB,EACnB,YAAoB,EACpB,aAA8D;IAE9D,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,iBAAiB,GAAG,iBAAO,CAAC,GAAG,CAAC,UAAU,EAAE,yBAAyB,EAAE,IAAI,CAAC,EAAE;QAClF,IAAI,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,2CAA2C;YAC3C,IAAI,iBAAiB,CAAC;YACtB,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACtC,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CACjC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EACzC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAC9C;iBACF,CAAC;YACJ,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACxC,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,4BAA4B,CAC1D,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAClE,CAAC;gBACF,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,YAAY,CAAC;iBAC7F,CAAC;YACJ,CAAC;iBAAM,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBAC7C,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CACjC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EACzC,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,8BAA8B,CAAC,aAAa,CAAC,CAAC,CACxF;iBACF,CAAC;YACJ,CAAC;YACD,OAAO,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,8BAA8B,CAAC,GAA2B;IACjE,MAAM,UAAU,GAAuC,EAAE,CAAC;IAC1D,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAA,4BAAa,EAAC,GAAG,CAAC,EAAE,CAAC;QAC9C,UAAU,CAAC,IAAI,CACb,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAChH,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC","sourcesContent":["import { objectEntries } from \"@trackunit/shared-utils\";\nimport { tsquery } from \"@phenomnomnominal/tsquery\";\nimport * as ts from \"typescript\";\n\n/**\n * This function removes a property with a given name from an object literal\n * provided as a text string.\n *\n * @param {string} fileContent The source code as text\n * @param {string} propertyNames The properties to remove from the object\n */\nexport function removePropertyWithName(fileContent: string, ...propertyNames: Array<string>) {\n  const sourceFile = tsquery.ast(fileContent);\n  const updatedSourceFile = tsquery.map(sourceFile, \"ObjectLiteralExpression\", node => {\n    if (ts.isObjectLiteralExpression(node)) {\n      // Filter out the given property\n      const updatedProperties = node.properties.filter(property => {\n        return !(\n          ts.isPropertyAssignment(property) &&\n          ts.isIdentifier(property.name) &&\n          propertyNames.includes(property.name.text)\n        );\n      });\n      return ts.factory.createObjectLiteralExpression(updatedProperties, true);\n    }\n    return node;\n  });\n  return ts.createPrinter().printFile(updatedSourceFile);\n}\n\n/**\n * This function adds a new property with a given name and value to an object literal\n * provided as a text string.\n *\n * @param {string} fileContent - the source code as text\n * @param {string} propertyName - the name of the property to add\n * @param {string | string[]} propertyValue - the value of the property to add\n */\nexport function addPropertyWithName(\n  fileContent: string,\n  propertyName: string,\n  propertyValue: string | Array<string> | Record<string, string>\n) {\n  const sourceFile = tsquery.ast(fileContent);\n  const updatedSourceFile = tsquery.map(sourceFile, \"ObjectLiteralExpression\", node => {\n    if (ts.isObjectLiteralExpression(node)) {\n      // Add a new property to the object literal\n      let updatedProperties;\n      if (typeof propertyValue === \"string\") {\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(\n            ts.factory.createIdentifier(propertyName),\n            ts.factory.createStringLiteral(propertyValue)\n          ),\n        ];\n      } else if (Array.isArray(propertyValue)) {\n        const arrayLiteral = ts.factory.createArrayLiteralExpression(\n          propertyValue.map(value => ts.factory.createStringLiteral(value))\n        );\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(ts.factory.createIdentifier(propertyName), arrayLiteral),\n        ];\n      } else if (typeof propertyValue === \"object\") {\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(\n            ts.factory.createIdentifier(propertyName),\n            ts.factory.createObjectLiteralExpression(createObjectLiteralElementLike(propertyValue))\n          ),\n        ];\n      }\n      return ts.factory.createObjectLiteralExpression(updatedProperties, true);\n    }\n    return node;\n  });\n  return ts.createPrinter().printFile(updatedSourceFile);\n}\n\nfunction createObjectLiteralElementLike(obj: Record<string, string>): Array<ts.ObjectLiteralElementLike> {\n  const properties: Array<ts.ObjectLiteralElementLike> = [];\n  for (const [key, value] of objectEntries(obj)) {\n    properties.push(\n      ts.factory.createPropertyAssignment(ts.factory.createStringLiteral(key), ts.factory.createStringLiteral(value))\n    );\n  }\n  return properties;\n}\n"]}
+{"version":3,"file":"astUtils.js","sourceRoot":"","sources":["../../../../../../../libs/iris-app-sdk/iris-app/src/utils/ast/astUtils.ts"],"names":[],"mappings":";;AAWA,wDAiBC;AAUD,kDAwCC;AAQD,kDA4BC;AASD,kDAwCC;;AAnKD,0DAAwD;AACxD,uDAAoD;AACpD,uDAAiC;AAEjC;;;;;;GAMG;AACH,SAAgB,sBAAsB,CAAC,WAAmB,EAAE,GAAG,aAA4B;IACzF,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,iBAAiB,GAAG,iBAAO,CAAC,GAAG,CAAC,UAAU,EAAE,yBAAyB,EAAE,IAAI,CAAC,EAAE;QAClF,IAAI,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,gCAAgC;YAChC,MAAM,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE;gBAC1D,OAAO,CAAC,CACN,EAAE,CAAC,oBAAoB,CAAC,QAAQ,CAAC;oBACjC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAC9B,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;AACzD,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,mBAAmB,CACjC,WAAmB,EACnB,YAAoB,EACpB,aAA8D;IAE9D,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,iBAAiB,GAAG,iBAAO,CAAC,GAAG,CAAC,UAAU,EAAE,yBAAyB,EAAE,IAAI,CAAC,EAAE;QAClF,IAAI,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,2CAA2C;YAC3C,IAAI,iBAAiB,CAAC;YACtB,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACtC,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CACjC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EACzC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAC9C;iBACF,CAAC;YACJ,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACxC,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,4BAA4B,CAC1D,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAClE,CAAC;gBACF,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,YAAY,CAAC;iBAC7F,CAAC;YACJ,CAAC;iBAAM,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBAC7C,iBAAiB,GAAG;oBAClB,GAAG,IAAI,CAAC,UAAU;oBAClB,EAAE,CAAC,OAAO,CAAC,wBAAwB,CACjC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,EACzC,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,8BAA8B,CAAC,aAAa,CAAC,CAAC,CACxF;iBACF,CAAC;YACJ,CAAC;YACD,OAAO,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;AACzD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,mBAAmB,CAAC,WAAmB,EAAE,YAAoB;IAC3E,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAE5C,2CAA2C;IAC3C,MAAM,eAAe,GAAG,IAAA,iBAAO,EAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;IAChE,MAAM,iBAAiB,GAAG,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE;QACtD,IAAI,EAAE,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC;YAClC,OAAO,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,YAAY,CAAC;QACtD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO,WAAW,CAAC,CAAC,4CAA4C;IAClE,CAAC;IAED,MAAM,iBAAiB,GAAG,iBAAO,CAAC,GAAG,CAAC,UAAU,EAAE,yBAAyB,EAAE,IAAI,CAAC,EAAE;QAClF,IAAI,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,kDAAkD;YAClD,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,sBAAsB,CACxD,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAC1C,CAAC;YACF,MAAM,iBAAiB,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;YACjE,OAAO,EAAE,CAAC,OAAO,CAAC,6BAA6B,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;AACzD,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CAAC,WAAmB,EAAE,YAAoB,EAAE,UAAkB;IAC/F,MAAM,UAAU,GAAG,iBAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAE5C,gDAAgD;IAChD,MAAM,gBAAgB,GAAG,IAAA,iBAAO,EAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;IAC/D,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACtD,IAAI,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACzB,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC;YAChC,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS;YAClC,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACnC,IAAI,QAAQ,IAAI,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7C,OAAO,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC;YACtC,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,WAAW,CAAC,CAAC,6CAA6C;IACnE,CAAC;IAED,kDAAkD;IAClD,MAAM,iBAAiB,GAAG,IAAA,iBAAO,EAAC,UAAU,EAAE,mBAAmB,CAAC;SAC/D,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC;IAEvD,MAAM,gBAAgB,GAAG,SAAS,YAAY,eAAe,UAAU,aAAa,CAAC;IAErF,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,uCAAuC;QACvC,MAAM,WAAW,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACpE,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,cAAc,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;YAE5C,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,GAAG,IAAI,GAAG,gBAAgB,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;QAC5G,CAAC;IACH,CAAC;IAED,sDAAsD;IACtD,OAAO,gBAAgB,GAAG,IAAI,GAAG,WAAW,CAAC;AAC/C,CAAC;AAED,SAAS,8BAA8B,CAAC,GAA2B;IACjE,MAAM,UAAU,GAAuC,EAAE,CAAC;IAC1D,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAA,4BAAa,EAAC,GAAG,CAAC,EAAE,CAAC;QAC9C,UAAU,CAAC,IAAI,CACb,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAChH,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC","sourcesContent":["import { objectEntries } from \"@trackunit/shared-utils\";\nimport { tsquery } from \"@phenomnomnominal/tsquery\";\nimport * as ts from \"typescript\";\n\n/**\n * This function removes a property with a given name from an object literal\n * provided as a text string.\n *\n * @param {string} fileContent The source code as text\n * @param {string} propertyNames The properties to remove from the object\n */\nexport function removePropertyWithName(fileContent: string, ...propertyNames: Array<string>) {\n  const sourceFile = tsquery.ast(fileContent);\n  const updatedSourceFile = tsquery.map(sourceFile, \"ObjectLiteralExpression\", node => {\n    if (ts.isObjectLiteralExpression(node)) {\n      // Filter out the given property\n      const updatedProperties = node.properties.filter(property => {\n        return !(\n          ts.isPropertyAssignment(property) &&\n          ts.isIdentifier(property.name) &&\n          propertyNames.includes(property.name.text)\n        );\n      });\n      return ts.factory.createObjectLiteralExpression(updatedProperties, true);\n    }\n    return node;\n  });\n  return ts.createPrinter().printFile(updatedSourceFile);\n}\n\n/**\n * This function adds a new property with a given name and value to an object literal\n * provided as a text string.\n *\n * @param {string} fileContent - the source code as text\n * @param {string} propertyName - the name of the property to add\n * @param {string | string[]} propertyValue - the value of the property to add\n */\nexport function addPropertyWithName(\n  fileContent: string,\n  propertyName: string,\n  propertyValue: string | Array<string> | Record<string, string>\n) {\n  const sourceFile = tsquery.ast(fileContent);\n  const updatedSourceFile = tsquery.map(sourceFile, \"ObjectLiteralExpression\", node => {\n    if (ts.isObjectLiteralExpression(node)) {\n      // Add a new property to the object literal\n      let updatedProperties;\n      if (typeof propertyValue === \"string\") {\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(\n            ts.factory.createIdentifier(propertyName),\n            ts.factory.createStringLiteral(propertyValue)\n          ),\n        ];\n      } else if (Array.isArray(propertyValue)) {\n        const arrayLiteral = ts.factory.createArrayLiteralExpression(\n          propertyValue.map(value => ts.factory.createStringLiteral(value))\n        );\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(ts.factory.createIdentifier(propertyName), arrayLiteral),\n        ];\n      } else if (typeof propertyValue === \"object\") {\n        updatedProperties = [\n          ...node.properties,\n          ts.factory.createPropertyAssignment(\n            ts.factory.createIdentifier(propertyName),\n            ts.factory.createObjectLiteralExpression(createObjectLiteralElementLike(propertyValue))\n          ),\n        ];\n      }\n      return ts.factory.createObjectLiteralExpression(updatedProperties, true);\n    }\n    return node;\n  });\n  return ts.createPrinter().printFile(updatedSourceFile);\n}\n\n/**\n * This function adds a spread assignment to an object literal.\n *\n * @param {string} fileContent - the source code as text\n * @param {string} variableName - the name of the variable to spread\n */\nexport function addSpreadAssignment(fileContent: string, variableName: string) {\n  const sourceFile = tsquery.ast(fileContent);\n  \n  // First check if the spread already exists\n  const existingSpreads = tsquery(sourceFile, \"SpreadAssignment\");\n  const hasExistingSpread = existingSpreads.some(spread => {\n    if (ts.isSpreadAssignment(spread)) {\n      return spread.expression.getText() === variableName;\n    }\n    return false;\n  });\n  \n  if (hasExistingSpread) {\n    return fileContent; // Already has the spread, no changes needed\n  }\n  \n  const updatedSourceFile = tsquery.map(sourceFile, \"ObjectLiteralExpression\", node => {\n    if (ts.isObjectLiteralExpression(node)) {\n      // Add the spread assignment to the object literal\n      const spreadAssignment = ts.factory.createSpreadAssignment(\n        ts.factory.createIdentifier(variableName)\n      );\n      const updatedProperties = [...node.properties, spreadAssignment];\n      return ts.factory.createObjectLiteralExpression(updatedProperties, true);\n    }\n    return node;\n  });\n  return ts.createPrinter().printFile(updatedSourceFile);\n}\n\n/**\n * This function adds a require statement to a file after existing require statements.\n *\n * @param {string} fileContent - the source code as text\n * @param {string} variableName - the name of the variable to assign to\n * @param {string} modulePath - the module path to require\n */\nexport function addRequireStatement(fileContent: string, variableName: string, modulePath: string) {\n  const sourceFile = tsquery.ast(fileContent);\n  \n  // Check if the require statement already exists\n  const existingRequires = tsquery(sourceFile, \"CallExpression\");\n  const hasExistingRequire = existingRequires.some(call => {\n    if (ts.isCallExpression(call) && \n        ts.isIdentifier(call.expression) && \n        call.expression.text === \"require\" &&\n        call.arguments.length > 0) {\n      const firstArg = call.arguments[0];\n      if (firstArg && ts.isStringLiteral(firstArg)) {\n        return firstArg.text === modulePath;\n      }\n    }\n    return false;\n  });\n  \n  if (hasExistingRequire) {\n    return fileContent; // Already has the require, no changes needed\n  }\n  \n  // Find all variable statements with require calls\n  const requireStatements = tsquery(sourceFile, \"VariableStatement\")\n    .filter(stmt => stmt.getText().includes(\"require(\"));\n  \n  const requireStatement = `const ${variableName} = require(\"${modulePath}\").default;`;\n  \n  if (requireStatements.length > 0) {\n    // Add after the last require statement\n    const lastRequire = requireStatements[requireStatements.length - 1];\n    if (lastRequire) {\n      const lastRequireEnd = lastRequire.getEnd();\n      \n      return fileContent.slice(0, lastRequireEnd) + '\\n' + requireStatement + fileContent.slice(lastRequireEnd);\n    }\n  }\n  \n  // Add at the beginning if no require statements found\n  return requireStatement + '\\n' + fileContent;\n}\n\nfunction createObjectLiteralElementLike(obj: Record<string, string>): Array<ts.ObjectLiteralElementLike> {\n  const properties: Array<ts.ObjectLiteralElementLike> = [];\n  for (const [key, value] of objectEntries(obj)) {\n    properties.push(\n      ts.factory.createPropertyAssignment(ts.factory.createStringLiteral(key), ts.factory.createStringLiteral(value))\n    );\n  }\n  return properties;\n}\n"]}