industrial-model 0.9.0 → 0.10.0

package/README.md

@@ -503,7 +503,7 @@ do {
 } while (cursor !== null);
 ```
 
-Pass `limit: -1` when you want the SDK to follow all root cursors automatically. The SDK issues multiple `instances.query` calls, using 1000 root items per page by default, and returns `cursor: null`.
+Pass `limit: -1` when you want the SDK to follow all root cursors automatically. The SDK issues multiple `instances.query` calls, using 10000 root items per page, and returns `cursor: null`.
 
 ```ts
 const { items } = await model.query<CogniteAsset>()({

package/dist/cli/index.cjs

@@ -3964,6 +3964,7 @@ function processMappedProperty(propertyName, fieldName, prop) {
   const isList = prop.type.list === true;
   const isRelation = cogniteType === "direct";
   const relationSource = getDirectRelationSource(prop);
+  const enumValues = cogniteType === "enum" && prop.type.values ? Object.keys(prop.type.values) : null;
   let relationTarget = null;
   let relationTargetSpace = null;
   let relationTargetExternalId = null;
@@ -3986,7 +3987,8 @@ function processMappedProperty(propertyName, fieldName, prop) {
     isListDirectRelation: isRelation && isList,
     relationTarget,
     relationTargetSpace,
-    relationTargetExternalId
+    relationTargetExternalId,
+    enumValues
   };
 }
 function processEdgeProperty(propertyName, fieldName, prop) {
@@ -4003,7 +4005,8 @@ function processEdgeProperty(propertyName, fieldName, prop) {
     isListDirectRelation: false,
     relationTarget: toPascal(prop.source.externalId),
     relationTargetSpace: prop.source.space,
-    relationTargetExternalId: prop.source.externalId
+    relationTargetExternalId: prop.source.externalId,
+    enumValues: null
   };
 }
 function processReverseProperty(propertyName, fieldName, prop) {
@@ -4021,7 +4024,8 @@ function processReverseProperty(propertyName, fieldName, prop) {
     isListDirectRelation: false,
     relationTarget: toPascal(prop.source.externalId),
     relationTargetSpace: prop.source.space,
-    relationTargetExternalId: prop.source.externalId
+    relationTargetExternalId: prop.source.externalId,
+    enumValues: null
   };
 }
 
@@ -4034,11 +4038,16 @@ function getRelationTypeFields(view) {
     (f) => (f.isRelation || f.isEdge || f.isReverseRelation) && f.relationTarget
   );
 }
-function getInterfaceType(field) {
+function getInterfaceType(field, viewName) {
   if (field.isRelation && !field.isList) return "NodeId";
   if (field.isEdge || field.isRelation && field.isList) return "NodeId[]";
-  if (field.isList) return `${field.mappedType}[]`;
-  return field.mappedType;
+  const baseType = field.enumValues && viewName ? getEnumTypeName(viewName, field) : field.mappedType;
+  if (field.isList) return `${baseType}[]`;
+  return baseType;
+}
+function getEnumTypeName(viewName, field) {
+  const capitalized = field.fieldName.charAt(0).toUpperCase() + field.fieldName.slice(1);
+  return `${viewName}${capitalized}`;
 }
 function getRelationResolvedType(field) {
   const target = field.relationTarget ?? "unknown";
@@ -4187,9 +4196,14 @@ function renderTypes(views, config) {
     "  UpsertOptions,",
     "  UpsertResult,",
     '} from "industrial-model";',
-    "",
-    renderViewExternalIdUnion(views, config)
+    ""
   ];
+  const enumAliases = renderEnumTypeAliases(views);
+  if (enumAliases) {
+    lines.push(enumAliases);
+    lines.push("");
+  }
+  lines.push(renderViewExternalIdUnion(views, config));
   for (const view of views) {
     lines.push("");
     lines.push(renderView(view));
@@ -4205,19 +4219,32 @@ function renderViewExternalIdUnion(views, config) {
   return `export type ${config.clientName}ViewExternalId =
 ${views.map((view) => `  | "${view.viewExternalId}"`).join("\n")};`;
 }
+function renderEnumTypeAliases(views) {
+  const aliases = [];
+  for (const view of views) {
+    for (const field of view.fields) {
+      if (field.enumValues && field.enumValues.length > 0) {
+        const typeName = getEnumTypeName(view.viewName, field);
+        const union = field.enumValues.map((v) => `"${v}"`).join(" | ");
+        aliases.push(`export type ${typeName} = ${union};`);
+      }
+    }
+  }
+  return aliases.join("\n");
+}
 function renderView(view) {
   const propsFields = getPropsFields(view);
   const relationFields = getRelationTypeFields(view);
   if (relationFields.length === 0) {
     const propsLines2 = propsFields.map(
-      (f) => `  ${f.fieldName}${f.isNullable ? "?" : ""}: ${getInterfaceType(f)};`
+      (f) => `  ${f.fieldName}${f.isNullable ? "?" : ""}: ${getInterfaceType(f, view.viewName)};`
     );
     return `export type ${view.viewName} = IndustrialModel<{
 ${propsLines2.join("\n")}
 }>;`;
   }
   const propsLines = propsFields.map(
-    (f) => `    ${f.fieldName}${f.isNullable ? "?" : ""}: ${getInterfaceType(f)};`
+    (f) => `    ${f.fieldName}${f.isNullable ? "?" : ""}: ${getInterfaceType(f, view.viewName)};`
   );
   const relLines = relationFields.map(
     (f) => `    ${f.fieldName}${f.isNullable ? "?" : ""}: ${getRelationResolvedType(f)};`
@@ -6771,7 +6798,7 @@ var generateCommand = new Command("generate").description("Generate TypeScript t
     dataModelVersion: dataModel.version,
     clientName: options.clientName,
     outputPath: options.outputPath,
-    packageVersion: "0.9.0"
+    packageVersion: "0.10.0"
   });
   console.log(
     `
@@ -6791,7 +6818,7 @@ Generating types for ${dataModel.space}/${dataModel.externalId}/${dataModel.vers
 });
 
 // src/cli/index.ts
-var program2 = new Command().name("industrial-model").description("Code generator for Cognite Data Fusion data models").version("0.9.0");
+var program2 = new Command().name("industrial-model").description("Code generator for Cognite Data Fusion data models").version("0.10.0");
 program2.addCommand(generateCommand);
 program2.parse();
 //# sourceMappingURL=index.cjs.map