@apollo/client-ai-apps 0.3.3 → 0.5.0

package/dist/vite/application_manifest_plugin.js

@@ -1,274 +0,0 @@
-import { readFileSync, writeFileSync, mkdirSync } from "fs";
-import { glob } from "glob";
-import { gqlPluckFromCodeStringSync } from "@graphql-tools/graphql-tag-pluck";
-import { createHash } from "crypto";
-import { Kind, parse, print } from "graphql";
-import { ApolloClient, ApolloLink, InMemoryCache } from "@apollo/client";
-import { removeDirectivesFromDocument } from "@apollo/client/utilities/internal";
-import { of } from "rxjs";
-import path from "path";
-const root = process.cwd();
-function getRawValue(node) {
-    switch (node.kind) {
-        case Kind.STRING:
-        case Kind.BOOLEAN:
-            return node.value;
-        case Kind.LIST:
-            return node.values.map(getRawValue);
-        case Kind.OBJECT:
-            return node.fields.reduce((acc, field) => {
-                acc[field.name.value] = getRawValue(field.value);
-                return acc;
-            }, {});
-        default:
-            throw new Error(`Error when parsing directive values: unexpected type '${node.kind}'`);
-    }
-}
-function getArgumentValue(argument, expectedType) {
-    const argumentType = argument.value.kind;
-    invariant(argumentType === expectedType, `Expected argument '${argument.name.value}' to be of type '${expectedType}' but found '${argumentType}' instead.`);
-    return getRawValue(argument.value);
-}
-function getDirectiveArgument(argumentName, directive, { required = false } = {}) {
-    const argument = directive.arguments?.find((directiveArgument) => directiveArgument.name.value === argumentName);
-    invariant(argument || !required, `'${argumentName}' argument must be supplied for @tool`);
-    return argument;
-}
-function getTypeName(type) {
-    let t = type;
-    while (t.kind === "NonNullType" || t.kind === "ListType") {
-        t = t.type;
-    }
-    return t.name.value;
-}
-export const ApplicationManifestPlugin = () => {
-    const cache = new Map();
-    let packageJson = null;
-    let config = null;
-    const clientCache = new InMemoryCache();
-    const client = new ApolloClient({
-        cache: clientCache,
-        link: new ApolloLink((operation) => {
-            const body = print(removeClientDirective(sortTopLevelDefinitions(operation.query)));
-            const name = operation.operationName;
-            const definition = operation.query.definitions.find((d) => d.kind === "OperationDefinition");
-            // Use `operation.query` so that the error reflects the end-user defined
-            // document, not our sorted one
-            invariant(definition, `Document does not contain an operation:\n${print(operation.query)}`);
-            const { directives, operation: type } = definition;
-            const variables = definition.variableDefinitions?.reduce((obj, varDef) => ({
-                ...obj,
-                [varDef.variable.name.value]: getTypeName(varDef.type),
-            }), {});
-            const prefetch = directives?.some((d) => d.name.value === "prefetch");
-            const id = createHash("sha256").update(body).digest("hex");
-            // TODO: For now, you can only have 1 operation marked as prefetch. In the future, we'll likely support more than 1, and the "prefetchId" will be defined on the `@prefetch` itself as an argument
-            const prefetchID = prefetch ? "__anonymous" : undefined;
-            const tools = directives
-                ?.filter((d) => d.name.value === "tool")
-                .map((directive) => {
-                const name = getArgumentValue(getDirectiveArgument("name", directive, { required: true }), Kind.STRING);
-                invariant(name.indexOf(" ") === -1, `Tool with name "${name}" contains spaces which is not allowed.`);
-                const description = getArgumentValue(getDirectiveArgument("description", directive, { required: true }), Kind.STRING);
-                const extraInputsNode = getDirectiveArgument("extraInputs", directive);
-                const toolOptions = {
-                    name,
-                    description,
-                };
-                if (extraInputsNode) {
-                    toolOptions.extraInputs = getArgumentValue(extraInputsNode, Kind.LIST);
-                }
-                return toolOptions;
-            });
-            // TODO: Make this object satisfy the `ManifestOperation` type. Currently
-            // it errors because we need more validation on a few of these fields
-            return of({
-                data: { id, name, type, body, variables, prefetch, prefetchID, tools },
-            });
-        }),
-    });
-    const processFile = async (file) => {
-        const code = readFileSync(file, "utf-8");
-        if (!code.includes("gql"))
-            return;
-        const fileHash = createHash("md5").update(code).digest("hex");
-        if (cache.get("file")?.hash === fileHash)
-            return;
-        const sources = gqlPluckFromCodeStringSync(file, code, {
-            modules: [
-                { name: "graphql-tag", identifier: "gql" },
-                { name: "@apollo/client", identifier: "gql" },
-            ],
-        }).map((source) => ({
-            node: parse(source.body),
-            file,
-            location: source.locationOffset,
-        }));
-        const operations = [];
-        for (const source of sources) {
-            const type = source.node.definitions.find((d) => d.kind === "OperationDefinition").operation;
-            let result;
-            if (type === "query") {
-                result = await client.query({
-                    query: source.node,
-                    fetchPolicy: "no-cache",
-                });
-            }
-            else if (type === "mutation") {
-                result = await client.mutate({
-                    mutation: source.node,
-                    fetchPolicy: "no-cache",
-                });
-            }
-            else {
-                throw new Error("Found an unsupported operation type. Only Query and Mutation are supported.");
-            }
-            operations.push(result.data);
-        }
-        cache.set(file, {
-            file: file,
-            hash: fileHash,
-            operations,
-        });
-    };
-    const generateManifest = async () => {
-        const operations = Array.from(cache.values()).flatMap((entry) => entry.operations);
-        invariant(operations.filter((o) => o.prefetch).length <= 1, "Found multiple operations marked as `@prefetch`. You can only mark 1 operation with `@prefetch`.");
-        let resource = "";
-        if (config.command === "serve") {
-            resource =
-                packageJson.entry?.[config.mode] ??
-                    `http${config.server.https ? "s" : ""}://${config.server.host ?? "localhost"}:${config.server.port}`;
-        }
-        else {
-            let entryPoint = packageJson.entry?.[config.mode];
-            if (entryPoint) {
-                resource = entryPoint;
-            }
-            else if (config.mode === "production") {
-                resource = "index.html";
-            }
-            else {
-                throw new Error(`No entry point found for mode "${config.mode}". Entry points other than "development" and "production" must be defined in package.json file.`);
-            }
-        }
-        const manifest = {
-            format: "apollo-ai-app-manifest",
-            version: "1",
-            name: packageJson.name,
-            description: packageJson.description,
-            hash: createHash("sha256").update(Date.now().toString()).digest("hex"),
-            operations: Array.from(cache.values()).flatMap((entry) => entry.operations),
-            resource,
-            csp: {
-                connectDomains: packageJson.csp?.connectDomains ?? [],
-                resourceDomains: packageJson.csp?.resourceDomains ?? [],
-            },
-        };
-        if (packageJson.widgetSettings &&
-            Object.keys(packageJson.widgetSettings).length > 0) {
-            function validateWidgetSetting(key, type) {
-                if (key in widgetSettings) {
-                    invariant(typeof widgetSettings[key] === type, `Expected 'widgetSettings.${key}' to be of type '${type}' but found '${typeof widgetSettings[key]}' instead.`);
-                }
-            }
-            const widgetSettings = packageJson.widgetSettings;
-            validateWidgetSetting("prefersBorder", "boolean");
-            validateWidgetSetting("description", "string");
-            validateWidgetSetting("domain", "string");
-            manifest.widgetSettings = packageJson.widgetSettings;
-        }
-        // Always write to build directory so the MCP server picks it up
-        const dest = path.resolve(root, config.build.outDir, ".application-manifest.json");
-        mkdirSync(path.dirname(dest), { recursive: true });
-        writeFileSync(dest, JSON.stringify(manifest));
-        // Always write to the dev location so that the app can bundle the manifest content
-        writeFileSync(".application-manifest.json", JSON.stringify(manifest));
-    };
-    return {
-        name: "OperationManifest",
-        async configResolved(resolvedConfig) {
-            config = resolvedConfig;
-        },
-        async buildStart() {
-            // Read package.json on start
-            packageJson = JSON.parse(readFileSync("package.json", "utf-8"));
-            // Scan all files on startup
-            const files = await glob("src/**/*.{ts,tsx,js,jsx}");
-            for (const file of files) {
-                const fullPath = path.resolve(root, file);
-                await processFile(fullPath);
-            }
-            // We don't want to do this here on builds cause it just gets overwritten anyways. We'll call it on writeBundle instead.
-            if (config.command === "serve") {
-                await generateManifest();
-            }
-        },
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        configureServer(server) {
-            server.watcher.on("change", async (file) => {
-                if (file.endsWith("package.json")) {
-                    packageJson = JSON.parse(readFileSync("package.json", "utf-8"));
-                    await generateManifest();
-                }
-                else if (file.match(/\.(jsx?|tsx?)$/)) {
-                    await processFile(file);
-                    await generateManifest();
-                }
-            });
-        },
-        async writeBundle() {
-            await generateManifest();
-        },
-    };
-};
-// Sort the definitions in this document so that operations come before fragments,
-// and so that each kind of definition is sorted by name.
-export function sortTopLevelDefinitions(query) {
-    const definitions = [...query.definitions];
-    // We want to avoid unnecessary dependencies, so write out a comparison
-    // function instead of using _.orderBy.
-    definitions.sort((a, b) => {
-        // This is a reverse sort by kind, so that OperationDefinition precedes FragmentDefinition.
-        if (a.kind > b.kind) {
-            return -1;
-        }
-        if (a.kind < b.kind) {
-            return 1;
-        }
-        // Extract the name from each definition. Jump through some hoops because
-        // non-executable definitions don't have to have names (even though any
-        // DocumentNode actually passed here should only have executable
-        // definitions).
-        const aName = a.kind === "OperationDefinition" || a.kind === "FragmentDefinition" ?
-            (a.name?.value ?? "")
-            : "";
-        const bName = b.kind === "OperationDefinition" || b.kind === "FragmentDefinition" ?
-            (b.name?.value ?? "")
-            : "";
-        // Sort by name ascending.
-        if (aName < bName) {
-            return -1;
-        }
-        if (aName > bName) {
-            return 1;
-        }
-        // Assuming that the document is "valid", no operation or fragment name can appear
-        // more than once, so we don't need to differentiate further to have a deterministic
-        // sort.
-        return 0;
-    });
-    return {
-        ...query,
-        definitions,
-    };
-}
-function removeClientDirective(doc) {
-    return removeDirectivesFromDocument([{ name: "prefetch" }, { name: "tool" }], doc);
-}
-function invariant(condition, message) {
-    if (!condition) {
-        throw new Error(message);
-    }
-}
-//# sourceMappingURL=application_manifest_plugin.js.map

package/dist/vite/application_manifest_plugin.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"application_manifest_plugin.js","sourceRoot":"","sources":["../../src/vite/application_manifest_plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC5D,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAYpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AACzE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC1B,OAAO,IAAI,MAAM,MAAM,CAAC;AAQxB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;AAE3B,SAAS,WAAW,CAAC,IAAe;IAClC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,IAAI,CAAC,MAAM,CAAC;QACjB,KAAK,IAAI,CAAC,OAAO;YACf,OAAO,IAAI,CAAC,KAAK,CAAC;QACpB,KAAK,IAAI,CAAC,IAAI;YACZ,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,MAAM;YACd,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAsB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;gBAC5D,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACjD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,EAAE,CAAC,CAAC;QACT;YACE,MAAM,IAAI,KAAK,CACb,yDAAyD,IAAI,CAAC,IAAI,GAAG,CACtE,CAAC;IACN,CAAC;AACH,CAAC;AAsBD,SAAS,gBAAgB,CAAC,QAAsB,EAAE,YAAkB;IAClE,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;IAEzC,SAAS,CACP,YAAY,KAAK,YAAY,EAC7B,sBAAsB,QAAQ,CAAC,IAAI,CAAC,KAAK,oBAAoB,YAAY,gBAAgB,YAAY,YAAY,CAClH,CAAC;IAEF,OAAO,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AACrC,CAAC;AAkBD,SAAS,oBAAoB,CAC3B,YAAoB,EACpB,SAAwB,EACxB,EAAE,QAAQ,GAAG,KAAK,KAA6B,EAAE;IAEjD,MAAM,QAAQ,GAAG,SAAS,CAAC,SAAS,EAAE,IAAI,CACxC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,KAAK,YAAY,CACrE,CAAC;IAEF,SAAS,CACP,QAAQ,IAAI,CAAC,QAAQ,EACrB,IAAI,YAAY,uCAAuC,CACxD,CAAC;IAEF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,IAAc;IACjC,IAAI,CAAC,GAAG,IAAI,CAAC;IACb,OAAO,CAAC,CAAC,IAAI,KAAK,aAAa,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QACzD,CAAC,GAAI,CAAoC,CAAC,IAAI,CAAC;IACjD,CAAC;IACD,OAAQ,CAAmB,CAAC,IAAI,CAAC,KAAK,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,EAAE;IAC5C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;IACxB,IAAI,WAAW,GAAQ,IAAI,CAAC;IAC5B,IAAI,MAAM,GAAQ,IAAI,CAAC;IAEvB,MAAM,WAAW,GAAG,IAAI,aAAa,EAAE,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC;QAC9B,KAAK,EAAE,WAAW;QAClB,IAAI,EAAE,IAAI,UAAU,CAAC,CAAC,SAAS,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,KAAK,CAChB,qBAAqB,CAAC,uBAAuB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAChE,CAAC;YACF,MAAM,IAAI,GAAG,SAAS,CAAC,aAAa,CAAC;YACrC,MAAM,UAAU,GAAG,SAAS,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CACjD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,qBAAqB,CACxC,CAAC;YAEF,wEAAwE;YACxE,+BAA+B;YAC/B,SAAS,CACP,UAAU,EACV,4CAA4C,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CACrE,CAAC;YAEF,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC;YAEnD,MAAM,SAAS,GAAG,UAAU,CAAC,mBAAmB,EAAE,MAAM,CACtD,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBAChB,GAAG,GAAG;gBACN,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;aACvD,CAAC,EACF,EAAE,CACH,CAAC;YAEF,MAAM,QAAQ,GAAG,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC;YACtE,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC3D,kMAAkM;YAClM,MAAM,UAAU,GAAG,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC;YAExD,MAAM,KAAK,GAAG,UAAU;gBACtB,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,MAAM,CAAC;iBACvC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;gBACjB,MAAM,IAAI,GAAG,gBAAgB,CAC3B,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,EAC3D,IAAI,CAAC,MAAM,CACZ,CAAC;gBAEF,SAAS,CACP,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EACxB,mBAAmB,IAAI,yCAAyC,CACjE,CAAC;gBAEF,MAAM,WAAW,GAAG,gBAAgB,CAClC,oBAAoB,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,EAClE,IAAI,CAAC,MAAM,CACZ,CAAC;gBAEF,MAAM,eAAe,GAAG,oBAAoB,CAC1C,aAAa,EACb,SAAS,CACV,CAAC;gBAEF,MAAM,WAAW,GAAiB;oBAChC,IAAI;oBACJ,WAAW;iBACZ,CAAC;gBAEF,IAAI,eAAe,EAAE,CAAC;oBACpB,WAAW,CAAC,WAAW,GAAG,gBAAgB,CACxC,eAAe,EACf,IAAI,CAAC,IAAI,CACc,CAAC;gBAC5B,CAAC;gBAED,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC,CAAC;YAEL,yEAAyE;YACzE,qEAAqE;YACrE,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE;aACvE,CAAC,CAAC;QACL,CAAC,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,KAAK,EAAE,IAAY,EAAE,EAAE;QACzC,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAEzC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO;QAElC,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9D,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO;QACjD,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,EAAE,IAAI,EAAE;YACrD,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE;gBAC1C,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,KAAK,EAAE;aAC9C;SACF,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAClB,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,IAAI;YACJ,QAAQ,EAAE,MAAM,CAAC,cAAc;SAChC,CAAC,CAAC,CAAC;QAEJ,MAAM,UAAU,GAAG,EAAE,CAAC;QACtB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,IAAI,GACR,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAC1B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,qBAAqB,CAE1C,CAAC,SAAS,CAAC;YAEZ,IAAI,MAAM,CAAC;YACX,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;gBACrB,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;oBAC1B,KAAK,EAAE,MAAM,CAAC,IAAI;oBAClB,WAAW,EAAE,UAAU;iBACxB,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC;oBAC3B,QAAQ,EAAE,MAAM,CAAC,IAAI;oBACrB,WAAW,EAAE,UAAU;iBACxB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;YACJ,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;YACd,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,UAAU;SACX,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,gBAAgB,GAAG,KAAK,IAAI,EAAE;QAClC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CACnD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAC5B,CAAC;QAEF,SAAS,CACP,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM,IAAI,CAAC,EAChD,kGAAkG,CACnG,CAAC;QAEF,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,IAAI,MAAM,CAAC,OAAO,KAAK,OAAO,EAAE,CAAC;YAC/B,QAAQ;gBACN,WAAW,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;oBAChC,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,IAAI,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACzG,CAAC;aAAM,CAAC;YACN,IAAI,UAAU,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClD,IAAI,UAAU,EAAE,CAAC;gBACf,QAAQ,GAAG,UAAU,CAAC;YACxB,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxC,QAAQ,GAAG,YAAY,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,kCAAkC,MAAM,CAAC,IAAI,iGAAiG,CAC/I,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAwB;YACpC,MAAM,EAAE,wBAAwB;YAChC,OAAO,EAAE,GAAG;YACZ,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,WAAW,EAAE,WAAW,CAAC,WAAW;YACpC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YACtE,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAC5C,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAC5B;YACD,QAAQ;YACR,GAAG,EAAE;gBACH,cAAc,EAAE,WAAW,CAAC,GAAG,EAAE,cAAc,IAAI,EAAE;gBACrD,eAAe,EAAE,WAAW,CAAC,GAAG,EAAE,eAAe,IAAI,EAAE;aACxD;SACF,CAAC;QAEF,IACE,WAAW,CAAC,cAAc;YAC1B,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,EAClD,CAAC;YACD,SAAS,qBAAqB,CAC5B,GAAiC,EACjC,IAA0B;gBAE1B,IAAI,GAAG,IAAI,cAAc,EAAE,CAAC;oBAC1B,SAAS,CACP,OAAO,cAAc,CAAC,GAAG,CAAC,KAAK,IAAI,EACnC,4BAA4B,GAAG,oBAAoB,IAAI,gBAAgB,OAAO,cAAc,CAAC,GAAG,CAAC,YAAY,CAC9G,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,cAAc,GAClB,WAAW,CAAC,cAAwC,CAAC;YAEvD,qBAAqB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;YAClD,qBAAqB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YAC/C,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAE1C,QAAQ,CAAC,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC;QACvD,CAAC;QAED,gEAAgE;QAChE,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CACvB,IAAI,EACJ,MAAM,CAAC,KAAK,CAAC,MAAM,EACnB,4BAA4B,CAC7B,CAAC;QACF,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE9C,mFAAmF;QACnF,aAAa,CAAC,4BAA4B,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,mBAAmB;QAEzB,KAAK,CAAC,cAAc,CAAC,cAAmB;YACtC,MAAM,GAAG,cAAc,CAAC;QAC1B,CAAC;QAED,KAAK,CAAC,UAAU;YACd,6BAA6B;YAC7B,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,4BAA4B;YAC5B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAErD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;YAED,wHAAwH;YACxH,IAAI,MAAM,CAAC,OAAO,KAAK,OAAO,EAAE,CAAC;gBAC/B,MAAM,gBAAgB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,8DAA8D;QAC9D,eAAe,CAAC,MAAW;YACzB,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE;gBACjD,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;oBAClC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC;oBAChE,MAAM,gBAAgB,EAAE,CAAC;gBAC3B,CAAC;qBAAM,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBACxC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;oBACxB,MAAM,gBAAgB,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,WAAW;YACf,MAAM,gBAAgB,EAAE,CAAC;QAC3B,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,kFAAkF;AAClF,yDAAyD;AACzD,MAAM,UAAU,uBAAuB,CAAC,KAAmB;IACzD,MAAM,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC;IAC3C,uEAAuE;IACvE,uCAAuC;IACvC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxB,2FAA2F;QAC3F,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC,CAAC;QACZ,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,yEAAyE;QACzE,uEAAuE;QACvE,gEAAgE;QAChE,gBAAgB;QAChB,MAAM,KAAK,GACT,CAAC,CAAC,IAAI,KAAK,qBAAqB,IAAI,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,KAAK,GACT,CAAC,CAAC,IAAI,KAAK,qBAAqB,IAAI,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;QAEP,0BAA0B;QAC1B,IAAI,KAAK,GAAG,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,CAAC,CAAC;QACZ,CAAC;QACD,IAAI,KAAK,GAAG,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,kFAAkF;QAClF,oFAAoF;QACpF,QAAQ;QACR,OAAO,CAAC,CAAC;IACX,CAAC,CAAC,CAAC;IACH,OAAO;QACL,GAAG,KAAK;QACR,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAiB;IAC9C,OAAO,4BAA4B,CACjC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EACxC,GAAG,CACH,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,SAAc,EAAE,OAAe;IAChD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC","sourcesContent":["import { readFileSync, writeFileSync, mkdirSync } from \"fs\";\nimport { glob } from \"glob\";\nimport { gqlPluckFromCodeStringSync } from \"@graphql-tools/graphql-tag-pluck\";\nimport { createHash } from \"crypto\";\nimport type {\n  ArgumentNode,\n  ListTypeNode,\n  NamedTypeNode,\n  NonNullTypeNode,\n  TypeNode,\n  ValueNode,\n  DocumentNode,\n  OperationDefinitionNode,\n  DirectiveNode,\n} from \"graphql\";\nimport { Kind, parse, print } from \"graphql\";\nimport { ApolloClient, ApolloLink, InMemoryCache } from \"@apollo/client\";\nimport { removeDirectivesFromDocument } from \"@apollo/client/utilities/internal\";\nimport { of } from \"rxjs\";\nimport path from \"path\";\nimport type {\n  ApplicationManifest,\n  ManifestExtraInput,\n  ManifestTool,\n  ManifestWidgetSettings,\n} from \"../types/application-manifest.js\";\n\nconst root = process.cwd();\n\nfunction getRawValue(node: ValueNode): unknown {\n  switch (node.kind) {\n    case Kind.STRING:\n    case Kind.BOOLEAN:\n      return node.value;\n    case Kind.LIST:\n      return node.values.map(getRawValue);\n    case Kind.OBJECT:\n      return node.fields.reduce<Record<string, any>>((acc, field) => {\n        acc[field.name.value] = getRawValue(field.value);\n        return acc;\n      }, {});\n    default:\n      throw new Error(\n        `Error when parsing directive values: unexpected type '${node.kind}'`\n      );\n  }\n}\n\nfunction getArgumentValue(\n  argument: ArgumentNode,\n  expectedType: Kind.STRING\n): string;\n\nfunction getArgumentValue(\n  argument: ArgumentNode,\n  expectedType: Kind.BOOLEAN\n): boolean;\n\nfunction getArgumentValue(\n  argument: ArgumentNode,\n  expectedType: Kind.LIST\n): unknown[];\n\nfunction getArgumentValue(\n  argument: ArgumentNode,\n  expectedType: Kind.OBJECT\n): Record<string, unknown>;\n\nfunction getArgumentValue(argument: ArgumentNode, expectedType: Kind) {\n  const argumentType = argument.value.kind;\n\n  invariant(\n    argumentType === expectedType,\n    `Expected argument '${argument.name.value}' to be of type '${expectedType}' but found '${argumentType}' instead.`\n  );\n\n  return getRawValue(argument.value);\n}\n\ninterface GetArgumentNodeOptions {\n  required?: boolean;\n}\n\nfunction getDirectiveArgument(\n  argumentName: string,\n  directive: DirectiveNode,\n  opts: GetArgumentNodeOptions & { required: true }\n): ArgumentNode;\n\nfunction getDirectiveArgument(\n  argumentName: string,\n  directive: DirectiveNode,\n  opts?: GetArgumentNodeOptions\n): ArgumentNode | undefined;\n\nfunction getDirectiveArgument(\n  argumentName: string,\n  directive: DirectiveNode,\n  { required = false }: { required?: boolean } = {}\n) {\n  const argument = directive.arguments?.find(\n    (directiveArgument) => directiveArgument.name.value === argumentName\n  );\n\n  invariant(\n    argument || !required,\n    `'${argumentName}' argument must be supplied for @tool`\n  );\n\n  return argument;\n}\n\nfunction getTypeName(type: TypeNode): string {\n  let t = type;\n  while (t.kind === \"NonNullType\" || t.kind === \"ListType\") {\n    t = (t as NonNullTypeNode | ListTypeNode).type;\n  }\n  return (t as NamedTypeNode).name.value;\n}\n\nexport const ApplicationManifestPlugin = () => {\n  const cache = new Map();\n  let packageJson: any = null;\n  let config: any = null;\n\n  const clientCache = new InMemoryCache();\n  const client = new ApolloClient({\n    cache: clientCache,\n    link: new ApolloLink((operation) => {\n      const body = print(\n        removeClientDirective(sortTopLevelDefinitions(operation.query))\n      );\n      const name = operation.operationName;\n      const definition = operation.query.definitions.find(\n        (d) => d.kind === \"OperationDefinition\"\n      );\n\n      // Use `operation.query` so that the error reflects the end-user defined\n      // document, not our sorted one\n      invariant(\n        definition,\n        `Document does not contain an operation:\\n${print(operation.query)}`\n      );\n\n      const { directives, operation: type } = definition;\n\n      const variables = definition.variableDefinitions?.reduce(\n        (obj, varDef) => ({\n          ...obj,\n          [varDef.variable.name.value]: getTypeName(varDef.type),\n        }),\n        {}\n      );\n\n      const prefetch = directives?.some((d) => d.name.value === \"prefetch\");\n      const id = createHash(\"sha256\").update(body).digest(\"hex\");\n      // TODO: For now, you can only have 1 operation marked as prefetch. In the future, we'll likely support more than 1, and the \"prefetchId\" will be defined on the `@prefetch` itself as an argument\n      const prefetchID = prefetch ? \"__anonymous\" : undefined;\n\n      const tools = directives\n        ?.filter((d) => d.name.value === \"tool\")\n        .map((directive) => {\n          const name = getArgumentValue(\n            getDirectiveArgument(\"name\", directive, { required: true }),\n            Kind.STRING\n          );\n\n          invariant(\n            name.indexOf(\" \") === -1,\n            `Tool with name \"${name}\" contains spaces which is not allowed.`\n          );\n\n          const description = getArgumentValue(\n            getDirectiveArgument(\"description\", directive, { required: true }),\n            Kind.STRING\n          );\n\n          const extraInputsNode = getDirectiveArgument(\n            \"extraInputs\",\n            directive\n          );\n\n          const toolOptions: ManifestTool = {\n            name,\n            description,\n          };\n\n          if (extraInputsNode) {\n            toolOptions.extraInputs = getArgumentValue(\n              extraInputsNode,\n              Kind.LIST\n            ) as ManifestExtraInput[];\n          }\n\n          return toolOptions;\n        });\n\n      // TODO: Make this object satisfy the `ManifestOperation` type. Currently\n      // it errors because we need more validation on a few of these fields\n      return of({\n        data: { id, name, type, body, variables, prefetch, prefetchID, tools },\n      });\n    }),\n  });\n\n  const processFile = async (file: string) => {\n    const code = readFileSync(file, \"utf-8\");\n\n    if (!code.includes(\"gql\")) return;\n\n    const fileHash = createHash(\"md5\").update(code).digest(\"hex\");\n    if (cache.get(\"file\")?.hash === fileHash) return;\n    const sources = gqlPluckFromCodeStringSync(file, code, {\n      modules: [\n        { name: \"graphql-tag\", identifier: \"gql\" },\n        { name: \"@apollo/client\", identifier: \"gql\" },\n      ],\n    }).map((source) => ({\n      node: parse(source.body),\n      file,\n      location: source.locationOffset,\n    }));\n\n    const operations = [];\n    for (const source of sources) {\n      const type = (\n        source.node.definitions.find(\n          (d) => d.kind === \"OperationDefinition\"\n        ) as OperationDefinitionNode\n      ).operation;\n\n      let result;\n      if (type === \"query\") {\n        result = await client.query({\n          query: source.node,\n          fetchPolicy: \"no-cache\",\n        });\n      } else if (type === \"mutation\") {\n        result = await client.mutate({\n          mutation: source.node,\n          fetchPolicy: \"no-cache\",\n        });\n      } else {\n        throw new Error(\n          \"Found an unsupported operation type. Only Query and Mutation are supported.\"\n        );\n      }\n      operations.push(result.data);\n    }\n\n    cache.set(file, {\n      file: file,\n      hash: fileHash,\n      operations,\n    });\n  };\n\n  const generateManifest = async () => {\n    const operations = Array.from(cache.values()).flatMap(\n      (entry) => entry.operations\n    );\n\n    invariant(\n      operations.filter((o) => o.prefetch).length <= 1,\n      \"Found multiple operations marked as `@prefetch`. You can only mark 1 operation with `@prefetch`.\"\n    );\n\n    let resource = \"\";\n    if (config.command === \"serve\") {\n      resource =\n        packageJson.entry?.[config.mode] ??\n        `http${config.server.https ? \"s\" : \"\"}://${config.server.host ?? \"localhost\"}:${config.server.port}`;\n    } else {\n      let entryPoint = packageJson.entry?.[config.mode];\n      if (entryPoint) {\n        resource = entryPoint;\n      } else if (config.mode === \"production\") {\n        resource = \"index.html\";\n      } else {\n        throw new Error(\n          `No entry point found for mode \"${config.mode}\". Entry points other than \"development\" and \"production\" must be defined in package.json file.`\n        );\n      }\n    }\n\n    const manifest: ApplicationManifest = {\n      format: \"apollo-ai-app-manifest\",\n      version: \"1\",\n      name: packageJson.name,\n      description: packageJson.description,\n      hash: createHash(\"sha256\").update(Date.now().toString()).digest(\"hex\"),\n      operations: Array.from(cache.values()).flatMap(\n        (entry) => entry.operations\n      ),\n      resource,\n      csp: {\n        connectDomains: packageJson.csp?.connectDomains ?? [],\n        resourceDomains: packageJson.csp?.resourceDomains ?? [],\n      },\n    };\n\n    if (\n      packageJson.widgetSettings &&\n      Object.keys(packageJson.widgetSettings).length > 0\n    ) {\n      function validateWidgetSetting(\n        key: keyof ManifestWidgetSettings,\n        type: \"string\" | \"boolean\"\n      ) {\n        if (key in widgetSettings) {\n          invariant(\n            typeof widgetSettings[key] === type,\n            `Expected 'widgetSettings.${key}' to be of type '${type}' but found '${typeof widgetSettings[key]}' instead.`\n          );\n        }\n      }\n\n      const widgetSettings =\n        packageJson.widgetSettings as ManifestWidgetSettings;\n\n      validateWidgetSetting(\"prefersBorder\", \"boolean\");\n      validateWidgetSetting(\"description\", \"string\");\n      validateWidgetSetting(\"domain\", \"string\");\n\n      manifest.widgetSettings = packageJson.widgetSettings;\n    }\n\n    // Always write to build directory so the MCP server picks it up\n    const dest = path.resolve(\n      root,\n      config.build.outDir,\n      \".application-manifest.json\"\n    );\n    mkdirSync(path.dirname(dest), { recursive: true });\n    writeFileSync(dest, JSON.stringify(manifest));\n\n    // Always write to the dev location so that the app can bundle the manifest content\n    writeFileSync(\".application-manifest.json\", JSON.stringify(manifest));\n  };\n\n  return {\n    name: \"OperationManifest\",\n\n    async configResolved(resolvedConfig: any) {\n      config = resolvedConfig;\n    },\n\n    async buildStart() {\n      // Read package.json on start\n      packageJson = JSON.parse(readFileSync(\"package.json\", \"utf-8\"));\n\n      // Scan all files on startup\n      const files = await glob(\"src/**/*.{ts,tsx,js,jsx}\");\n\n      for (const file of files) {\n        const fullPath = path.resolve(root, file);\n        await processFile(fullPath);\n      }\n\n      // We don't want to do this here on builds cause it just gets overwritten anyways. We'll call it on writeBundle instead.\n      if (config.command === \"serve\") {\n        await generateManifest();\n      }\n    },\n\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    configureServer(server: any) {\n      server.watcher.on(\"change\", async (file: string) => {\n        if (file.endsWith(\"package.json\")) {\n          packageJson = JSON.parse(readFileSync(\"package.json\", \"utf-8\"));\n          await generateManifest();\n        } else if (file.match(/\\.(jsx?|tsx?)$/)) {\n          await processFile(file);\n          await generateManifest();\n        }\n      });\n    },\n\n    async writeBundle() {\n      await generateManifest();\n    },\n  };\n};\n\n// Sort the definitions in this document so that operations come before fragments,\n// and so that each kind of definition is sorted by name.\nexport function sortTopLevelDefinitions(query: DocumentNode): DocumentNode {\n  const definitions = [...query.definitions];\n  // We want to avoid unnecessary dependencies, so write out a comparison\n  // function instead of using _.orderBy.\n  definitions.sort((a, b) => {\n    // This is a reverse sort by kind, so that OperationDefinition precedes FragmentDefinition.\n    if (a.kind > b.kind) {\n      return -1;\n    }\n    if (a.kind < b.kind) {\n      return 1;\n    }\n\n    // Extract the name from each definition. Jump through some hoops because\n    // non-executable definitions don't have to have names (even though any\n    // DocumentNode actually passed here should only have executable\n    // definitions).\n    const aName =\n      a.kind === \"OperationDefinition\" || a.kind === \"FragmentDefinition\" ?\n        (a.name?.value ?? \"\")\n      : \"\";\n    const bName =\n      b.kind === \"OperationDefinition\" || b.kind === \"FragmentDefinition\" ?\n        (b.name?.value ?? \"\")\n      : \"\";\n\n    // Sort by name ascending.\n    if (aName < bName) {\n      return -1;\n    }\n    if (aName > bName) {\n      return 1;\n    }\n\n    // Assuming that the document is \"valid\", no operation or fragment name can appear\n    // more than once, so we don't need to differentiate further to have a deterministic\n    // sort.\n    return 0;\n  });\n  return {\n    ...query,\n    definitions,\n  };\n}\n\nfunction removeClientDirective(doc: DocumentNode) {\n  return removeDirectivesFromDocument(\n    [{ name: \"prefetch\" }, { name: \"tool\" }],\n    doc\n  )!;\n}\n\nfunction invariant(condition: any, message: string): asserts condition {\n  if (!condition) {\n    throw new Error(message);\n  }\n}\n"]}