@cogitator-ai/studio 0.1.0

package/dist/server/server.js

@@ -0,0 +1,437 @@
+import { randomBytes, randomUUID, timingSafeEqual } from 'node:crypto';
+import { existsSync, readFileSync, statSync } from 'node:fs';
+import { createServer } from 'node:http';
+import { basename, extname, join, normalize, resolve, sep } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { calculateCost, initializeModels } from '@cogitator-ai/models';
+import { HostManager } from './host-manager.js';
+import { StudioStore } from './store.js';
+const LOOPBACK = new Set(['127.0.0.1', '::1', 'localhost']);
+const MAX_BODY_BYTES = 1024 * 1024;
+const MIME = {
+    '.html': 'text/html; charset=utf-8',
+    '.js': 'text/javascript; charset=utf-8',
+    '.css': 'text/css; charset=utf-8',
+    '.svg': 'image/svg+xml',
+    '.png': 'image/png',
+    '.ico': 'image/x-icon',
+    '.json': 'application/json',
+    '.woff2': 'font/woff2',
+    '.woff': 'font/woff',
+};
+class HttpError extends Error {
+    status;
+    constructor(status, message) {
+        super(message);
+        this.status = status;
+    }
+}
+function uiDirectory() {
+    return fileURLToPath(new URL('../ui/', import.meta.url));
+}
+function send(res, status, body) {
+    const json = JSON.stringify(body);
+    res.writeHead(status, {
+        'content-type': 'application/json; charset=utf-8',
+        'cache-control': 'no-store',
+        'content-length': Buffer.byteLength(json),
+    });
+    res.end(json);
+}
+async function readJson(req) {
+    if (!(req.headers['content-type'] ?? '').includes('application/json')) {
+        throw new HttpError(415, 'Send JSON with content-type application/json');
+    }
+    let size = 0;
+    const chunks = [];
+    for await (const chunk of req) {
+        const buffer = chunk;
+        size += buffer.length;
+        if (size > MAX_BODY_BYTES)
+            throw new HttpError(413, 'The request body is too large');
+        chunks.push(buffer);
+    }
+    const text = Buffer.concat(chunks).toString('utf-8');
+    if (!text.trim())
+        return {};
+    const parsed = JSON.parse(text);
+    if (typeof parsed !== 'object' || parsed === null || Array.isArray(parsed)) {
+        throw new HttpError(400, 'The request body must be a JSON object');
+    }
+    return parsed;
+}
+function requireString(body, field) {
+    const value = body[field];
+    if (typeof value !== 'string' || value.trim() === '')
+        throw new HttpError(400, `"${field}" is required`);
+    return value;
+}
+function optionalString(body, field) {
+    const value = body[field];
+    if (value === undefined || value === null)
+        return undefined;
+    if (typeof value !== 'string')
+        throw new HttpError(400, `"${field}" must be a string`);
+    return value;
+}
+function sameToken(given, expected) {
+    if (!given)
+        return false;
+    const a = Buffer.from(given);
+    const b = Buffer.from(expected);
+    return a.length === b.length && timingSafeEqual(a, b);
+}
+function cookieToken(req) {
+    const cookie = req.headers.cookie ?? '';
+    return /(?:^|;\s*)cogitator_studio=([^;]+)/.exec(cookie)?.[1];
+}
+function hostnameOf(header) {
+    if (!header)
+        return undefined;
+    if (header.startsWith('['))
+        return header.slice(1, header.indexOf(']'));
+    return header.split(':')[0];
+}
+function listen(server, host, port, strict) {
+    return new Promise((resolvePort, reject) => {
+        const attempt = (candidate, left) => {
+            const onError = (error) => {
+                server.off('listening', onListening);
+                if (error.code === 'EADDRINUSE' && !strict && left > 0)
+                    attempt(candidate + 1, left - 1);
+                else
+                    reject(error);
+            };
+            const onListening = () => {
+                server.off('error', onError);
+                const address = server.address();
+                resolvePort(typeof address === 'object' && address ? address.port : candidate);
+            };
+            server.once('error', onError);
+            server.once('listening', onListening);
+            server.listen(candidate, host);
+        };
+        attempt(port, 20);
+    });
+}
+/**
+ * Starts Cogitator Studio for the project in `projectDir`: the runtime host
+ * that runs the project, and the web app to chat with its agents, read
+ * traces and costs, approve tool calls, run workflows and fork runs.
+ */
+export async function startStudio(options) {
+    const projectDir = resolve(options.projectDir);
+    const studioDir = resolve(options.studioDir ?? join(projectDir, '.cogitator', 'studio'));
+    const host = options.host ?? '127.0.0.1';
+    const exposed = !LOOPBACK.has(host);
+    const token = exposed ? (options.token ?? randomBytes(18).toString('base64url')) : undefined;
+    const log = options.log ?? (() => undefined);
+    if (!options.price) {
+        initializeModels().catch(() => undefined);
+    }
+    const price = options.price ?? ((model, usage) => calculateCost(model, usage));
+    const store = new StudioStore(studioDir, price);
+    const projectName = (() => {
+        try {
+            const manifest = JSON.parse(readFileSync(join(projectDir, 'package.json'), 'utf-8'));
+            return typeof manifest.name === 'string' ? manifest.name : basename(projectDir);
+        }
+        catch {
+            return basename(projectDir);
+        }
+    })();
+    const clients = new Set();
+    const broadcast = (event) => {
+        const data = `data: ${JSON.stringify(event)}\n\n`;
+        for (const client of clients)
+            client.write(data);
+    };
+    const broadcastRuns = (runs) => {
+        for (const run of runs)
+            broadcast({ type: 'run', run });
+    };
+    const broadcastThread = (thread) => {
+        if (thread)
+            broadcast({ type: 'thread', thread });
+    };
+    const manager = new HostManager({
+        projectDir,
+        studioDir,
+        watch: options.watch,
+        onStatus: (status) => {
+            if (status.state === 'ready')
+                store.setRegistry(status.registry);
+            broadcast({ type: 'host', status });
+        },
+        onEvent: (event) => {
+            const change = store.apply(event);
+            broadcastRuns(change.runs);
+            if (change.token)
+                broadcast({ type: 'token', ...change.token });
+            if (change.reasoning)
+                broadcast({ type: 'reasoning', ...change.reasoning });
+            if (event.type === 'run.started' &&
+                event.threadId &&
+                event.kind === 'agent' &&
+                event.rootRunId === event.runId) {
+                broadcastThread(store.addRunToThread(event.threadId, event.runId));
+            }
+        },
+        onLost: (reason) => broadcastRuns(store.abandonRunning(reason)),
+        onOutput: log,
+    });
+    const allowedHosts = (port) => {
+        const names = new Set(['127.0.0.1', 'localhost', '::1']);
+        if (exposed && host !== '0.0.0.0' && host !== '::')
+            names.add(host);
+        return { names, port };
+    };
+    let port = 0;
+    const server = createServer((req, res) => {
+        void handle(req, res).catch((error) => {
+            const status = error instanceof HttpError ? error.status : 500;
+            if (!res.headersSent)
+                send(res, status, { error: error instanceof Error ? error.message : String(error) });
+            else
+                res.end();
+        });
+    });
+    async function handle(req, res) {
+        const url = new URL(req.url ?? '/', `http://${req.headers.host ?? 'localhost'}`);
+        const path = url.pathname;
+        if (!exposed) {
+            const allowed = allowedHosts(port);
+            const hostname = hostnameOf(req.headers.host);
+            if (!hostname || !allowed.names.has(hostname)) {
+                throw new HttpError(403, 'Cogitator Studio answers on localhost only');
+            }
+        }
+        else if (token) {
+            const given = url.searchParams.get('token') ??
+                cookieToken(req) ??
+                req.headers.authorization?.replace(/^Bearer\s+/i, '');
+            if (!sameToken(given ?? undefined, token))
+                throw new HttpError(401, 'Open the studio with the URL cogitator dev printed');
+            if (url.searchParams.has('token')) {
+                res.setHeader('set-cookie', `cogitator_studio=${token}; HttpOnly; SameSite=Strict; Path=/`);
+            }
+        }
+        if (req.method === 'POST') {
+            const origin = req.headers.origin;
+            if (origin) {
+                let originHost;
+                try {
+                    originHost = new URL(origin).host;
+                }
+                catch {
+                    originHost = undefined;
+                }
+                if (originHost !== req.headers.host)
+                    throw new HttpError(403, 'Requests from other sites are refused');
+            }
+        }
+        if (path.startsWith('/api/'))
+            return api(req, res, url);
+        return serveUi(res, path);
+    }
+    function serveUi(res, path) {
+        const root = uiDirectory();
+        const index = join(root, 'index.html');
+        if (!existsSync(index)) {
+            res.writeHead(500, { 'content-type': 'text/plain' });
+            res.end('The studio UI is not built: run pnpm --filter @cogitator-ai/studio build');
+            return;
+        }
+        const candidate = normalize(join(root, decodeURIComponent(path)));
+        const file = candidate.startsWith(root) && existsSync(candidate) && statSync(candidate).isFile()
+            ? candidate
+            : index;
+        const type = MIME[extname(file)] ?? 'application/octet-stream';
+        const immutable = basename(file) !== 'index.html' && file.includes(`${sep}assets${sep}`);
+        res.writeHead(200, {
+            'content-type': type,
+            'cache-control': immutable ? 'public, max-age=31536000, immutable' : 'no-store',
+            'x-content-type-options': 'nosniff',
+            ...(type.startsWith('text/html') && {
+                'content-security-policy': "default-src 'self'; style-src 'self' 'unsafe-inline'; img-src 'self' data:; connect-src 'self'; frame-ancestors 'none'",
+            }),
+        });
+        res.end(readFileSync(file));
+    }
+    function runOr404(id) {
+        const run = store.getRun(id);
+        if (!run)
+            throw new HttpError(404, `There is no run ${id}`);
+        return run;
+    }
+    async function api(req, res, url) {
+        const parts = url.pathname.split('/').filter(Boolean).slice(1).map(decodeURIComponent);
+        const method = req.method ?? 'GET';
+        const route = `${method} /${parts.map((part, i) => (i % 2 === 1 && parts[0] !== 'workflows' ? ':id' : part)).join('/')}`;
+        if (method === 'GET' && parts[0] === 'events') {
+            res.writeHead(200, {
+                'content-type': 'text/event-stream',
+                'cache-control': 'no-store',
+                connection: 'keep-alive',
+            });
+            res.write(`data: ${JSON.stringify({ type: 'host', status: manager.current })}\n\n`);
+            clients.add(res);
+            const ping = setInterval(() => res.write(': ping\n\n'), 15_000);
+            req.on('close', () => {
+                clearInterval(ping);
+                clients.delete(res);
+            });
+            return;
+        }
+        switch (route) {
+            case 'GET /state':
+                return send(res, 200, {
+                    host: manager.current,
+                    project: projectName,
+                    runs: store.listRuns({ limit: 100 }),
+                    threads: store.listThreads(),
+                });
+            case 'GET /stats':
+                return send(res, 200, store.stats());
+            case 'GET /runs': {
+                const kind = url.searchParams.get('kind');
+                return send(res, 200, {
+                    runs: store.listRuns({
+                        target: url.searchParams.get('target') ?? undefined,
+                        kind: kind === 'agent' || kind === 'workflow' || kind === 'fork' ? kind : undefined,
+                        q: url.searchParams.get('q') ?? undefined,
+                        limit: Number(url.searchParams.get('limit') ?? 200) || 200,
+                    }),
+                });
+            }
+            case 'GET /runs/:id': {
+                const tree = store.getRunTree(parts[1]);
+                if (!tree)
+                    throw new HttpError(404, `There is no run ${parts[1]}`);
+                return send(res, 200, tree);
+            }
+            case 'GET /threads':
+                return send(res, 200, {
+                    threads: store.listThreads(url.searchParams.get('agent') ?? undefined),
+                });
+            case 'GET /threads/:id': {
+                const thread = store.getThread(parts[1]);
+                if (!thread)
+                    throw new HttpError(404, `There is no thread ${parts[1]}`);
+                const runs = thread.runs
+                    .map((id) => store.getRun(id))
+                    .filter((run) => run !== undefined);
+                return send(res, 200, { thread, runs });
+            }
+            case 'POST /chat': {
+                const body = await readJson(req);
+                const agent = requireString(body, 'agent');
+                const input = requireString(body, 'input');
+                const threadId = optionalString(body, 'threadId') ?? `studio_${randomUUID()}`;
+                const thread = store.touchThread(threadId, agent, input);
+                broadcastThread(thread);
+                const { runId } = await manager.request({
+                    type: 'chat',
+                    agent,
+                    input,
+                    threadId,
+                    history: store.history(threadId),
+                });
+                broadcastThread(store.addRunToThread(threadId, runId));
+                return send(res, 202, { runId, threadId });
+            }
+            case 'POST /runs/:id/stop': {
+                runOr404(parts[1]);
+                return send(res, 200, await manager.request({ type: 'stop', runId: parts[1] }));
+            }
+            case 'POST /approvals/:id': {
+                const body = await readJson(req);
+                if (typeof body.approved !== 'boolean')
+                    throw new HttpError(400, '"approved" must be true or false');
+                return send(res, 200, await manager.request({
+                    type: 'approve',
+                    approvalId: parts[1],
+                    approved: body.approved,
+                    ...(optionalString(body, 'reason') && { reason: optionalString(body, 'reason') }),
+                }));
+            }
+            case 'POST /runs/:id/fork': {
+                const run = runOr404(parts[1]);
+                if (run.kind === 'workflow')
+                    throw new HttpError(400, 'Workflow runs are rerun from a node, not forked');
+                if (run.status !== 'completed' || !run.steps?.length) {
+                    throw new HttpError(409, 'Only a completed run with steps can be forked');
+                }
+                const body = await readJson(req);
+                const step = body.step;
+                if (typeof step !== 'number' || !Number.isInteger(step) || step < 0) {
+                    throw new HttpError(400, '"step" must be a step index of the run');
+                }
+                const toolResults = body.toolResults;
+                if (toolResults !== undefined &&
+                    (typeof toolResults !== 'object' || toolResults === null || Array.isArray(toolResults))) {
+                    throw new HttpError(400, '"toolResults" must map tool names to results');
+                }
+                const fork = {
+                    step,
+                    ...(optionalString(body, 'input') !== undefined && {
+                        input: optionalString(body, 'input'),
+                    }),
+                    ...(optionalString(body, 'context') !== undefined && {
+                        context: optionalString(body, 'context'),
+                    }),
+                    ...(toolResults !== undefined && { toolResults: toolResults }),
+                };
+                return send(res, 202, await manager.request({ type: 'fork', run, fork }));
+            }
+            case 'POST /runs/:id/rerun': {
+                const run = runOr404(parts[1]);
+                if (run.kind !== 'workflow')
+                    throw new HttpError(400, 'Only workflow runs are rerun from a node');
+                const body = await readJson(req);
+                return send(res, 202, await manager.request({
+                    type: 'workflow.rerun',
+                    run,
+                    fromNode: requireString(body, 'fromNode'),
+                }));
+            }
+            default:
+                if (method === 'POST' &&
+                    parts[0] === 'workflows' &&
+                    parts.length === 3 &&
+                    parts[2] === 'run') {
+                    const body = await readJson(req);
+                    return send(res, 202, await manager.request({
+                        type: 'workflow.run',
+                        workflow: parts[1],
+                        input: body.input ?? {},
+                    }));
+                }
+                throw new HttpError(404, `No route ${method} ${url.pathname}`);
+        }
+    }
+    port = await listen(server, host, options.port ?? 4321, options.strictPort === true);
+    manager.start();
+    const displayHost = exposed
+        ? host === '0.0.0.0' || host === '::'
+            ? 'localhost'
+            : host
+        : 'localhost';
+    const url = `http://${displayHost.includes(':') ? `[${displayHost}]` : displayHost}:${port}/${token ? `?token=${token}` : ''}`;
+    return {
+        url,
+        port,
+        ...(token && { token }),
+        store,
+        hostStatus: () => manager.current,
+        ready: () => manager.ready(),
+        async close() {
+            for (const client of clients)
+                client.end();
+            clients.clear();
+            await manager.close();
+            await new Promise((done) => server.close(() => done()));
+        },
+    };
+}
+//# sourceMappingURL=server.js.map

package/dist/server/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/server/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACvE,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,YAAY,EAA0D,MAAM,WAAW,CAAC;AACjG,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAC7E,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAEvE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAe,MAAM,YAAY,CAAC;AAoCtD,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC;AAC5D,MAAM,cAAc,GAAG,IAAI,GAAG,IAAI,CAAC;AACnC,MAAM,IAAI,GAA2B;IACnC,OAAO,EAAE,0BAA0B;IACnC,KAAK,EAAE,gCAAgC;IACvC,MAAM,EAAE,yBAAyB;IACjC,MAAM,EAAE,eAAe;IACvB,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,cAAc;IACtB,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE,YAAY;IACtB,OAAO,EAAE,WAAW;CACrB,CAAC;AAEF,MAAM,SAAU,SAAQ,KAAK;IAEhB;IADX,YACW,MAAc,EACvB,OAAe;QAEf,KAAK,CAAC,OAAO,CAAC,CAAC;QAHN,WAAM,GAAN,MAAM,CAAQ;IAIzB,CAAC;CACF;AAED,SAAS,WAAW;IAClB,OAAO,aAAa,CAAC,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,IAAI,CAAC,GAAmB,EAAE,MAAc,EAAE,IAAa;IAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAClC,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE;QACpB,cAAc,EAAE,iCAAiC;QACjD,eAAe,EAAE,UAAU;QAC3B,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;KAC1C,CAAC,CAAC;IACH,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,GAAoB;IAC1C,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,8CAA8C,CAAC,CAAC;IAC3E,CAAC;IACD,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,KAAe,CAAC;QAC/B,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC;QACtB,IAAI,IAAI,GAAG,cAAc;YAAE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,+BAA+B,CAAC,CAAC;QACrF,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACrD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACzC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3E,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,wCAAwC,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,MAAiC,CAAC;AAC3C,CAAC;AAED,SAAS,aAAa,CAAC,IAA6B,EAAE,KAAa;IACjE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE;QAClD,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,IAAI,KAAK,eAAe,CAAC,CAAC;IACrD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,cAAc,CAAC,IAA6B,EAAE,KAAa;IAClE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5D,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,IAAI,KAAK,oBAAoB,CAAC,CAAC;IACvF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,SAAS,CAAC,KAAyB,EAAE,QAAgB;IAC5D,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,eAAe,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,WAAW,CAAC,GAAoB;IACvC,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;IACxC,OAAO,oCAAoC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,UAAU,CAAC,MAA0B;IAC5C,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IACxE,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,MAAM,CAAC,MAAc,EAAE,IAAY,EAAE,IAAY,EAAE,MAAe;IACzE,OAAO,IAAI,OAAO,CAAC,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE;QACzC,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAE,IAAY,EAAE,EAAE;YAClD,MAAM,OAAO,GAAG,CAAC,KAA4B,EAAE,EAAE;gBAC/C,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,MAAM,IAAI,IAAI,GAAG,CAAC;oBAAE,OAAO,CAAC,SAAS,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;;oBACpF,MAAM,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,MAAM,WAAW,GAAG,GAAG,EAAE;gBACvB,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjC,WAAW,CAAC,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACjF,CAAC,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9B,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACjC,CAAC,CAAC;QACF,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,OAAsB;IACtD,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,CAAC;IACzF,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC;IACzC,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7F,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,gBAAgB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IACD,MAAM,KAAK,GAAW,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;IACvF,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAChD,MAAM,WAAW,GAAG,CAAC,GAAG,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,OAAO,CAAC,CAElF,CAAC;YACF,OAAO,OAAO,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAClF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE1C,MAAM,SAAS,GAAG,CAAC,KAAkB,EAAE,EAAE;QACvC,MAAM,IAAI,GAAG,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC;QAClD,KAAK,MAAM,MAAM,IAAI,OAAO;YAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC,CAAC;IACF,MAAM,aAAa,GAAG,CAAC,IAA0B,EAAE,EAAE;QACnD,KAAK,MAAM,GAAG,IAAI,IAAI;YAAE,SAAS,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC;IACF,MAAM,eAAe,GAAG,CAAC,MAAgC,EAAE,EAAE;QAC3D,IAAI,MAAM;YAAE,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC;QAC9B,UAAU;QACV,SAAS;QACT,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE;YACnB,IAAI,MAAM,CAAC,KAAK,KAAK,OAAO;gBAAE,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjE,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAClC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,MAAM,CAAC,KAAK;gBAAE,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;YAChE,IAAI,MAAM,CAAC,SAAS;gBAAE,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;YAC5E,IACE,KAAK,CAAC,IAAI,KAAK,aAAa;gBAC5B,KAAK,CAAC,QAAQ;gBACd,KAAK,CAAC,IAAI,KAAK,OAAO;gBACtB,KAAK,CAAC,SAAS,KAAK,KAAK,CAAC,KAAK,EAC/B,CAAC;gBACD,eAAe,CAAC,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QACD,MAAM,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC/D,QAAQ,EAAE,GAAG;KACd,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,CAAC,IAAY,EAAE,EAAE;QACpC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAS,CAAC,WAAW,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;QACjE,IAAI,OAAO,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI;YAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC,CAAC;IAEF,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACvC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC7C,MAAM,MAAM,GAAG,KAAK,YAAY,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC;YAC/D,IAAI,CAAC,GAAG,CAAC,WAAW;gBAClB,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;;gBAClF,GAAG,CAAC,GAAG,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,MAAM,CAAC,GAAoB,EAAE,GAAmB;QAC7D,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,EAAE,UAAU,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,WAAW,EAAE,CAAC,CAAC;QACjF,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC;QAE1B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9C,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,4CAA4C,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,EAAE,CAAC;YACjB,MAAM,KAAK,GACT,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC;gBAC7B,WAAW,CAAC,GAAG,CAAC;gBAChB,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YACxD,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,SAAS,EAAE,KAAK,CAAC;gBACvC,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,oDAAoD,CAAC,CAAC;YACjF,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,oBAAoB,KAAK,qCAAqC,CAAC,CAAC;YAC9F,CAAC;QACH,CAAC;QAED,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC;YAClC,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,UAA8B,CAAC;gBACnC,IAAI,CAAC;oBACH,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC;gBACpC,CAAC;gBAAC,MAAM,CAAC;oBACP,UAAU,GAAG,SAAS,CAAC;gBACzB,CAAC;gBACD,IAAI,UAAU,KAAK,GAAG,CAAC,OAAO,CAAC,IAAI;oBACjC,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,uCAAuC,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACxD,OAAO,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED,SAAS,OAAO,CAAC,GAAmB,EAAE,IAAY;QAChD,MAAM,IAAI,GAAG,WAAW,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YACrD,GAAG,CAAC,GAAG,CAAC,0EAA0E,CAAC,CAAC;YACpF,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClE,MAAM,IAAI,GACR,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE;YACjF,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,KAAK,CAAC;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,0BAA0B,CAAC;QAC/D,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,GAAG,SAAS,GAAG,EAAE,CAAC,CAAC;QACzF,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;YACjB,cAAc,EAAE,IAAI;YACpB,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC,qCAAqC,CAAC,CAAC,CAAC,UAAU;YAC/E,wBAAwB,EAAE,SAAS;YACnC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI;gBAClC,yBAAyB,EACvB,wHAAwH;aAC3H,CAAC;SACH,CAAC,CAAC;QACH,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,SAAS,QAAQ,CAAC,EAAU;QAC1B,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC7B,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE,CAAC,CAAC;QAC5D,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,UAAU,GAAG,CAAC,GAAoB,EAAE,GAAmB,EAAE,GAAQ;QACpE,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACvF,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC;QACnC,MAAM,KAAK,GAAG,GAAG,MAAM,KAAK,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAEzH,IAAI,MAAM,KAAK,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC9C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;gBACjB,cAAc,EAAE,mBAAmB;gBACnC,eAAe,EAAE,UAAU;gBAC3B,UAAU,EAAE,YAAY;aACzB,CAAC,CAAC;YACH,GAAG,CAAC,KAAK,CACP,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,EAAwB,CAAC,MAAM,CAC/F,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjB,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,CAAC;YAChE,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACnB,aAAa,CAAC,IAAI,CAAC,CAAC;gBACpB,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtB,CAAC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;oBACpB,IAAI,EAAE,OAAO,CAAC,OAAO;oBACrB,OAAO,EAAE,WAAW;oBACpB,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;oBACpC,OAAO,EAAE,KAAK,CAAC,WAAW,EAAE;iBAC7B,CAAC,CAAC;YACL,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;YACvC,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC1C,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;oBACpB,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC;wBACnB,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,SAAS;wBACnD,IAAI,EAAE,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;wBACnF,CAAC,EAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,SAAS;wBACzC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG;qBAC3D,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACxC,IAAI,CAAC,IAAI;oBAAE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,mBAAmB,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACnE,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YAC9B,CAAC;YACD,KAAK,cAAc;gBACjB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;oBACpB,OAAO,EAAE,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC;iBACvE,CAAC,CAAC;YACL,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzC,IAAI,CAAC,MAAM;oBAAE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,sBAAsB,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACxE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI;qBACrB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;qBAC7B,MAAM,CAAC,CAAC,GAAG,EAAoB,EAAE,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC;gBACxD,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;gBACjC,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC3C,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC3C,MAAM,QAAQ,GAAG,cAAc,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,UAAU,UAAU,EAAE,EAAE,CAAC;gBAC9E,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;gBACzD,eAAe,CAAC,MAAM,CAAC,CAAC;gBACxB,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,OAAO,CAAC,OAAO,CAAoB;oBACzD,IAAI,EAAE,MAAM;oBACZ,KAAK;oBACL,KAAK;oBACL,QAAQ;oBACR,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;iBACjC,CAAC,CAAC;gBACH,eAAe,CAAC,KAAK,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;gBACvD,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC7C,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAClF,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;gBACjC,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS;oBACpC,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,kCAAkC,CAAC,CAAC;gBAC/D,OAAO,IAAI,CACT,GAAG,EACH,GAAG,EACH,MAAM,OAAO,CAAC,OAAO,CAAC;oBACpB,IAAI,EAAE,SAAS;oBACf,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;oBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;iBAClF,CAAC,CACH,CAAC;YACJ,CAAC;YACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/B,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU;oBACzB,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,iDAAiD,CAAC,CAAC;gBAC9E,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;oBACrD,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,+CAA+C,CAAC,CAAC;gBAC5E,CAAC;gBACD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;gBACjC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;oBACpE,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,wCAAwC,CAAC,CAAC;gBACrE,CAAC;gBACD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;gBACrC,IACE,WAAW,KAAK,SAAS;oBACzB,CAAC,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,EACvF,CAAC;oBACD,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,8CAA8C,CAAC,CAAC;gBAC3E,CAAC;gBACD,MAAM,IAAI,GAAgB;oBACxB,IAAI;oBACJ,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,SAAS,IAAI;wBACjD,KAAK,EAAE,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC;qBACrC,CAAC;oBACF,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,KAAK,SAAS,IAAI;wBACnD,OAAO,EAAE,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC;qBACzC,CAAC;oBACF,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,WAAsC,EAAE,CAAC;iBAC1F,CAAC;gBACF,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC5E,CAAC;YACD,KAAK,sBAAsB,CAAC,CAAC,CAAC;gBAC5B,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/B,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU;oBACzB,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,0CAA0C,CAAC,CAAC;gBACvE,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;gBACjC,OAAO,IAAI,CACT,GAAG,EACH,GAAG,EACH,MAAM,OAAO,CAAC,OAAO,CAAC;oBACpB,IAAI,EAAE,gBAAgB;oBACtB,GAAG;oBACH,QAAQ,EAAE,aAAa,CAAC,IAAI,EAAE,UAAU,CAAC;iBAC1C,CAAC,CACH,CAAC;YACJ,CAAC;YACD;gBACE,IACE,MAAM,KAAK,MAAM;oBACjB,KAAK,CAAC,CAAC,CAAC,KAAK,WAAW;oBACxB,KAAK,CAAC,MAAM,KAAK,CAAC;oBAClB,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,EAClB,CAAC;oBACD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;oBACjC,OAAO,IAAI,CACT,GAAG,EACH,GAAG,EACH,MAAM,OAAO,CAAC,OAAO,CAAC;wBACpB,IAAI,EAAE,cAAc;wBACpB,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;wBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;qBACxB,CAAC,CACH,CAAC;gBACJ,CAAC;gBACD,MAAM,IAAI,SAAS,CAAC,GAAG,EAAE,YAAY,MAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,IAAI,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,EAAE,OAAO,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;IACrF,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,MAAM,WAAW,GAAG,OAAO;QACzB,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI;YACnC,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,IAAI;QACR,CAAC,CAAC,WAAW,CAAC;IAChB,MAAM,GAAG,GAAG,UAAU,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC,WAAW,IAAI,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IAE/H,OAAO;QACL,GAAG;QACH,IAAI;QACJ,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;QACvB,KAAK;QACL,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO;QACjC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE;QAC5B,KAAK,CAAC,KAAK;YACT,KAAK,MAAM,MAAM,IAAI,OAAO;gBAAE,MAAM,CAAC,GAAG,EAAE,CAAC;YAC3C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChE,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/server/store.d.ts

@@ -0,0 +1,88 @@
+import type { HistoryMessage, HostEvent, RegistryInfo, RunRecord, StudioStats, ThreadRecord } from '../protocol.js';
+/** Prices a model call, `null` when the model's price is unknown. */
+export type Pricer = (model: string, usage: {
+    inputTokens: number;
+    outputTokens: number;
+}) => number | null;
+export interface RunQuery {
+    /** Registry key of an agent or workflow. */
+    target?: string;
+    kind?: RunRecord['kind'];
+    /** Words looked up in the input, the output, the target and the error. */
+    q?: string;
+    limit?: number;
+}
+/** What changed after an event, for the server to push to the UI. */
+export interface StoreChange {
+    runs: RunRecord[];
+    token?: {
+        runId: string;
+        text: string;
+    };
+    reasoning?: {
+        runId: string;
+        text: string;
+    };
+}
+/**
+ * The runs and threads of the studio, kept as JSON files under
+ * `.cogitator/studio` so the history survives restarts without a database.
+ * It turns what the host reports into run records: spans priced per model
+ * call, nested runs under the tool call that started them, workflow nodes
+ * with their status.
+ */
+export declare class StudioStore {
+    private readonly directory;
+    private readonly price;
+    private readonly runs;
+    private readonly threads;
+    private readonly roots;
+    private registry?;
+    constructor(directory: string, price: Pricer);
+    private load;
+    private persist;
+    private persistThreads;
+    setRegistry(registry: RegistryInfo): void;
+    getRun(id: string): RunRecord | undefined;
+    /** A run with every run it started, nested ones included, and its forks. */
+    getRunTree(id: string): {
+        run: RunRecord;
+        descendants: RunRecord[];
+        forks: RunRecord[];
+    } | undefined;
+    /** Runs the user started (not the nested ones), newest first. */
+    listRuns(query?: RunQuery): RunRecord[];
+    /**
+     * Whether a workflow started the run, directly or through other runs: the
+     * workflow's usage already sums it.
+     */
+    private insideWorkflow;
+    /** Counts, spend, latency and daily activity over every run kept. */
+    stats(now?: number, days?: number): StudioStats;
+    listThreads(agent?: string): ThreadRecord[];
+    getThread(id: string): ThreadRecord | undefined;
+    /** The thread a chat message goes to, created on the first message. */
+    touchThread(id: string, agent: string, input: string): ThreadRecord;
+    addRunToThread(threadId: string, runId: string): ThreadRecord | undefined;
+    /** The finished turns of a thread, to replay into a fresh memory. */
+    history(threadId: string): HistoryMessage[];
+    /** Marks the runs still going as failed, when the host they ran in went away. */
+    abandonRunning(reason: string): RunRecord[];
+    /**
+     * A span as the trace shows it. Model calls name the model without its
+     * provider; the run's `provider/model` qualifies it, so it is priced at
+     * that provider's price.
+     */
+    private spanRecord;
+    private initialNodes;
+    /**
+     * Puts every nested run of a finished root under the run whose tool call
+     * started it: the run with the latest tool span that was open when the
+     * nested run started. Runs no tool span holds stay under the root.
+     */
+    private linkNested;
+    private sumChildren;
+    /** Applies an event of the host and returns what changed. */
+    apply(event: HostEvent): StoreChange;
+}
+//# sourceMappingURL=store.d.ts.map

package/dist/server/store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../src/server/store.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EACV,cAAc,EACd,SAAS,EAET,YAAY,EACZ,SAAS,EAGT,WAAW,EAEX,YAAY,EAGb,MAAM,gBAAgB,CAAC;AAExB,qEAAqE;AACrE,MAAM,MAAM,MAAM,GAAG,CACnB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAA;CAAE,KACjD,MAAM,GAAG,IAAI,CAAC;AAEnB,MAAM,WAAW,QAAQ;IACvB,4CAA4C;IAC5C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACzB,0EAA0E;IAC1E,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,qEAAqE;AACrE,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,KAAK,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,SAAS,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;CAC7C;AAwBD;;;;;;GAMG;AACH,qBAAa,WAAW;IAOpB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,KAAK;IAPxB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgC;IACrD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmC;IAC3D,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA6B;IACnD,OAAO,CAAC,QAAQ,CAAC,CAAe;gBAGb,SAAS,EAAE,MAAM,EACjB,KAAK,EAAE,MAAM;IAMhC,OAAO,CAAC,IAAI;IA8BZ,OAAO,CAAC,OAAO;IAOf,OAAO,CAAC,cAAc;IAItB,WAAW,CAAC,QAAQ,EAAE,YAAY,GAAG,IAAI;IAIzC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAIzC,4EAA4E;IAC5E,UAAU,CACR,EAAE,EAAE,MAAM,GACT;QAAE,GAAG,EAAE,SAAS,CAAC;QAAC,WAAW,EAAE,SAAS,EAAE,CAAC;QAAC,KAAK,EAAE,SAAS,EAAE,CAAA;KAAE,GAAG,SAAS;IAoB/E,iEAAiE;IACjE,QAAQ,CAAC,KAAK,GAAE,QAAa,GAAG,SAAS,EAAE;IAiB3C;;;OAGG;IACH,OAAO,CAAC,cAAc;IAatB,qEAAqE;IACrE,KAAK,CAAC,GAAG,SAAa,EAAE,IAAI,SAAK,GAAG,WAAW;IAuF/C,WAAW,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE;IAM3C,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAI/C,uEAAuE;IACvE,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,YAAY;IAoBnE,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IASzE,qEAAqE;IACrE,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc,EAAE;IAa3C,iFAAiF;IACjF,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,EAAE;IAiB3C;;;;OAIG;IACH,OAAO,CAAC,UAAU;IA4BlB,OAAO,CAAC,YAAY;IAKpB;;;;OAIG;IACH,OAAO,CAAC,UAAU;IA2ClB,OAAO,CAAC,WAAW;IAuBnB,6DAA6D;IAC7D,KAAK,CAAC,KAAK,EAAE,SAAS,GAAG,WAAW;CAiLrC"}