@osdk/client 0.9.0 → 0.11.0

package/build/js/chunk-GOBQUVQZ.mjs

@@ -0,0 +1,1138 @@
+import { aggregateObjectSetV2, loadObjectSetV2, getObjectTypeV2 } from '@osdk/gateway/requests';
+import { createOpenApiRequest, PalantirApiError } from '@osdk/shared.net';
+import invariant2 from 'tiny-invariant';
+import WebSocket from 'isomorphic-ws';
+import { conjureFetch } from 'conjure-lite';
+
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+function legacyToModernSingleAggregationResult(entry) {
+  return entry.metrics.reduce((accumulator, curValue) => {
+    const parts = curValue.name.split(".");
+    if (parts[0] === "count") {
+      return accumulator;
+    }
+    !(parts.length === 2) ? process.env.NODE_ENV !== "production" ? invariant2(false, "assumed we were getting a `${key}.${type}`") : invariant2(false) : void 0;
+    if (!(parts[0] in accumulator)) {
+      accumulator[parts[0]] = {};
+    }
+    accumulator[parts[0]][parts[1]] = curValue.value;
+    return accumulator;
+  }, {});
+}
+
+// src/internal/conversions/modernToLegacyAggregationClause.ts
+function modernToLegacyAggregationClause(select) {
+  return Object.entries(select).flatMap(([k, v]) => {
+    if (k === "$count") {
+      if (v)
+        return {
+          type: "count",
+          name: "count"
+        };
+      return [];
+    } else if (Array.isArray(v)) {
+      return v.map((v2) => {
+        return {
+          type: v2,
+          name: `${k}.${v2}`,
+          field: k
+        };
+      });
+    } else {
+      return [{
+        type: v,
+        // FIXME v has additional possible values
+        name: `${k}.${v}`,
+        field: k
+      }];
+    }
+  });
+}
+
+// src/internal/conversions/modernToLegacyGroupByClause.ts
+function modernToLegacyGroupByClause(groupByClause) {
+  if (!groupByClause)
+    return [];
+  return Object.entries(groupByClause).flatMap(([field, type]) => {
+    if (type === "exact") {
+      return [{
+        type,
+        field
+      }];
+    } else if ("exactWithLimit" in type) {
+      {
+        return [{
+          type: "exact",
+          field,
+          maxGroupCount: type.exactWithLimit
+        }];
+      }
+    } else if ("fixedWidth" in type) {
+      return [{
+        type: "fixedWidth",
+        field,
+        fixedWidth: type.fixedWidth
+      }];
+    } else if ("ranges" in type) {
+      return [{
+        type: "ranges",
+        field,
+        ranges: type.ranges.map((range) => convertRange(range))
+      }];
+    } else
+      return [];
+  });
+}
+function convertRange(range) {
+  return {
+    startValue: range[0],
+    endValue: range[1]
+  };
+}
+
+// src/query/WhereClause.ts
+var DistanceUnitMapping = {
+  "centimeter": "CENTIMETERS",
+  "centimeters": "CENTIMETERS",
+  "cm": "CENTIMETERS",
+  "meter": "METERS",
+  "meters": "METERS",
+  "m": "METERS",
+  "kilometer": "KILOMETERS",
+  "kilometers": "KILOMETERS",
+  "km": "KILOMETERS",
+  "inch": "INCHES",
+  "inches": "INCHES",
+  "foot": "FEET",
+  "feet": "FEET",
+  "yard": "YARDS",
+  "yards": "YARDS",
+  "mile": "MILES",
+  "miles": "MILES",
+  "nautical_mile": "NAUTICAL_MILES",
+  "nauticalMile": "NAUTICAL_MILES",
+  "nautical miles": "NAUTICAL_MILES"
+};
+
+// src/internal/conversions/modernToLegacyWhereClause.ts
+function modernToLegacyWhereClause(whereClause) {
+  if ("$and" in whereClause) {
+    return {
+      type: "and",
+      value: whereClause.$and.map(modernToLegacyWhereClause)
+    };
+  } else if ("$or" in whereClause) {
+    return {
+      type: "or",
+      value: whereClause.$or.map(modernToLegacyWhereClause)
+    };
+  } else if ("$not" in whereClause) {
+    return {
+      type: "not",
+      value: modernToLegacyWhereClause(whereClause.$not)
+    };
+  }
+  const parts = Object.entries(whereClause);
+  if (parts.length === 1) {
+    return handleWherePair(parts[0]);
+  }
+  return {
+    type: "and",
+    value: parts.map(handleWherePair)
+  };
+}
+function makeGeoFilterBbox(field, bbox, filterType) {
+  return {
+    type: filterType === "$within" ? "withinBoundingBox" : "intersectsBoundingBox",
+    field,
+    value: {
+      topLeft: {
+        type: "Point",
+        coordinates: [bbox[0], bbox[3]]
+      },
+      bottomRight: {
+        type: "Point",
+        coordinates: [bbox[2], bbox[1]]
+      }
+    }
+  };
+}
+function makeGeoFilterPolygon(field, coordinates, filterType) {
+  return {
+    type: filterType,
+    field,
+    value: {
+      type: "Polygon",
+      coordinates
+    }
+  };
+}
+function handleWherePair([field, filter]) {
+  !(filter != null) ? process.env.NODE_ENV !== "production" ? invariant2(false, "Defined key values are only allowed when they are not undefined.") : invariant2(false) : void 0;
+  if (typeof filter === "string" || typeof filter === "number") {
+    return {
+      type: "eq",
+      field,
+      value: filter
+    };
+  }
+  const keysOfFilter = Object.keys(filter);
+  const hasDollarSign = keysOfFilter.some((key) => key.startsWith("$"));
+  !(!hasDollarSign || keysOfFilter.length === 1) ? process.env.NODE_ENV !== "production" ? invariant2(false, "WhereClause Filter with multiple clauses isn't allowed") : invariant2(false) : void 0;
+  if (!hasDollarSign) {
+    throw new Error("Unsupported filter. Did you forget to use a $-prefixed filter?");
+  }
+  const firstKey = keysOfFilter[0];
+  !(filter[firstKey] != null) ? process.env.NODE_ENV !== "production" ? invariant2(false) : invariant2(false) : void 0;
+  if (firstKey === "$ne") {
+    return {
+      type: "not",
+      value: {
+        type: "eq",
+        field,
+        value: filter[firstKey]
+      }
+    };
+  }
+  if (firstKey === "$within") {
+    const withinBody = filter[firstKey];
+    if (Array.isArray(withinBody)) {
+      return makeGeoFilterBbox(field, withinBody, firstKey);
+    } else if ("bbox" in withinBody && !("type" in withinBody)) {
+      return makeGeoFilterBbox(field, withinBody.bbox, firstKey);
+    } else if ("distance" in withinBody && "of" in withinBody) {
+      return {
+        type: "withinDistanceOf",
+        field,
+        value: {
+          center: Array.isArray(withinBody.of) ? {
+            type: "Point",
+            coordinates: withinBody.of
+          } : withinBody.of,
+          distance: {
+            value: withinBody.distance[0],
+            unit: DistanceUnitMapping[withinBody.distance[1]]
+          }
+        }
+      };
+    } else {
+      const coordinates = "polygon" in withinBody ? withinBody.polygon : withinBody.coordinates;
+      return makeGeoFilterPolygon(field, coordinates, "withinPolygon");
+    }
+  }
+  if (firstKey === "$intersects") {
+    const intersectsBody = filter[firstKey];
+    if (Array.isArray(intersectsBody)) {
+      return makeGeoFilterBbox(field, intersectsBody, firstKey);
+    } else if ("bbox" in intersectsBody && !("type" in intersectsBody)) {
+      return makeGeoFilterBbox(field, intersectsBody.bbox, firstKey);
+    } else {
+      const coordinates = "polygon" in intersectsBody ? intersectsBody.polygon : intersectsBody.coordinates;
+      return makeGeoFilterPolygon(field, coordinates, "intersectsPolygon");
+    }
+  }
+  return {
+    type: firstKey.substring(1),
+    field,
+    value: filter[firstKey]
+  };
+}
+async function fetchSingle(client, objectType, args, objectSet) {
+  const result = await fetchPage(client, objectType, {
+    ...args,
+    pageSize: 1
+  }, objectSet);
+  if (result.data.length !== 1 || result.nextPageToken != null) {
+    throw new PalantirApiError(`Expected a single result but got ${result.data.length} instead${result.nextPageToken != null ? " with nextPageToken set" : ""}`);
+  }
+  return result.data[0];
+}
+
+// src/object/index.ts
+var object_exports = {};
+__export(object_exports, {
+  aggregate: () => aggregate,
+  fetchPage: () => fetchPage
+});
+async function aggregate(clientCtx, objectType, objectSet = {
+  type: "base",
+  objectType: objectType["apiName"]
+}, req) {
+  const body = {
+    aggregation: modernToLegacyAggregationClause(req.select),
+    groupBy: [],
+    where: void 0
+  };
+  if (req.groupBy) {
+    body.groupBy = modernToLegacyGroupByClause(req.groupBy);
+  }
+  if (req.where) {
+    body.where = modernToLegacyWhereClause(req.where);
+  }
+  const result = await aggregateObjectSetV2(createOpenApiRequest(clientCtx.stack, clientCtx.fetch), clientCtx.ontology.metadata.ontologyApiName, {
+    objectSet,
+    groupBy: body.groupBy,
+    aggregation: body.aggregation
+  });
+  if (!req.groupBy) {
+    !(result.data.length === 1) ? process.env.NODE_ENV !== "production" ? invariant2(false, "no group by clause should mean only one data result") : invariant2(false) : void 0;
+    return {
+      ...aggregationToCountResult(result.data[0]),
+      ...legacyToModernSingleAggregationResult(result.data[0])
+    };
+  }
+  const ret = result.data.map((entry) => {
+    return {
+      $group: entry.group,
+      ...aggregationToCountResult(entry),
+      ...legacyToModernSingleAggregationResult(entry)
+    };
+  });
+  return ret;
+}
+function aggregationToCountResult(entry) {
+  for (const aggregateResult of entry.metrics) {
+    if (aggregateResult.name === "count") {
+      return {
+        $count: aggregateResult.value
+      };
+    }
+  }
+}
+async function createTemporaryObjectSet(ctx, request) {
+  return conjureFetch(ctx, `/objectSets/temporary`, "POST", request);
+}
+async function batchEnableWatcher(ctx, request) {
+  return conjureFetch(ctx, `/object-set-watcher/batchEnableWatcher`, "POST", request);
+}
+async function loadAllOntologies(ctx, request) {
+  return conjureFetch(ctx, `/ontology/ontology/ontologies/load/all`, "POST", request);
+}
+async function loadOntologyEntities(ctx, request) {
+  return conjureFetch(ctx, `/ontology/ontology/loadEntities`, "POST", request);
+}
+
+// src/objectSet/toConjureObjectSet.ts
+function toConjureObjectSet(objectSet, objectPropertyMapping) {
+  switch (objectSet.type) {
+    case "base":
+      return {
+        type: "base",
+        base: {
+          objectTypeId: objectPropertyMapping.id
+        }
+      };
+    case "static":
+      return {
+        type: "static",
+        static: {
+          objectRids: objectSet.objects,
+          provenance: void 0
+        }
+      };
+    case "reference":
+      return {
+        type: "referenced",
+        referenced: {
+          objectSetRid: objectSet.reference
+        }
+      };
+    case "filter":
+      return {
+        type: "filtered",
+        filtered: {
+          objectSet: toConjureObjectSet(objectSet.objectSet, objectPropertyMapping),
+          runtimeDerivedProperties: void 0,
+          filter: mapWhereClauseToObjectSetFilter(objectSet.where, objectPropertyMapping)
+        }
+      };
+    case "union":
+      return {
+        type: "unioned",
+        unioned: {
+          objectSets: objectSet.objectSets.map((os) => toConjureObjectSet(os, objectPropertyMapping))
+        }
+      };
+    case "intersect":
+      return {
+        type: "intersected",
+        intersected: {
+          objectSets: objectSet.objectSets.map((os) => toConjureObjectSet(os, objectPropertyMapping))
+        }
+      };
+    case "subtract":
+      return {
+        type: "subtracted",
+        subtracted: {
+          objectSets: objectSet.objectSets.map((os) => toConjureObjectSet(os, objectPropertyMapping))
+        }
+      };
+    case "searchAround":
+      throw new Error("not implemented");
+  }
+}
+async function getObjectSetBaseType(objectSet) {
+  switch (objectSet.type) {
+    case "base":
+      return objectSet.objectType;
+    case "static":
+      throw new Error("not implemented");
+    case "reference":
+      throw new Error("not implemented");
+    case "filter":
+      return getObjectSetBaseType(objectSet.objectSet);
+    case "union":
+      return getObjectSetBaseType(objectSet.objectSets[0]);
+    case "intersect":
+      return getObjectSetBaseType(objectSet.objectSets[0]);
+    case "subtract":
+      return getObjectSetBaseType(objectSet.objectSets[0]);
+    case "searchAround":
+      throw new Error("not implemented");
+  }
+}
+function mapWhereClauseToObjectSetFilter(objectSetFilter, propertyMapping) {
+  switch (objectSetFilter.type) {
+    case "lt":
+      return {
+        type: "range",
+        range: {
+          propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field],
+          lt: objectSetFilter.value,
+          lte: void 0,
+          gt: void 0,
+          gte: void 0
+        }
+      };
+    case "gt":
+      return {
+        type: "range",
+        range: {
+          propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field],
+          gt: objectSetFilter.value,
+          lte: void 0,
+          lt: void 0,
+          gte: void 0
+        }
+      };
+    case "lte":
+      return {
+        type: "range",
+        range: {
+          propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field],
+          lte: objectSetFilter.value,
+          lt: void 0,
+          gt: void 0,
+          gte: void 0
+        }
+      };
+    case "gte":
+      return {
+        type: "range",
+        range: {
+          propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field],
+          gte: objectSetFilter.value,
+          lt: void 0,
+          lte: void 0,
+          gt: void 0
+        }
+      };
+    case "eq":
+      return {
+        type: "exactMatch",
+        exactMatch: {
+          propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field],
+          terms: []
+        }
+      };
+    case "and":
+      return {
+        type: "and",
+        and: {
+          filters: objectSetFilter.value.map((filter) => mapWhereClauseToObjectSetFilter(filter, propertyMapping))
+        }
+      };
+    case "or":
+      return {
+        type: "or",
+        or: {
+          filters: objectSetFilter.value.map((filter) => mapWhereClauseToObjectSetFilter(filter, propertyMapping))
+        }
+      };
+    case "isNull":
+      return {
+        type: "not",
+        not: {
+          filter: {
+            type: "hasProperty",
+            hasProperty: {
+              propertyId: propertyMapping.propertyApiNameToIdMapping[objectSetFilter.field]
+            }
+          }
+        }
+      };
+    case "not":
+      return {
+        type: "not",
+        not: {
+          filter: mapWhereClauseToObjectSetFilter(objectSetFilter.value, propertyMapping)
+        }
+      };
+    case "contains":
+    case "startsWith":
+    case "containsAllTermsInOrder":
+    case "containsAnyTerm":
+    case "containsAllTerms":
+    case "withinDistanceOf":
+    case "withinBoundingBox":
+    case "intersectsBoundingBox":
+    case "doesNotIntersectBoundingBox":
+    case "withinPolygon":
+    case "intersectsPolygon":
+    case "doesNotIntersectPolygon":
+    case "containsAllTermsInOrderPrefixLastTerm":
+      throw new Error("not implemented");
+  }
+}
+
+// src/objectSet/ObjectSetListenerWebsocket.ts
+var ONE_DAY_MS = 24 * 60 * 60 * 1e3;
+var MINIMUM_RECONNECT_DELAY_MS = 5 * 1e3;
+var ObjectSetListenerWebsocket = class _ObjectSetListenerWebsocket {
+  static #instances = /* @__PURE__ */ new WeakMap();
+  // FIXME
+  static getInstance(client) {
+    let instance = _ObjectSetListenerWebsocket.#instances.get(client);
+    if (instance == null) {
+      instance = new _ObjectSetListenerWebsocket(client);
+      _ObjectSetListenerWebsocket.#instances.set(client, instance);
+    }
+    return instance;
+  }
+  #ws;
+  #lastWsConnect = 0;
+  #client;
+  /** map of listenerId to listener */
+  #listeners = /* @__PURE__ */ new Map();
+  /** map of subscriptionId to listenerId */
+  #subscriptionToRequestId = /* @__PURE__ */ new Map();
+  #conjureContext;
+  #metadataContext;
+  #ossContext;
+  constructor(client) {
+    this.#client = client;
+    const stackUrl = new URL(client.stack);
+    this.#conjureContext = {
+      baseUrl: stackUrl.origin,
+      servicePath: "/object-set-watcher/api",
+      fetchFn: client.fetch,
+      tokenProvider: async () => await client.tokenProvider()
+    };
+    this.#ossContext = {
+      baseUrl: stackUrl.origin,
+      servicePath: "/object-set-service/api",
+      fetchFn: client.fetch,
+      tokenProvider: async () => await client.tokenProvider()
+    };
+    this.#metadataContext = {
+      baseUrl: stackUrl.origin,
+      servicePath: "/ontology-metadata/api",
+      fetchFn: client.fetch,
+      tokenProvider: async () => await client.tokenProvider()
+    };
+  }
+  subscribe(objectSet, listener) {
+    const requestId = crypto.randomUUID();
+    const expiry = setTimeout(() => {
+      this.#expire(requestId);
+    }, ONE_DAY_MS);
+    this.#listeners.set(requestId, {
+      listener,
+      objectSet,
+      expiry
+    });
+    this.#subscribe(requestId, objectSet);
+    return () => {
+      this.#unsubscribe(requestId);
+    };
+  }
+  async #subscribe(requestId, objectSet) {
+    try {
+      const [temporaryObjectSet] = await Promise.all([
+        // create a time-bounded object set representation for watching
+        this.#createTemporaryObjectSet(objectSet),
+        this.#ensureWebsocket(),
+        // look up the object type's rid and ensure that we have enabled object set watcher for that rid
+        // TODO ???
+        getObjectSetBaseType(objectSet).then((baseType) => getObjectTypeV2(createOpenApiRequest(this.#client.stack, this.#client.fetch), this.#client.ontology.metadata.ontologyApiName, baseType)).then((objectType) => this.#enableObjectSetsWatcher([objectType.rid]))
+      ]);
+      if (!this.#listeners.has(requestId)) {
+        return;
+      }
+      const subscribe = {
+        id: requestId,
+        requests: [{
+          objectSet: temporaryObjectSet.objectSetRid,
+          objectSetContext: {
+            objectSetFilterContext: {
+              parameterOverrides: {}
+            }
+          },
+          watchAllLinks: false
+        }]
+      };
+      this.#ws?.send(JSON.stringify(subscribe));
+    } catch (error) {
+      this.#getCallbackByRequestId(requestId, "onError")?.(error);
+    }
+  }
+  #expire(requestId) {
+    const state = this.#listeners.get(requestId);
+    if (state) {
+      const {
+        subscriptionId,
+        objectSet
+      } = state;
+      if (subscriptionId) {
+        state.subscriptionId = void 0;
+        this.#subscriptionToRequestId.delete(subscriptionId);
+      }
+      this.#subscribe(requestId, objectSet);
+    }
+  }
+  #unsubscribe(requestId) {
+    const data = this.#listeners.get(requestId);
+    if (data == null) {
+      return;
+    }
+    this.#listeners.delete(requestId);
+    clearTimeout(data.expiry);
+    const {
+      subscriptionId
+    } = data;
+    if (subscriptionId != null) {
+      this.#subscriptionToRequestId.delete(subscriptionId);
+    }
+    if (this.#listeners.size === 0) {
+      this.#destroyWebsocket();
+    }
+  }
+  async #ensureWebsocket() {
+    if (this.#ws == null) {
+      const {
+        stack,
+        tokenProvider
+      } = this.#client;
+      const base = new URL(stack);
+      const url = `wss://${base.host}/object-set-watcher/ws/subscriptions`;
+      const token = await tokenProvider();
+      if (this.#ws == null) {
+        const nextConnectTime = (this.#lastWsConnect ?? 0) + MINIMUM_RECONNECT_DELAY_MS;
+        if (nextConnectTime > Date.now()) {
+          await new Promise((resolve) => {
+            setTimeout(resolve, nextConnectTime - Date.now());
+          });
+        }
+        this.#lastWsConnect = Date.now();
+        if (this.#ws == null) {
+          this.#ws = new WebSocket(url, [`Bearer-${token}`]);
+          this.#ws.addEventListener("close", this.#onClose);
+          this.#ws.addEventListener("message", this.#onMessage);
+          this.#ws.addEventListener("open", this.#onOpen);
+        }
+      }
+      if (this.#ws.readyState === WebSocket.CONNECTING) {
+        return new Promise((resolve, reject) => {
+          this.#ws.addEventListener("open", () => {
+            resolve();
+          });
+          this.#ws.addEventListener("error", (event) => {
+            reject(new Error(event.toString()));
+          });
+        });
+      }
+    }
+  }
+  #onOpen = () => {
+    for (const [requestId, state] of this.#listeners) {
+      this.#subscribe(requestId, state.objectSet);
+    }
+  };
+  #onMessage = async (message) => {
+    const data = JSON.parse(message.data.toString());
+    switch (data.type) {
+      case "objectSetChanged": {
+        if (data.objectSetChanged.confidenceValue) {
+          this.#getCallback(data.objectSetChanged.id, "onOutOfDate")?.();
+          break;
+        }
+        const {
+          id: subscriptionId,
+          objects
+        } = data.objectSetChanged;
+        const callback = this.#getCallback(subscriptionId, "onChange");
+        if (callback) {
+          callback(await convertFoundryToOsdkObjects(this.#client, this.#metadataContext, objects));
+        }
+        break;
+      }
+      case "refreshObjectSet": {
+        const {
+          id: subscriptionId
+        } = data.refreshObjectSet;
+        this.#getCallback(subscriptionId, "onOutOfDate")?.();
+        break;
+      }
+      case "subscribeResponses": {
+        const {
+          id: requestId,
+          responses
+        } = data.subscribeResponses;
+        const listenerData = this.#listeners.get(requestId);
+        if (listenerData == null) {
+          return;
+        }
+        if (responses.length !== 1) {
+          throw new Error("Got more than one response but we only expect a single one");
+        }
+        const response = responses[0];
+        switch (response.type) {
+          case "error":
+            this.#getCallbackByRequestId(requestId, "onError")?.(response.error);
+            this.#unsubscribe(requestId);
+            return;
+          case "qos":
+            this.#destroyWebsocket();
+            this.#ensureWebsocket();
+            return;
+          case "success":
+            const {
+              id: subscriptionId
+            } = response.success;
+            listenerData.subscriptionId = subscriptionId;
+            this.#subscriptionToRequestId.set(subscriptionId, requestId);
+            this.#getCallbackByRequestId(requestId, "onOutOfDate")?.();
+            break;
+          default:
+            this.#getCallbackByRequestId(requestId, "onError")?.(response);
+        }
+        break;
+      }
+    }
+  };
+  #onClose = () => {
+    this.#destroyWebsocket();
+  };
+  async #enableObjectSetsWatcher(objectTypeRids) {
+    return batchEnableWatcher(this.#conjureContext, {
+      requests: objectTypeRids
+    });
+  }
+  async #createTemporaryObjectSet(objectSet) {
+    const objectSetBaseType = await getObjectSetBaseType(objectSet);
+    const mapping = await getOntologyPropertyMappingForApiName(this.#client, this.#metadataContext, objectSetBaseType);
+    const temporaryObjectSet = await createTemporaryObjectSet(this.#ossContext, {
+      objectSet: toConjureObjectSet(objectSet, mapping),
+      timeToLive: "ONE_DAY",
+      objectSetFilterContext: {
+        parameterOverrides: {}
+      }
+    });
+    return {
+      objectSetRid: temporaryObjectSet.objectSetRid
+    };
+  }
+  #destroyWebsocket = () => {
+    if (this.#ws) {
+      this.#ws.removeEventListener("open", this.#onOpen);
+      this.#ws.removeEventListener("message", this.#onMessage);
+      this.#ws.removeEventListener("close", this.#onClose);
+      if (this.#ws.readyState !== WebSocket.CLOSING && this.#ws.readyState !== WebSocket.CLOSED) {
+        this.#ws.close();
+      }
+      this.#ws = void 0;
+    }
+    this.#subscriptionToRequestId.clear();
+    for (const state of this.#listeners.values()) {
+      state.subscriptionId = void 0;
+    }
+    if (this.#listeners.size > 0) {
+      this.#ensureWebsocket();
+    }
+  };
+  #getCallbackByRequestId(requestId, type) {
+    const maybeListener = this.#listeners.get(requestId);
+    return maybeListener?.listener?.[type];
+  }
+  #getCallback(subscriptionId, type) {
+    const requestId = this.#subscriptionToRequestId.get(subscriptionId);
+    if (requestId) {
+      return this.#getCallbackByRequestId(requestId, type);
+    }
+    return;
+  }
+};
+async function convertFoundryToOsdkObjects(client, ctx, objects) {
+  const osdkObjects = await Promise.all(objects.map(async (object) => {
+    const propertyMapping = await getOntologyPropertyMappingForRid(ctx, client.ontology.metadata.ontologyRid, object.type);
+    const convertedObject = Object.fromEntries([...Object.entries(object.properties).map(([key, value]) => {
+      return [propertyMapping?.propertyIdToApiNameMapping[key], value];
+    }), [propertyMapping?.propertyIdToApiNameMapping[Object.entries(object.key)[0][0]], Object.entries(object.key)[0][1]], ["__apiName", propertyMapping?.apiName], ["$apiName", propertyMapping?.apiName]]);
+    return convertedObject;
+  }));
+  await convertWireToOsdkObjectsInPlace(client, osdkObjects);
+  return osdkObjects;
+}
+var objectTypeMapping = /* @__PURE__ */ new WeakMap();
+var objectApiNameToRid = /* @__PURE__ */ new Map();
+async function getOntologyPropertyMappingForApiName(client, ctx, objectApiName) {
+  if (objectApiNameToRid.has(objectApiName)) {
+    return objectTypeMapping.get(ctx)?.get(objectApiNameToRid.get(objectApiName));
+  }
+  const wireObjectType = await getObjectTypeV2(createOpenApiRequest(client.stack, client.fetch), client.ontology.metadata.ontologyApiName, objectApiName);
+  return getOntologyPropertyMappingForRid(ctx, client.ontology.metadata.ontologyRid, wireObjectType.rid);
+}
+var cachedAllOntologies;
+async function getOntologyVersionForRid(ctx, ontologyRid) {
+  cachedAllOntologies ??= await loadAllOntologies(ctx, {});
+  !cachedAllOntologies.ontologies[ontologyRid] ? process.env.NODE_ENV !== "production" ? invariant2(false, "ontology should be loaded") : invariant2(false) : void 0;
+  return cachedAllOntologies.ontologies[ontologyRid].currentOntologyVersion;
+}
+async function getOntologyPropertyMappingForRid(ctx, ontologyRid, objectRid) {
+  if (!objectTypeMapping.has(ctx)) {
+    objectTypeMapping.set(ctx, /* @__PURE__ */ new Map());
+  }
+  if (!objectTypeMapping.get(ctx).has(objectRid)) {
+    const ontologyVersion = await getOntologyVersionForRid(ctx, ontologyRid);
+    const body = {
+      objectTypeVersions: {
+        // TODO: Undefined drops this in the body
+        [objectRid]: ontologyVersion
+      },
+      linkTypeVersions: {},
+      loadRedacted: false,
+      includeObjectTypesWithoutSearchableDatasources: true
+    };
+    const entities = await loadOntologyEntities(ctx, body);
+    !entities.objectTypes[objectRid] ? process.env.NODE_ENV !== "production" ? invariant2(false, "object type should be loaded") : invariant2(false) : void 0;
+    const propertyIdToApiNameMapping = Object.fromEntries(Object.values(entities.objectTypes[objectRid].propertyTypes).map((property) => {
+      return [property.id, property.apiName];
+    }));
+    const propertyApiNameToIdMapping = Object.fromEntries(Object.values(entities.objectTypes[objectRid].propertyTypes).map((property) => {
+      return [property.id, property.apiName];
+    }));
+    objectTypeMapping.get(ctx)?.set(objectRid, {
+      apiName: entities.objectTypes[objectRid].apiName,
+      id: entities.objectTypes[objectRid].id,
+      propertyIdToApiNameMapping,
+      propertyApiNameToIdMapping
+    });
+    objectApiNameToRid.set(entities.objectTypes[objectRid].apiName, objectRid);
+  }
+  return objectTypeMapping.get(ctx)?.get(objectRid);
+}
+
+// src/objectSet/createObjectSet.ts
+function isObjectTypeDefinition(def) {
+  return def.type === "object";
+}
+var searchAroundPrefix = "searchAround_";
+function createObjectSet2(objectType, clientCtx, objectSet = {
+  type: "base",
+  objectType: objectType["apiName"]
+}) {
+  const base = {
+    definition: objectSet,
+    aggregate: (req) => {
+      return aggregate(clientCtx, objectType, objectSet, req);
+    },
+    aggregateOrThrow: (req) => {
+      return aggregate(clientCtx, objectType, objectSet, req);
+    },
+    fetchPage: async (args) => {
+      return fetchPage(clientCtx, objectType, args ?? {}, objectSet);
+    },
+    fetchPageOrThrow: async (args) => {
+      return fetchPage(clientCtx, objectType, args ?? {}, objectSet);
+    },
+    where: (clause) => {
+      return createObjectSet2(objectType, clientCtx, {
+        type: "filter",
+        objectSet,
+        where: modernToLegacyWhereClause(clause)
+      });
+    },
+    pivotTo: function(type) {
+      return createSearchAround(type)();
+    },
+    union: (...objectSets) => {
+      return createObjectSet2(objectType, clientCtx, {
+        type: "union",
+        objectSets: [objectSet, ...objectSets.map((os) => os.definition)]
+      });
+    },
+    intersect: (...objectSets) => {
+      return createObjectSet2(objectType, clientCtx, {
+        type: "intersect",
+        objectSets: [objectSet, ...objectSets.map((os) => os.definition)]
+      });
+    },
+    subtract: (...objectSets) => {
+      return createObjectSet2(objectType, clientCtx, {
+        type: "subtract",
+        objectSets: [objectSet, ...objectSets.map((os) => os.definition)]
+      });
+    },
+    subscribe(listener) {
+      const instance = ObjectSetListenerWebsocket.getInstance(clientCtx);
+      return instance.subscribe(objectSet, listener);
+    },
+    asyncIter: async function* () {
+      let nextPageToken = void 0;
+      do {
+        const result = await base.fetchPage({
+          nextPageToken
+        });
+        await convertWireToOsdkObjectsInPlace(clientCtx, result.data);
+        for (const obj of result.data) {
+          yield obj;
+        }
+      } while (nextPageToken != null);
+    }
+  };
+  function createSearchAround(link) {
+    return () => {
+      return createBaseObjectSet(objectType, clientCtx, {
+        type: "searchAround",
+        objectSet,
+        link
+      });
+    };
+  }
+  return new Proxy(base, {
+    get(target, p, receiver) {
+      if (typeof p === "string" && p.startsWith(searchAroundPrefix)) {
+        return createSearchAround(p.substring(searchAroundPrefix.length));
+      }
+      return target[p];
+    }
+  });
+}
+function createBaseObjectSet(objectType, clientCtx, objectSet = {
+  type: "base",
+  objectType: objectType["apiName"]
+}) {
+  return {
+    ...createObjectSet2(objectType, clientCtx, objectSet),
+    get: isObjectTypeDefinition(objectType) ? async (primaryKey, options) => {
+      const withPk = {
+        type: "filter",
+        objectSet,
+        where: modernToLegacyWhereClause({
+          [objectType.primaryKeyApiName]: primaryKey
+        })
+      };
+      return await fetchSingle(clientCtx, objectType, options, withPk);
+    } : void 0
+  };
+}
+
+// src/object/Attachment.ts
+var Attachment = class {
+  constructor(rid) {
+    this.rid = rid;
+  }
+};
+function isAttachment(o) {
+  return o instanceof Attachment;
+}
+
+// src/object/Cache.ts
+function createCache(fn) {
+  const cache = /* @__PURE__ */ new WeakMap();
+  function get(client, key) {
+    if (cache.get(client) == null) {
+      cache.set(client, /* @__PURE__ */ new Map());
+    }
+    let r = cache.get(client).get(key);
+    if (r === void 0 && fn !== void 0) {
+      return set(client, key, fn(client, key));
+    } else {
+      return r;
+    }
+  }
+  function set(client, key, value) {
+    if (cache.get(client) == null) {
+      cache.set(client, /* @__PURE__ */ new Map());
+    }
+    cache.get(client).set(key, value);
+    return value;
+  }
+  return {
+    get,
+    set
+  };
+}
+function createAsyncCache(fn) {
+  const cache = createCache();
+  function getOrUndefined(client, key) {
+    return cache.get(client, key);
+  }
+  async function get(client, key) {
+    return cache.get(client, key) ?? set(client, key, fn(client, key));
+  }
+  async function set(client, k, v) {
+    return cache.set(client, k, await v);
+  }
+  return {
+    get,
+    set,
+    getOrUndefined
+  };
+}
+
+// src/object/convertWireToOsdkObjects.ts
+var OriginClient = Symbol();
+function createPrototype(objDef, client) {
+  const proto = {};
+  Object.defineProperty(proto, "$link", {
+    get: function() {
+      const primaryKey = this["__primaryKey"];
+      return new Proxy({}, {
+        get(_target, p, _receiver) {
+          const linkDef = objDef.links[p];
+          if (linkDef == null) {
+            return;
+          }
+          const objectSet = createBaseObjectSet(objDef, client).where({
+            [objDef.primaryKeyApiName]: primaryKey
+          }).pivotTo(p);
+          if (!linkDef.multiplicity) {
+            return {
+              get: (options) => fetchSingle(client, objDef, options ?? {}, objectSet.definition)
+            };
+          } else {
+            return objectSet;
+          }
+        }
+      });
+    },
+    enumerable: false,
+    configurable: false
+  });
+  return proto;
+}
+function createConverter(objDef) {
+  const steps = [];
+  for (const [key, value] of Object.entries(objDef.properties)) {
+    if (value.type === "attachment") {
+      steps.push((o) => {
+        if (o[key] != null) {
+          if (Array.isArray(o[key])) {
+            o[key] = o[key].map((a) => new Attachment(a.rid));
+          } else {
+            o[key] = new Attachment(o[key].rid);
+          }
+        }
+      });
+    }
+  }
+  return steps.length > 0 ? (o) => {
+    for (const step of steps) {
+      step(o);
+    }
+  } : false;
+}
+var protoConverterCache = createCache((client, objectDef) => {
+  const proto = createPrototype(objectDef, client);
+  const converter = createConverter(objectDef);
+  return {
+    proto,
+    converter
+  };
+});
+async function convertWireToOsdkObjectsInPlace(client, objs) {
+  for (const obj of objs) {
+    if (obj.__rid) {
+      obj.$rid = obj.__rid;
+      delete obj.__rid;
+    }
+    obj.$apiName = obj.__apiName;
+    obj.$objectType = obj.__apiName;
+    obj.$primaryKey = obj.__primaryKey;
+    {
+      Object.defineProperties(obj, {
+        "__apiName": {
+          enumerable: false
+        },
+        "__primaryKey": {
+          enumerable: false
+        }
+      });
+    }
+  }
+  const localObjectCache = createAsyncCache((client2, apiName) => client2.ontology.provider.getObjectOrInterfaceDefinition(apiName));
+  const uniqueApiNames = new Set(objs.map((o) => o.$apiName));
+  await Promise.all(Array.from(uniqueApiNames).map((n) => localObjectCache.get(client, n)));
+  for (const obj of objs) {
+    const objectDef = await localObjectCache.get(client, obj.$apiName);
+    if (objectDef == null) {
+      throw new Error(`Failed to find ontology definition for '${obj.$apiName}'`);
+    } else if (objectDef.type === "interface") {
+      throw new Error("Interface objects are not supported currently");
+    }
+    const {
+      proto,
+      converter
+    } = protoConverterCache.get(client, objectDef);
+    Object.setPrototypeOf(obj, proto);
+    Object.defineProperty(obj, OriginClient, {
+      value: client,
+      enumerable: false,
+      configurable: false,
+      writable: false
+    });
+    if (converter) {
+      converter(obj);
+    }
+  }
+}
+
+// src/object/fetchPage.ts
+async function fetchPage(client, objectType, args, objectSet = {
+  type: "base",
+  objectType: objectType["apiName"]
+}) {
+  const body = {
+    objectSet,
+    // We have to do the following case because LoadObjectSetRequestV2 isnt readonly
+    select: args?.select ?? [],
+    // FIXME?
+    excludeRid: !args?.includeRid
+  };
+  if (args?.nextPageToken) {
+    body.pageToken = args.nextPageToken;
+  }
+  if (args?.pageSize != null) {
+    body.pageSize = args.pageSize;
+  }
+  if (args?.orderBy != null) {
+    body.orderBy = {
+      fields: Object.entries(args.orderBy).map(([field, direction]) => ({
+        field,
+        direction
+      }))
+    };
+  }
+  const r = await loadObjectSetV2(createOpenApiRequest(client.stack, client.fetch), client.ontology.metadata.ontologyApiName, body);
+  await convertWireToOsdkObjectsInPlace(client, r.data);
+  return r;
+}
+
+export { aggregate, createAsyncCache, createBaseObjectSet, createObjectSet2 as createObjectSet, fetchPage, isAttachment, object_exports };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=chunk-GOBQUVQZ.mjs.map