vega-frontend-monorepo/libs/wallet/src/use-json-rpc-connect.ts
2022-12-09 12:08:26 +01:00

74 lines
2.3 KiB
TypeScript

import { useCallback, useState } from 'react';
import type { JsonRpcConnector } from './connectors';
import { ClientErrors } from './connectors';
import { WalletError } from './connectors';
import { useVegaWallet } from './use-vega-wallet';
export enum Status {
Idle = 'Idle',
CheckingVersion = 'CheckingVersion',
GettingChainId = 'GettingChainId',
Connecting = 'Connecting',
GettingPerms = 'GettingPerms',
ListingKeys = 'ListingKeys',
Connected = 'Connected',
Error = 'Error',
}
export const useJsonRpcConnect = (onConnect: () => void) => {
const { connect } = useVegaWallet();
const [status, setStatus] = useState(Status.Idle);
const [error, setError] = useState<WalletError | null>(null);
const attemptConnect = useCallback(
async (connector: JsonRpcConnector, appChainId: string) => {
try {
// Check that the running wallet is compatible with this connector
setStatus(Status.CheckingVersion);
await connector.checkCompat();
// Check if wallet is configured for the same chain as the app
setStatus(Status.GettingChainId);
// Dont throw in when cypress is running as trading app relies on
// mocks which result in a mismatch between chainId for app and
// chainId for wallet
if (!('Cypress' in window)) {
const chainIdResult = await connector.getChainId();
if (chainIdResult.chainID !== appChainId) {
// Throw wallet error for consitent error handling
throw ClientErrors.WRONG_NETWORK;
}
}
// Start connection flow. User will be prompted to select a wallet and enter
// its password in the wallet application, promise will resolve once successful
// or it will throw
setStatus(Status.Connecting);
await connector.connectWallet();
setStatus(Status.GettingPerms);
// Call connect in the wallet provider. The connector will be stored for
// future actions such as sending transactions
await connect(connector);
setStatus(Status.Connected);
onConnect();
} catch (err) {
if (err instanceof WalletError) {
setError(err);
}
setStatus(Status.Error);
}
},
[onConnect, connect]
);
return {
status,
error,
connect: attemptConnect,
};
};