@kosdev-code/kos-codegen-core 3.0.6 → 3.0.7

package/index.mjs

@@ -585,6 +585,13 @@ function updateModelIndex(codegenFs, indexPath, modelPath) {
     ts.ScriptTarget.Latest,
     true
   );
+  const alreadyExported = sourceFile.statements.some(
+    (st) => ts.isExportDeclaration(st) && st.moduleSpecifier !== void 0 && ts.isStringLiteral(st.moduleSpecifier) && st.moduleSpecifier.text === `./${modelPath}`
+  );
+  if (alreadyExported) {
+    logger.info(`Export for ${modelPath} already present in ${indexPath}`);
+    return;
+  }
   const exportDeclaration = ts.factory.createExportDeclaration(
     void 0,
     false,
@@ -1759,10 +1766,11 @@ function addDecoratedProperty(cls, spec) {
   cls.insertProperty(propertyInsertIndex(cls), {
     name: spec.name,
     type: spec.type,
+    initializer: spec.initializer,
     scope: spec.scope ? SCOPE_MAP[spec.scope] : void 0,
-    hasExclamationToken: !!spec.hasExclamation,
+    hasExclamationToken: spec.initializer ? false : !!spec.hasExclamation,
     decorators: [
-      {
+      spec.bare && !spec.decoratorArgsText ? { name: spec.decoratorName } : {
         name: spec.decoratorName,
         arguments: spec.decoratorArgsText ? [spec.decoratorArgsText] : []
       }
@@ -1865,7 +1873,7 @@ function addModelEffectToModel(codegenFs, options, projects) {
   });
   return { modelFilePath };
 }
-const FRAMEWORK_TYPES = /* @__PURE__ */ new Set(["IKosDataModel", "IKosIdentifiable"]);
+const FRAMEWORK_TYPES$1 = /* @__PURE__ */ new Set(["IKosDataModel", "IKosIdentifiable"]);
 function addDependencyToModel(codegenFs, options, projects) {
   const logger = getCodegenLogger();
   const { modelFilePath } = resolveModelFilePath(codegenFs, options, projects);
@@ -1887,7 +1895,7 @@ function addDependencyToModel(codegenFs, options, projects) {
       )?.[1];
       if (bean) named.push({ name: bean });
       const depType = options.dependencyType?.trim();
-      if (depType && /^[A-Za-z_$][\w$]*$/.test(depType) && !FRAMEWORK_TYPES.has(depType)) {
+      if (depType && /^[A-Za-z_$][\w$]*$/.test(depType) && !FRAMEWORK_TYPES$1.has(depType)) {
         named.push({ name: depType, isTypeOnly: true });
       }
       if (named.length) ensureNamedImport(sf, options.dependencyPackage, named);
@@ -1904,6 +1912,61 @@ function addDependencyToModel(codegenFs, options, projects) {
   });
   return { modelFilePath };
 }
+const FRAMEWORK_TYPES = /* @__PURE__ */ new Set(["IKosDataModel", "IKosIdentifiable"]);
+function addChildToModel(codegenFs, options, projects) {
+  const logger = getCodegenLogger();
+  const shape = options.shape ?? "single";
+  const { modelFilePath } = resolveModelFilePath(codegenFs, options, projects);
+  if (!codegenFs.exists(modelFilePath)) {
+    throw new Error(`Model file not found: ${modelFilePath}`);
+  }
+  logger.info(
+    `Adding @kosChild "${options.propertyName}" (${shape}) to ${options.modelName}`
+  );
+  const childType = options.childType?.trim();
+  transformSourceFile(codegenFs, modelFilePath, (sf) => {
+    const sdk = resolveSdkModuleSpecifier(sf);
+    const sdkImports = [
+      { name: "kosChild" }
+    ];
+    if (shape === "container") sdkImports.push({ name: "KosModelContainer" });
+    ensureNamedImport(sf, sdk, sdkImports);
+    if (options.childPackage && childType && /^[A-Za-z_$][\w$]*$/.test(childType) && !FRAMEWORK_TYPES.has(childType)) {
+      ensureNamedImport(sf, options.childPackage, [
+        { name: childType, isTypeOnly: true }
+      ]);
+    }
+    const cls = getModelClass(sf);
+    if (shape === "container") {
+      addDecoratedProperty(cls, {
+        name: options.propertyName,
+        decoratorName: "kosChild",
+        bare: true,
+        initializer: `new KosModelContainer<${childType}>()`,
+        scope: "private"
+      });
+    } else if (shape === "array") {
+      addDecoratedProperty(cls, {
+        name: options.propertyName,
+        decoratorName: "kosChild",
+        bare: true,
+        type: `${childType}[]`,
+        initializer: "[]",
+        scope: "private"
+      });
+    } else {
+      addDecoratedProperty(cls, {
+        name: options.propertyName,
+        decoratorName: "kosChild",
+        bare: true,
+        type: childType,
+        scope: "private",
+        hasExclamation: true
+      });
+    }
+  });
+  return { modelFilePath };
+}
 function addTopicHandlerToModel(codegenFs, options, projects) {
   const logger = getCodegenLogger();
   const { modelFilePath } = resolveModelFilePath(codegenFs, options, projects);
@@ -2993,6 +3056,7 @@ export {
   PluginHandlerFactory,
   TrackingFileSystem,
   ValidationError,
+  addChildToModel,
   addComputedToModel,
   addConfigPropertyToModel,
   addContainerSupportToModel,