| // Copyright (c) 2015 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| /** |
| * @implements {Protocol.InspectorBackend.Connection} |
| * @unrestricted |
| */ |
| SDK.MainConnection = class { |
| /** |
| * @param {!Protocol.InspectorBackend.Connection.Params} params |
| */ |
| constructor(params) { |
| this._onMessage = params.onMessage; |
| this._onDisconnect = params.onDisconnect; |
| this._disconnected = false; |
| this._eventListeners = [ |
| InspectorFrontendHost.events.addEventListener( |
| InspectorFrontendHostAPI.Events.DispatchMessage, this._dispatchMessage, this), |
| InspectorFrontendHost.events.addEventListener( |
| InspectorFrontendHostAPI.Events.DispatchMessageChunk, this._dispatchMessageChunk, this), |
| ]; |
| } |
| |
| /** |
| * @override |
| * @param {string} message |
| */ |
| sendMessage(message) { |
| if (!this._disconnected) |
| InspectorFrontendHost.sendMessageToBackend(message); |
| } |
| |
| /** |
| * @param {!Common.Event} event |
| */ |
| _dispatchMessage(event) { |
| this._onMessage.call(null, /** @type {string} */ (event.data)); |
| } |
| |
| /** |
| * @param {!Common.Event} event |
| */ |
| _dispatchMessageChunk(event) { |
| const messageChunk = /** @type {string} */ (event.data['messageChunk']); |
| const messageSize = /** @type {number} */ (event.data['messageSize']); |
| if (messageSize) { |
| this._messageBuffer = ''; |
| this._messageSize = messageSize; |
| } |
| this._messageBuffer += messageChunk; |
| if (this._messageBuffer.length === this._messageSize) { |
| this._onMessage.call(null, this._messageBuffer); |
| this._messageBuffer = ''; |
| this._messageSize = 0; |
| } |
| } |
| |
| /** |
| * @override |
| * @return {!Promise} |
| */ |
| disconnect() { |
| const onDisconnect = this._onDisconnect; |
| Common.EventTarget.removeEventListeners(this._eventListeners); |
| this._onDisconnect = null; |
| this._onMessage = null; |
| this._disconnected = true; |
| |
| let fulfill; |
| const promise = new Promise(f => fulfill = f); |
| InspectorFrontendHost.reattach(() => { |
| onDisconnect.call(null, 'force disconnect'); |
| fulfill(); |
| }); |
| return promise; |
| } |
| }; |
| |
| /** |
| * @implements {Protocol.InspectorBackend.Connection} |
| * @unrestricted |
| */ |
| SDK.WebSocketConnection = class { |
| /** |
| * @param {string} url |
| * @param {function()} onWebSocketDisconnect |
| * @param {!Protocol.InspectorBackend.Connection.Params} params |
| */ |
| constructor(url, onWebSocketDisconnect, params) { |
| this._socket = new WebSocket(url); |
| this._socket.onerror = this._onError.bind(this); |
| this._socket.onopen = this._onOpen.bind(this); |
| this._socket.onmessage = messageEvent => params.onMessage.call(null, /** @type {string} */ (messageEvent.data)); |
| this._socket.onclose = this._onClose.bind(this); |
| |
| this._onDisconnect = params.onDisconnect; |
| this._onWebSocketDisconnect = onWebSocketDisconnect; |
| this._connected = false; |
| this._messages = []; |
| } |
| |
| _onError() { |
| this._onWebSocketDisconnect.call(null); |
| // This is called if error occurred while connecting. |
| this._onDisconnect.call(null, 'connection failed'); |
| this._close(); |
| } |
| |
| _onOpen() { |
| this._socket.onerror = console.error; |
| this._connected = true; |
| for (const message of this._messages) |
| this._socket.send(message); |
| this._messages = []; |
| } |
| |
| _onClose() { |
| this._onWebSocketDisconnect.call(null); |
| this._onDisconnect.call(null, 'websocket closed'); |
| this._close(); |
| } |
| |
| /** |
| * @param {function()=} callback |
| */ |
| _close(callback) { |
| this._socket.onerror = null; |
| this._socket.onopen = null; |
| this._socket.onclose = callback || null; |
| this._socket.onmessage = null; |
| this._socket.close(); |
| this._socket = null; |
| this._onWebSocketDisconnect = null; |
| } |
| |
| /** |
| * @override |
| * @param {string} message |
| */ |
| sendMessage(message) { |
| if (this._connected) |
| this._socket.send(message); |
| else |
| this._messages.push(message); |
| } |
| |
| /** |
| * @override |
| * @return {!Promise} |
| */ |
| disconnect() { |
| let fulfill; |
| const promise = new Promise(f => fulfill = f); |
| this._close(() => { |
| this._onDisconnect.call(null, 'force disconnect'); |
| fulfill(); |
| }); |
| return promise; |
| } |
| }; |
| |
| /** |
| * @implements {Protocol.InspectorBackend.Connection} |
| * @unrestricted |
| */ |
| SDK.StubConnection = class { |
| /** |
| * @param {!Protocol.InspectorBackend.Connection.Params} params |
| */ |
| constructor(params) { |
| this._onMessage = params.onMessage; |
| this._onDisconnect = params.onDisconnect; |
| } |
| |
| /** |
| * @override |
| * @param {string} message |
| */ |
| sendMessage(message) { |
| setTimeout(this._respondWithError.bind(this, message), 0); |
| } |
| |
| /** |
| * @param {string} message |
| */ |
| _respondWithError(message) { |
| const messageObject = JSON.parse(message); |
| const error = { |
| message: 'This is a stub connection, can\'t dispatch message.', |
| code: Protocol.InspectorBackend.DevToolsStubErrorCode, |
| data: messageObject |
| }; |
| this._onMessage.call(null, {id: messageObject.id, error: error}); |
| } |
| |
| /** |
| * @override |
| * @return {!Promise} |
| */ |
| disconnect() { |
| this._onDisconnect.call(null, 'force disconnect'); |
| this._onDisconnect = null; |
| this._onMessage = null; |
| return Promise.resolve(); |
| } |
| }; |
| |
| /** |
| * @implements {Protocol.InspectorBackend.Connection} |
| */ |
| SDK.ChildConnection = class { |
| /** |
| * @param {!Protocol.TargetAgent} agent |
| * @param {string} sessionId |
| * @param {!Protocol.InspectorBackend.Connection.Params} params |
| */ |
| constructor(agent, sessionId, params) { |
| this._agent = agent; |
| this._sessionId = sessionId; |
| this.onMessage = params.onMessage; |
| this.onDisconnect = params.onDisconnect; |
| } |
| |
| /** |
| * @override |
| * @param {string} message |
| */ |
| sendMessage(message) { |
| this._agent.sendMessageToTarget(message, this._sessionId); |
| } |
| |
| /** |
| * @override |
| * @return {!Promise} |
| */ |
| disconnect() { |
| throw 'Not implemented'; |
| } |
| }; |
| |
| /** |
| * @param {!Protocol.InspectorBackend.Connection.Params} params |
| * @param {function()} connectionLostCallback |
| * @return {!Protocol.InspectorBackend.Connection} |
| */ |
| SDK.createMainConnection = function(params, connectionLostCallback) { |
| const wsParam = Runtime.queryParam('ws'); |
| const wssParam = Runtime.queryParam('wss'); |
| |
| if (wsParam || wssParam) { |
| const ws = wsParam ? `ws://${wsParam}` : `wss://${wssParam}`; |
| return new SDK.WebSocketConnection(ws, connectionLostCallback, params); |
| } |
| |
| if (InspectorFrontendHost.isHostedMode()) |
| return new SDK.StubConnection(params); |
| return new SDK.MainConnection(params); |
| }; |