@@ -8,13 +8,22 @@ import React, {
8import * as azimuth from 'azimuth-js';
8import * as azimuth from 'azimuth-js';
9import Web3 from 'web3';
9import Web3 from 'web3';
10import Maybe from 'folktale/maybe';
10import Maybe from 'folktale/maybe';
11+import { includes } from 'lodash';
12import { CONTRACT_ADDRESSES } from '../lib/contracts';
13import { CONTRACT_ADDRESSES } from '../lib/contracts';
13import { NETWORK_TYPES } from '../lib/network';
14import { NETWORK_TYPES } from '../lib/network';
14import { isDevelopment } from '../lib/flags';
15import { isDevelopment } from '../lib/flags';
16+import { BRIDGE_ERROR } from '../lib/error';
16function _useNetwork(initialNetworkType = null) {
18function _useNetwork(initialNetworkType = null) {
17− const [networkType, setNetworkType] = useState(initialNetworkType);
19+ const [networkType, _setNetworkType] = useState(initialNetworkType);
21+ const setNetworkType = networkType => {
22+ if (!includes(NETWORK_TYPES, networkType)) {
23+ throw BRIDGE_ERROR.INVALID_NETWORK_TYPE;
25+ _setNetworkType(networkType);
19 const { web3, contracts } = useMemo(() => {
28 const { web3, contracts } = useMemo(() => {
20 if (networkType === NETWORK_TYPES.LOCAL) {
29 if (networkType === NETWORK_TYPES.LOCAL) {