@@ -8,13 +8,22 @@ import React, {
88import * as azimuth from 'azimuth-js';
99import Web3 from 'web3';
1010import Maybe from 'folktale/maybe';
11+import { includes } from 'lodash';
1112
1213import { CONTRACT_ADDRESSES } from '../lib/contracts';
1314import { NETWORK_TYPES } from '../lib/network';
1415import { isDevelopment } from '../lib/flags';
16+import { BRIDGE_ERROR } from '../lib/error';
1517
1618function _useNetwork(initialNetworkType = null) {
17− const [networkType, setNetworkType] = useState(initialNetworkType);
19+ const [networkType, _setNetworkType] = useState(initialNetworkType);
20+
21+ const setNetworkType = networkType => {
22+ if (!includes(NETWORK_TYPES, networkType)) {
23+ throw BRIDGE_ERROR.INVALID_NETWORK_TYPE;
24+ }
25+ _setNetworkType(networkType);
26+ };
1827
1928 const { web3, contracts } = useMemo(() => {
2029 if (networkType === NETWORK_TYPES.LOCAL) {