blob: 781194945e2cb4f08d5bbd69b751633a1da4a410 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
import { useRef } from 'react';
import { useDispatch, useSelector, connect } from 'react-redux';
import { useSnackbar } from 'notistack';
import { useHistory } from 'react-router-dom';
import { positionsActions, devicesActions, sessionActions } from './store';
import { useEffectAsync } from './reactHelper';
import { useTranslation } from './LocalizationProvider';
import { prefixString } from './common/stringUtils';
const SocketController = () => {
const dispatch = useDispatch();
const history = useHistory();
const t = useTranslation();
const { enqueueSnackbar } = useSnackbar();
const authenticated = useSelector((state) => !!state.session.user);
const socketRef = useRef();
const enqueueEvents = (events) => events.forEach((event) => enqueueSnackbar(t(prefixString('event', event.type))));
const connectSocket = () => {
const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
const socket = new WebSocket(`${protocol}//${window.location.host}/api/socket`);
socketRef.current = socket;
socket.onerror = () => {
setTimeout(() => connectSocket(), 60 * 1000);
};
socket.onmessage = (event) => {
const data = JSON.parse(event.data);
if (data.devices) {
dispatch(devicesActions.update(data.devices));
}
if (data.positions) {
dispatch(positionsActions.update(data.positions));
}
if (data.events) {
enqueueEvents(data.events);
}
};
};
useEffectAsync(async () => {
const response = await fetch('/api/server');
if (response.ok) {
dispatch(sessionActions.updateServer(await response.json()));
}
}, []);
useEffectAsync(async () => {
if (authenticated) {
const response = await fetch('/api/devices');
if (response.ok) {
dispatch(devicesActions.refresh(await response.json()));
}
connectSocket();
return () => {
const socket = socketRef.current;
if (socket) {
socket.close();
}
};
}
const response = await fetch('/api/session');
if (response.ok) {
dispatch(sessionActions.updateUser(await response.json()));
} else {
history.push('/login');
}
return null;
}, [authenticated]);
return null;
};
export default connect()(SocketController);
|
