-
Notifications
You must be signed in to change notification settings - Fork 127
/
socketMiddleware.ts
108 lines (88 loc) · 2.27 KB
/
socketMiddleware.ts
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import { Middleware } from 'redux';
import { RootState, AppThunkAction, AppThunkDispatch } from './../reducers';
import { getRobotStatus } from '../actions/status';
import {
connect,
receiveConnectionStatus,
receivePingTime,
} from '../actions/socket';
import {
CONNECT,
DISCONNECT,
GET_CONFIG,
GET_ROBOT_STATUS,
INIT_OP_MODE,
RECEIVE_GAMEPAD_STATE,
RECEIVE_ROBOT_STATUS,
SAVE_CONFIG,
START_OP_MODE,
STOP_OP_MODE,
} from '../types';
let socket: WebSocket;
let statusSentTime: number;
const robotStatusLoop = (): AppThunkAction => (
dispatch: AppThunkDispatch,
getState: () => RootState,
) => {
const { isConnected } = getState().socket;
if (!isConnected) {
return;
}
statusSentTime = Date.now();
dispatch(getRobotStatus());
setTimeout(() => {
dispatch(robotStatusLoop());
}, 1000);
};
const socketMiddleware: Middleware<Record<string, unknown>, RootState> = (
store,
) => (next) => (action) => {
switch (action.type) {
case CONNECT:
socket = new WebSocket(`ws://${action.host}:${action.port}`);
socket.onmessage = (evt) => {
const msg = JSON.parse(evt.data);
store.dispatch(msg);
};
socket.onopen = () => {
(store.dispatch as AppThunkDispatch)(receiveConnectionStatus(true));
(store.dispatch as AppThunkDispatch)(robotStatusLoop());
};
socket.onclose = () => {
(store.dispatch as AppThunkDispatch)(receiveConnectionStatus(false));
setTimeout(
() => store.dispatch(connect(action.host, action.port)),
500,
);
};
break;
case DISCONNECT:
socket.close();
break;
case RECEIVE_ROBOT_STATUS: {
const pingTime = Date.now() - statusSentTime;
store.dispatch(receivePingTime(pingTime));
next(action);
break;
}
// messages forwarded to the server
case RECEIVE_GAMEPAD_STATE:
case GET_ROBOT_STATUS:
case SAVE_CONFIG:
case GET_CONFIG:
case INIT_OP_MODE:
case START_OP_MODE:
case STOP_OP_MODE: {
const { isConnected } = store.getState().socket;
if (isConnected) {
socket.send(JSON.stringify(action));
}
next(action);
break;
}
default:
next(action);
break;
}
};
export default socketMiddleware;