apollo-conn-gen 0.0.2

package/dist/oas/io/writer.js

@@ -0,0 +1,313 @@
+import _ from 'lodash';
+import { CircularRef } from '../nodes/circularRef.js';
+import { Composed } from '../nodes/comp.js';
+import { Obj } from '../nodes/obj.js';
+import { Prop } from '../nodes/props/prop.js';
+import { PropArray } from '../nodes/props/propArray.js';
+import { PropScalar } from '../nodes/props/propScalar.js';
+import { Union } from '../nodes/union.js';
+import { Naming } from '../utils/naming.js';
+import { T } from '../utils/typeUtils.js';
+export class Writer {
+    generator;
+    static findNonPropParent(type) {
+        let parent = type;
+        while (parent instanceof Prop) {
+            parent = parent.parent;
+        }
+        return parent;
+    }
+    static progressiveSplits(input) {
+        const parts = input.split('>');
+        const results = [];
+        for (let i = 1; i <= parts.length; i++) {
+            results.push(parts.slice(0, i).join('>'));
+        }
+        return results;
+    }
+    buffer;
+    constructor(generator) {
+        this.generator = generator;
+        this.buffer = [];
+    }
+    write(input) {
+        this.buffer.push(input);
+        return this;
+    }
+    append(input) {
+        this.buffer.push(input);
+        return this;
+    }
+    flush() {
+        return this.buffer.join('');
+    }
+    writeSchema(writer, pending, selection) {
+        const context = this.generator.context;
+        const generatedSet = context.generatedSet;
+        // generatedSet.clear();
+        this.writeDirectives(writer);
+        this.writeJSONScalar(writer);
+        pending.forEach((type) => {
+            if (!generatedSet.has(type.id)) {
+                type.generate(context, this, selection);
+                generatedSet.add(type.id);
+            }
+        });
+        // TODO: Pending
+        // const counter = new RefCounter(this.context);
+        // counter.addAll(this.collected);
+        // const refs = counter.getCount();
+        // this.printRefs(refs);
+        // for (const type of this.context.types.ts.values()) {
+        //   if (counter.getCount().has(type.name)) {
+        //     await type.generate(this.context, writer);
+        //     generatedSet.add(type.name);
+        //   }
+        // }
+        this.writeQuery(context, writer, this.generator.paths, selection);
+        writer.flush();
+    }
+    generate(selection) {
+        const pending = new Map();
+        selection = this.collectExpandedPaths(selection);
+        for (const path of selection) {
+            let collection = Array.from(this.generator.paths.values());
+            let current;
+            let last;
+            let i = 0;
+            const parts = path.split('>');
+            do {
+                const part = parts[i].replace(/#\/c\/s/g, '#/components/schemas');
+                if (part === '*') {
+                    // remove the current path from the selection array
+                    selection = selection.filter((s) => s !== path);
+                    if (current instanceof Composed) {
+                        current.consolidate(selection);
+                    }
+                    // add all the props from the current node and exit loop
+                    current?.props.forEach((child) => {
+                        if (T.isLeaf(child)) {
+                            selection.push(child.path());
+                        }
+                    });
+                    break;
+                }
+                current = collection.find((t) => t.id === part);
+                if (!current) {
+                    throw new Error('Could not find type: ' + part + ' from ' + path + ', last: ' + last?.pathToRoot());
+                }
+                // make sure we expand it before we move on to the next part
+                this.generator.expand(current);
+                last = current;
+                collection = Array.from(current.children.values()) || Array.from(current.props.values()) || [];
+                i++;
+            } while (i < parts.length);
+            if (current) {
+                // TODO: this seems redundant, we've already walked the parent AND can be also
+                // contained in the context stack
+                const parentType = Writer.findNonPropParent(current);
+                if (!pending.has(parentType.id)) {
+                    pending.set(parentType.id, parentType);
+                }
+                parentType
+                    .ancestors()
+                    .filter((t) => !pending.has(t.id) && this.isContainer(t))
+                    .forEach((dep) => {
+                    // TODO: potential merge needed?
+                    pending.set(dep.id, dep);
+                });
+            }
+        }
+        if (!_.isEmpty(pending)) {
+            // first pass is to consolidate all Composed & Union nodes
+            const composed = Array.from(pending.values())
+                .filter((t) => t instanceof Composed || t instanceof Union)
+                .map((t) => t);
+            for (const comp of composed) {
+                comp.consolidate(selection).forEach((id) => pending.delete(id));
+            }
+            this.writeSchema(this, pending, selection);
+        }
+    }
+    collectPaths(path, collection) {
+        const stack = [];
+        let current;
+        let last;
+        let i = 0;
+        const parts = path.split('>');
+        do {
+            const part = parts[i].replace(/#\/c\/s/g, '#/components/schemas');
+            current = collection.find((t) => t.id === part);
+            if (!current) {
+                throw new Error('Could not find type: ' + part + ' from ' + path + ', last: ' + last?.pathToRoot());
+            }
+            // make sure we expand it before we move on to the next part
+            this.generator.expand(current);
+            last = current;
+            collection = Array.from(current.children.values()) || Array.from(current.props.values()) || [];
+            stack.push(current);
+            i++;
+        } while (i < parts.length);
+        return stack;
+    }
+    writeJSONScalar(writer) {
+        writer.write('\nscalar JSON\n\n');
+    }
+    writeQuery(context, writer, collected, selection) {
+        writer.write('type Query {\n');
+        const selectionSet = new Set(selection.map((s) => s.split('>')[0]));
+        const paths = Array.from(collected.values()).filter((path) => selectionSet.has(path.id));
+        for (const path of paths) {
+            path.generate(context, writer, []);
+            this.writeConnector(context, writer, path, selection);
+            context.generatedSet.add(path.id);
+        }
+        writer.write('}\n\n');
+    }
+    writeConnector(context, writer, type, selection) {
+        const indent = 0;
+        const get = type; // assume type is GetOp
+        let spacing = ' '.repeat(indent + 4);
+        writer.append(spacing).append('@connect(\n');
+        const request = this.buildRequestMethodAndArgs(get);
+        spacing = ' '.repeat(indent + 6);
+        writer
+            .append(spacing)
+            .append('source: "api"\n')
+            .append(spacing)
+            .append('http: ' + request + '\n')
+            .append(spacing)
+            .append('selection: """\n');
+        if (get.resultType) {
+            this.writeSelection(context, writer, get.resultType, selection);
+        }
+        writer.append(spacing).append('"""\n');
+        spacing = ' '.repeat(indent + 4);
+        writer.append(spacing).append(')\n');
+    }
+    buildRequestMethodAndArgs(get) {
+        let builder = '';
+        builder += '"' + get.operation.path.replace(/\{([a-zA-Z0-9]+)\}/g, '{$args.$1}');
+        if (get.params.length > 0) {
+            const params = get.params.filter((p) => {
+                return p.required && p.parameter.in && p.parameter.in.toLowerCase() === 'query';
+            });
+            if (params.length > 0) {
+                builder += '?' + params.map((p) => `${p.name}={$args.${Naming.genParamName(p.name)}}`).join('&');
+            }
+            const headers = get.operation.getParameters().filter((p) => p.in && p.in.toLowerCase() === 'header');
+            builder += '"\n';
+            if (headers.length > 0) {
+                let spacing = ' '.repeat(6);
+                builder += spacing + 'headers: [\n';
+                spacing = ' '.repeat(8);
+                for (const p of headers) {
+                    let value = null;
+                    if (p.example != null) {
+                        value = p.example.toString();
+                    }
+                    if (p.examples && Object.keys(p.examples).length > 0) {
+                        value = Object.keys(p.examples).join(',');
+                    }
+                    if (value == null) {
+                        value = '<placeholder>';
+                    }
+                    builder += spacing + `{ name: "${p.name}", value: "${value}" }\n`;
+                }
+                spacing = ' '.repeat(6);
+                builder += spacing + ']';
+            }
+        }
+        else {
+            builder += '"';
+        }
+        return `{ GET: ${builder} }`;
+    }
+    writeSelection(context, writer, type, selection) {
+        context.indent = 6;
+        type.select(context, writer, selection);
+    }
+    writeDirectives(writer) {
+        const api = this.generator.parser;
+        const host = this.getServerUrl(api.getDefinition().servers?.[0]);
+        writer
+            .append('extend schema\n')
+            .append('  @link(url: "https://specs.apollo.dev/federation/v2.10", import: ["@key"])\n')
+            .append('  @link(\n')
+            .append('    url: "https://specs.apollo.dev/connect/v0.1"\n')
+            .append('    import: ["@connect", "@source"]\n')
+            .append('  )\n')
+            .append('  @source(name: "api", http: { baseURL: "')
+            .append(host)
+            .append('" })\n\n');
+    }
+    getServerUrl(server) {
+        if (!server) {
+            return 'http://localhost:4010';
+        }
+        let url = server.url;
+        if (server.variables) {
+            for (const key in server.variables) {
+                url = url.replace('{' + key + '}', server.variables[key].default);
+            }
+        }
+        return url;
+    }
+    traverseTree(current, selection, pending) {
+        // we might be in a node far from the root, so we need to traverse upwards
+        // as well and add the props that we can find on the way
+        const source = current;
+        source
+            .ancestors()
+            .filter((t) => t instanceof Prop)
+            .map((p) => Writer.findNonPropParent(p))
+            .forEach((parent) => pending.set(parent.id, parent));
+        T.traverse(source, (child) => {
+            if (T.isLeaf(child)) {
+                // this is a weird take but if the child is an array of scalars
+                // then we want to avoid adding it twice
+                if (T.isLeaf(child.parent)) {
+                    return;
+                }
+                selection.push(child.path());
+                const parentType = Writer.findNonPropParent(child);
+                if (!pending.has(parentType.id)) {
+                    pending.set(parentType.id, parentType);
+                }
+            }
+            else {
+                this.generator.expand(child);
+                if (child instanceof Composed) {
+                    child.consolidate(selection);
+                }
+            }
+        });
+    }
+    isContainer(type) {
+        return type instanceof Obj || type instanceof Union || type instanceof Composed || type instanceof CircularRef;
+    }
+    collectExpandedPaths(selection) {
+        const newSelection = new Set();
+        const expands = selection.filter((p) => p.endsWith('>**'));
+        const filtered = expands.map((p) => p.replace('>**', ''));
+        const paths = Array.from(this.generator.paths.values());
+        const nodes = filtered.map((p) => this.collectPaths(p, paths));
+        /*.map(stack => _.last(stack)!.id)
+          .filter(id => selection.includes(id))
+          .forEach(id => newSelection.add(id));*/
+        nodes.forEach((stack) => {
+            const root = _.last(stack);
+            T.traverse(root, (child) => {
+                if (T.isPropScalar(child) || (child instanceof PropArray && child.items instanceof PropScalar)) {
+                    newSelection.add(child.path());
+                }
+                else {
+                    this.generator.expand(child);
+                }
+            });
+        });
+        // finally remove the expanded paths from the selection
+        return [...newSelection, ...selection.filter((p) => !expands.includes(p))];
+    }
+}
+//# sourceMappingURL=writer.js.map

package/dist/oas/io/writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"writer.js","sourceRoot":"","sources":["../../../src/oas/io/writer.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,QAAQ,CAAC;AAKvB,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAE5C,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAC;AAEtC,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,8BAA8B,CAAC;AAE1D,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,uBAAuB,CAAC;AAE1C,MAAM,OAAO,MAAM;IAmBE;IAlBZ,MAAM,CAAC,iBAAiB,CAAC,IAAW;QACzC,IAAI,MAAM,GAAG,IAAI,CAAC;QAClB,OAAO,MAAM,YAAY,IAAI,EAAE,CAAC;YAC9B,MAAM,GAAG,MAAM,CAAC,MAAO,CAAC;QAC1B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,MAAM,CAAC,iBAAiB,CAAC,KAAa;QAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IACM,MAAM,CAAW;IAExB,YAAmB,SAAiB;QAAjB,cAAS,GAAT,SAAS,CAAQ;QAClC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,KAAa;QACxB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,MAAM,CAAC,KAAa;QACzB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,KAAK;QACV,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC;IAEM,WAAW,CAAC,MAAc,EAAE,OAA2B,EAAE,SAAmB;QACjF,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAQ,CAAC;QACxC,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QAC1C,wBAAwB;QAExB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC7B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAE7B,OAAO,CAAC,OAAO,CAAC,CAAC,IAAW,EAAE,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC/B,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;gBACxC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,gBAAgB;QAChB,gDAAgD;QAChD,kCAAkC;QAElC,mCAAmC;QACnC,wBAAwB;QAExB,uDAAuD;QACvD,6CAA6C;QAC7C,iDAAiD;QACjD,mCAAmC;QACnC,MAAM;QACN,IAAI;QAEJ,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;IAEM,QAAQ,CAAC,SAAmB;QACjC,MAAM,OAAO,GAAuB,IAAI,GAAG,EAAE,CAAC;QAE9C,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAEjD,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;YAC3D,IAAI,OAA0B,CAAC;YAC/B,IAAI,IAAuB,CAAC;YAE5B,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,GAAG,CAAC;gBACF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;gBAClE,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;oBACjB,mDAAmD;oBACnD,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;oBAEhD,IAAI,OAAO,YAAY,QAAQ,EAAE,CAAC;wBAChC,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;oBACjC,CAAC;oBAED,wDAAwD;oBACxD,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;wBAC/B,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;4BACpB,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;wBAC/B,CAAC;oBACH,CAAC,CAAC,CAAC;oBACH,MAAM;gBACR,CAAC;gBAED,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC;gBAChD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,IAAI,GAAG,QAAQ,GAAG,IAAI,GAAG,UAAU,GAAG,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;gBACtG,CAAC;gBAED,4DAA4D;gBAC5D,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC/B,IAAI,GAAG,OAAO,CAAC;gBAEf,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,OAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;gBAEjG,CAAC,EAAE,CAAC;YACN,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE;YAE3B,IAAI,OAAO,EAAE,CAAC;gBACZ,8EAA8E;gBAC9E,iCAAiC;gBACjC,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,OAAgB,CAAC,CAAC;gBAE9D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;oBAChC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;gBACzC,CAAC;gBAED,UAAU;qBACP,SAAS,EAAE;qBACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;qBACxD,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;oBACf,gCAAgC;oBAChC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC3B,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAED,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACxB,0DAA0D;YAC1D,MAAM,QAAQ,GAA4B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;iBACnE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,QAAQ,IAAI,CAAC,YAAY,KAAK,CAAC;iBAC1D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAa,CAAC,CAAC;YAE7B,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;YAClE,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEM,YAAY,CAAC,IAAY,EAAE,UAAmB;QACnD,MAAM,KAAK,GAAY,EAAE,CAAC;QAC1B,IAAI,OAA0B,CAAC;QAC/B,IAAI,IAAuB,CAAC;QAE5B,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC9B,GAAG,CAAC;YACF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;YAElE,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,IAAI,GAAG,QAAQ,GAAG,IAAI,GAAG,UAAU,GAAG,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YACtG,CAAC;YAED,4DAA4D;YAC5D,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,GAAG,OAAO,CAAC;YAEf,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,OAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YAEjG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpB,CAAC,EAAE,CAAC;QACN,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE;QAE3B,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,eAAe,CAAC,MAAc;QACpC,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;IACpC,CAAC;IAEO,UAAU,CAAC,OAAmB,EAAE,MAAc,EAAE,SAA6B,EAAE,SAAmB;QACxG,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAE/B,MAAM,YAAY,GAAG,IAAI,GAAG,CAAS,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5E,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAEzF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACtD,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAEO,cAAc,CAAC,OAAmB,EAAE,MAAc,EAAE,IAAW,EAAE,SAAmB;QAC1F,MAAM,MAAM,GAAG,CAAC,CAAC;QACjB,MAAM,GAAG,GAAG,IAAsB,CAAC,CAAC,uBAAuB;QAC3D,IAAI,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC;QACpD,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjC,MAAM;aACH,MAAM,CAAC,OAAO,CAAC;aACf,MAAM,CAAC,iBAAiB,CAAC;aACzB,MAAM,CAAC,OAAO,CAAC;aACf,MAAM,CAAC,QAAQ,GAAG,OAAO,GAAG,IAAI,CAAC;aACjC,MAAM,CAAC,OAAO,CAAC;aACf,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAE9B,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACnB,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACvC,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAEO,yBAAyB,CAAC,GAAQ;QACxC,IAAI,OAAO,GAAG,EAAE,CAAC;QAEjB,OAAO,IAAI,GAAG,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QAEjF,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAQ,EAAE,EAAE;gBAC5C,OAAO,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC;YAClF,CAAC,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,OAAO,IAAI,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,WAAW,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC1G,CAAC;YACD,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,QAAQ,CAAC,CAAC;YAErG,OAAO,IAAI,KAAK,CAAC;YAEjB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,IAAI,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAC5B,OAAO,IAAI,OAAO,GAAG,cAAc,CAAC;gBACpC,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAExB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;oBACxB,IAAI,KAAK,GAAkB,IAAI,CAAC;oBAEhC,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;wBACtB,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;oBAC/B,CAAC;oBAED,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACrD,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC5C,CAAC;oBAED,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;wBAClB,KAAK,GAAG,eAAe,CAAC;oBAC1B,CAAC;oBAED,OAAO,IAAI,OAAO,GAAG,YAAY,CAAC,CAAC,IAAI,cAAc,KAAK,OAAO,CAAC;gBACpE,CAAC;gBAED,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACxB,OAAO,IAAI,OAAO,GAAG,GAAG,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,GAAG,CAAC;QACjB,CAAC;QAED,OAAO,UAAU,OAAO,IAAI,CAAC;IAC/B,CAAC;IAEO,cAAc,CAAC,OAAmB,EAAE,MAAc,EAAE,IAAW,EAAE,SAAmB;QAC1F,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACnB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC;IAEO,eAAe,CAAC,MAAc;QACpC,MAAM,GAAG,GAAQ,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM;aACH,MAAM,CAAC,iBAAiB,CAAC;aACzB,MAAM,CAAC,+EAA+E,CAAC;aACvF,MAAM,CAAC,YAAY,CAAC;aACpB,MAAM,CAAC,oDAAoD,CAAC;aAC5D,MAAM,CAAC,uCAAuC,CAAC;aAC/C,MAAM,CAAC,OAAO,CAAC;aACf,MAAM,CAAC,2CAA2C,CAAC;aACnD,MAAM,CAAC,IAAI,CAAC;aACZ,MAAM,CAAC,UAAU,CAAC,CAAC;IACxB,CAAC;IAEO,YAAY,CAAC,MAAgC;QACnD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,uBAAuB,CAAC;QACjC,CAAC;QACD,IAAI,GAAG,GAAW,MAAM,CAAC,GAAG,CAAC;QAC7B,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACnC,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,YAAY,CAAC,OAAc,EAAE,SAAmB,EAAE,OAA2B;QACnF,0EAA0E;QAC1E,wDAAwD;QACxD,MAAM,MAAM,GAAG,OAAe,CAAC;QAE/B,MAAM;aACH,SAAS,EAAE;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,IAAI,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;aACvC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAEvD,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3B,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpB,+DAA+D;gBAC/D,wCAAwC;gBACxC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,MAAO,CAAC,EAAE,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBAE7B,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACnD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;oBAChC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAE7B,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;oBAC9B,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,IAAW;QAC7B,OAAO,IAAI,YAAY,GAAG,IAAI,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,QAAQ,IAAI,IAAI,YAAY,WAAW,CAAC;IACjH,CAAC;IAEO,oBAAoB,CAAC,SAAmB;QAC9C,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QACvC,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;QAE1D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACxD,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;QAC/D;;iDAEyC;QAEzC,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACtB,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAE,CAAC;YAC5B,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE;gBACzB,IAAI,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,YAAY,SAAS,IAAI,KAAK,CAAC,KAAK,YAAY,UAAU,CAAC,EAAE,CAAC;oBAC/F,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACjC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,uDAAuD;QACvD,OAAO,CAAC,GAAG,YAAY,EAAE,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,CAAC;CACF","sourcesContent":["import _ from 'lodash';\nimport Oas from 'oas';\nimport { ServerObject } from 'oas/types';\nimport { OasContext } from '../oasContext.js';\nimport { OasGen } from '../oasGen.js';\nimport { CircularRef } from '../nodes/circularRef.js';\nimport { Composed } from '../nodes/comp.js';\nimport { Get } from '../nodes/get.js';\nimport { Obj } from '../nodes/obj.js';\nimport { Param } from '../nodes/param/param.js';\nimport { Prop } from '../nodes/props/prop.js';\nimport { PropArray } from '../nodes/props/propArray.js';\nimport { PropScalar } from '../nodes/props/propScalar.js';\nimport { IType, Type } from '../nodes/type.js';\nimport { Union } from '../nodes/union.js';\nimport { Naming } from '../utils/naming.js';\nimport { T } from '../utils/typeUtils.js';\n\nexport class Writer {\n  public static findNonPropParent(type: IType) {\n    let parent = type;\n    while (parent instanceof Prop) {\n      parent = parent.parent!;\n    }\n    return parent;\n  }\n\n  public static progressiveSplits(input: string): string[] {\n    const parts = input.split('>');\n    const results: string[] = [];\n    for (let i = 1; i <= parts.length; i++) {\n      results.push(parts.slice(0, i).join('>'));\n    }\n    return results;\n  }\n  public buffer: string[];\n\n  constructor(public generator: OasGen) {\n    this.buffer = [];\n  }\n\n  public write(input: string): Writer {\n    this.buffer.push(input);\n    return this;\n  }\n\n  public append(input: string): Writer {\n    this.buffer.push(input);\n    return this;\n  }\n\n  public flush(): string {\n    return this.buffer.join('');\n  }\n\n  public writeSchema(writer: Writer, pending: Map<string, IType>, selection: string[]): void {\n    const context = this.generator.context!;\n    const generatedSet = context.generatedSet;\n    // generatedSet.clear();\n\n    this.writeDirectives(writer);\n    this.writeJSONScalar(writer);\n\n    pending.forEach((type: IType) => {\n      if (!generatedSet.has(type.id)) {\n        type.generate(context, this, selection);\n        generatedSet.add(type.id);\n      }\n    });\n\n    // TODO: Pending\n    // const counter = new RefCounter(this.context);\n    // counter.addAll(this.collected);\n\n    // const refs = counter.getCount();\n    // this.printRefs(refs);\n\n    // for (const type of this.context.types.ts.values()) {\n    //   if (counter.getCount().has(type.name)) {\n    //     await type.generate(this.context, writer);\n    //     generatedSet.add(type.name);\n    //   }\n    // }\n\n    this.writeQuery(context, writer, this.generator.paths, selection);\n    writer.flush();\n  }\n\n  public generate(selection: string[]) {\n    const pending: Map<string, IType> = new Map();\n\n    selection = this.collectExpandedPaths(selection);\n\n    for (const path of selection) {\n      let collection = Array.from(this.generator.paths.values());\n      let current: IType | undefined;\n      let last: IType | undefined;\n\n      let i = 0;\n      const parts = path.split('>');\n      do {\n        const part = parts[i].replace(/#\\/c\\/s/g, '#/components/schemas');\n        if (part === '*') {\n          // remove the current path from the selection array\n          selection = selection.filter((s) => s !== path);\n\n          if (current instanceof Composed) {\n            current.consolidate(selection);\n          }\n\n          // add all the props from the current node and exit loop\n          current?.props.forEach((child) => {\n            if (T.isLeaf(child)) {\n              selection.push(child.path());\n            }\n          });\n          break;\n        }\n\n        current = collection.find((t) => t.id === part);\n        if (!current) {\n          throw new Error('Could not find type: ' + part + ' from ' + path + ', last: ' + last?.pathToRoot());\n        }\n\n        // make sure we expand it before we move on to the next part\n        this.generator.expand(current);\n        last = current;\n\n        collection = Array.from(current!.children.values()) || Array.from(current!.props.values()) || [];\n\n        i++;\n      } while (i < parts.length);\n\n      if (current) {\n        // TODO: this seems redundant, we've already walked the parent AND can be also\n        // contained in the context stack\n        const parentType = Writer.findNonPropParent(current as IType);\n\n        if (!pending.has(parentType.id)) {\n          pending.set(parentType.id, parentType);\n        }\n\n        parentType\n          .ancestors()\n          .filter((t) => !pending.has(t.id) && this.isContainer(t))\n          .forEach((dep) => {\n            // TODO: potential merge needed?\n            pending.set(dep.id, dep);\n          });\n      }\n    }\n\n    if (!_.isEmpty(pending)) {\n      // first pass is to consolidate all Composed & Union nodes\n      const composed: Array<Composed | Union> = Array.from(pending.values())\n        .filter((t) => t instanceof Composed || t instanceof Union)\n        .map((t) => t as Composed);\n\n      for (const comp of composed) {\n        comp.consolidate(selection).forEach((id) => pending.delete(id));\n      }\n\n      this.writeSchema(this, pending, selection);\n    }\n  }\n\n  public collectPaths(path: string, collection: IType[]): IType[] {\n    const stack: IType[] = [];\n    let current: IType | undefined;\n    let last: IType | undefined;\n\n    let i = 0;\n    const parts = path.split('>');\n    do {\n      const part = parts[i].replace(/#\\/c\\/s/g, '#/components/schemas');\n\n      current = collection.find((t) => t.id === part);\n      if (!current) {\n        throw new Error('Could not find type: ' + part + ' from ' + path + ', last: ' + last?.pathToRoot());\n      }\n\n      // make sure we expand it before we move on to the next part\n      this.generator.expand(current);\n      last = current;\n\n      collection = Array.from(current!.children.values()) || Array.from(current!.props.values()) || [];\n\n      stack.push(current);\n      i++;\n    } while (i < parts.length);\n\n    return stack;\n  }\n\n  private writeJSONScalar(writer: Writer): void {\n    writer.write('\\nscalar JSON\\n\\n');\n  }\n\n  private writeQuery(context: OasContext, writer: Writer, collected: Map<string, IType>, selection: string[]): void {\n    writer.write('type Query {\\n');\n\n    const selectionSet = new Set<string>(selection.map((s) => s.split('>')[0]));\n\n    const paths = Array.from(collected.values()).filter((path) => selectionSet.has(path.id));\n\n    for (const path of paths) {\n      path.generate(context, writer, []);\n      this.writeConnector(context, writer, path, selection);\n      context.generatedSet.add(path.id);\n    }\n\n    writer.write('}\\n\\n');\n  }\n\n  private writeConnector(context: OasContext, writer: Writer, type: IType, selection: string[]): void {\n    const indent = 0;\n    const get = type as unknown as Get; // assume type is GetOp\n    let spacing = ' '.repeat(indent + 4);\n    writer.append(spacing).append('@connect(\\n');\n\n    const request = this.buildRequestMethodAndArgs(get);\n    spacing = ' '.repeat(indent + 6);\n    writer\n      .append(spacing)\n      .append('source: \"api\"\\n')\n      .append(spacing)\n      .append('http: ' + request + '\\n')\n      .append(spacing)\n      .append('selection: \"\"\"\\n');\n\n    if (get.resultType) {\n      this.writeSelection(context, writer, get.resultType, selection);\n    }\n\n    writer.append(spacing).append('\"\"\"\\n');\n    spacing = ' '.repeat(indent + 4);\n    writer.append(spacing).append(')\\n');\n  }\n\n  private buildRequestMethodAndArgs(get: Get): string {\n    let builder = '';\n\n    builder += '\"' + get.operation.path.replace(/\\{([a-zA-Z0-9]+)\\}/g, '{$args.$1}');\n\n    if (get.params.length > 0) {\n      const params = get.params.filter((p: Param) => {\n        return p.required && p.parameter.in && p.parameter.in.toLowerCase() === 'query';\n      });\n\n      if (params.length > 0) {\n        builder += '?' + params.map((p: Param) => `${p.name}={$args.${Naming.genParamName(p.name)}}`).join('&');\n      }\n      const headers = get.operation.getParameters().filter((p) => p.in && p.in.toLowerCase() === 'header');\n\n      builder += '\"\\n';\n\n      if (headers.length > 0) {\n        let spacing = ' '.repeat(6);\n        builder += spacing + 'headers: [\\n';\n        spacing = ' '.repeat(8);\n\n        for (const p of headers) {\n          let value: string | null = null;\n\n          if (p.example != null) {\n            value = p.example.toString();\n          }\n\n          if (p.examples && Object.keys(p.examples).length > 0) {\n            value = Object.keys(p.examples).join(',');\n          }\n\n          if (value == null) {\n            value = '<placeholder>';\n          }\n\n          builder += spacing + `{ name: \"${p.name}\", value: \"${value}\" }\\n`;\n        }\n\n        spacing = ' '.repeat(6);\n        builder += spacing + ']';\n      }\n    } else {\n      builder += '\"';\n    }\n\n    return `{ GET: ${builder} }`;\n  }\n\n  private writeSelection(context: OasContext, writer: Writer, type: IType, selection: string[]): void {\n    context.indent = 6;\n    type.select(context, writer, selection);\n  }\n\n  private writeDirectives(writer: Writer): void {\n    const api: Oas = this.generator.parser;\n    const host = this.getServerUrl(api.getDefinition().servers?.[0]);\n    writer\n      .append('extend schema\\n')\n      .append('  @link(url: \"https://specs.apollo.dev/federation/v2.10\", import: [\"@key\"])\\n')\n      .append('  @link(\\n')\n      .append('    url: \"https://specs.apollo.dev/connect/v0.1\"\\n')\n      .append('    import: [\"@connect\", \"@source\"]\\n')\n      .append('  )\\n')\n      .append('  @source(name: \"api\", http: { baseURL: \"')\n      .append(host)\n      .append('\" })\\n\\n');\n  }\n\n  private getServerUrl(server: ServerObject | undefined): string {\n    if (!server) {\n      return 'http://localhost:4010';\n    }\n    let url: string = server.url;\n    if (server.variables) {\n      for (const key in server.variables) {\n        url = url.replace('{' + key + '}', server.variables[key].default);\n      }\n    }\n    return url;\n  }\n\n  private traverseTree(current: IType, selection: string[], pending: Map<string, IType>) {\n    // we might be in a node far from the root, so we need to traverse upwards\n    // as well and add the props that we can find on the way\n    const source = current as Type;\n\n    source\n      .ancestors()\n      .filter((t) => t instanceof Prop)\n      .map((p) => Writer.findNonPropParent(p))\n      .forEach((parent) => pending.set(parent.id, parent));\n\n    T.traverse(source, (child) => {\n      if (T.isLeaf(child)) {\n        // this is a weird take but if the child is an array of scalars\n        // then we want to avoid adding it twice\n        if (T.isLeaf(child.parent!)) {\n          return;\n        }\n        selection.push(child.path());\n\n        const parentType = Writer.findNonPropParent(child);\n        if (!pending.has(parentType.id)) {\n          pending.set(parentType.id, parentType);\n        }\n      } else {\n        this.generator.expand(child);\n\n        if (child instanceof Composed) {\n          child.consolidate(selection);\n        }\n      }\n    });\n  }\n\n  private isContainer(type: IType) {\n    return type instanceof Obj || type instanceof Union || type instanceof Composed || type instanceof CircularRef;\n  }\n\n  private collectExpandedPaths(selection: string[]) {\n    const newSelection = new Set<string>();\n    const expands = selection.filter((p) => p.endsWith('>**'));\n    const filtered = expands.map((p) => p.replace('>**', ''));\n\n    const paths = Array.from(this.generator.paths.values());\n    const nodes = filtered.map((p) => this.collectPaths(p, paths));\n    /*.map(stack => _.last(stack)!.id)\n      .filter(id => selection.includes(id))\n      .forEach(id => newSelection.add(id));*/\n\n    nodes.forEach((stack) => {\n      const root = _.last(stack)!;\n      T.traverse(root, (child) => {\n        if (T.isPropScalar(child) || (child instanceof PropArray && child.items instanceof PropScalar)) {\n          newSelection.add(child.path());\n        } else {\n          this.generator.expand(child);\n        }\n      });\n    });\n\n    // finally remove the expanded paths from the selection\n    return [...newSelection, ...selection.filter((p) => !expands.includes(p))];\n  }\n}\n"]}

package/dist/oas/log/trace.d.ts

@@ -0,0 +1,3 @@
+import { OasContext } from '../oasContext.js';
+export declare function trace(ctx: OasContext | null, loc: string, log: string): void;
+export declare function warn(ctx: OasContext | null, loc: string, log: string): void;

package/dist/oas/log/trace.js

@@ -0,0 +1,7 @@
+export function trace(ctx, loc, log) {
+    console.log(' '.repeat(ctx?.size() ?? 0), loc, log);
+}
+export function warn(ctx, loc, log) {
+    console.error(' '.repeat(ctx?.size() ?? 0), loc, log);
+}
+//# sourceMappingURL=trace.js.map

package/dist/oas/log/trace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace.js","sourceRoot":"","sources":["../../../src/oas/log/trace.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,KAAK,CAAC,GAAsB,EAAE,GAAW,EAAE,GAAW;IACpE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,GAAsB,EAAE,GAAW,EAAE,GAAW;IACnE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import { OasContext } from '../oasContext.js';\n\nexport function trace(ctx: OasContext | null, loc: string, log: string) {\n  console.log(' '.repeat(ctx?.size() ?? 0), loc, log);\n}\n\nexport function warn(ctx: OasContext | null, loc: string, log: string) {\n  console.error(' '.repeat(ctx?.size() ?? 0), loc, log);\n}\n"]}

package/dist/oas/nodes/arr.d.ts

@@ -0,0 +1,15 @@
+import { OpenAPIV3 } from 'openapi-types';
+import ArraySchemaObject = OpenAPIV3.ArraySchemaObject;
+import { OasContext } from '../oasContext.js';
+import { Writer } from '../io/writer.js';
+import { IType, Type } from './type.js';
+export declare class Arr extends Type {
+    items: ArraySchemaObject;
+    itemsType?: IType;
+    constructor(parent: IType | undefined, name: string, items: ArraySchemaObject);
+    get id(): string;
+    visit(context: OasContext): void;
+    forPrompt(_context: OasContext): string;
+    generate(context: OasContext, writer: Writer, selection: string[]): void;
+    select(context: OasContext, writer: Writer, selection: string[]): void;
+}

package/dist/oas/nodes/arr.js

@@ -0,0 +1,47 @@
+import { trace } from '../log/trace.js';
+import { Naming } from '../utils/naming.js';
+import { Type } from './type.js';
+export class Arr extends Type {
+    items;
+    itemsType;
+    constructor(parent, name, items) {
+        super(parent, name);
+        this.items = items;
+    }
+    get id() {
+        return 'array:' + (this.itemsType ? this.itemsType.name : 'unknown-yet');
+    }
+    visit(context) {
+        if (this.visited) {
+            return;
+        }
+        context.enter(this);
+        trace(context, '-> [array:visit]', 'in');
+        this.itemsType?.visit(context);
+        this.visited = true;
+        trace(context, '-> [array:visit]', 'out');
+        context.leave(this);
+    }
+    forPrompt(_context) {
+        return `${Naming.getRefName(this.name)} (Array)`;
+    }
+    generate(context, writer, selection) {
+        context.enter(this);
+        trace(context, '-> [array::generate]', `-> in: ${this.name}`);
+        writer.append('[');
+        if (this.itemsType) {
+            this.itemsType.generate(context, writer, selection);
+        }
+        writer.append(']');
+        trace(context, '<- [array::generate]', `-> out: ${this.name}`);
+        context.leave(this);
+    }
+    select(context, writer, selection) {
+        trace(context, '-> [array::select]', `-> in: ${this.name}`);
+        if (this.itemsType) {
+            this.itemsType.select(context, writer, selection);
+        }
+        trace(context, '<- [array::select]', `-> out: ${this.name}`);
+    }
+}
+//# sourceMappingURL=arr.js.map

package/dist/oas/nodes/arr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"arr.js","sourceRoot":"","sources":["../../../src/oas/nodes/arr.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAGxC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,OAAO,EAAS,IAAI,EAAE,MAAM,WAAW,CAAC;AAExC,MAAM,OAAO,GAAI,SAAQ,IAAI;IAKlB;IAJF,SAAS,CAAS;IACzB,YACE,MAAyB,EACzB,IAAY,EACL,KAAwB;QAE/B,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAFb,UAAK,GAAL,KAAK,CAAmB;IAGjC,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,QAAQ,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;IAC3E,CAAC;IAEM,KAAK,CAAC,OAAmB;QAC9B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpB,KAAK,CAAC,OAAO,EAAE,kBAAkB,EAAE,IAAI,CAAC,CAAC;QAEzC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,KAAK,CAAC,OAAO,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC1C,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAEM,SAAS,CAAC,QAAoB;QACnC,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;IACnD,CAAC;IAEM,QAAQ,CAAC,OAAmB,EAAE,MAAc,EAAE,SAAmB;QACtE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpB,KAAK,CAAC,OAAO,EAAE,sBAAsB,EAAE,UAAU,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAE9D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAEnB,KAAK,CAAC,OAAO,EAAE,sBAAsB,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAEM,MAAM,CAAC,OAAmB,EAAE,MAAc,EAAE,SAAmB;QACpE,KAAK,CAAC,OAAO,EAAE,oBAAoB,EAAE,UAAU,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAE5D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACpD,CAAC;QAED,KAAK,CAAC,OAAO,EAAE,oBAAoB,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC;CACF","sourcesContent":["import { OpenAPIV3 } from 'openapi-types';\nimport ArraySchemaObject = OpenAPIV3.ArraySchemaObject;\n\nimport { trace } from '../log/trace.js';\nimport { OasContext } from '../oasContext.js';\nimport { Writer } from '../io/writer.js';\nimport { Naming } from '../utils/naming.js';\nimport { Factory } from './factory.js';\nimport { IType, Type } from './type.js';\n\nexport class Arr extends Type {\n  public itemsType?: IType;\n  constructor(\n    parent: IType | undefined,\n    name: string,\n    public items: ArraySchemaObject,\n  ) {\n    super(parent, name);\n  }\n\n  get id(): string {\n    return 'array:' + (this.itemsType ? this.itemsType.name : 'unknown-yet');\n  }\n\n  public visit(context: OasContext): void {\n    if (this.visited) {\n      return;\n    }\n\n    context.enter(this);\n    trace(context, '-> [array:visit]', 'in');\n\n    this.itemsType?.visit(context);\n    this.visited = true;\n\n    trace(context, '-> [array:visit]', 'out');\n    context.leave(this);\n  }\n\n  public forPrompt(_context: OasContext): string {\n    return `${Naming.getRefName(this.name)} (Array)`;\n  }\n\n  public generate(context: OasContext, writer: Writer, selection: string[]): void {\n    context.enter(this);\n    trace(context, '-> [array::generate]', `-> in: ${this.name}`);\n\n    writer.append('[');\n    if (this.itemsType) {\n      this.itemsType.generate(context, writer, selection);\n    }\n    writer.append(']');\n\n    trace(context, '<- [array::generate]', `-> out: ${this.name}`);\n    context.leave(this);\n  }\n\n  public select(context: OasContext, writer: Writer, selection: string[]) {\n    trace(context, '-> [array::select]', `-> in: ${this.name}`);\n\n    if (this.itemsType) {\n      this.itemsType.select(context, writer, selection);\n    }\n\n    trace(context, '<- [array::select]', `-> out: ${this.name}`);\n  }\n}\n"]}

package/dist/oas/nodes/circularRef.d.ts

@@ -0,0 +1,13 @@
+import { OasContext } from '../oasContext.js';
+import { Writer } from '../io/writer.js';
+import { IType, Type } from './type.js';
+export declare class CircularRef extends Type {
+    ref: IType;
+    constructor(parent: IType, ref: IType);
+    get id(): string;
+    forPrompt(_: OasContext): string;
+    add(child: IType): void;
+    visit(context: OasContext): void;
+    generate(context: OasContext, writer: Writer): void;
+    select(context: OasContext, writer: Writer): void;
+}

package/dist/oas/nodes/circularRef.js

@@ -0,0 +1,39 @@
+import { trace } from '../log/trace.js';
+import { Naming } from '../utils/naming.js';
+import { Type } from './type.js';
+export class CircularRef extends Type {
+    ref;
+    constructor(parent, ref) {
+        super(parent, ref.name);
+        this.ref = ref;
+        // this.children = parent.children;
+    }
+    get id() {
+        return `circular-ref:#${this.ref.id}`;
+        // return this.child.id;
+    }
+    forPrompt(_) {
+        return `${Naming.getRefName(this.ref.name)} (Circular Ref in: ${this.parent?.id})`;
+    }
+    add(child) {
+        throw new Error('Should not be adding a child to a circular ref');
+    }
+    visit(context) {
+        trace(context, '-> [circular-ref:visit]', `-> in: ${this.name}`);
+        // Do nothing, this type is always visited.
+        trace(context, '<- [circular-ref:visit]', `-> out: ${this.name}`);
+    }
+    generate(context, writer) {
+        trace(context, '-> [circular-ref:generate]', `-> in: ${this.name}`);
+        // Do nothing, we can't really generate a circular reference.
+        trace(context, '<- [circular-ref:generate]', `-> out: ${this.name}`);
+    }
+    select(context, writer) {
+        trace(context, '-> [circular-ref:select]', `-> in: ${this.name}`);
+        writer
+            .append(' '.repeat(context.indent + context.stack.length))
+            .append(`# Circular reference to '${this.name}' detected! Please re-visit the schema and remove the reference.\n`);
+        trace(context, '<- [circular-ref:select]', `-> out: ${this.name}`);
+    }
+}
+//# sourceMappingURL=circularRef.js.map

package/dist/oas/nodes/circularRef.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"circularRef.js","sourceRoot":"","sources":["../../../src/oas/nodes/circularRef.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAGxC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAS,IAAI,EAAE,MAAM,WAAW,CAAC;AAExC,MAAM,OAAO,WAAY,SAAQ,IAAI;IAG1B;IAFT,YACE,MAAa,EACN,GAAU;QAEjB,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAFjB,QAAG,GAAH,GAAG,CAAO;QAGjB,mCAAmC;IACrC,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,iBAAiB,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACtC,wBAAwB;IAC1B,CAAC;IAEM,SAAS,CAAC,CAAa;QAC5B,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,sBAAsB,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,CAAC;IACrF,CAAC;IAEM,GAAG,CAAC,KAAY;QACrB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IAEM,KAAK,CAAC,OAAmB;QAC9B,KAAK,CAAC,OAAO,EAAE,yBAAyB,EAAE,UAAU,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACjE,2CAA2C;QAC3C,KAAK,CAAC,OAAO,EAAE,yBAAyB,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC;IAEM,QAAQ,CAAC,OAAmB,EAAE,MAAc;QACjD,KAAK,CAAC,OAAO,EAAE,4BAA4B,EAAE,UAAU,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACpE,6DAA6D;QAC7D,KAAK,CAAC,OAAO,EAAE,4BAA4B,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACvE,CAAC;IAEM,MAAM,CAAC,OAAmB,EAAE,MAAc;QAC/C,KAAK,CAAC,OAAO,EAAE,0BAA0B,EAAE,UAAU,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAElE,MAAM;aACH,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;aACzD,MAAM,CACL,4BAA4B,IAAI,CAAC,IAAI,oEAAoE,CAC1G,CAAC;QAEJ,KAAK,CAAC,OAAO,EAAE,0BAA0B,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACrE,CAAC;CACF","sourcesContent":["import { trace } from '../log/trace.js';\nimport { OasContext } from '../oasContext.js';\nimport { Writer } from '../io/writer.js';\nimport { Naming } from '../utils/naming.js';\nimport { IType, Type } from './type.js';\n\nexport class CircularRef extends Type {\n  constructor(\n    parent: IType,\n    public ref: IType,\n  ) {\n    super(parent, ref.name);\n    // this.children = parent.children;\n  }\n\n  get id(): string {\n    return `circular-ref:#${this.ref.id}`;\n    // return this.child.id;\n  }\n\n  public forPrompt(_: OasContext): string {\n    return `${Naming.getRefName(this.ref.name)} (Circular Ref in: ${this.parent?.id})`;\n  }\n\n  public add(child: IType): void {\n    throw new Error('Should not be adding a child to a circular ref');\n  }\n\n  public visit(context: OasContext): void {\n    trace(context, '-> [circular-ref:visit]', `-> in: ${this.name}`);\n    // Do nothing, this type is always visited.\n    trace(context, '<- [circular-ref:visit]', `-> out: ${this.name}`);\n  }\n\n  public generate(context: OasContext, writer: Writer): void {\n    trace(context, '-> [circular-ref:generate]', `-> in: ${this.name}`);\n    // Do nothing, we can't really generate a circular reference.\n    trace(context, '<- [circular-ref:generate]', `-> out: ${this.name}`);\n  }\n\n  public select(context: OasContext, writer: Writer): void {\n    trace(context, '-> [circular-ref:select]', `-> in: ${this.name}`);\n\n    writer\n      .append(' '.repeat(context.indent + context.stack.length))\n      .append(\n        `# Circular reference to '${this.name}' detected! Please re-visit the schema and remove the reference.\\n`,\n      );\n\n    trace(context, '<- [circular-ref:select]', `-> out: ${this.name}`);\n  }\n}\n"]}

package/dist/oas/nodes/comp.d.ts

@@ -0,0 +1,18 @@
+import { SchemaObject } from 'oas/types';
+import { OasContext } from '../oasContext.js';
+import { Writer } from '../io/writer.js';
+import { IType, Type } from './type.js';
+export declare class Composed extends Type {
+    name: string;
+    schema: SchemaObject;
+    consolidated: boolean;
+    get id(): string;
+    constructor(parent: IType | undefined, name: string, schema: SchemaObject, consolidated?: boolean);
+    forPrompt(_context: OasContext): string;
+    visit(context: OasContext): void;
+    generate(context: OasContext, writer: Writer, selection: string[]): void;
+    select(context: OasContext, writer: Writer, selection: string[]): void;
+    consolidate(selection: string[]): Set<string>;
+    private visitAllOfNode;
+    private visitOneOfNode;
+}