mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 10:11:03 +00:00
494 lines
15 KiB
JavaScript
494 lines
15 KiB
JavaScript
import debug from 'debug';
|
|
import reqwest from 'reqwest';
|
|
import { defaults } from 'lodash';
|
|
import { Map as makeMap, List } from 'immutable';
|
|
|
|
import {
|
|
blurSearch, clearControlError, closeWebsocket, openWebsocket, receiveError,
|
|
receiveApiDetails, receiveNodesDelta, receiveNodeDetails, receiveControlError,
|
|
receiveControlNodeRemoved, receiveControlPipe, receiveControlPipeStatus,
|
|
receiveControlSuccess, receiveTopologies, receiveNotFound,
|
|
receiveNodesForTopology, receiveNodes,
|
|
} from '../actions/app-actions';
|
|
|
|
import { getCurrentTopologyUrl } from '../utils/topology-utils';
|
|
import { layersTopologyIdsSelector } from '../selectors/resource-view/layout';
|
|
import { activeTopologyOptionsSelector } from '../selectors/topology';
|
|
import { isPausedSelector } from '../selectors/time-travel';
|
|
|
|
import { API_REFRESH_INTERVAL, TOPOLOGY_REFRESH_INTERVAL } from '../constants/timer';
|
|
|
|
const log = debug('scope:web-api-utils');
|
|
|
|
const reconnectTimerInterval = 5000;
|
|
const updateFrequency = '5s';
|
|
const FIRST_RENDER_TOO_LONG_THRESHOLD = 100; // ms
|
|
const csrfToken = (() => {
|
|
// Check for token at window level or parent level (for iframe);
|
|
/* eslint-disable no-underscore-dangle */
|
|
const token = typeof window !== 'undefined'
|
|
? window.__WEAVEWORKS_CSRF_TOKEN || window.parent.__WEAVEWORKS_CSRF_TOKEN
|
|
: null;
|
|
/* eslint-enable no-underscore-dangle */
|
|
if (!token || token === '$__CSRF_TOKEN_PLACEHOLDER__') {
|
|
// Authfe did not replace the token in the static html.
|
|
return null;
|
|
}
|
|
|
|
return token;
|
|
})();
|
|
|
|
let socket;
|
|
let reconnectTimer = 0;
|
|
let topologyTimer = 0;
|
|
let apiDetailsTimer = 0;
|
|
let controlErrorTimer = 0;
|
|
let currentUrl = null;
|
|
let createWebsocketAt = null;
|
|
let firstMessageOnWebsocketAt = null;
|
|
let continuePolling = true;
|
|
|
|
|
|
export function buildUrlQuery(params = makeMap(), state) {
|
|
// Attach the time travel timestamp to every request to the backend.
|
|
params = params.set('timestamp', state.get('pausedAt'));
|
|
|
|
// Ignore the entries with values `null` or `undefined`.
|
|
return params.map((value, param) => {
|
|
if (value === undefined || value === null) return null;
|
|
if (List.isList(value)) {
|
|
value = value.join(',');
|
|
}
|
|
return `${param}=${value}`;
|
|
}).filter(s => s).join('&');
|
|
}
|
|
|
|
export function basePath(urlPath) {
|
|
//
|
|
// "/scope/terminal.html" -> "/scope"
|
|
// "/scope/" -> "/scope"
|
|
// "/scope" -> "/scope"
|
|
// "/" -> ""
|
|
//
|
|
const parts = urlPath.split('/');
|
|
// if the last item has a "." in it, e.g. foo.html...
|
|
if (parts[parts.length - 1].indexOf('.') !== -1) {
|
|
return parts.slice(0, -1).join('/');
|
|
}
|
|
return parts.join('/').replace(/\/$/, '');
|
|
}
|
|
|
|
export function basePathSlash(urlPath) {
|
|
//
|
|
// "/scope/terminal.html" -> "/scope/"
|
|
// "/scope/" -> "/scope/"
|
|
// "/scope" -> "/scope/"
|
|
// "/" -> "/"
|
|
//
|
|
return `${basePath(urlPath)}/`;
|
|
}
|
|
|
|
export function getApiPath(pathname = window.location.pathname) {
|
|
if (process.env.SCOPE_API_PREFIX) {
|
|
return basePath(`${process.env.SCOPE_API_PREFIX}${pathname}`);
|
|
}
|
|
|
|
return basePath(pathname);
|
|
}
|
|
|
|
function topologiesUrl(state) {
|
|
const activeTopologyOptions = activeTopologyOptionsSelector(state);
|
|
const optionsQuery = buildUrlQuery(activeTopologyOptions, state);
|
|
return `${getApiPath()}/api/topology?${optionsQuery}`;
|
|
}
|
|
|
|
export function getWebsocketUrl(host = window.location.host, pathname = window.location.pathname) {
|
|
const wsProto = window.location.protocol === 'https:' ? 'wss' : 'ws';
|
|
return `${wsProto}://${host}${process.env.SCOPE_API_PREFIX || ''}${basePath(pathname)}`;
|
|
}
|
|
|
|
function buildWebsocketUrl(topologyUrl, topologyOptions = makeMap(), state) {
|
|
topologyOptions = topologyOptions.set('t', updateFrequency);
|
|
const optionsQuery = buildUrlQuery(topologyOptions, state);
|
|
return `${getWebsocketUrl()}${topologyUrl}/ws?${optionsQuery}`;
|
|
}
|
|
|
|
function createWebsocket(websocketUrl, getState, dispatch) {
|
|
if (socket) {
|
|
socket.onclose = null;
|
|
socket.onerror = null;
|
|
socket.close();
|
|
// onclose() is not called, but that's fine since we're opening a new one
|
|
// right away
|
|
}
|
|
|
|
// profiling
|
|
createWebsocketAt = new Date();
|
|
firstMessageOnWebsocketAt = null;
|
|
|
|
socket = new WebSocket(websocketUrl);
|
|
|
|
socket.onopen = () => {
|
|
log(`Opening websocket to ${websocketUrl}`);
|
|
dispatch(openWebsocket());
|
|
};
|
|
|
|
socket.onclose = () => {
|
|
clearTimeout(reconnectTimer);
|
|
log(`Closing websocket to ${websocketUrl}`, socket.readyState);
|
|
socket = null;
|
|
dispatch(closeWebsocket());
|
|
|
|
if (continuePolling && !isPausedSelector(getState())) {
|
|
reconnectTimer = setTimeout(() => {
|
|
createWebsocket(websocketUrl, getState, dispatch);
|
|
}, reconnectTimerInterval);
|
|
}
|
|
};
|
|
|
|
socket.onerror = () => {
|
|
log(`Error in websocket to ${websocketUrl}`);
|
|
dispatch(receiveError(websocketUrl));
|
|
};
|
|
|
|
socket.onmessage = (event) => {
|
|
const msg = JSON.parse(event.data);
|
|
dispatch(receiveNodesDelta(msg));
|
|
|
|
// profiling (receiveNodesDelta triggers synchronous render)
|
|
if (!firstMessageOnWebsocketAt) {
|
|
firstMessageOnWebsocketAt = new Date();
|
|
const timeToFirstMessage = firstMessageOnWebsocketAt - createWebsocketAt;
|
|
if (timeToFirstMessage > FIRST_RENDER_TOO_LONG_THRESHOLD) {
|
|
log(
|
|
'Time (ms) to first nodes render after websocket was created',
|
|
firstMessageOnWebsocketAt - createWebsocketAt
|
|
);
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
/**
|
|
* XHR wrapper. Applies a CSRF token (if it exists) and content-type to all requests.
|
|
* Any opts that get passed in will override the defaults.
|
|
*/
|
|
function doRequest(opts) {
|
|
const config = defaults(opts, {
|
|
contentType: 'application/json',
|
|
type: 'json'
|
|
});
|
|
if (csrfToken) {
|
|
config.headers = Object.assign({}, config.headers, { 'X-CSRF-Token': csrfToken });
|
|
}
|
|
|
|
return reqwest(config);
|
|
}
|
|
|
|
/**
|
|
* Does a one-time fetch of all the nodes for a custom list of topologies.
|
|
*/
|
|
function getNodesForTopologies(state, dispatch, topologyIds, topologyOptions = makeMap()) {
|
|
// fetch sequentially
|
|
state.get('topologyUrlsById')
|
|
.filter((_, topologyId) => topologyIds.contains(topologyId))
|
|
.reduce(
|
|
(sequence, topologyUrl, topologyId) => sequence
|
|
.then(() => {
|
|
const optionsQuery = buildUrlQuery(topologyOptions.get(topologyId), state);
|
|
return doRequest({ url: `${getApiPath()}${topologyUrl}?${optionsQuery}` });
|
|
})
|
|
.then(json => dispatch(receiveNodesForTopology(json.nodes, topologyId))),
|
|
Promise.resolve()
|
|
);
|
|
}
|
|
|
|
function getNodesOnce(getState, dispatch) {
|
|
const state = getState();
|
|
const topologyUrl = getCurrentTopologyUrl(state);
|
|
const topologyOptions = activeTopologyOptionsSelector(state);
|
|
const optionsQuery = buildUrlQuery(topologyOptions, state);
|
|
const url = `${getApiPath()}${topologyUrl}?${optionsQuery}`;
|
|
doRequest({
|
|
url,
|
|
success: (res) => {
|
|
dispatch(receiveNodes(res.nodes));
|
|
},
|
|
error: (req) => {
|
|
log(`Error in nodes request: ${req.responseText}`);
|
|
dispatch(receiveError(url));
|
|
}
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Gets nodes for all topologies (for search).
|
|
*/
|
|
export function getAllNodes(state, dispatch) {
|
|
const topologyOptions = state.get('topologyOptions');
|
|
const topologyIds = state.get('topologyUrlsById').keySeq();
|
|
getNodesForTopologies(state, dispatch, topologyIds, topologyOptions);
|
|
}
|
|
|
|
/**
|
|
* One-time update of all the nodes of topologies that appear in the current resource view.
|
|
* TODO: Replace the one-time snapshot with periodic polling.
|
|
*/
|
|
export function getResourceViewNodesSnapshot(state, dispatch) {
|
|
const topologyIds = layersTopologyIdsSelector(state);
|
|
getNodesForTopologies(state, dispatch, topologyIds);
|
|
}
|
|
|
|
function pollTopologies(getState, dispatch, initialPoll = false) {
|
|
// Used to resume polling when navigating between pages in Weave Cloud.
|
|
continuePolling = initialPoll === true ? true : continuePolling;
|
|
clearTimeout(topologyTimer);
|
|
// NOTE: getState is called every time to make sure the up-to-date state is used.
|
|
const url = topologiesUrl(getState());
|
|
doRequest({
|
|
url,
|
|
success: (res) => {
|
|
if (continuePolling && !isPausedSelector(getState())) {
|
|
dispatch(receiveTopologies(res));
|
|
topologyTimer = setTimeout(() => {
|
|
pollTopologies(getState, dispatch);
|
|
}, TOPOLOGY_REFRESH_INTERVAL);
|
|
}
|
|
},
|
|
error: (req) => {
|
|
log(`Error in topology request: ${req.responseText}`);
|
|
dispatch(receiveError(url));
|
|
// Only retry in stand-alone mode
|
|
if (continuePolling && !isPausedSelector(getState())) {
|
|
topologyTimer = setTimeout(() => {
|
|
pollTopologies(getState, dispatch);
|
|
}, TOPOLOGY_REFRESH_INTERVAL);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
function getTopologiesOnce(getState, dispatch) {
|
|
const url = topologiesUrl(getState());
|
|
doRequest({
|
|
url,
|
|
success: (res) => {
|
|
dispatch(receiveTopologies(res));
|
|
},
|
|
error: (req) => {
|
|
log(`Error in topology request: ${req.responseText}`);
|
|
dispatch(receiveError(url));
|
|
}
|
|
});
|
|
}
|
|
|
|
function updateWebsocketChannel(getState, dispatch, forceRequest) {
|
|
const topologyUrl = getCurrentTopologyUrl(getState());
|
|
const topologyOptions = activeTopologyOptionsSelector(getState());
|
|
const websocketUrl = buildWebsocketUrl(topologyUrl, topologyOptions, getState());
|
|
// Only recreate websocket if url changed or if forced (weave cloud instance reload);
|
|
const isNewUrl = websocketUrl !== currentUrl;
|
|
// `topologyUrl` can be undefined initially, so only create a socket if it is truthy
|
|
// and no socket exists, or if we get a new url.
|
|
if (topologyUrl && (!socket || isNewUrl || forceRequest)) {
|
|
createWebsocket(websocketUrl, getState, dispatch);
|
|
currentUrl = websocketUrl;
|
|
}
|
|
}
|
|
|
|
export function getNodeDetails(getState, dispatch) {
|
|
const state = getState();
|
|
const nodeMap = state.get('nodeDetails');
|
|
const topologyUrlsById = state.get('topologyUrlsById');
|
|
const currentTopologyId = state.get('currentTopologyId');
|
|
const requestTimestamp = state.get('pausedAt');
|
|
|
|
// get details for all opened nodes
|
|
const obj = nodeMap.last();
|
|
if (obj && topologyUrlsById.has(obj.topologyId)) {
|
|
const topologyUrl = topologyUrlsById.get(obj.topologyId);
|
|
let urlComponents = [getApiPath(), topologyUrl, '/', encodeURIComponent(obj.id)];
|
|
|
|
// Only forward filters for nodes in the current topology.
|
|
const topologyOptions = currentTopologyId === obj.topologyId
|
|
? activeTopologyOptionsSelector(state) : makeMap();
|
|
|
|
const query = buildUrlQuery(topologyOptions, state);
|
|
if (query) {
|
|
urlComponents = urlComponents.concat(['?', query]);
|
|
}
|
|
const url = urlComponents.join('');
|
|
|
|
doRequest({
|
|
url,
|
|
success: (res) => {
|
|
// make sure node is still selected
|
|
if (nodeMap.has(res.node.id)) {
|
|
dispatch(receiveNodeDetails(res.node, requestTimestamp));
|
|
}
|
|
},
|
|
error: (err) => {
|
|
log(`Error in node details request: ${err.responseText}`);
|
|
// dont treat missing node as error
|
|
if (err.status === 404) {
|
|
dispatch(receiveNotFound(obj.id, requestTimestamp));
|
|
} else {
|
|
dispatch(receiveError(topologyUrl));
|
|
}
|
|
}
|
|
});
|
|
} else if (obj) {
|
|
log('No details or url found for ', obj);
|
|
}
|
|
}
|
|
|
|
export function getTopologies(getState, dispatch, forceRequest) {
|
|
if (isPausedSelector(getState())) {
|
|
getTopologiesOnce(getState, dispatch);
|
|
} else {
|
|
pollTopologies(getState, dispatch, forceRequest);
|
|
}
|
|
}
|
|
|
|
export function getNodes(getState, dispatch, forceRequest = false) {
|
|
if (isPausedSelector(getState())) {
|
|
getNodesOnce(getState, dispatch);
|
|
} else {
|
|
updateWebsocketChannel(getState, dispatch, forceRequest);
|
|
}
|
|
getNodeDetails(getState, dispatch);
|
|
}
|
|
|
|
export function getApiDetails(dispatch) {
|
|
clearTimeout(apiDetailsTimer);
|
|
const url = `${getApiPath()}/api`;
|
|
doRequest({
|
|
url,
|
|
success: (res) => {
|
|
dispatch(receiveApiDetails(res));
|
|
if (continuePolling) {
|
|
apiDetailsTimer = setTimeout(() => {
|
|
getApiDetails(dispatch);
|
|
}, API_REFRESH_INTERVAL);
|
|
}
|
|
},
|
|
error: (req) => {
|
|
log(`Error in api details request: ${req.responseText}`);
|
|
receiveError(url);
|
|
if (continuePolling) {
|
|
apiDetailsTimer = setTimeout(() => {
|
|
getApiDetails(dispatch);
|
|
}, API_REFRESH_INTERVAL / 2);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
export function doControlRequest(nodeId, control, dispatch) {
|
|
clearTimeout(controlErrorTimer);
|
|
const url = `${getApiPath()}/api/control/${encodeURIComponent(control.probeId)}/`
|
|
+ `${encodeURIComponent(control.nodeId)}/${control.id}`;
|
|
doRequest({
|
|
method: 'POST',
|
|
url,
|
|
success: (res) => {
|
|
dispatch(receiveControlSuccess(nodeId));
|
|
if (res) {
|
|
if (res.pipe) {
|
|
dispatch(blurSearch());
|
|
const resizeTtyControl = res.resize_tty_control &&
|
|
{id: res.resize_tty_control, probeId: control.probeId, nodeId: control.nodeId};
|
|
dispatch(receiveControlPipe(
|
|
res.pipe,
|
|
nodeId,
|
|
res.raw_tty,
|
|
resizeTtyControl,
|
|
control
|
|
));
|
|
}
|
|
if (res.removedNode) {
|
|
dispatch(receiveControlNodeRemoved(nodeId));
|
|
}
|
|
}
|
|
},
|
|
error: (err) => {
|
|
dispatch(receiveControlError(nodeId, err.response));
|
|
controlErrorTimer = setTimeout(() => {
|
|
dispatch(clearControlError(nodeId));
|
|
}, 10000);
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
export function doResizeTty(pipeId, control, cols, rows) {
|
|
const url = `${getApiPath()}/api/control/${encodeURIComponent(control.probeId)}/`
|
|
+ `${encodeURIComponent(control.nodeId)}/${control.id}`;
|
|
|
|
return doRequest({
|
|
method: 'POST',
|
|
url,
|
|
data: JSON.stringify({pipeID: pipeId, width: cols.toString(), height: rows.toString()}),
|
|
})
|
|
.fail((err) => {
|
|
log(`Error resizing pipe: ${err}`);
|
|
});
|
|
}
|
|
|
|
|
|
export function deletePipe(pipeId, dispatch) {
|
|
const url = `${getApiPath()}/api/pipe/${encodeURIComponent(pipeId)}`;
|
|
doRequest({
|
|
method: 'DELETE',
|
|
url,
|
|
success: () => {
|
|
log('Closed the pipe!');
|
|
},
|
|
error: (err) => {
|
|
log(`Error closing pipe:${err}`);
|
|
dispatch(receiveError(url));
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
export function getPipeStatus(pipeId, dispatch) {
|
|
const url = `${getApiPath()}/api/pipe/${encodeURIComponent(pipeId)}/check`;
|
|
doRequest({
|
|
method: 'GET',
|
|
url,
|
|
complete: (res) => {
|
|
const status = {
|
|
204: 'PIPE_ALIVE',
|
|
404: 'PIPE_DELETED'
|
|
}[res.status];
|
|
|
|
if (!status) {
|
|
log('Unexpected pipe status:', res.status);
|
|
return;
|
|
}
|
|
|
|
dispatch(receiveControlPipeStatus(pipeId, status));
|
|
}
|
|
});
|
|
}
|
|
|
|
export function stopPolling() {
|
|
clearTimeout(apiDetailsTimer);
|
|
clearTimeout(topologyTimer);
|
|
continuePolling = false;
|
|
}
|
|
|
|
export function teardownWebsockets() {
|
|
clearTimeout(reconnectTimer);
|
|
if (socket) {
|
|
socket.onerror = null;
|
|
socket.onclose = null;
|
|
socket.onmessage = null;
|
|
socket.onopen = null;
|
|
socket.close();
|
|
socket = null;
|
|
currentUrl = null;
|
|
}
|
|
}
|