HuolalaTech / page-spy / 9394622037

import { getRandomId, stringifyData } from 'base/src';

import atom from 'base/src/atom';

import Client, { combineName } from 'base/src/client';

import { makeMessage } from 'base/src/message';

import { UPDATE_ROOM_INFO } from 'base/src/message/server-type';

import {

  SocketStoreBase,

  SocketState,

  SocketWrapper,

  WebSocketEvents,

} from 'base/src/socket-base';

import { SpyBase } from 'packages/page-spy-types';

import { InitConfig } from 'page-spy-react-native/types';

export class RNWebSocketWrapper extends SocketWrapper {

  private socketInstance: WebSocket | null = null;

  init(url: string) {

    this.socketInstance = new WebSocket(url);

    const eventNames: WebSocketEvents[] = ['open', 'close', 'error', 'message'];

    eventNames.forEach((eventName) => {

      this.socketInstance!.addEventListener(eventName, (data) => {

        this.events[eventName].forEach((cb) => {

          cb(data);

        });

      });

    });

  }

  send(data: string) {

    this.socketInstance?.send(stringifyData(data));

  }

  close() {

    this.socketInstance?.close();

  }

  getState(): SocketState {

    return this.socketInstance?.readyState as SocketState;

  }

}

export class RNWebSocketStore extends SocketStoreBase {

  // websocket instance

  protected socketWrapper: RNWebSocketWrapper = new RNWebSocketWrapper();

  public getPageSpyConfig: (() => Required<InitConfig>) | null = null;

  public getSocket() {

    return this.socketWrapper;

  }

  // disable lint: this is an abstract method of parent class, so it cannot be static

  // eslint-disable-next-line class-methods-use-this

  onOffline(): void {}

  // eslint-disable-next-line @typescript-eslint/no-useless-constructor

  constructor() {

    super();

    this.addListener('debug', RNWebSocketStore.handleDebugger);

  }

  updateRoomInfo() {

    if (this.getPageSpyConfig) {

      const { project, title } = this.getPageSpyConfig();

      const device = combineName(Client.info);

      this.send(

        {

          type: UPDATE_ROOM_INFO,

          content: {

            info: {

              name: device,

              group: project,

              tags: {

                title,

                name: device,

                group: project,

              },

            },

          },

        },

        true,

      );

    }

  }

  // run executable code which received from remote and send back the result

  private static handleDebugger(

    { source }: SpyBase.InteractiveEvent<string>,

    reply: (data: any) => void,

  ) {

    const { type, data } = source;

    if (type === 'debug') {

      const originMsg = makeMessage('console', {

        logType: 'debug-origin',

        logs: [

          {

            id: getRandomId(),

            type: 'debug-origin',

            value: data,

          },

        ],

      });

      reply(originMsg);

      try {

        // eslint-disable-next-line no-new-func, @typescript-eslint/no-implied-eval

        const result = new Function(`return ${data}`)();

        const evalMsg = makeMessage('console', {

          logType: 'debug-eval',

          logs: [atom.transformToAtom(result)],

        });

        reply(evalMsg);

      } catch (err) {

        const errMsg = makeMessage('console', {

          logType: 'error',

          logs: [

            {

              type: 'error',

              value: (err as Error).stack,

            },

          ],

        });

        reply(errMsg);

      }

    }

  }

}

const socketStore = new RNWebSocketStore();

export default socketStore;