industrial-model 0.2.0 → 0.4.0

package/dist/index.js

@@ -1,3 +1,5 @@
+import { z } from 'zod';
+
 // src/cognite/adapter.ts
 function createCogniteAdapter(client) {
   return new CogniteSdkAdapter(client);
@@ -24,6 +26,14 @@ var CogniteSdkAdapter = class {
       nextCursor: response.nextCursor
     };
   }
+  async aggregateInstances(request) {
+    const response = await this.client.instances.aggregate(
+      request
+    );
+    return {
+      items: response.items
+    };
+  }
 };
 
 // src/constants.ts
@@ -32,6 +42,8 @@ var EDGE_MARKER = "<EdgeMarker>";
 var MAX_LIMIT = 1e4;
 var DEFAULT_LIMIT = 1e3;
 var MAX_DEPENDENCY_DEPTH = 3;
+var AGGREGATE_LIMIT = 1e3;
+var MAX_GROUP_BY = 5;
 
 // src/mappers/utils.ts
 var NODE_PROPERTIES = /* @__PURE__ */ new Set([
@@ -73,6 +85,511 @@ function buildSelect(source, properties) {
   if (properties.length === 0) return {};
   return { sources: [{ source, properties }] };
 }
+var GROUPABLE_PROPERTY_TYPES = /* @__PURE__ */ new Set([
+  "text",
+  "direct",
+  "int32",
+  "int64",
+  "float32",
+  "float64",
+  "boolean",
+  "enum"
+]);
+var NUMERIC_PROPERTY_TYPES = /* @__PURE__ */ new Set(["int32", "int64", "float32", "float64"]);
+function isGroupableProperty(property) {
+  if (!isViewPropertyDefinition(property)) return false;
+  if (property.type.list === true) return false;
+  const type = property.type.type;
+  return type != null && GROUPABLE_PROPERTY_TYPES.has(type);
+}
+function isNumericProperty(property) {
+  const type = property.type.type;
+  return type != null && NUMERIC_PROPERTY_TYPES.has(type);
+}
+function getSelectedGroupByKeys(groupBy) {
+  return Object.entries(groupBy).filter((entry) => entry[1] === true).map(([key]) => key);
+}
+
+// src/validation.ts
+var nodeIdSchema = z.object({
+  space: z.string().min(1),
+  externalId: z.string().min(1)
+});
+function dateSchema(dateMode) {
+  if (dateMode === "coerce") {
+    return z.preprocess(
+      (value) => typeof value === "string" || typeof value === "number" ? new Date(value) : value,
+      z.date()
+    );
+  }
+  return z.union([z.string(), z.date()]);
+}
+function propertyValueSchema(property, options = {}) {
+  const type = property.type;
+  let schema;
+  switch (type.type) {
+    case "text":
+    case "enum":
+      schema = z.string();
+      break;
+    case "int32":
+    case "int64":
+      schema = z.number().int();
+      break;
+    case "float32":
+    case "float64":
+      schema = z.number();
+      break;
+    case "boolean":
+      schema = z.boolean();
+      break;
+    case "date":
+    case "timestamp":
+      schema = dateSchema(options.dateMode);
+      break;
+    case "direct":
+      schema = nodeIdSchema;
+      break;
+    case "json":
+      schema = z.unknown();
+      break;
+    default:
+      schema = z.unknown();
+      break;
+  }
+  return type.list === true ? z.array(schema) : schema;
+}
+function buildViewSchema(view, options = {}) {
+  const shape = {};
+  for (const [name, property] of Object.entries(view.properties)) {
+    if (isViewPropertyDefinition(property)) {
+      shape[name] = propertyValueSchema(property, options).optional();
+    }
+  }
+  return z.object(shape).strict();
+}
+
+// src/mappers/query-validator.ts
+var NODE_STRING_PROPERTIES = ["externalId", "space"];
+var NODE_NUMBER_PROPERTIES = ["createdTime", "deletedTime", "lastUpdatedTime"];
+var NODE_PROPERTIES2 = /* @__PURE__ */ new Set([...NODE_STRING_PROPERTIES, ...NODE_NUMBER_PROPERTIES]);
+var SORT_DIRECTION_SCHEMA = z.enum(["ascending", "descending"]);
+var recordSchema = z.record(z.string(), z.unknown());
+var leafOps = /* @__PURE__ */ new Set([
+  "eq",
+  "in",
+  "gt",
+  "gte",
+  "lt",
+  "lte",
+  "exists",
+  "prefix",
+  "containsAny",
+  "containsAll"
+]);
+function isRecord(value) {
+  return value != null && typeof value === "object" && !Array.isArray(value);
+}
+function isLeafFilter(value) {
+  return Object.keys(value).some((key) => leafOps.has(key));
+}
+function issuePath(path) {
+  return path.length === 0 ? "query" : path.map(String).join(".");
+}
+function formatZodIssues(error, path) {
+  return error.issues.map((issue) => `${issuePath([...path, ...issue.path])}: ${issue.message}`);
+}
+function getRelationTarget(property) {
+  if (isViewPropertyDefinition(property)) {
+    return getDirectRelationSource(property)?.externalId ?? null;
+  }
+  if (isReverseDirectRelation(property) || isEdgeConnection(property)) {
+    return property.source.externalId;
+  }
+  return null;
+}
+function baseValueSchema(property) {
+  if (property === "node-string") return z.string();
+  if (property === "node-number") return z.number();
+  switch (property.type.type) {
+    case "text":
+    case "enum":
+      return z.string();
+    case "int32":
+    case "int64":
+      return z.number().int();
+    case "float32":
+    case "float64":
+      return z.number();
+    case "boolean":
+      return z.boolean();
+    case "date":
+    case "timestamp":
+      return z.union([z.string(), z.date()]);
+    case "direct":
+      return nodeIdSchema;
+    default:
+      return z.union([z.string(), z.number(), z.boolean()]);
+  }
+}
+function leafFilterSchema(property) {
+  const value = baseValueSchema(property);
+  const isList = typeof property !== "string" && property.type.list === true;
+  if (isList) {
+    return z.object({
+      containsAny: z.array(value).optional(),
+      containsAll: z.array(value).optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (property === "node-string" || typeof property !== "string" && property.type.type === "text") {
+    return z.object({
+      eq: z.string().optional(),
+      in: z.array(z.string()).optional(),
+      prefix: z.string().optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (typeof property !== "string" && property.type.type === "enum") {
+    return z.object({
+      eq: z.string().optional(),
+      in: z.array(z.string()).optional(),
+      prefix: z.string().optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (property === "node-number" || typeof property !== "string" && ["int32", "int64", "float32", "float64"].includes(property.type.type ?? "")) {
+    return z.object({
+      eq: value.optional(),
+      in: z.array(value).optional(),
+      gt: value.optional(),
+      gte: value.optional(),
+      lt: value.optional(),
+      lte: value.optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (typeof property !== "string" && ["date", "timestamp"].includes(property.type.type ?? "")) {
+    return z.object({
+      eq: value.optional(),
+      in: z.array(value).optional(),
+      gt: value.optional(),
+      gte: value.optional(),
+      lt: value.optional(),
+      lte: value.optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (typeof property !== "string" && property.type.type === "boolean") {
+    return z.object({
+      eq: z.boolean().optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  if (typeof property !== "string" && property.type.type === "direct") {
+    return z.object({
+      eq: nodeIdSchema.optional(),
+      in: z.array(nodeIdSchema).optional(),
+      exists: z.boolean().optional()
+    }).strict();
+  }
+  return z.object({
+    eq: value.optional(),
+    in: z.array(value).optional(),
+    exists: z.boolean().optional()
+  }).strict();
+}
+var QueryValidator = class {
+  constructor(viewMapper) {
+    this.viewMapper = viewMapper;
+  }
+  async validate(options, rootView) {
+    const errors = [];
+    errors.push(...this.validateOptionsShape(options, rootView));
+    if (options.select !== void 0) {
+      errors.push(...await this.validateSelect(options.select, rootView, ["select"]));
+    }
+    if (options.filters !== void 0) {
+      errors.push(...await this.validateWhereInput(options.filters, rootView, ["filters"]));
+    }
+    if (options.sort !== void 0) {
+      errors.push(...this.validateSort(options.sort, rootView, ["sort"]));
+    }
+    if (errors.length > 0) {
+      throw new Error(`Invalid query options:
+${errors.map((error) => `- ${error}`).join("\n")}`);
+    }
+  }
+  validateOptionsShape(options, rootView) {
+    const schema = z.object({
+      viewExternalId: z.literal(rootView.externalId),
+      select: z.unknown().optional(),
+      filters: z.unknown().optional(),
+      sort: z.unknown().optional(),
+      limit: z.union([z.literal(-1), z.number().int().positive().max(MAX_LIMIT)]).optional(),
+      cursor: z.string().nullable().optional()
+    }).strict();
+    const result = schema.safeParse(options);
+    return result.success ? [] : formatZodIssues(result.error, []);
+  }
+  async validateSelect(select, view, path) {
+    const shape = {
+      _all: z.literal(true).optional()
+    };
+    for (const property of NODE_PROPERTIES2) {
+      shape[property] = z.boolean().optional();
+    }
+    for (const [name, property] of Object.entries(view.properties)) {
+      const target = getRelationTarget(property);
+      if (target != null) {
+        const nestedSelect = recordSchema;
+        shape[name] = isViewPropertyDefinition(property) ? z.union([z.boolean(), nestedSelect]).optional() : nestedSelect.optional();
+      } else {
+        shape[name] = z.boolean().optional();
+      }
+    }
+    const result = z.object(shape).strict().safeParse(select);
+    if (!result.success) return formatZodIssues(result.error, path);
+    if (!isRecord(select)) return [];
+    const errors = [];
+    for (const [name, value] of Object.entries(select)) {
+      if (name === "_all" || value == null || typeof value !== "object" || Array.isArray(value)) {
+        continue;
+      }
+      const property = view.properties[name];
+      if (!property) continue;
+      const target = getRelationTarget(property);
+      if (target == null) {
+        errors.push(
+          `${issuePath([...path, name])}: property "${name}" does not support nested select`
+        );
+        continue;
+      }
+      const targetView = await this.viewMapper.getView(target);
+      errors.push(...await this.validateSelect(value, targetView, [...path, name]));
+    }
+    return errors;
+  }
+  async validateWhereInput(filters, view, path) {
+    return this.validateFilters(filters, view, path);
+  }
+  async validateFilters(filters, view, path) {
+    const shape = {
+      AND: z.union([recordSchema, z.array(recordSchema)]).optional(),
+      OR: z.array(recordSchema).optional(),
+      NOT: z.union([recordSchema, z.array(recordSchema)]).optional()
+    };
+    for (const property of NODE_STRING_PROPERTIES) {
+      shape[property] = z.unknown().optional();
+    }
+    for (const property of NODE_NUMBER_PROPERTIES) {
+      shape[property] = z.unknown().optional();
+    }
+    for (const property of Object.keys(view.properties)) {
+      shape[property] = z.unknown().optional();
+    }
+    const result = z.object(shape).strict().safeParse(filters);
+    if (!result.success) return formatZodIssues(result.error, path);
+    if (!isRecord(filters)) return [];
+    const errors = [];
+    for (const [name, value] of Object.entries(filters)) {
+      if (value == null) continue;
+      if (name === "AND" || name === "OR" || name === "NOT") {
+        const clauses = Array.isArray(value) ? value : [value];
+        for (const [index, clause] of clauses.entries()) {
+          errors.push(...await this.validateFilters(clause, view, [...path, name, index]));
+        }
+        continue;
+      }
+      if (!isRecord(value)) {
+        errors.push(`${issuePath([...path, name])}: Expected object`);
+        continue;
+      }
+      const nodePropertyType = NODE_STRING_PROPERTIES.includes(
+        name
+      ) ? "node-string" : NODE_NUMBER_PROPERTIES.includes(name) ? "node-number" : null;
+      if (nodePropertyType != null) {
+        errors.push(...this.validateLeafFilter(value, nodePropertyType, [...path, name]));
+        continue;
+      }
+      const property = view.properties[name];
+      if (!property) continue;
+      if (isViewPropertyDefinition(property)) {
+        const target2 = getDirectRelationSource(property);
+        if (target2 != null && !isLeafFilter(value)) {
+          const targetView = await this.viewMapper.getView(target2.externalId);
+          errors.push(...await this.validateFilters(value, targetView, [...path, name]));
+        } else {
+          errors.push(...this.validateLeafFilter(value, property, [...path, name]));
+        }
+        continue;
+      }
+      const target = getRelationTarget(property);
+      if (target == null) {
+        errors.push(`${issuePath([...path, name])}: property "${name}" does not support filters`);
+        continue;
+      }
+      errors.push(
+        `${issuePath([...path, name])}: filtering through "${name}" is not supported by the query mapper`
+      );
+    }
+    return errors;
+  }
+  validateLeafFilter(value, property, path) {
+    const result = leafFilterSchema(property).safeParse(value);
+    return result.success ? [] : formatZodIssues(result.error, path);
+  }
+  validateSort(sort, view, path) {
+    const shape = {};
+    for (const property of NODE_PROPERTIES2) {
+      shape[property] = SORT_DIRECTION_SCHEMA.optional();
+    }
+    for (const [name, property] of Object.entries(view.properties)) {
+      if (isViewPropertyDefinition(property) && property.type.list !== true) {
+        shape[name] = SORT_DIRECTION_SCHEMA.optional();
+      }
+    }
+    const result = z.object(shape).strict().safeParse(sort);
+    return result.success ? [] : formatZodIssues(result.error, path);
+  }
+};
+
+// src/mappers/aggregate-validator.ts
+var NODE_COUNT_PROPERTIES = /* @__PURE__ */ new Set(["externalId", "space"]);
+function issuePath2(path) {
+  return path.length === 0 ? "aggregate" : path.map(String).join(".");
+}
+function formatZodIssues2(error, path) {
+  return error.issues.map((issue) => `${issuePath2([...path, ...issue.path])}: ${issue.message}`);
+}
+function isEmptyObject(value) {
+  return value != null && typeof value === "object" && !Array.isArray(value) && Object.keys(value).length === 0;
+}
+var AggregateValidator = class {
+  constructor(viewMapper) {
+    this.queryValidator = new QueryValidator(viewMapper);
+  }
+  async validate(options, rootView) {
+    const errors = [];
+    errors.push(...this.validateOptionsShape(options, rootView));
+    const selectedGroupBy = options.groupBy ? getSelectedGroupByKeys(options.groupBy) : [];
+    if (selectedGroupBy.length === 0 && options.aggregate === void 0) {
+      errors.push("aggregate: either groupBy or aggregate must be provided");
+    }
+    if (options.filters !== void 0) {
+      errors.push(
+        ...await this.queryValidator.validateWhereInput(options.filters, rootView, ["filters"])
+      );
+    }
+    if (options.groupBy !== void 0) {
+      errors.push(...this.validateGroupBy(options.groupBy, rootView, ["groupBy"]));
+    }
+    if (options.aggregate !== void 0) {
+      errors.push(
+        ...this.validateAggregate(
+          options.aggregate,
+          rootView,
+          ["aggregate"]
+        )
+      );
+    }
+    if (errors.length > 0) {
+      throw new Error(
+        `Invalid aggregate options:
+${errors.map((error) => `- ${error}`).join("\n")}`
+      );
+    }
+  }
+  validateOptionsShape(options, rootView) {
+    const schema = z.object({
+      viewExternalId: z.literal(rootView.externalId),
+      filters: z.unknown().optional(),
+      groupBy: z.unknown().optional(),
+      aggregate: z.unknown().optional()
+    }).strict();
+    const result = schema.safeParse(options);
+    return result.success ? [] : formatZodIssues2(result.error, []);
+  }
+  validateGroupBy(groupBy, view, path) {
+    const shape = {};
+    for (const [name, property] of Object.entries(view.properties)) {
+      if (isGroupableProperty(property)) {
+        shape[name] = z.literal(true).optional();
+      }
+    }
+    const result = z.object(shape).strict().safeParse(groupBy);
+    if (!result.success) {
+      return formatZodIssues2(result.error, path);
+    }
+    const selected = getSelectedGroupByKeys(groupBy);
+    const errors = [];
+    if (selected.length === 0) {
+      errors.push(`${issuePath2(path)}: at least one property must be set to true`);
+    }
+    if (selected.length > MAX_GROUP_BY) {
+      errors.push(`${issuePath2(path)}: at most ${MAX_GROUP_BY} properties can be grouped`);
+    }
+    for (const name of selected) {
+      const property = view.properties[name];
+      if (!property || !isGroupableProperty(property)) {
+        errors.push(`${issuePath2([...path, name])}: property "${name}" cannot be used in groupBy`);
+      }
+    }
+    return errors;
+  }
+  validateAggregate(aggregate, view, path) {
+    if ("count" in aggregate) {
+      const property2 = aggregate.count;
+      if (isEmptyObject(property2)) {
+        return [];
+      }
+      if (typeof property2 === "string") {
+        if (NODE_COUNT_PROPERTIES.has(property2)) {
+          return [];
+        }
+        const viewProperty = view.properties[property2];
+        if (!viewProperty || !isGroupableProperty(viewProperty)) {
+          return [`${issuePath2([...path, "count"])}: property "${property2}" cannot be counted`];
+        }
+        return [];
+      }
+      return [`${issuePath2([...path, "count"])}: invalid count property`];
+    }
+    let propertyName;
+    let numericOp = null;
+    if ("avg" in aggregate) {
+      numericOp = "avg";
+      propertyName = aggregate.avg;
+    } else if ("min" in aggregate) {
+      numericOp = "min";
+      propertyName = aggregate.min;
+    } else if ("max" in aggregate) {
+      numericOp = "max";
+      propertyName = aggregate.max;
+    } else if ("sum" in aggregate) {
+      numericOp = "sum";
+      propertyName = aggregate.sum;
+    }
+    if (numericOp == null) {
+      return [`${issuePath2(path)}: unknown aggregate operation`];
+    }
+    if (typeof propertyName !== "string") {
+      return [`${issuePath2(path)}: aggregate property must be a string`];
+    }
+    const property = view.properties[propertyName];
+    if (!property || !isViewPropertyDefinition(property) || !isNumericProperty(property)) {
+      return [
+        `${issuePath2([...path, numericOp])}: property "${propertyName}" must be a numeric view property`
+      ];
+    }
+    if (getDirectRelationSource(property) != null) {
+      return [
+        `${issuePath2([...path, numericOp])}: property "${propertyName}" is a relation and cannot be aggregated`
+      ];
+    }
+    return [];
+  }
+};
 
 // src/mappers/filter-mapper.ts
 var LEAF_OPS = /* @__PURE__ */ new Set([
@@ -87,7 +604,7 @@ var LEAF_OPS = /* @__PURE__ */ new Set([
   "containsAny",
   "containsAll"
 ]);
-function isLeafFilter(value) {
+function isLeafFilter2(value) {
   return Object.keys(value).some((k) => LEAF_OPS.has(k));
 }
 var FilterMapper = class {
@@ -116,7 +633,7 @@ var FilterMapper = class {
       } else {
         const filterValue = value;
         const property = getPropertyRef(key, rootView);
-        if (isLeafFilter(filterValue)) {
+        if (isLeafFilter2(filterValue)) {
           result.push(...this.leafToFilterDefs(property, filterValue));
         } else {
           const targetView = await this.getNestedTargetView(key, rootView);
@@ -203,6 +720,85 @@ var FilterMapper = class {
   }
 };
 
+// src/mappers/aggregate-mapper.ts
+var AggregateMapper = class {
+  constructor(viewMapper) {
+    this.viewMapper = viewMapper;
+    this.filterMapper = new FilterMapper(viewMapper);
+    this.validator = new AggregateValidator(viewMapper);
+  }
+  async map(options) {
+    const { viewExternalId, filters, groupBy, aggregate } = options;
+    const rootView = await this.viewMapper.getView(viewExternalId);
+    await this.validator.validate(options, rootView);
+    const filterParts = filters ? await this.filterMapper.map(filters, rootView) : [];
+    const filter = filterParts.length === 0 ? void 0 : filterParts.length === 1 ? filterParts[0] : { and: filterParts };
+    return {
+      view: toViewReference(rootView),
+      instanceType: "node",
+      limit: AGGREGATE_LIMIT,
+      ...filter !== void 0 ? { filter } : {},
+      ...groupBy ? { groupBy: getSelectedGroupByKeys(groupBy) } : {},
+      ...aggregate ? { aggregates: [mapAggregateDefinition(aggregate)] } : {}
+    };
+  }
+};
+function mapAggregateDefinition(aggregate) {
+  if ("count" in aggregate) {
+    const property = aggregate.count;
+    if (property != null && typeof property === "object" && !Array.isArray(property) && Object.keys(property).length === 0) {
+      return { count: {} };
+    }
+    if (typeof property === "string") {
+      return { count: { property } };
+    }
+    return { count: {} };
+  }
+  if ("avg" in aggregate) {
+    return { avg: { property: aggregate.avg } };
+  }
+  if ("min" in aggregate) {
+    return { min: { property: aggregate.min } };
+  }
+  if ("max" in aggregate) {
+    return { max: { property: aggregate.max } };
+  }
+  if ("sum" in aggregate) {
+    return { sum: { property: aggregate.sum } };
+  }
+  throw new Error("Invalid aggregate definition");
+}
+
+// src/mappers/aggregate-result-mapper.ts
+function isNodeId(value) {
+  return value != null && typeof value === "object" && "space" in value && "externalId" in value && typeof value.space === "string" && typeof value.externalId === "string";
+}
+var AggregateResultMapper = class {
+  map(response, options) {
+    const groupByKeys = options.groupBy ? getSelectedGroupByKeys(options.groupBy) : [];
+    return response.items.map((item) => {
+      let group;
+      if (item.group != null && groupByKeys.length > 0) {
+        group = {};
+        for (const key of groupByKeys) {
+          const value = item.group[key];
+          if (value === void 0) continue;
+          group[key] = isNodeId(value) ? { space: value.space, externalId: value.externalId } : value;
+        }
+        if (Object.keys(group).length === 0) {
+          group = void 0;
+        }
+      }
+      const aggregateValue = item.aggregates[0];
+      const aggregate = aggregateValue?.value !== void 0 ? aggregateValue.property != null ? { property: aggregateValue.property, value: aggregateValue.value } : { value: aggregateValue.value } : void 0;
+      return {
+        ...group !== void 0 ? { group } : {},
+        ...aggregate !== void 0 ? { aggregate } : {}
+      };
+    });
+  }
+};
+
 // src/mappers/sort-mapper.ts
 var SortMapper = class {
   map(sort, rootView) {
@@ -227,6 +823,7 @@ var QueryMapper = class {
     this.viewMapper = viewMapper;
     this.filterMapper = new FilterMapper(viewMapper);
     this.sortMapper = new SortMapper();
+    this.validator = new QueryValidator(viewMapper);
   }
   async map(options) {
     const {
@@ -239,6 +836,7 @@ var QueryMapper = class {
     } = options;
     const limit = requestedLimit === -1 ? DEFAULT_LIMIT : requestedLimit;
     const rootView = await this.viewMapper.getView(viewExternalId);
+    await this.validator.validate(options, rootView);
     const rootViewRef = toViewReference(rootView);
     const whereFilters = filters ? await this.filterMapper.map(filters, rootView) : [];
     const baseFilters = [{ hasData: [rootViewRef] }, ...whereFilters];
@@ -529,6 +1127,92 @@ var QueryResultMapper = class {
     return entry;
   }
 };
+var nodeMetadataSchema = {
+  instanceType: z.literal("node").optional(),
+  space: z.string(),
+  externalId: z.string(),
+  version: z.number().optional(),
+  createdTime: z.number().optional(),
+  deletedTime: z.number().optional(),
+  lastUpdatedTime: z.number().optional(),
+  _edges: z.record(z.string(), z.unknown()).optional()
+};
+function isListRelation(property) {
+  if (isViewPropertyDefinition(property)) {
+    return isListDirectRelation(property);
+  }
+  if (isReverseDirectRelation(property)) {
+    return property.connectionType === "multi_reverse_direct_relation" || property.targetsList === true;
+  }
+  return isEdgeConnection(property);
+}
+function isRecord2(value) {
+  return value != null && typeof value === "object" && !Array.isArray(value);
+}
+var QueryResultValidator = class {
+  constructor(viewMapper) {
+    this.viewMapper = viewMapper;
+  }
+  async parseItems(rootViewExternalId, items, select) {
+    const rootView = await this.viewMapper.getView(rootViewExternalId);
+    const schema = await this.buildResultSchema(rootView, MAX_DEPENDENCY_DEPTH, select);
+    const result = z.array(schema).safeParse(items);
+    if (!result.success) {
+      throw new Error(
+        `Invalid query result:
+${result.error.issues.map((issue) => `- ${issue.path.map(String).join(".")}: ${issue.message}`).join("\n")}`
+      );
+    }
+    return result.data;
+  }
+  async buildResultSchema(view, remainingDepth, select) {
+    const shape = { ...nodeMetadataSchema };
+    const includeAllProperties = select == null || select._all === true;
+    for (const [name, property] of Object.entries(view.properties)) {
+      const isSelected = includeAllProperties || name in select;
+      if (!isSelected) continue;
+      const nestedSelect = isRecord2(select?.[name]) ? select[name] : void 0;
+      if (isViewPropertyDefinition(property)) {
+        const relationSource = getDirectRelationSource(property);
+        if (relationSource) {
+          shape[name] = await this.buildRelationSchema(
+            property,
+            relationSource.externalId,
+            remainingDepth,
+            nestedSelect
+          );
+        } else {
+          shape[name] = propertyValueSchema(property, { dateMode: "coerce" }).optional();
+        }
+        continue;
+      }
+      if (isReverseDirectRelation(property) || isEdgeConnection(property)) {
+        shape[name] = await this.buildRelationSchema(
+          property,
+          property.source.externalId,
+          remainingDepth,
+          nestedSelect
+        );
+      }
+    }
+    const schema = z.object(shape);
+    return includeAllProperties ? schema.strict() : schema;
+  }
+  async buildRelationSchema(property, targetViewExternalId, remainingDepth, select) {
+    const isList = isListRelation(property);
+    const fallbackSchema = isViewPropertyDefinition(property) ? propertyValueSchema(property, { dateMode: "coerce" }) : z.unknown();
+    if (remainingDepth <= 0 || select == null) {
+      return fallbackSchema.optional();
+    }
+    const targetView = await this.viewMapper.getView(targetViewExternalId);
+    const nestedSchema = await this.buildResultSchema(targetView, remainingDepth - 1, select);
+    if (isViewPropertyDefinition(property)) {
+      const nestedRelationSchema = isList ? z.array(nestedSchema) : nestedSchema;
+      return z.union([nestedRelationSchema, fallbackSchema]).optional();
+    }
+    return (isList ? z.array(nestedSchema) : nestedSchema).optional();
+  }
+};
 
 // src/mappers/view-mapper.ts
 var ViewMapper = class {
@@ -679,15 +1363,33 @@ function buildDependenciesQuery(previousQuery, nodesParent, nodesChildren, leafC
 
 // src/client.ts
 var IndustrialModelClient = class {
-  constructor(client, dataModelId) {
+  constructor(client, dataModelId, options = {}) {
     const cognite = createCogniteAdapter(client);
     this.cognite = cognite;
     const viewMapper = new ViewMapper(cognite, dataModelId);
     this.queryMapper = new QueryMapper(viewMapper);
+    this.aggregateMapper = new AggregateMapper(viewMapper);
+    this.aggregateResultMapper = new AggregateResultMapper();
     this.resultMapper = new QueryResultMapper(viewMapper);
+    this.resultValidator = new QueryResultValidator(viewMapper);
+    this.validateResults = options.validateResults ?? false;
   }
   query() {
-    return (options) => this.queryInternal(options);
+    const execute = (options) => this.queryInternal(options);
+    return execute;
+  }
+  aggregate() {
+    const execute = (options) => this.aggregateInternal(options);
+    return execute;
+  }
+  async aggregateInternal(options) {
+    const cogniteRequest = await this.aggregateMapper.map(options);
+    const response = await this.cognite.aggregateInstances(cogniteRequest);
+    const items = this.aggregateResultMapper.map(
+      response,
+      options
+    );
+    return { items };
   }
   async queryInternal(options) {
     const { viewExternalId, limit = DEFAULT_LIMIT } = options;
@@ -705,7 +1407,8 @@ var IndustrialModelClient = class {
         mapNodesAndEdges(queryResult),
         dependenciesData
       );
-      const pageResult = await this.resultMapper.mapNodes(viewExternalId, queryResultData);
+      const mappedPageResult = await this.resultMapper.mapNodes(viewExternalId, queryResultData);
+      const pageResult = this.validateResults ? await this.resultValidator.parseItems(viewExternalId, mappedPageResult, options.select) : mappedPageResult;
       const nextCursor = queryResult.nextCursor[viewExternalId] ?? null;
       const isLastPage = pageResult.length < limit || !nextCursor;
       const resolvedCursor = isLastPage ? null : nextCursor;
@@ -736,6 +1439,6 @@ var IndustrialModelClient = class {
   }
 };
 
-export { IndustrialModelClient };
+export { IndustrialModelClient, buildViewSchema, nodeIdSchema };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map