@takeshape/schema 11.40.0 → 11.41.0

package/dist/cjs/src/refs.js

@@ -0,0 +1,662 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.propertyRefItemToPath = exports.queryRefToLocalName = exports.propertyRefItemToLocalName = exports.propertyRefItemToArgsPath = exports.propertyRefItemToResolverPath = exports.applyNamespace = exports.formatShapeRef = exports.propertyRefToShapeRef = exports.normalizePropertyRef = exports.serializePropertyRef = exports.parsePropertyRef = exports.createGetNamespace = exports.dereferenceObjectSchema = exports.dereferenceSchema = exports.hasResolvableRef = exports.hasRef = exports.followRef = exports.omitRefAndExtend = exports.anyRefToShapeSchemaPath = exports.getRefShapeName = exports.refItemToShapeSchema = exports.refItemToShapeWithPath = exports.refItemListToShapeList = exports.refItemToShape = exports.refItemToShapePath = exports.refItemToNamespacedShapeName = exports.refItemToAtRef = exports.getRefOrItemsRef = exports.getOneOfRefs = exports.hasRefProperty = exports.getRefWithPath = exports.getRef = exports.shapeToRefString = exports.shapeToRefItem = exports.parseName = exports.refToRefItem = exports.atRefToRefItem = exports.splitAtRef = exports.refExpressionToShape = exports.refExpressionListToRefItemList = exports.refExpressionToRefItem = exports.refSchemaToPath = exports.$refToAtRef = exports.getFlattenedTemplateShapeName = exports.untemplate = exports.createTemplateShapeName = exports.returnShapeToSchema = exports.parseReturnShape = exports.parseTemplateShape = exports.isRefItemWithPath = void 0;
+exports.normalizeRefs = exports.ensureQuery = exports.getQuery = exports.getAllPropertyRefs = exports.normalizeLocalRefString = exports.normalizeRefExpression = exports.propertyRefToArgs = void 0;
+const util_1 = require("@takeshape/util");
+const get_js_1 = __importDefault(require("lodash/get.js"));
+const assign_js_1 = __importDefault(require("lodash/fp/assign.js"));
+const omit_js_1 = __importDefault(require("lodash/fp/omit.js"));
+const isString_js_1 = __importDefault(require("lodash/isString.js"));
+const errors_1 = require("@takeshape/errors");
+const compact_js_1 = __importDefault(require("lodash/compact.js"));
+const util_2 = require("./services/util");
+const merge_1 = require("./util/merge");
+const ai_tools_1 = require("./util/ai-tools");
+const scalars_1 = require("./scalars");
+const types_1 = require("./types");
+/**
+ * Guard for RefItemWithPath. Tests for presence of all required props.
+ */
+function isRefItemWithPath(ref) {
+    return Boolean(ref.typeName && ref.serviceKey && ref.path && (0, util_1.isDefined)(ref.isForeign));
+}
+exports.isRefItemWithPath = isRefItemWithPath;
+const templateShapeRegex = /^(\w+)<([\w:-]+)>$/;
+/**
+ * Parse a template like `PaginatedList<Post>` and return both the template and the shape name.
+ */
+function parseTemplateShape(shapeExpression) {
+    const matches = templateShapeRegex.exec(shapeExpression);
+    if (matches) {
+        return { shapeName: matches[2], template: matches[1] };
+    }
+    return { shapeName: shapeExpression, template: undefined };
+}
+exports.parseTemplateShape = parseTemplateShape;
+function parseReturnShape(projectSchema, shape) {
+    if (typeof shape === 'object') {
+        if ((0, types_1.isRefSchema)(shape.items)) {
+            const ref = getRef(projectSchema, shape.items);
+            if (!ref) {
+                throw new Error(`Could not parse ${JSON.stringify(shape.items)}: invalid ref`);
+            }
+            return { isArray: true, ref, shapeName: refItemToNamespacedShapeName(ref) };
+        }
+        if (typeof shape.items.type !== 'string') {
+            throw new Error(`Could not parse ${JSON.stringify(shape.items)}: invalid type`);
+        }
+        return { isArray: true, shapeName: shape.items.type };
+    }
+    const ref = refExpressionToRefItem(projectSchema, shape);
+    const shapeName = refItemToNamespacedShapeName(ref);
+    return { isArray: false, shapeName, ref, template: ref.template };
+}
+exports.parseReturnShape = parseReturnShape;
+/**
+ * Convert Query.shape to a PropertySchema
+ */
+function returnShapeToSchema(projectSchema, returnShape) {
+    if (typeof returnShape === 'string') {
+        if ((0, scalars_1.isScalar)(returnShape) || returnShape === 'object') {
+            return { type: returnShape };
+        }
+        return { '@ref': refItemToAtRef(refExpressionToRefItem(projectSchema, returnShape)) };
+    }
+    return returnShape;
+}
+exports.returnShapeToSchema = returnShapeToSchema;
+/**
+ * Create a template string like `PaginatedList<Post>` from a template and shape name.
+ */
+function createTemplateShapeName(template, shapeName) {
+    return `${template}<${shapeName}>`;
+}
+exports.createTemplateShapeName = createTemplateShapeName;
+/**
+ * If the string is a template like `PaginatedList<Post>`, return the shape name,
+ * in this case `Post`. Otherwise return the input.
+ */
+const untemplate = (input) => {
+    return parseTemplateShape(input).shapeName;
+};
+exports.untemplate = untemplate;
+function getFlattenedTemplateShapeName(shapeName, template) {
+    return `${shapeName}${template ?? ''}`;
+}
+exports.getFlattenedTemplateShapeName = getFlattenedTemplateShapeName;
+function $refToPath(ref) {
+    return ref.substr(2).split('/');
+}
+function $refToShapeName(ref) {
+    return $refToPath(ref)[1];
+}
+function $refToAtRef($ref, service, namespace) {
+    const shapeName = $refToShapeName($ref);
+    if (service === 'local') {
+        return `local:${shapeName}`;
+    }
+    return `${service}:${namespace === undefined ? shapeName : shapeName.replace(`${namespace}_`, '')}`;
+}
+exports.$refToAtRef = $refToAtRef;
+function refSchemaToPath(context, refSchema) {
+    const refItem = getRef(context, refSchema);
+    if (refItem === undefined) {
+        throw new Error('Invalid ref in refSchemaToPath');
+    }
+    return refItemToShapePath(refItem);
+}
+exports.refSchemaToPath = refSchemaToPath;
+/**
+ * Safely turn a "template" shape reference (`@args`, `args`, `shape`) into a `RefItem`
+ *
+ * @param context A partial project schema
+ * @param refExpression
+ * A ref expression might include "template" name and refer to a local or remote
+ * shape, `Foo` or `CreateArgs<Foo>` or `CreateArgs<remote:Foo>`.*
+ */
+function refExpressionToRefItem(context, refExpression) {
+    const { shapeName, template } = parseTemplateShape(refExpression);
+    if (shapeName.includes(':')) {
+        return atRefToRefItem(context, shapeName, template);
+    }
+    return shapeNameToRefItem(context, shapeName, template);
+}
+exports.refExpressionToRefItem = refExpressionToRefItem;
+/**
+ * Converts a list of ref expressions into a list of ref items.
+ */
+function refExpressionListToRefItemList(context, refExpressionList) {
+    return refExpressionList.map(ref => refExpressionToRefItem(context, ref));
+}
+exports.refExpressionListToRefItemList = refExpressionListToRefItemList;
+/**
+ * Sugar for converting a `refExpression` directly into a shape, without the
+ * intermediate `refItem`.
+ *
+ * @param context A partial project schema
+ * @param refExpression
+ * A ref expression might include "template" name and refer to a local or remote
+ * shape, `Foo` or `CreateArgs<Foo>` or `CreateArgs<remote:Foo>`.
+ */
+function refExpressionToShape(context, refExpression) {
+    return refItemToShape(context, refExpressionToRefItem(context, refExpression));
+}
+exports.refExpressionToShape = refExpressionToShape;
+/**
+ * Provided a service name and a referenced type. Supports 2 or 3 part
+ * references, e.g., `shopify:my-store:ProductInput` or `my-store:ProductInput`.
+ *
+ * @returns
+ * A 2 member array, where 0 is the service name, 1 is the referenced type.
+ */
+function splitAtRef(atRef) {
+    const index = atRef.lastIndexOf(':');
+    const serviceKey = index === -1 ? 'local' : atRef.substring(0, index);
+    const referencedType = index === -1 ? atRef : atRef.substring(index + 1);
+    return [serviceKey, referencedType];
+}
+exports.splitAtRef = splitAtRef;
+function isValidServiceId(context, serviceId) {
+    return serviceId === 'local' || Boolean(context.services?.[serviceId]);
+}
+/**
+ * Parses and returns a typeName, serviceKey, and potentially a namespace for
+ * the type.
+ */
+function atRefToRefItem(context, atRef, template) {
+    const [serviceKey, typeName] = splitAtRef(atRef);
+    // It's possible the service doesn't have a namespace, or is local and has
+    // no service config.
+    const serviceConfig = context.services?.[serviceKey];
+    const serviceNamespace = serviceConfig?.namespace;
+    return {
+        typeName,
+        serviceKey,
+        serviceNamespace,
+        template,
+        isForeign: Boolean(serviceConfig),
+        isValidService: isValidServiceId(context, serviceKey)
+    };
+}
+exports.atRefToRefItem = atRefToRefItem;
+function refToRefItem(context, $ref, template) {
+    return shapeNameToRefItem(context, $refToShapeName($ref), template);
+}
+exports.refToRefItem = refToRefItem;
+/**
+ * Parse a name (query, shape, etc) that might have a namespace and find the corresponding service layer id
+ */
+function parseName(context, name) {
+    const shapeNameParts = name.split('_');
+    let serviceId;
+    let namespace;
+    if (shapeNameParts.length > 1) {
+        const namespaces = (0, util_2.getServiceNamespaces)(context);
+        const maybeNamespace = shapeNameParts[0];
+        serviceId = namespaces.get(maybeNamespace);
+        if (serviceId) {
+            name = shapeNameParts.slice(1).join('_');
+            namespace = maybeNamespace;
+        }
+    }
+    return {
+        name,
+        serviceId: serviceId ?? 'local',
+        namespace
+    };
+}
+exports.parseName = parseName;
+function shapeNameToRefItem(context, shapeName, template) {
+    const parsed = parseName(context, shapeName);
+    return {
+        typeName: parsed.name,
+        serviceKey: parsed.serviceId,
+        serviceNamespace: parsed.namespace,
+        template,
+        isForeign: parsed.serviceId !== 'local',
+        isValidService: isValidServiceId(context, parsed.serviceId)
+    };
+}
+function shapeToRefItem(context, shape) {
+    return shapeNameToRefItem(context, shape.name);
+}
+exports.shapeToRefItem = shapeToRefItem;
+function shapeToRefString(context, shape) {
+    return refItemToAtRef(shapeNameToRefItem(context, shape.name));
+}
+exports.shapeToRefString = shapeToRefString;
+/**
+ * Given a property schema and project schema, finds a `@ref` or `$ref` and
+ * returns a `RefItem` suitable for loading a referenced schema...
+ */
+function getRef(context, refSchema) {
+    if (refSchema['@ref']) {
+        // @ref: my-service:ShapeName
+        const { shapeName, template } = parseTemplateShape(refSchema['@ref']);
+        return atRefToRefItem(context, shapeName, template);
+    }
+    if (refSchema.$ref) {
+        // $ref: #/shapes/[ShapeName OR Namespace_TypeName]/schema
+        return refToRefItem(context, refSchema.$ref);
+    }
+}
+exports.getRef = getRef;
+function getRefWithPath(context, refSchema, schemaPath = [], isInterfaceRef = false) {
+    const refItem = getRef(context, refSchema);
+    if (refItem) {
+        const refType = (0, types_1.getRefType)(refSchema);
+        return {
+            ...refItem,
+            path: schemaPath.concat([refType]),
+            isInterfaceRef
+        };
+    }
+}
+exports.getRefWithPath = getRefWithPath;
+/**
+ * Tests for a `@ref` or `$ref` property directly on the passed schema.
+ *
+ * @param schema Any schema that might could have a ref property.
+ */
+function hasRefProperty(schema) {
+    return Boolean(schema['@ref'] ?? schema.$ref);
+}
+exports.hasRefProperty = hasRefProperty;
+/**
+ * From a property that has oneOf or items with oneOf, get all refs
+ */
+function getOneOfRefs(context, propertySchema, schemaPath = []) {
+    const oneOf = propertySchema.oneOf ?? propertySchema.items?.oneOf;
+    const path = propertySchema.oneOf ? schemaPath : [...schemaPath, 'items'];
+    if (oneOf) {
+        return (0, compact_js_1.default)(oneOf.map(one => {
+            return getRefWithPath(context, one, path);
+        }));
+    }
+    return [];
+}
+exports.getOneOfRefs = getOneOfRefs;
+/**
+ * From a schema with a ref, will return the top-level ref, and potentially
+ * a ref item from a nested `items` property.
+ */
+function getRefOrItemsRef(context, refSchema, schemaPath = []) {
+    const refItem = getRefWithPath(context, refSchema, schemaPath);
+    if (refItem) {
+        return refItem;
+    }
+    if (refSchema.items) {
+        return getRefOrItemsRef(context, refSchema.items, schemaPath.concat(['items']));
+    }
+}
+exports.getRefOrItemsRef = getRefOrItemsRef;
+/**
+ * Turns a `RefItem` into an `@ref`.
+ */
+function refItemToAtRef(refItem) {
+    const shapeRef = `${refItem.serviceKey}:${refItem.typeName}`;
+    return refItem.template ? createTemplateShapeName(refItem.template, shapeRef) : shapeRef;
+}
+exports.refItemToAtRef = refItemToAtRef;
+/**
+ * Turns a `RefItem` into a standard shape name, namespacing if present.
+ */
+function refItemToNamespacedShapeName(refItem) {
+    const { serviceNamespace, typeName } = refItem;
+    return serviceNamespace ? `${serviceNamespace}_${typeName}` : typeName;
+}
+exports.refItemToNamespacedShapeName = refItemToNamespacedShapeName;
+/**
+ * Turns a `RefItem` into a path suitable for `lodash.get`.
+ */
+function refItemToShapePath(refItem) {
+    return ['shapes', refItemToNamespacedShapeName(refItem)];
+}
+exports.refItemToShapePath = refItemToShapePath;
+/**
+ * Get a Shape referenced by a `RefItem`.
+ */
+function refItemToShape(context, refItem) {
+    const shapePath = refItemToShapePath(refItem);
+    return (0, get_js_1.default)(context, shapePath);
+}
+exports.refItemToShape = refItemToShape;
+/**
+ * Get all shapes referenced by a RefItem array.
+ */
+function refItemListToShapeList(context, refItems) {
+    return refItems.map(ref => refItemToShape(context, ref)).filter(util_1.isDefined);
+}
+exports.refItemListToShapeList = refItemListToShapeList;
+/**
+ * Get a shape referenced by a `RefItem`, also returning the path to the new shape.
+ */
+function refItemToShapeWithPath(context, refItem) {
+    const shapePath = refItemToShapePath(refItem);
+    const shape = (0, get_js_1.default)(context, shapePath);
+    return shape ? { shape, path: shapePath } : undefined;
+}
+exports.refItemToShapeWithPath = refItemToShapeWithPath;
+/**
+ * Get a Shape schema referenced by a `RefItem`.
+ */
+function refItemToShapeSchema(context, refItem) {
+    const shapePath = refItemToShapePath(refItem);
+    return (0, get_js_1.default)(context, [...shapePath, 'schema']);
+}
+exports.refItemToShapeSchema = refItemToShapeSchema;
+/**
+ * Sugar, for when you really just need a string...
+ */
+function getRefShapeName(context, refSchema) {
+    const refItem = getRef(context, refSchema);
+    return refItem ? refItemToNamespacedShapeName(refItem) : undefined;
+}
+exports.getRefShapeName = getRefShapeName;
+/**
+ * Sugar, for when you really just need a path...
+ */
+function anyRefToShapeSchemaPath(context, refSchema) {
+    const refItem = getRef(context, refSchema);
+    return refItem ? [...refItemToShapePath(refItem), 'schema'] : undefined;
+}
+exports.anyRefToShapeSchemaPath = anyRefToShapeSchemaPath;
+/**
+ * Helper fn to omit `ref` props from the target schema, and then extend it with the source schema.
+ */
+const omitRefAndExtend = (targetSchema, sourceSchema) => (0, assign_js_1.default)((0, omit_js_1.default)(['@ref', '$ref'], targetSchema), sourceSchema);
+exports.omitRefAndExtend = omitRefAndExtend;
+/**
+ * If there is a $ref, this will insert a property schema from the top level
+ * in place of the reference.
+ */
+function followRef(context, schema, ref) {
+    const refItem = ref ?? getRef(context, schema);
+    if (refItem) {
+        const referencedSchema = refItemToShapeSchema(context, refItem);
+        if (referencedSchema) {
+            return (0, exports.omitRefAndExtend)(schema, referencedSchema);
+        }
+    }
+    return schema;
+}
+exports.followRef = followRef;
+/**
+ * Returns whether a schema has refs.
+ */
+function hasRef(schema) {
+    if ((0, types_1.isRefSchema)(schema) || (0, types_1.isRefSchemaLegacy)(schema)) {
+        return true;
+    }
+    if ((0, types_1.isAllOfSchema)(schema)) {
+        return Boolean(schema.allOf.find(hasRef));
+    }
+    if ((0, types_1.isOneOfSchema)(schema)) {
+        return Boolean(schema.oneOf.find(hasRef));
+    }
+    return false;
+}
+exports.hasRef = hasRef;
+/**
+ * Returns whether a schema has refs which can be followed.
+ */
+function hasResolvableRef(context, schema) {
+    if ((0, types_1.isRefSchema)(schema) || (0, types_1.isRefSchemaLegacy)(schema)) {
+        return Boolean(refItemToShapeSchema(context, getRef(context, schema)));
+    }
+    if ((0, types_1.isAllOfSchema)(schema)) {
+        return Boolean(schema.allOf.find(s => hasResolvableRef(context, s)));
+    }
+    if ((0, types_1.isOneOfSchema)(schema)) {
+        return Boolean(schema.oneOf.find(s => hasResolvableRef(context, s)));
+    }
+    return false;
+}
+exports.hasResolvableRef = hasResolvableRef;
+function isSelfReferential(path, shapeName) {
+    if (path[0] !== 'shapes' || path[1] !== shapeName) {
+        return false;
+    }
+    const index = path.findIndex(part => part === 'allOf' || part === 'extends');
+    return index !== -1 && (0, util_1.isIntegerLike)(path[Number(index) + 1]);
+}
+/**
+ * Given a schema, will resolve any `allOf` or `ref` objects. This function has
+ * to fully resolve all refs, because attempting to extend an object schema with
+ * a ref property would create an invalid schema, potentially containing both
+ * a ref and properties.
+ *
+ * If it can't create a valid schema it returns the input.
+ */
+function dereferenceSchema(context, shapeOrFieldSchema, schemaPath = []) {
+    const resolveAll = (propertySchema, resolvePath) => {
+        try {
+            if ((0, types_1.isAllOfSchema)(propertySchema)) {
+                return merge(propertySchema.allOf, 'allOf', resolvePath);
+            }
+            if ((0, types_1.isExtendsSchema)(propertySchema)) {
+                return merge(propertySchema.extends, 'extends', resolvePath);
+            }
+            if ((0, types_1.isRefSchema)(propertySchema)) {
+                const refItem = getRefWithPath(context, propertySchema);
+                if (!refItem) {
+                    throw new Error('ref could not be parsed');
+                }
+                const shapeWithPath = refItemToShapeWithPath(context, refItem);
+                if (shapeWithPath && !isSelfReferential(resolvePath, shapeWithPath.shape.name)) {
+                    /**
+                     * Pass the new shape's path rather than the parent's path, which
+                     * will have been truncated to something opaque, like `@ref`.
+                     */
+                    return resolveAll((0, exports.omitRefAndExtend)(propertySchema, shapeWithPath.shape.schema), shapeWithPath.path);
+                }
+                return;
+            }
+            return propertySchema;
+        }
+        catch (err) {
+            throw new Error(`error at '${resolvePath.join('.')}', ${err.message}`);
+        }
+    };
+    const merge = (schemas, propName, path) => {
+        const resolvedSchemas = schemas.map((schema, index) => {
+            const resolved = resolveAll(schema, [...path, propName, index]);
+            return resolved && (0, types_1.isObjectSchema)(resolved) ? resolved : undefined;
+        });
+        return (propName === 'allOf' ? merge_1.mergeDeepObjectSchemas : merge_1.mergeObjectSchemas)(...resolvedSchemas);
+    };
+    return resolveAll(shapeOrFieldSchema, schemaPath) ?? shapeOrFieldSchema;
+}
+exports.dereferenceSchema = dereferenceSchema;
+function dereferenceObjectSchema(context, shapeOrFieldSchema, schemaPath = []) {
+    const schema = dereferenceSchema(context, shapeOrFieldSchema, schemaPath);
+    if (!(0, types_1.isObjectSchema)(schema)) {
+        throw new Error(`provided schema at '${schemaPath.join('.')}' could not be resolved to an object schema`);
+    }
+    return schema;
+}
+exports.dereferenceObjectSchema = dereferenceObjectSchema;
+function createGetNamespace(context) {
+    return serviceId => context.services?.[serviceId]?.namespace;
+}
+exports.createGetNamespace = createGetNamespace;
+function parsePropertyRef(refStr) {
+    const [serviceId, coordinate] = splitAtRef(refStr);
+    const coordinateParts = coordinate.split('.');
+    if (coordinateParts.length === 1) {
+        return { serviceId, shapeName: 'Query', propertyName: coordinateParts[0] };
+    }
+    if (coordinateParts.length === 2) {
+        return { serviceId, shapeName: coordinateParts[0], propertyName: coordinateParts[1] };
+    }
+}
+exports.parsePropertyRef = parsePropertyRef;
+function serializePropertyRef({ shapeName, propertyName, serviceId }) {
+    const coordinate = `${shapeName}.${propertyName}`;
+    return serviceId === 'local' ? coordinate : `${serviceId}:${coordinate}`;
+}
+exports.serializePropertyRef = serializePropertyRef;
+function normalizePropertyRef(refStr) {
+    const parsed = parsePropertyRef(refStr);
+    return parsed ? serializePropertyRef(parsed) : refStr;
+}
+exports.normalizePropertyRef = normalizePropertyRef;
+function propertyRefToShapeRef(propertyRef) {
+    const parsed = parsePropertyRef(propertyRef);
+    if (!parsed) {
+        throw new Error(`Invalid property ref "${propertyRef}`);
+    }
+    return formatShapeRef(parsed.serviceId, parsed.shapeName);
+}
+exports.propertyRefToShapeRef = propertyRefToShapeRef;
+function formatShapeRef(serviceId, shapeName) {
+    return `${serviceId}:${shapeName}`;
+}
+exports.formatShapeRef = formatShapeRef;
+function applyNamespace(namespace, name) {
+    return namespace ? `${namespace}_${name}` : name;
+}
+exports.applyNamespace = applyNamespace;
+function propertyRefItemToResolverPath(getNamespace, item) {
+    const path = propertyRefItemToPath(getNamespace, item);
+    return path.concat(path[0] === 'shapes' ? '@resolver' : 'resolver');
+}
+exports.propertyRefItemToResolverPath = propertyRefItemToResolverPath;
+function propertyRefItemToArgsPath(getNamespace, item) {
+    const path = propertyRefItemToPath(getNamespace, item);
+    return path.concat(path[0] === 'shapes' ? '@args' : 'args');
+}
+exports.propertyRefItemToArgsPath = propertyRefItemToArgsPath;
+function propertyRefItemToLocalName(getNamespace, item) {
+    const { shapeName, serviceId, propertyName } = item;
+    if (shapeName === 'Query' || shapeName === 'Mutation') {
+        const namespaced = applyNamespace(getNamespace(serviceId), propertyName);
+        return `${shapeName}.${namespaced}`;
+    }
+    return propertyName;
+}
+exports.propertyRefItemToLocalName = propertyRefItemToLocalName;
+function queryRefToLocalName(context, queryRef) {
+    const parsedQueryRef = parsePropertyRef(queryRef);
+    if (parsedQueryRef) {
+        return parsedQueryRef.serviceId === 'local'
+            ? parsedQueryRef.propertyName
+            : propertyRefItemToLocalName(createGetNamespace(context), parsedQueryRef);
+    }
+}
+exports.queryRefToLocalName = queryRefToLocalName;
+function propertyRefItemToPath(getNamespace, item) {
+    const { shapeName, serviceId, propertyName } = item;
+    const namespace = getNamespace(serviceId);
+    if (shapeName === 'Query' || shapeName === 'Mutation') {
+        return [shapeName === 'Query' ? 'queries' : 'mutations', applyNamespace(namespace, propertyName)];
+    }
+    return ['shapes', applyNamespace(namespace, shapeName), 'schema', 'properties', propertyName];
+}
+exports.propertyRefItemToPath = propertyRefItemToPath;
+function propertyRefToArgs(projectSchema, ref) {
+    const parsed = typeof ref === 'object' ? ref : parsePropertyRef(ref);
+    if (parsed) {
+        const path = propertyRefItemToArgsPath(createGetNamespace(projectSchema), parsed);
+        return (0, get_js_1.default)(projectSchema, path);
+    }
+}
+exports.propertyRefToArgs = propertyRefToArgs;
+/**
+ * Get a ref to the given shape.
+ */
+function normalizeRefExpression(context, refExpression) {
+    return refItemToAtRef(refExpressionToRefItem(context, refExpression));
+}
+exports.normalizeRefExpression = normalizeRefExpression;
+function normalizeLocalRefString(ref) {
+    return ref.startsWith('local:') ? ref.substring(6) : ref;
+}
+exports.normalizeLocalRefString = normalizeLocalRefString;
+function getAllPropertyRefs(projectSchema) {
+    const result = [];
+    const parentIsResolver = (path, resolverName) => {
+        const parent = (0, get_js_1.default)(projectSchema, path.slice(0, -1));
+        return ((0, util_1.isRecord)(parent) &&
+            (Array.isArray(resolverName) ? resolverName.some(name => name === parent.name) : parent.name === resolverName));
+    };
+    const parentIsGenerateState = (path) => {
+        if (path[0] !== 'ai-experimental' || path[path.length - 2] !== 'execution') {
+            return false;
+        }
+        const parent = (0, get_js_1.default)(projectSchema, path.slice(0, -1));
+        return (0, util_1.isRecord)(parent) && parent.type === 'generate';
+    };
+    const pushRef = (refStr) => {
+        const parsed = parsePropertyRef(refStr);
+        if (parsed) {
+            result.push(parsed);
+        }
+    };
+    (0, util_1.visit)(projectSchema, ['to', 'tools'], (value, path) => {
+        const propName = path[path.length - 1];
+        if (propName === 'to' && (0, isString_js_1.default)(value) && parentIsResolver(path, 'delegate')) {
+            pushRef(value);
+        }
+        else if (propName === 'tools' &&
+            Array.isArray(value) &&
+            (parentIsGenerateState(path) || parentIsResolver(path, ['ai:generateText', 'ai:chat']))) {
+            const tools = value;
+            for (const tool of tools) {
+                pushRef((0, ai_tools_1.getToolRef)(tool));
+            }
+        }
+    });
+    return result;
+}
+exports.getAllPropertyRefs = getAllPropertyRefs;
+function getQuery(projectSchema, queryRef) {
+    const propertyRef = parsePropertyRef(queryRef);
+    if (!propertyRef || (propertyRef.shapeName !== 'Query' && propertyRef.shapeName !== 'Mutation')) {
+        throw new Error(`Invalid query ref "${queryRef}"`);
+    }
+    const path = propertyRefItemToPath(createGetNamespace(projectSchema), propertyRef);
+    const query = (0, get_js_1.default)(projectSchema, path);
+    return query ? { name: path[path.length - 1], query, ref: propertyRef } : undefined;
+}
+exports.getQuery = getQuery;
+function ensureQuery(projectSchema, queryRef) {
+    const query = getQuery(projectSchema, queryRef);
+    if (!query) {
+        throw new errors_1.NotFoundError(`Query "${queryRef}" not found`);
+    }
+    return query;
+}
+exports.ensureQuery = ensureQuery;
+function normalizeRefs(projectSchema, property) {
+    let updatedProperty = {
+        ...property
+    };
+    // Array items
+    if (property.items) {
+        return {
+            ...updatedProperty,
+            items: normalizeRefs(projectSchema, property.items)
+        };
+    }
+    // Unions
+    if (property.oneOf) {
+        return {
+            ...updatedProperty,
+            oneOf: property.oneOf.map(item => normalizeRefs(projectSchema, item))
+        };
+    }
+    // Convert $ref to @ref
+    const refItem = getRef(projectSchema, updatedProperty);
+    if (refItem && updatedProperty.$ref) {
+        updatedProperty = {
+            ...(0, omit_js_1.default)(['$ref'], updatedProperty),
+            '@ref': refItemToAtRef(refItem)
+        };
+    }
+    return updatedProperty;
+}
+exports.normalizeRefs = normalizeRefs;