Mark McDowall
1 month ago
parent
6f871a1bfb
commit
c3f9cd12af
@ -1,422 +0,0 @@
|
|||||||
import * as signalR from '@microsoft/signalr/dist/browser/signalr.js';
|
|
||||||
import PropTypes from 'prop-types';
|
|
||||||
import { Component } from 'react';
|
|
||||||
import { connect } from 'react-redux';
|
|
||||||
import { createSelector } from 'reselect';
|
|
||||||
import { setAppValue, setVersion } from 'Store/Actions/appActions';
|
|
||||||
import { removeItem, update, updateItem } from 'Store/Actions/baseActions';
|
|
||||||
import { fetchCommands, finishCommand, updateCommand } from 'Store/Actions/commandActions';
|
|
||||||
import { fetchQueue, fetchQueueDetails } from 'Store/Actions/queueActions';
|
|
||||||
import { fetchRootFolders } from 'Store/Actions/rootFolderActions';
|
|
||||||
import { fetchSeries } from 'Store/Actions/seriesActions';
|
|
||||||
import { fetchQualityDefinitions } from 'Store/Actions/settingsActions';
|
|
||||||
import { fetchHealth } from 'Store/Actions/systemActions';
|
|
||||||
import { fetchTagDetails, fetchTags } from 'Store/Actions/tagActions';
|
|
||||||
import { repopulatePage } from 'Utilities/pagePopulator';
|
|
||||||
import titleCase from 'Utilities/String/titleCase';
|
|
||||||
|
|
||||||
function getHandlerName(name) {
|
|
||||||
name = titleCase(name);
|
|
||||||
name = name.replace('/', '');
|
|
||||||
|
|
||||||
return `handle${name}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function createMapStateToProps() {
|
|
||||||
return createSelector(
|
|
||||||
(state) => state.app.isReconnecting,
|
|
||||||
(state) => state.app.isDisconnected,
|
|
||||||
(state) => state.queue.paged.isPopulated,
|
|
||||||
(isReconnecting, isDisconnected, isQueuePopulated) => {
|
|
||||||
return {
|
|
||||||
isReconnecting,
|
|
||||||
isDisconnected,
|
|
||||||
isQueuePopulated
|
|
||||||
};
|
|
||||||
}
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
const mapDispatchToProps = {
|
|
||||||
dispatchFetchCommands: fetchCommands,
|
|
||||||
dispatchUpdateCommand: updateCommand,
|
|
||||||
dispatchFinishCommand: finishCommand,
|
|
||||||
dispatchSetAppValue: setAppValue,
|
|
||||||
dispatchSetVersion: setVersion,
|
|
||||||
dispatchUpdate: update,
|
|
||||||
dispatchUpdateItem: updateItem,
|
|
||||||
dispatchRemoveItem: removeItem,
|
|
||||||
dispatchFetchHealth: fetchHealth,
|
|
||||||
dispatchFetchQualityDefinitions: fetchQualityDefinitions,
|
|
||||||
dispatchFetchQueue: fetchQueue,
|
|
||||||
dispatchFetchQueueDetails: fetchQueueDetails,
|
|
||||||
dispatchFetchRootFolders: fetchRootFolders,
|
|
||||||
dispatchFetchSeries: fetchSeries,
|
|
||||||
dispatchFetchTags: fetchTags,
|
|
||||||
dispatchFetchTagDetails: fetchTagDetails
|
|
||||||
};
|
|
||||||
|
|
||||||
function Logger(minimumLogLevel) {
|
|
||||||
this.minimumLogLevel = minimumLogLevel;
|
|
||||||
}
|
|
||||||
|
|
||||||
Logger.prototype.cleanse = function(message) {
|
|
||||||
const apikey = new RegExp(`access_token=${encodeURIComponent(window.Sonarr.apiKey)}`, 'g');
|
|
||||||
return message.replace(apikey, 'access_token=(removed)');
|
|
||||||
};
|
|
||||||
|
|
||||||
Logger.prototype.log = function(logLevel, message) {
|
|
||||||
// see https://github.com/aspnet/AspNetCore/blob/21c9e2cc954c10719878839cd3f766aca5f57b34/src/SignalR/clients/ts/signalr/src/Utils.ts#L147
|
|
||||||
if (logLevel >= this.minimumLogLevel) {
|
|
||||||
switch (logLevel) {
|
|
||||||
case signalR.LogLevel.Critical:
|
|
||||||
case signalR.LogLevel.Error:
|
|
||||||
console.error(`[signalR] ${signalR.LogLevel[logLevel]}: ${this.cleanse(message)}`);
|
|
||||||
break;
|
|
||||||
case signalR.LogLevel.Warning:
|
|
||||||
console.warn(`[signalR] ${signalR.LogLevel[logLevel]}: ${this.cleanse(message)}`);
|
|
||||||
break;
|
|
||||||
case signalR.LogLevel.Information:
|
|
||||||
console.info(`[signalR] ${signalR.LogLevel[logLevel]}: ${this.cleanse(message)}`);
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
// console.debug only goes to attached debuggers in Node, so we use console.log for Trace and Debug
|
|
||||||
console.log(`[signalR] ${signalR.LogLevel[logLevel]}: ${this.cleanse(message)}`);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
class SignalRConnector extends Component {
|
|
||||||
|
|
||||||
//
|
|
||||||
// Lifecycle
|
|
||||||
|
|
||||||
constructor(props, context) {
|
|
||||||
super(props, context);
|
|
||||||
|
|
||||||
this.connection = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
componentDidMount() {
|
|
||||||
console.log('[signalR] starting');
|
|
||||||
|
|
||||||
const url = `${window.Sonarr.urlBase}/signalr/messages`;
|
|
||||||
|
|
||||||
this.connection = new signalR.HubConnectionBuilder()
|
|
||||||
.configureLogging(new Logger(signalR.LogLevel.Information))
|
|
||||||
.withUrl(`${url}?access_token=${encodeURIComponent(window.Sonarr.apiKey)}`)
|
|
||||||
.withAutomaticReconnect({
|
|
||||||
nextRetryDelayInMilliseconds: (retryContext) => {
|
|
||||||
if (retryContext.elapsedMilliseconds > 180000) {
|
|
||||||
this.props.dispatchSetAppValue({ isDisconnected: true });
|
|
||||||
}
|
|
||||||
return Math.min(retryContext.previousRetryCount, 10) * 1000;
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.build();
|
|
||||||
|
|
||||||
this.connection.onreconnecting(this.onReconnecting);
|
|
||||||
this.connection.onreconnected(this.onReconnected);
|
|
||||||
this.connection.onclose(this.onClose);
|
|
||||||
|
|
||||||
this.connection.on('receiveMessage', this.onReceiveMessage);
|
|
||||||
|
|
||||||
this.connection.start().then(this.onStart, this.onStartFail);
|
|
||||||
}
|
|
||||||
|
|
||||||
componentWillUnmount() {
|
|
||||||
this.connection.stop();
|
|
||||||
this.connection = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
|
||||||
// Control
|
|
||||||
handleMessage = (message) => {
|
|
||||||
const {
|
|
||||||
name,
|
|
||||||
body
|
|
||||||
} = message;
|
|
||||||
|
|
||||||
const handler = this[getHandlerName(name)];
|
|
||||||
|
|
||||||
if (handler) {
|
|
||||||
handler(body);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
console.error(`signalR: Unable to find handler for ${name}`);
|
|
||||||
};
|
|
||||||
|
|
||||||
handleCalendar = (body) => {
|
|
||||||
if (body.action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({
|
|
||||||
section: 'calendar',
|
|
||||||
updateOnly: true,
|
|
||||||
...body.resource
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleCommand = (body) => {
|
|
||||||
if (body.action === 'sync') {
|
|
||||||
this.props.dispatchFetchCommands();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const resource = body.resource;
|
|
||||||
const status = resource.status;
|
|
||||||
|
|
||||||
// Both successful and failed commands need to be
|
|
||||||
// completed, otherwise they spin until they time out.
|
|
||||||
|
|
||||||
if (status === 'completed' || status === 'failed') {
|
|
||||||
this.props.dispatchFinishCommand(resource);
|
|
||||||
} else {
|
|
||||||
this.props.dispatchUpdateCommand(resource);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleEpisode = (body) => {
|
|
||||||
if (body.action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({
|
|
||||||
section: 'episodes',
|
|
||||||
updateOnly: true,
|
|
||||||
...body.resource
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleEpisodefile = (body) => {
|
|
||||||
const section = 'episodeFiles';
|
|
||||||
|
|
||||||
if (body.action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...body.resource });
|
|
||||||
|
|
||||||
// Repopulate the page to handle recently imported file
|
|
||||||
repopulatePage('episodeFileUpdated');
|
|
||||||
} else if (body.action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: body.resource.id });
|
|
||||||
|
|
||||||
repopulatePage('episodeFileDeleted');
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleDownloadclient = ({ action, resource }) => {
|
|
||||||
const section = 'settings.downloadClients';
|
|
||||||
|
|
||||||
if (action === 'created' || action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...resource });
|
|
||||||
} else if (action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: resource.id });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleHealth = () => {
|
|
||||||
this.props.dispatchFetchHealth();
|
|
||||||
};
|
|
||||||
|
|
||||||
handleImportlist = ({ action, resource }) => {
|
|
||||||
const section = 'settings.importLists';
|
|
||||||
|
|
||||||
if (action === 'created' || action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...resource });
|
|
||||||
} else if (action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: resource.id });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleIndexer = ({ action, resource }) => {
|
|
||||||
const section = 'settings.indexers';
|
|
||||||
|
|
||||||
if (action === 'created' || action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...resource });
|
|
||||||
} else if (action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: resource.id });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleMetadata = ({ action, resource }) => {
|
|
||||||
const section = 'settings.metadata';
|
|
||||||
|
|
||||||
if (action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...resource });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleNotification = ({ action, resource }) => {
|
|
||||||
const section = 'settings.notifications';
|
|
||||||
|
|
||||||
if (action === 'created' || action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...resource });
|
|
||||||
} else if (action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: resource.id });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleSeries = (body) => {
|
|
||||||
const action = body.action;
|
|
||||||
const section = 'series';
|
|
||||||
|
|
||||||
if (action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({ section, ...body.resource });
|
|
||||||
|
|
||||||
repopulatePage('seriesUpdated');
|
|
||||||
} else if (action === 'deleted') {
|
|
||||||
this.props.dispatchRemoveItem({ section, id: body.resource.id });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleQualitydefinition = () => {
|
|
||||||
this.props.dispatchFetchQualityDefinitions();
|
|
||||||
};
|
|
||||||
|
|
||||||
handleQueue = () => {
|
|
||||||
if (this.props.isQueuePopulated) {
|
|
||||||
this.props.dispatchFetchQueue();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleQueueDetails = () => {
|
|
||||||
this.props.dispatchFetchQueueDetails();
|
|
||||||
};
|
|
||||||
|
|
||||||
handleQueueStatus = (body) => {
|
|
||||||
this.props.dispatchUpdate({ section: 'queue.status', data: body.resource });
|
|
||||||
};
|
|
||||||
|
|
||||||
handleVersion = (body) => {
|
|
||||||
const version = body.version;
|
|
||||||
|
|
||||||
this.props.dispatchSetVersion({ version });
|
|
||||||
};
|
|
||||||
|
|
||||||
handleWantedCutoff = (body) => {
|
|
||||||
if (body.action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({
|
|
||||||
section: 'wanted.cutoffUnmet',
|
|
||||||
updateOnly: true,
|
|
||||||
...body.resource
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleWantedMissing = (body) => {
|
|
||||||
if (body.action === 'updated') {
|
|
||||||
this.props.dispatchUpdateItem({
|
|
||||||
section: 'wanted.missing',
|
|
||||||
updateOnly: true,
|
|
||||||
...body.resource
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
handleSystemTask = () => {
|
|
||||||
this.props.dispatchFetchCommands();
|
|
||||||
};
|
|
||||||
|
|
||||||
handleRootfolder = () => {
|
|
||||||
this.props.dispatchFetchRootFolders();
|
|
||||||
};
|
|
||||||
|
|
||||||
handleTag = (body) => {
|
|
||||||
if (body.action === 'sync') {
|
|
||||||
this.props.dispatchFetchTags();
|
|
||||||
this.props.dispatchFetchTagDetails();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
//
|
|
||||||
// Listeners
|
|
||||||
|
|
||||||
onStartFail = (error) => {
|
|
||||||
console.error('[signalR] failed to connect');
|
|
||||||
console.error(error);
|
|
||||||
|
|
||||||
this.props.dispatchSetAppValue({
|
|
||||||
isConnected: false,
|
|
||||||
isReconnecting: false,
|
|
||||||
isDisconnected: false,
|
|
||||||
isRestarting: false
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
onStart = () => {
|
|
||||||
console.debug('[signalR] connected');
|
|
||||||
|
|
||||||
this.props.dispatchSetAppValue({
|
|
||||||
isConnected: true,
|
|
||||||
isReconnecting: false,
|
|
||||||
isDisconnected: false,
|
|
||||||
isRestarting: false
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
onReconnecting = () => {
|
|
||||||
this.props.dispatchSetAppValue({ isReconnecting: true });
|
|
||||||
};
|
|
||||||
|
|
||||||
onReconnected = () => {
|
|
||||||
|
|
||||||
const {
|
|
||||||
dispatchFetchCommands,
|
|
||||||
dispatchFetchSeries,
|
|
||||||
dispatchSetAppValue
|
|
||||||
} = this.props;
|
|
||||||
|
|
||||||
dispatchSetAppValue({
|
|
||||||
isConnected: true,
|
|
||||||
isReconnecting: false,
|
|
||||||
isDisconnected: false,
|
|
||||||
isRestarting: false
|
|
||||||
});
|
|
||||||
|
|
||||||
// Repopulate the page (if a repopulator is set) to ensure things
|
|
||||||
// are in sync after reconnecting.
|
|
||||||
dispatchFetchSeries();
|
|
||||||
dispatchFetchCommands();
|
|
||||||
repopulatePage();
|
|
||||||
};
|
|
||||||
|
|
||||||
onClose = () => {
|
|
||||||
console.debug('[signalR] connection closed');
|
|
||||||
};
|
|
||||||
|
|
||||||
onReceiveMessage = (message) => {
|
|
||||||
console.debug('[signalR] received', message.name, message.body);
|
|
||||||
|
|
||||||
this.handleMessage(message);
|
|
||||||
};
|
|
||||||
|
|
||||||
//
|
|
||||||
// Render
|
|
||||||
|
|
||||||
render() {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
SignalRConnector.propTypes = {
|
|
||||||
isReconnecting: PropTypes.bool.isRequired,
|
|
||||||
isDisconnected: PropTypes.bool.isRequired,
|
|
||||||
isQueuePopulated: PropTypes.bool.isRequired,
|
|
||||||
dispatchFetchCommands: PropTypes.func.isRequired,
|
|
||||||
dispatchUpdateCommand: PropTypes.func.isRequired,
|
|
||||||
dispatchFinishCommand: PropTypes.func.isRequired,
|
|
||||||
dispatchSetAppValue: PropTypes.func.isRequired,
|
|
||||||
dispatchSetVersion: PropTypes.func.isRequired,
|
|
||||||
dispatchUpdate: PropTypes.func.isRequired,
|
|
||||||
dispatchUpdateItem: PropTypes.func.isRequired,
|
|
||||||
dispatchRemoveItem: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchHealth: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchQualityDefinitions: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchQueue: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchQueueDetails: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchRootFolders: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchSeries: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchTags: PropTypes.func.isRequired,
|
|
||||||
dispatchFetchTagDetails: PropTypes.func.isRequired
|
|
||||||
};
|
|
||||||
|
|
||||||
export default connect(createMapStateToProps, mapDispatchToProps)(SignalRConnector);
|
|
||||||
@ -0,0 +1,358 @@
|
|||||||
|
import signalR, { HubConnection } from '@microsoft/signalr';
|
||||||
|
import { useEffect, useRef } from 'react';
|
||||||
|
import { useDispatch, useSelector } from 'react-redux';
|
||||||
|
import ModelBase from 'App/ModelBase';
|
||||||
|
import AppState from 'App/State/AppState';
|
||||||
|
import Command from 'Commands/Command';
|
||||||
|
import { setAppValue, setVersion } from 'Store/Actions/appActions';
|
||||||
|
import { removeItem, update, updateItem } from 'Store/Actions/baseActions';
|
||||||
|
import {
|
||||||
|
fetchCommands,
|
||||||
|
finishCommand,
|
||||||
|
updateCommand,
|
||||||
|
} from 'Store/Actions/commandActions';
|
||||||
|
import { fetchQueue, fetchQueueDetails } from 'Store/Actions/queueActions';
|
||||||
|
import { fetchRootFolders } from 'Store/Actions/rootFolderActions';
|
||||||
|
import { fetchSeries } from 'Store/Actions/seriesActions';
|
||||||
|
import { fetchQualityDefinitions } from 'Store/Actions/settingsActions';
|
||||||
|
import { fetchHealth } from 'Store/Actions/systemActions';
|
||||||
|
import { fetchTagDetails, fetchTags } from 'Store/Actions/tagActions';
|
||||||
|
import { repopulatePage } from 'Utilities/pagePopulator';
|
||||||
|
import SignalRLogger from 'Utilities/SignalRLogger';
|
||||||
|
|
||||||
|
type SignalRAction = 'sync' | 'created' | 'updated' | 'deleted';
|
||||||
|
|
||||||
|
interface SignalRMessage {
|
||||||
|
name: string;
|
||||||
|
body: {
|
||||||
|
action: SignalRAction;
|
||||||
|
resource: ModelBase;
|
||||||
|
version: string;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function SignalRListener() {
|
||||||
|
const dispatch = useDispatch();
|
||||||
|
|
||||||
|
const isQueuePopulated = useSelector(
|
||||||
|
(state: AppState) => state.queue.paged.isPopulated
|
||||||
|
);
|
||||||
|
|
||||||
|
const connection = useRef<HubConnection | null>(null);
|
||||||
|
|
||||||
|
const handleStartFail = useRef((error: unknown) => {
|
||||||
|
console.error('[signalR] failed to connect');
|
||||||
|
console.error(error);
|
||||||
|
|
||||||
|
dispatch(
|
||||||
|
setAppValue({
|
||||||
|
isConnected: false,
|
||||||
|
isReconnecting: false,
|
||||||
|
isDisconnected: false,
|
||||||
|
isRestarting: false,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleStart = useRef(() => {
|
||||||
|
console.debug('[signalR] connected');
|
||||||
|
|
||||||
|
dispatch(
|
||||||
|
setAppValue({
|
||||||
|
isConnected: true,
|
||||||
|
isReconnecting: false,
|
||||||
|
isDisconnected: false,
|
||||||
|
isRestarting: false,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleReconnecting = useRef(() => {
|
||||||
|
dispatch(setAppValue({ isReconnecting: true }));
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleReconnected = useRef(() => {
|
||||||
|
dispatch(
|
||||||
|
setAppValue({
|
||||||
|
isConnected: true,
|
||||||
|
isReconnecting: false,
|
||||||
|
isDisconnected: false,
|
||||||
|
isRestarting: false,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
// Repopulate the page (if a repopulator is set) to ensure things
|
||||||
|
// are in sync after reconnecting.
|
||||||
|
dispatch(fetchSeries());
|
||||||
|
dispatch(fetchCommands());
|
||||||
|
repopulatePage();
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleClose = useRef(() => {
|
||||||
|
console.debug('[signalR] connection closed');
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleReceiveMessage = useRef((message: SignalRMessage) => {
|
||||||
|
console.debug('[signalR] received', message.name, message.body);
|
||||||
|
|
||||||
|
const { name, body } = message;
|
||||||
|
|
||||||
|
if (name === 'calendar') {
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(
|
||||||
|
updateItem({
|
||||||
|
section: 'calendar',
|
||||||
|
updateOnly: true,
|
||||||
|
...body.resource,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'command') {
|
||||||
|
if (body.action === 'sync') {
|
||||||
|
dispatch(fetchCommands());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const resource = body.resource as Command;
|
||||||
|
const status = resource.status;
|
||||||
|
|
||||||
|
// Both successful and failed commands need to be
|
||||||
|
// completed, otherwise they spin until they time out.
|
||||||
|
|
||||||
|
if (status === 'completed' || status === 'failed') {
|
||||||
|
dispatch(finishCommand(resource));
|
||||||
|
} else {
|
||||||
|
dispatch(updateCommand(resource));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'downloadclient') {
|
||||||
|
const section = 'settings.downloadClients';
|
||||||
|
|
||||||
|
if (body.action === 'created' || body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section, id: body.resource.id }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'episode') {
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(
|
||||||
|
updateItem({
|
||||||
|
section: 'episodes',
|
||||||
|
updateOnly: true,
|
||||||
|
...body.resource,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'episodefile') {
|
||||||
|
const section = 'episodeFiles';
|
||||||
|
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
|
||||||
|
// Repopulate the page to handle recently imported file
|
||||||
|
repopulatePage('episodeFileUpdated');
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section, id: body.resource.id }));
|
||||||
|
|
||||||
|
repopulatePage('episodeFileDeleted');
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'health') {
|
||||||
|
dispatch(fetchHealth());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'importlist') {
|
||||||
|
const section = 'settings.importLists';
|
||||||
|
|
||||||
|
if (body.action === 'created' || body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section, id: body.resource.id }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'indexer') {
|
||||||
|
const section = 'settings.indexers';
|
||||||
|
|
||||||
|
if (body.action === 'created' || body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section, id: body.resource.id }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'metadata') {
|
||||||
|
const section = 'settings.metadata';
|
||||||
|
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'notification') {
|
||||||
|
const section = 'settings.notifications';
|
||||||
|
|
||||||
|
if (body.action === 'created' || body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section, ...body.resource }));
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section, id: body.resource.id }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'qualitydefinition') {
|
||||||
|
dispatch(fetchQualityDefinitions());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'queue') {
|
||||||
|
if (isQueuePopulated) {
|
||||||
|
dispatch(fetchQueue());
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'queue/details') {
|
||||||
|
dispatch(fetchQueueDetails());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'queue/status') {
|
||||||
|
dispatch(update({ section: 'queue.status', data: body.resource }));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'rootfolder') {
|
||||||
|
dispatch(fetchRootFolders());
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'series') {
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(updateItem({ section: 'series', ...body.resource }));
|
||||||
|
|
||||||
|
repopulatePage('seriesUpdated');
|
||||||
|
} else if (body.action === 'deleted') {
|
||||||
|
dispatch(removeItem({ section: 'series', id: body.resource.id }));
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'system/task') {
|
||||||
|
dispatch(fetchCommands());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'tag') {
|
||||||
|
if (body.action === 'sync') {
|
||||||
|
dispatch(fetchTags());
|
||||||
|
dispatch(fetchTagDetails());
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'version') {
|
||||||
|
dispatch(setVersion({ version: body.version }));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'wanted/cutoff') {
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(
|
||||||
|
updateItem({
|
||||||
|
section: 'wanted.cutoffUnmet',
|
||||||
|
updateOnly: true,
|
||||||
|
...body.resource,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name === 'wanted/missing') {
|
||||||
|
if (body.action === 'updated') {
|
||||||
|
dispatch(
|
||||||
|
updateItem({
|
||||||
|
section: 'wanted.missing',
|
||||||
|
updateOnly: true,
|
||||||
|
...body.resource,
|
||||||
|
})
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.error(`signalR: Unable to find handler for ${name}`);
|
||||||
|
});
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
console.log('[signalR] starting');
|
||||||
|
|
||||||
|
const url = `${window.Sonarr.urlBase}/signalr/messages`;
|
||||||
|
|
||||||
|
connection.current = new signalR.HubConnectionBuilder()
|
||||||
|
.configureLogging(new SignalRLogger(signalR.LogLevel.Information))
|
||||||
|
.withUrl(
|
||||||
|
`${url}?access_token=${encodeURIComponent(window.Sonarr.apiKey)}`
|
||||||
|
)
|
||||||
|
.withAutomaticReconnect({
|
||||||
|
nextRetryDelayInMilliseconds: (retryContext) => {
|
||||||
|
if (retryContext.elapsedMilliseconds > 180000) {
|
||||||
|
dispatch(setAppValue({ isDisconnected: true }));
|
||||||
|
}
|
||||||
|
return Math.min(retryContext.previousRetryCount, 10) * 1000;
|
||||||
|
},
|
||||||
|
})
|
||||||
|
.build();
|
||||||
|
|
||||||
|
connection.current.onreconnecting(handleReconnecting.current);
|
||||||
|
connection.current.onreconnected(handleReconnected.current);
|
||||||
|
connection.current.onclose(handleClose.current);
|
||||||
|
|
||||||
|
connection.current.on('receiveMessage', handleReceiveMessage.current);
|
||||||
|
|
||||||
|
connection.current
|
||||||
|
.start()
|
||||||
|
.then(handleStart.current, handleStartFail.current);
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
connection.current?.stop();
|
||||||
|
connection.current = null;
|
||||||
|
};
|
||||||
|
}, [dispatch]);
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
export default SignalRListener;
|
||||||
@ -0,0 +1,47 @@
|
|||||||
|
import { LogLevel } from '@microsoft/signalr';
|
||||||
|
|
||||||
|
export default class SignalRLogger {
|
||||||
|
private _minimumLogLevel = LogLevel.Information;
|
||||||
|
|
||||||
|
constructor(minimumLogLevel: LogLevel) {
|
||||||
|
this._minimumLogLevel = minimumLogLevel;
|
||||||
|
}
|
||||||
|
|
||||||
|
public log(logLevel: LogLevel, message: string) {
|
||||||
|
// see https://github.com/aspnet/AspNetCore/blob/21c9e2cc954c10719878839cd3f766aca5f57b34/src/SignalR/clients/ts/signalr/src/Utils.ts#L147
|
||||||
|
if (logLevel >= this._minimumLogLevel) {
|
||||||
|
switch (logLevel) {
|
||||||
|
case LogLevel.Critical:
|
||||||
|
case LogLevel.Error:
|
||||||
|
console.error(
|
||||||
|
`[signalR] ${LogLevel[logLevel]}: ${this._cleanse(message)}`
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
case LogLevel.Warning:
|
||||||
|
console.warn(
|
||||||
|
`[signalR] ${LogLevel[logLevel]}: ${this._cleanse(message)}`
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
case LogLevel.Information:
|
||||||
|
console.info(
|
||||||
|
`[signalR] ${LogLevel[logLevel]}: ${this._cleanse(message)}`
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
// console.debug only goes to attached debuggers in Node, so we use console.log for Trace and Debug
|
||||||
|
console.log(
|
||||||
|
`[signalR] ${LogLevel[logLevel]}: ${this._cleanse(message)}`
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private _cleanse(message: string) {
|
||||||
|
const apikey = new RegExp(
|
||||||
|
`access_token=${encodeURIComponent(window.Sonarr.apiKey)}`,
|
||||||
|
'g'
|
||||||
|
);
|
||||||
|
return message.replace(apikey, 'access_token=(removed)');
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue