Class DartToolingDaemon

src/shared/services/tooling_daemon.ts:16–326

import { ActiveLocation, ActiveLocationChangedEvent, DebugSessionChangedEvent, DebugSessionStartedEvent, DebugSessionStoppedEvent, DeviceAddedEvent, DeviceChangedEvent, DeviceRemovedEvent, DeviceSelectedEvent, DtdMessage, DtdNotification, DtdRequest, DtdResponse, DtdResult, EnablePlatformTypeParams, Event, EventKind, GetDebugSessionsResult, GetDevicesResult, GetIDEWorkspaceRootsParams, GetIDEWorkspaceRootsResult, GetVmServicesResult, HotReloadParams, HotRestartParams, NavigateToCodeParams, OpenDevToolsPageParams, ReadFileAsStringParams, ReadFileAsStringResult, RegisterServiceParams, RegisterServiceResult, RegisterVmServiceParams, RegisterVmServiceResult, SelectDeviceParams, Service, ServiceMethod, ServiceRegisteredEventData, ServiceUnregisteredEventData, SetIDEWorkspaceRootsParams, SetIDEWorkspaceRootsResult, Stream, SuccessResult, UnregisterVmServiceParams, UnregisterVmServiceResult } from "./tooling_daemon_services";

export class DartToolingDaemon implements IAmDisposable {
	protected readonly disposables: IAmDisposable[] = [];
	protected readonly logger: CategoryLogger;

	private readonly dtdProcess: DartToolingDaemonProcess;
	private connection: ConnectionInfo | undefined;
	private nextId = 1;
	private completers: Record<string, PromiseCompleter<DtdResult>> = {};
	private serviceHandlers: Record<string, (params?: object) => PromiseOr<DtdResult>> = {};

	private hasShownTerminatedError = false;
	private isShuttingDown = false;

	private connectedCompleter = new PromiseCompleter<ConnectionInfo | undefined>();
	public get connected() { return this.connectedCompleter.promise; }

	private readonly notificationsEmitters: Record<string, EventsEmitter<any>> = {};

	/// The set of services that have been seen registered over the Services stream
	/// but not yet unregistered.
	public readonly registeredServiceMethods = new Set<string>();

	constructor(
		logger: Logger,
		sdks: DartSdks,
		protected readonly dartCapabilities: DartCapabilities,
		additionalArgs: string[],
		maxLogLineLength: number | undefined,
		getToolEnv: () => any,
		private readonly promptToReloadExtension: (logger: Logger, options: { prompt?: string; buttonText?: string; offerLog?: boolean; specificLog?: string; useError?: boolean; restartReason: ExtensionRestartReason }) => Promise<void>,
	) {
		this.logger = new CategoryLogger(logger, LogCategory.DartToolingDaemon);
		this.dtdProcess = new DartToolingDaemonProcess(this.logger, sdks, additionalArgs, maxLogLineLength, getToolEnv);
		this.disposables.push(this.dtdProcess);

		void this.dtdProcess.dtdUri.then(() => this.connect());
		void this.dtdProcess.processExit.then((codes) => this.handleProcessExit(codes));
	}

	public get dtdUri(): Promise<string | undefined> {
		return this.dtdProcess.dtdUri;
	}

	private async connect() {
		const dtdUri = await this.dtdProcess.dtdUri;
		if (!dtdUri)
			return;
		const dtdSecret = await this.dtdProcess.dtdSecret;

		this.logger.info(`Connecting to DTD at ${dtdUri}...`);
		const socket = new ws.WebSocket(dtdUri, { followRedirects: true });
		attachPing(socket);
		socket.on("open", () => this.handleOpen());
		// eslint-disable-next-line @typescript-eslint/no-base-to-string
		socket.on("message", (data) => this.handleData(data.toString()));
		socket.on("close", () => this.handleWebSocketClose());
		socket.on("error", (e) => this.handleError(e));