Function fetch

services/cloud-agent-next/test/test-worker.ts:83–148

(request: Request, env: TestEnv)

  CLOUD_AGENT_REPORT_QUEUE: Queue<CloudAgentQueueReport>;
  USER_KILO_FACADE: DurableObjectNamespace<UserKiloFacade>;
  SANDBOX_CONTROL: Env['SANDBOX_CONTROL'];
  SANDBOX_SESSION: Env['SANDBOX_SESSION'];
  NEXTAUTH_SECRET: Env['NEXTAUTH_SECRET'];
};

function routeToUserKiloFacade(request: Request, env: TestEnv, userId: string): Promise<Response> {
  const facade = env.USER_KILO_FACADE.get(env.USER_KILO_FACADE.idFromName(userId));
  const headers = new Headers(request.headers);
  headers.set(KILO_FACADE_USER_ID_HEADER, userId);
  headers.set(KILO_FACADE_AUTH_TOKEN_HEADER, 'test-token');
  return facade.fetch(new Request(request, { headers }));
}

function createTerminalUpgradeRequest(
  request: Request,
  role: 'browser' | 'wrapper',
  ptyId: string
): Request {
  const url = new URL(request.url);
  url.pathname = `/terminal/${role}`;
  url.search = '';
  url.searchParams.set('ptyId', ptyId);
  const headers = new Headers({ Upgrade: 'websocket' });
  if (role === 'wrapper') {
    const authorization = request.headers.get('Authorization');
    if (authorization !== null) headers.set('Authorization', authorization);
  }
  return new Request(url, { method: 'GET', headers });
}

function isTerminalRouteIdentity(ownerId: string, sessionId: string, ptyId: string): boolean {
  return (
    ownerId.length > 0 &&
    sessionId.startsWith('workspace_') &&
    SESSION_ID_RE.test(sessionId) &&
    terminalPtyIdSchema.safeParse(ptyId).success
  );
}

export default {
  async fetch(request: Request, env: TestEnv): Promise<Response> {
    const url = new URL(request.url);

    if (url.pathname.startsWith('/sandbox-control/')) {
      const sandboxId = decodeURIComponent(url.pathname.slice('/sandbox-control/'.length));
      return admitSandboxWrapperUpgrade(request, env, sandboxId);
    }

    if (url.pathname.startsWith('/sandbox-control-v2/')) {
      const upgradeHeader = request.headers.get('Upgrade');
      if (upgradeHeader?.toLowerCase() !== 'websocket') {
        return new Response('Expected WebSocket upgrade', { status: 426 });
      }
      const sandboxId = decodeURIComponent(url.pathname.slice('/sandbox-control-v2/'.length));
      if (!sandboxId || sandboxId.includes('/') || !isSandboxControlId(sandboxId)) {
        return new Response('Invalid sandboxId', { status: 400 });
      }
      return env.SANDBOX_CONTROL.getByName(sandboxId).fetch(request);