@nirvana-labs/nirvana-mcp 1.26.1 → 1.26.2

package/src/code-tool-worker.ts

@@ -1,377 +0,0 @@
-// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-import path from 'node:path';
-import util from 'node:util';
-
-import Fuse from 'fuse.js';
-import ts from 'typescript';
-
-import { WorkerInput, WorkerSuccess, WorkerError } from './code-tool-types';
-import { NirvanaLabs } from '@nirvana-labs/nirvana';
-
-function getRunFunctionSource(code: string): {
-  type: 'declaration' | 'expression';
-  client: string | undefined;
-  code: string;
-} | null {
-  const sourceFile = ts.createSourceFile('code.ts', code, ts.ScriptTarget.Latest, true);
-  const printer = ts.createPrinter();
-
-  for (const statement of sourceFile.statements) {
-    // Check for top-level function declarations
-    if (ts.isFunctionDeclaration(statement)) {
-      if (statement.name?.text === 'run') {
-        return {
-          type: 'declaration',
-          client: statement.parameters[0]?.name.getText(),
-          code: printer.printNode(ts.EmitHint.Unspecified, statement.body!, sourceFile),
-        };
-      }
-    }
-
-    // Check for variable declarations: const run = () => {} or const run = function() {}
-    if (ts.isVariableStatement(statement)) {
-      for (const declaration of statement.declarationList.declarations) {
-        if (
-          ts.isIdentifier(declaration.name) &&
-          declaration.name.text === 'run' &&
-          // Check if it's initialized with a function
-          declaration.initializer &&
-          (ts.isFunctionExpression(declaration.initializer) || ts.isArrowFunction(declaration.initializer))
-        ) {
-          return {
-            type: 'expression',
-            client: declaration.initializer.parameters[0]?.name.getText(),
-            code: printer.printNode(ts.EmitHint.Unspecified, declaration.initializer, sourceFile),
-          };
-        }
-      }
-    }
-  }
-
-  return null;
-}
-
-function getTSDiagnostics(code: string): string[] {
-  const functionSource = getRunFunctionSource(code)!;
-  const codeWithImport = [
-    'import { NirvanaLabs } from "@nirvana-labs/nirvana";',
-    functionSource.type === 'declaration' ?
-      `async function run(${functionSource.client}: NirvanaLabs)`
-    : `const run: (${functionSource.client}: NirvanaLabs) => Promise<unknown> =`,
-    functionSource.code,
-  ].join('\n');
-  const sourcePath = path.resolve('code.ts');
-  const ast = ts.createSourceFile(sourcePath, codeWithImport, ts.ScriptTarget.Latest, true);
-  const options = ts.getDefaultCompilerOptions();
-  options.target = ts.ScriptTarget.Latest;
-  options.module = ts.ModuleKind.NodeNext;
-  options.moduleResolution = ts.ModuleResolutionKind.NodeNext;
-  const host = ts.createCompilerHost(options, true);
-  const newHost: typeof host = {
-    ...host,
-    getSourceFile: (...args) => {
-      if (path.resolve(args[0]) === sourcePath) {
-        return ast;
-      }
-      return host.getSourceFile(...args);
-    },
-    readFile: (...args) => {
-      if (path.resolve(args[0]) === sourcePath) {
-        return codeWithImport;
-      }
-      return host.readFile(...args);
-    },
-    fileExists: (...args) => {
-      if (path.resolve(args[0]) === sourcePath) {
-        return true;
-      }
-      return host.fileExists(...args);
-    },
-  };
-  const program = ts.createProgram({
-    options,
-    rootNames: [sourcePath],
-    host: newHost,
-  });
-  const diagnostics = ts.getPreEmitDiagnostics(program, ast);
-  return diagnostics.map((d) => {
-    const message = ts.flattenDiagnosticMessageText(d.messageText, '\n');
-    if (!d.file || !d.start) return `- ${message}`;
-    const { line: tsLine } = ts.getLineAndCharacterOfPosition(d.file, d.start);
-    // We add two lines in the beginning, for the client import and the function declaration.
-    // So the actual (zero-based) line number is tsLine - 2.
-    const lineNumber = tsLine - 2;
-    const line = code.split('\n').at(lineNumber)?.trim();
-    return line ? `- ${message}\n    at line ${lineNumber + 1}\n      ${line}` : `- ${message}`;
-  });
-}
-
-const fuse = new Fuse(
-  [
-    'client.user.get',
-    'client.apiKeys.create',
-    'client.apiKeys.delete',
-    'client.apiKeys.get',
-    'client.apiKeys.list',
-    'client.apiKeys.update',
-    'client.operations.get',
-    'client.operations.list',
-    'client.projects.create',
-    'client.projects.delete',
-    'client.projects.get',
-    'client.projects.list',
-    'client.projects.update',
-    'client.compute.vms.create',
-    'client.compute.vms.delete',
-    'client.compute.vms.get',
-    'client.compute.vms.list',
-    'client.compute.vms.restart',
-    'client.compute.vms.update',
-    'client.compute.vms.availability.create',
-    'client.compute.vms.availability.update',
-    'client.compute.vms.volumes.list',
-    'client.compute.vms.osImages.list',
-    'client.compute.volumes.create',
-    'client.compute.volumes.delete',
-    'client.compute.volumes.get',
-    'client.compute.volumes.list',
-    'client.compute.volumes.update',
-    'client.compute.volumes.availability.create',
-    'client.compute.volumes.availability.update',
-    'client.networking.vpcs.create',
-    'client.networking.vpcs.delete',
-    'client.networking.vpcs.get',
-    'client.networking.vpcs.list',
-    'client.networking.vpcs.update',
-    'client.networking.vpcs.availability.create',
-    'client.networking.vpcs.availability.update',
-    'client.networking.firewallRules.create',
-    'client.networking.firewallRules.delete',
-    'client.networking.firewallRules.get',
-    'client.networking.firewallRules.list',
-    'client.networking.firewallRules.update',
-    'client.networking.connect.connections.create',
-    'client.networking.connect.connections.delete',
-    'client.networking.connect.connections.get',
-    'client.networking.connect.connections.list',
-    'client.networking.connect.connections.update',
-    'client.networking.connect.routes.list',
-    'client.rpcNodes.flex.create',
-    'client.rpcNodes.flex.delete',
-    'client.rpcNodes.flex.get',
-    'client.rpcNodes.flex.list',
-    'client.rpcNodes.flex.update',
-    'client.rpcNodes.flex.blockchains.list',
-    'client.rpcNodes.dedicated.get',
-    'client.rpcNodes.dedicated.list',
-    'client.rpcNodes.dedicated.blockchains.list',
-    'client.vektor.registry.assets.list',
-    'client.vektor.registry.blockchains.list',
-    'client.vektor.registry.venues.list',
-    'client.vektor.registry.errors.list',
-    'client.vektor.registry.lendMarkets.list',
-    'client.vektor.registry.borrowMarkets.list',
-    'client.vektor.registry.lpPools.list',
-    'client.vektor.balances.list',
-    'client.vektor.balances.listHistorical',
-    'client.vektor.prices.list',
-    'client.vektor.prices.listHistorical',
-    'client.vektor.lend.markets.list',
-    'client.vektor.lend.markets.listHistorical',
-    'client.vektor.lend.positions.list',
-    'client.vektor.lend.positions.listHistorical',
-    'client.vektor.lend.lend.create',
-    'client.vektor.lend.withdraw.create',
-    'client.vektor.lend.setCollateral.create',
-    'client.vektor.borrow.markets.list',
-    'client.vektor.borrow.markets.listHistorical',
-    'client.vektor.borrow.positions.list',
-    'client.vektor.borrow.positions.listHistorical',
-    'client.vektor.borrow.accounts.list',
-    'client.vektor.borrow.accounts.listHistorical',
-    'client.vektor.borrow.borrow.create',
-    'client.vektor.borrow.repay.create',
-    'client.vektor.lp.pools.list',
-    'client.vektor.lp.pools.listHistorical',
-    'client.vektor.lp.positions.list',
-    'client.vektor.lp.positions.listHistorical',
-    'client.vektor.lp.depositQuote.create',
-    'client.vektor.lp.withdrawQuote.create',
-    'client.vektor.buy.quotes.list',
-    'client.vektor.buy.buy.create',
-    'client.vektor.sell.quotes.list',
-    'client.vektor.sell.sell.create',
-    'client.vektor.move.create',
-    'client.vektor.wrap.wrap.create',
-    'client.vektor.wrap.unwrap.create',
-    'client.vektor.bridge.quotes.list',
-    'client.vektor.lock.markets.list',
-    'client.vektor.lock.positions.list',
-    'client.vektor.vote.markets.list',
-    'client.vektor.vote.rewards.list',
-    'client.vektor.incentivize.list',
-    'client.vektor.executions.get',
-    'client.vektor.executions.list',
-    'client.vektor.executions.steps.get',
-    'client.vektor.executions.steps.sign',
-  ],
-  { threshold: 1, shouldSort: true },
-);
-
-function getMethodSuggestions(fullyQualifiedMethodName: string): string[] {
-  return fuse
-    .search(fullyQualifiedMethodName)
-    .map(({ item }) => item)
-    .slice(0, 5);
-}
-
-const proxyToObj = new WeakMap<any, any>();
-const objToProxy = new WeakMap<any, any>();
-
-type ClientProxyConfig = {
-  path: string[];
-  isBelievedBad?: boolean;
-};
-
-function makeSdkProxy<T extends object>(obj: T, { path, isBelievedBad = false }: ClientProxyConfig): T {
-  let proxy: T = objToProxy.get(obj);
-
-  if (!proxy) {
-    proxy = new Proxy(obj, {
-      get(target, prop, receiver) {
-        const propPath = [...path, String(prop)];
-        const value = Reflect.get(target, prop, receiver);
-
-        if (isBelievedBad || (!(prop in target) && value === undefined)) {
-          // If we're accessing a path that doesn't exist, it will probably eventually error.
-          // Let's proxy it and mark it bad so that we can control the error message.
-          // We proxy an empty class so that an invocation or construction attempt is possible.
-          return makeSdkProxy(class {}, { path: propPath, isBelievedBad: true });
-        }
-
-        if (value !== null && (typeof value === 'object' || typeof value === 'function')) {
-          return makeSdkProxy(value, { path: propPath, isBelievedBad });
-        }
-
-        return value;
-      },
-
-      apply(target, thisArg, args) {
-        if (isBelievedBad || typeof target !== 'function') {
-          const fullyQualifiedMethodName = path.join('.');
-          const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
-          throw new Error(
-            `${fullyQualifiedMethodName} is not a function. Did you mean: ${suggestions.join(', ')}`,
-          );
-        }
-
-        return Reflect.apply(target, proxyToObj.get(thisArg) ?? thisArg, args);
-      },
-
-      construct(target, args, newTarget) {
-        if (isBelievedBad || typeof target !== 'function') {
-          const fullyQualifiedMethodName = path.join('.');
-          const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
-          throw new Error(
-            `${fullyQualifiedMethodName} is not a constructor. Did you mean: ${suggestions.join(', ')}`,
-          );
-        }
-
-        return Reflect.construct(target, args, newTarget);
-      },
-    });
-
-    objToProxy.set(obj, proxy);
-    proxyToObj.set(proxy, obj);
-  }
-
-  return proxy;
-}
-
-function parseError(code: string, error: unknown): string | undefined {
-  if (!(error instanceof Error)) return;
-  const message = error.name ? `${error.name}: ${error.message}` : error.message;
-  try {
-    // Deno uses V8; the first "<anonymous>:LINE:COLUMN" is the top of stack.
-    const lineNumber = error.stack?.match(/<anonymous>:([0-9]+):[0-9]+/)?.[1];
-    // -1 for the zero-based indexing
-    const line =
-      lineNumber &&
-      code
-        .split('\n')
-        .at(parseInt(lineNumber, 10) - 1)
-        ?.trim();
-    return line ? `${message}\n  at line ${lineNumber}\n    ${line}` : message;
-  } catch {
-    return message;
-  }
-}
-
-const fetch = async (req: Request): Promise<Response> => {
-  const { opts, code } = (await req.json()) as WorkerInput;
-
-  const runFunctionSource = code ? getRunFunctionSource(code) : null;
-  if (!runFunctionSource) {
-    const message =
-      code ?
-        'The code is missing a top-level `run` function.'
-      : 'The code argument is missing. Provide one containing a top-level `run` function.';
-    return Response.json(
-      {
-        message: `${message} Write code within this template:\n\n\`\`\`\nasync function run(client) {\n  // Fill this out\n}\n\`\`\``,
-        logLines: [],
-        errLines: [],
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-
-  const diagnostics = getTSDiagnostics(code);
-  if (diagnostics.length > 0) {
-    return Response.json(
-      {
-        message: `The code contains TypeScript diagnostics:\n${diagnostics.join('\n')}`,
-        logLines: [],
-        errLines: [],
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-
-  const client = new NirvanaLabs({
-    ...opts,
-  });
-
-  const logLines: string[] = [];
-  const errLines: string[] = [];
-  const console = {
-    log: (...args: unknown[]) => {
-      logLines.push(util.format(...args));
-    },
-    error: (...args: unknown[]) => {
-      errLines.push(util.format(...args));
-    },
-  };
-  try {
-    let run_ = async (client: any) => {};
-    eval(`${code}\nrun_ = run;`);
-    const result = await run_(makeSdkProxy(client, { path: ['client'] }));
-    return Response.json({
-      result,
-      logLines,
-      errLines,
-    } satisfies WorkerSuccess);
-  } catch (e) {
-    return Response.json(
-      {
-        message: parseError(code, e),
-        logLines,
-        errLines,
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-};
-
-export default { fetch };