git.iko.soy/bridge
on 86fddd6 help sign in
bridge/src/lib/authToken.ts
102 lines 2.7 kB history raw
1import { crypto } from 'bitcoinjs-lib';
2import { ecdsaSign } from 'secp256k1';
3import Web3 from 'web3';
4import WalletConnect from '@walletconnect/client';
6import { WALLET_TYPES } from './constants';
7import { ledgerSignMessage } from './ledger';
8import { trezorSignMessage } from './trezor';
9import BridgeWallet from './types/BridgeWallet';
11const MESSAGE = 'Bridge Authentication Token';
13function signMessage(privateKey: Buffer) {
14 const msg = '\x19Ethereum Signed Message:\n' + MESSAGE.length + MESSAGE;
15 // #ecdsaSign requires a 32-byte buffer, hence sha256
16 const hashed = crypto.sha256(Buffer.from(msg));
17 const { signature } = ecdsaSign(Buffer.from(hashed), privateKey);
19 // add key recovery parameter
20 const ethSignature = new Uint8Array(65);
21 ethSignature.set(signature);
22 const v = (ethSignature[32] & 1) + 27;
23 ethSignature[64] = v;
25 return ethSignature;
28type MetamaskAuthTokenArgs = {
29 address: string;
30 web3: Web3;
31 walletType: symbol;
32};
34type LedgerAuthTokenArgs = {
35 walletHdPath: string;
36 walletType: symbol;
37};
39type TrezorAuthTokenArgs = LedgerAuthTokenArgs;
41type WalletConnectAuthTokenArgs = {
42 address: string;
43 connector: WalletConnect;
44 walletType: symbol;
45};
47type DefaultAuthTokenArgs = {
48 wallet: BridgeWallet;
49 walletType?: symbol;
50};
52type GetAuthTokenArgs =
53 | MetamaskAuthTokenArgs
54 | LedgerAuthTokenArgs
55 | TrezorAuthTokenArgs
56 | WalletConnectAuthTokenArgs
57 | DefaultAuthTokenArgs;
59const getMetamaskAuthToken = ({ address, web3 }: MetamaskAuthTokenArgs) => {
60 return web3.eth.personal.sign(MESSAGE, address, '');
61};
63const getLedgerAuthToken = ({ walletHdPath }: LedgerAuthTokenArgs) => {
64 return ledgerSignMessage(MESSAGE, walletHdPath);
65};
67const getTrezorAuthToken = ({ walletHdPath }: TrezorAuthTokenArgs) => {
68 return trezorSignMessage(MESSAGE, walletHdPath);
69};
71const getWalletConnectAuthToken = ({
72 address,
73 connector,
74}: WalletConnectAuthTokenArgs) => {
75 return connector.signPersonalMessage([MESSAGE, address]);
76};
78const getDefaultAuthToken = ({ wallet }: DefaultAuthTokenArgs) => {
79 const signature = signMessage(wallet.privateKey);
81 const token = `0x${Buffer.from(signature).toString('hex')}`;
83 return token;
84};
86export const getAuthToken = async ({
87 walletType,
88 ...args
89}: GetAuthTokenArgs) => {
90 switch (walletType) {
91 case WALLET_TYPES.METAMASK:
92 return getMetamaskAuthToken(args);
93 case WALLET_TYPES.LEDGER:
94 return getLedgerAuthToken(args);
95 case WALLET_TYPES.TREZOR:
96 return getTrezorAuthToken(args);
97 case WALLET_TYPES.WALLET_CONNECT:
98 return getWalletConnectAuthToken(args);
99 default:
100 return getDefaultAuthToken(args);
101 }
102};