Class ServerExecutor

packages/executors/src/serverExecutor.ts:63–237

const tick = () => new Promise<void>(r => setTimeout(() => r(), 0));

export class ServerExecutor implements IExecutor {
  socket?: SocketIOClient.Socket;
  connectTimeout: number | null = null;
  token: Promise<string | undefined>;
  host?: string;
  sandboxId?: string;
  lastSent?: IFiles;

  constructor() {
    this.token = this.retrieveSSEToken();
  }

  private async initializeSocket() {
    if (!this.sandboxId) {
      throw new Error('initializeSocket: sandboxId is not defined');
    }
    const usedHost = this.host || 'https://codesandbox.io';
    const sseLbHost = usedHost.replace('https://', 'https://sse-lb.');
    const res = await axios.get(`${sseLbHost}/api/cluster/${this.sandboxId}`);
    const sseHost = res.data.hostname;

    this.socket = io(sseHost, {
      autoConnect: false,
      transports: ['websocket', 'polling'],
    });
  }

  async initialize({ sandboxId, files, host }: ISetupParams) {
    if (this.sandboxId === sandboxId && this.socket?.connected) {
      return;
    }

    this.host = host;
    this.sandboxId = sandboxId;
    this.lastSent = files;

    await this.dispose();
    await tick();

    await this.initializeSocket();
  }

  public async setup() {
    debug('Setting up server executor...');

    return this.openSocket();
  }

  public async dispose() {
    if (this.socket) {
      this.socket.removeAllListeners();
      this.socket.close();
    }
  }

  public updateFiles(newFiles: IFiles) {
    const changedFiles = this.lastSent
      ? getDiff(this.lastSent, newFiles)