Class IDF

src/modules/idf.ts:12–172

import { getDeviceVersion } from "../schema/device.type";

class IDF {
	tunnelManager: TunnelManager;
	wdaStreamClient: WdaStreamClient;
	wdaControlClient?: WdaControlClient;
	wdaGoIOS?: WdaGoIOS;

	udid: string;
	version: number;
	socketClient: Socket;

	controlPort?: number;
	streamPort?: number;

	streamClient?: SubProcess;

	constructor(udid: string, version: string, socket: Socket) {
		this.socketClient = socket;
		this.udid = udid;
		this.tunnelManager = new TunnelManager(udid);
		this.wdaStreamClient = new WdaStreamClient();
		this.version = getDeviceVersion(version);
	}

	getPorts() {
		return [this.controlPort, this.streamPort];
	}

	async sendCommand(data: any) {
		logger.info(`send command ${JSON.stringify(data)}`);
		try {
			if (this.wdaControlClient) {
				const cmd = buildCommand(data);
				await this.wdaControlClient.performCommand(cmd);
			}
		} catch (error) {
			logger.error(`error in send command`, error);
		}
	}

	async start() {
		const [controlPort, streamPort] = await findFreePorts(2);
		this.controlPort = controlPort;
		this.streamPort = streamPort;
		/**
		 * start tunnel for control port
		 */
		try {
			await this.tunnelManager.startControlTunnel(controlPort);
		} catch (error) {
			logger.error(`error start tunnel for control port`, error);
			await this.stop();
			return false;
		}

		/**
		 * start tunnel for stream port
		 */
		try {