gristlabs_grist-core/app/server/lib/TestingHooks.ts

import * as net from 'net';

import {UserProfile} from 'app/common/LoginSessionAPI';
import {Deps as ActiveDocDeps} from 'app/server/lib/ActiveDoc';
import * as Comm from 'app/server/lib/Comm';
import {ILoginSession} from 'app/server/lib/ILoginSession';
import * as log from 'app/server/lib/log';
import {IMessage, Rpc} from 'grain-rpc';
import {createCheckers} from 'ts-interface-checker';
import {FlexServer} from './FlexServer';
import {IInstanceManager} from './IInstanceManager';
import {ITestingHooks} from './ITestingHooks';
import ITestingHooksTI from './ITestingHooks-ti';
import {connect, fromCallback} from './serverUtils';

export function startTestingHooks(socketPath: string, port: number, instanceManager: IInstanceManager,
                                  comm: Comm, flexServer: FlexServer,
                                  workerServers: FlexServer[]): Promise<net.Server> {
  // Create socket server listening on the given path for testing connections.
  return new Promise((resolve, reject) => {
    const server = net.createServer();
    server.on('error', reject);
    server.on('listening', () => resolve(server));
    server.on('connection', socket => {
      // On connection, create an Rpc object communicating over that socket.
      const rpc = connectToSocket(new Rpc({logger: {}}), socket);
      // Register the testing implementation.
      rpc.registerImpl('testing',
                       new TestingHooks(port, instanceManager, comm, flexServer, workerServers),
                       createCheckers(ITestingHooksTI).ITestingHooks);
    });
    server.listen(socketPath);
  });
}

function connectToSocket(rpc: Rpc, socket: net.Socket): Rpc {
  socket.setEncoding('utf8');
  socket.on('data', (buf: string) => rpc.receiveMessage(JSON.parse(buf)));
  rpc.setSendMessage((m: IMessage) => fromCallback(cb => socket.write(JSON.stringify(m), 'utf8', cb)));
  return rpc;
}

export interface TestingHooksClient extends ITestingHooks {
  close(): void;
}

export async function connectTestingHooks(socketPath: string): Promise<TestingHooksClient> {
  const socket = await connect(socketPath);
  const rpc = connectToSocket(new Rpc({logger: {}}), socket);
  return Object.assign(rpc.getStub<TestingHooks>('testing', createCheckers(ITestingHooksTI).ITestingHooks), {
    close: () => socket.end(),
  });
}

export class TestingHooks implements ITestingHooks {
  constructor(
    private _port: number,
    private _instanceManager: IInstanceManager,
    private _comm: Comm,
    private _server: FlexServer,
    private _workerServers: FlexServer[]
  ) {}

  public getOwnPort(): number {
    log.info("TestingHooks.getOwnPort called");
    return this._server.getOwnPort();
  }

  public getPort(): number {
    log.info("TestingHooks.getPort called");
    return this._port;
  }

  public async updateAuthToken(instId: string, authToken: string): Promise<void> {
    log.info("TestingHooks.updateAuthToken called with", instId, authToken);
    const loginSession = this._getLoginSession(instId);
    await loginSession.updateTokenForTesting(authToken);
  }

  public async getAuthToken(instId: string): Promise<string|null> {
    log.info("TestingHooks.getAuthToken called with", instId);
    const loginSession = this._getLoginSession(instId);
    return await loginSession.getCurrentTokenForTesting();
  }

  public async useTestToken(instId: string, token: string): Promise<void> {
    log.info("TestingHooks.useTestToken called with", token);
    const loginSession = this._getLoginSession(instId);
    return await loginSession.useTestToken(token);
  }

  public async setLoginSessionProfile(gristSidCookie: string, profile: UserProfile|null, org?: string): Promise<void> {
    log.info("TestingHooks.setLoginSessionProfile called with", gristSidCookie, profile, org);
    const sessionId = this._comm.getSessionIdFromCookie(gristSidCookie);
    const loginSession = this._comm.getOrCreateSession(sessionId, {org});
    return await loginSession.testSetProfile(profile);
  }

  public async setServerVersion(version: string|null): Promise<void> {
    log.info("TestingHooks.setServerVersion called with", version);
    this._comm.setServerVersion(version);
    for (const server of this._workerServers) {
      await server.comm.setServerVersion(version);
    }
  }

  public async disconnectClients(): Promise<void> {
    log.info("TestingHooks.disconnectClients called");
    this._comm.destroyAllClients();
    for (const server of this._workerServers) {
      await server.comm.destroyAllClients();
    }
  }

  public async commShutdown(): Promise<void> {
    log.info("TestingHooks.commShutdown called");
    await this._comm.testServerShutdown();
    for (const server of this._workerServers) {
      await server.comm.testServerShutdown();
    }
  }

  public async commRestart(): Promise<void> {
    log.info("TestingHooks.commRestart called");
    await this._comm.testServerRestart();
    for (const server of this._workerServers) {
      await server.comm.testServerRestart();
    }
  }

  // Set how long new clients will persist after disconnection.
  // Call with 0 to return to default duration.
  public async commSetClientPersistence(ttlMs: number) {
    log.info("TestingHooks.setClientPersistence called with", ttlMs);
    this._comm.testSetClientPersistence(ttlMs);
    for (const server of this._workerServers) {
      await server.comm.testSetClientPersistence(ttlMs);
    }
  }

  public async closeDocs(): Promise<void> {
    log.info("TestingHooks.closeDocs called");
    if (this._server) {
      await this._server.closeDocs();
    }
    for (const server of this._workerServers) {
      await server.closeDocs();
    }
  }

  public async setDocWorkerActivation(workerId: string, active: 'active'|'inactive'|'crash'):
    Promise<void> {
    log.info("TestingHooks.setDocWorkerActivation called with", workerId, active);
    for (const server of this._workerServers) {
      if (server.worker.id === workerId || server.worker.publicUrl === workerId) {
        switch (active) {
        case 'active':
          await server.restartListening();
          break;
        case 'inactive':
          await server.stopListening();
          break;
        case 'crash':
          await server.stopListening('crash');
          break;
        }
        return;
      }
    }
    throw new Error(`could not find worker: ${workerId}`);
  }

  public async flushAuthorizerCache(): Promise<void> {
    log.info("TestingHooks.flushAuthorizerCache called");
    this._server.dbManager.flushDocAuthCache();
    for (const server of this._workerServers) {
      await server.dbManager.flushDocAuthCache();
    }
  }

  // Returns a Map from docId to number of connected clients for all open docs across servers,
  // but represented as an array of pairs, to be serializable.
  public async getDocClientCounts(): Promise<Array<[string, number]>> {
    log.info("TestingHooks.getDocClientCounts called");
    const counts = new Map<string, number>();
    for (const server of [this._server, ...this._workerServers]) {
      const c = await server.getDocClientCounts();
      for (const [key, val] of c) {
        counts.set(key, (counts.get(key) || 0) + val);
      }
    }
    return Array.from(counts);
  }

  // Sets the seconds for ActiveDoc timeout, and returns the previous value.
  public async setActiveDocTimeout(seconds: number): Promise<number> {
    const prev = ActiveDocDeps.ACTIVEDOC_TIMEOUT;
    ActiveDocDeps.ACTIVEDOC_TIMEOUT = seconds;
    return prev;
  }

  private _getLoginSession(instId: string): ILoginSession {
    return this._instanceManager.getLoginSession(instId);
  }
}