meriadec
7 years ago
6 changed files with 8 additions and 483 deletions
@ -1,241 +0,0 @@ |
|||
// @flow
|
|||
|
|||
import { coinTypeForId } from 'internals/usb/wallet/accounts' |
|||
import ledger from 'ledger-test-library' |
|||
import bitcoin from 'bitcoinjs-lib' |
|||
import axios from 'axios' |
|||
import type { OperationRaw } from '@ledgerhq/live-common/lib/types' |
|||
|
|||
import groupBy from 'lodash/groupBy' |
|||
import noop from 'lodash/noop' |
|||
import uniqBy from 'lodash/uniqBy' |
|||
|
|||
const GAP_LIMIT_ADDRESSES = 20 |
|||
|
|||
export const networks = { |
|||
bitcoin: { |
|||
...bitcoin.networks.bitcoin, |
|||
family: 1, |
|||
}, |
|||
bitcoin_testnet: { |
|||
...bitcoin.networks.testnet, |
|||
family: 1, |
|||
}, |
|||
} |
|||
|
|||
export function computeOperation(addresses: Array<string>, accountId: string) { |
|||
return (t: Object) => { |
|||
const outputVal = t.outputs |
|||
.filter(o => addresses.includes(o.address)) |
|||
.reduce((acc, cur) => acc + cur.value, 0) |
|||
const inputVal = t.inputs |
|||
.filter(i => addresses.includes(i.address)) |
|||
.reduce((acc, cur) => acc + cur.value, 0) |
|||
const amount = outputVal - inputVal |
|||
return { |
|||
id: t.hash, |
|||
hash: t.hash, |
|||
address: t.amount > 0 ? t.inputs[0].address : t.outputs[0].address, |
|||
from: t.inputs.map(t => t.address), |
|||
to: t.outputs.map(t => t.address), |
|||
amount, |
|||
confirmations: t.confirmations, |
|||
date: t.received_at, |
|||
accountId, |
|||
blockHeight: t.block.height, |
|||
} |
|||
} |
|||
} |
|||
|
|||
export function getBalanceByDay(operations: OperationRaw[]) { |
|||
const txsByDate = groupBy(operations, tx => { |
|||
const [date] = new Date(tx.date).toISOString().split('T') |
|||
return date |
|||
}) |
|||
|
|||
let balance = 0 |
|||
|
|||
return Object.keys(txsByDate) |
|||
.sort() |
|||
.reduce((result, k) => { |
|||
const txs = txsByDate[k] |
|||
|
|||
balance += txs.reduce((r, v) => { |
|||
r += v.amount |
|||
return r |
|||
}, 0) |
|||
|
|||
result[k] = balance |
|||
|
|||
return result |
|||
}, {}) |
|||
} |
|||
|
|||
export async function getAccount({ |
|||
rootPath, |
|||
allAddresses = [], |
|||
allTxsHash = [], |
|||
currentIndex = 0, |
|||
hdnode, |
|||
segwit, |
|||
network, |
|||
currencyId, |
|||
accountId, |
|||
asyncDelay = 250, |
|||
onProgress = noop, |
|||
}: { |
|||
rootPath: string, |
|||
allAddresses?: Array<string>, |
|||
allTxsHash?: Array<string>, |
|||
currentIndex?: number, |
|||
hdnode: Object, |
|||
segwit: boolean, |
|||
currencyId: string, |
|||
accountId: string, |
|||
network: Object, |
|||
asyncDelay?: number, |
|||
onProgress?: Function, |
|||
}): Promise<any> { |
|||
const script = segwit ? parseInt(network.scriptHash, 10) : parseInt(network.pubKeyHash, 10) |
|||
|
|||
let balance = 0 |
|||
let operations = [] |
|||
let lastAddress = null |
|||
|
|||
const pubKeyToSegwitAddress = (pubKey, scriptVersion) => { |
|||
const script = [0x00, 0x14].concat(Array.from(bitcoin.crypto.hash160(pubKey))) |
|||
const hash160 = bitcoin.crypto.hash160(new Uint8Array(script)) |
|||
return bitcoin.address.toBase58Check(hash160, scriptVersion) |
|||
} |
|||
|
|||
const getPublicAddress = ({ hdnode, path, script, segwit }) => { |
|||
hdnode = hdnode.derivePath(path) |
|||
if (!segwit) { |
|||
return hdnode.getAddress().toString() |
|||
} |
|||
return pubKeyToSegwitAddress(hdnode.getPublicKeyBuffer(), script) |
|||
} |
|||
|
|||
const getPath = (type, index) => `${type === 'external' ? 0 : 1}/${index}` |
|||
|
|||
const getAddress = ({ type, index }) => { |
|||
const p = getPath(type, index) |
|||
return { |
|||
type, |
|||
index, |
|||
path: `${rootPath}/${p}`, |
|||
address: getPublicAddress({ hdnode, path: p, script, segwit }), |
|||
} |
|||
} |
|||
|
|||
const getAsyncAddress = params => |
|||
new Promise(resolve => setTimeout(() => resolve(getAddress(params)), asyncDelay)) |
|||
|
|||
const getLastAddress = (addresses, txs) => { |
|||
const txsAddresses = [...txs.inputs.map(t => t.address), ...txs.outputs.map(t => t.address)] |
|||
return addresses.find(a => txsAddresses.includes(a.address)) || null |
|||
} |
|||
|
|||
const nextPath = (index = 0) => |
|||
Array.from(new Array(GAP_LIMIT_ADDRESSES).keys()) |
|||
.reduce( |
|||
(promise, v) => |
|||
promise.then(async results => { |
|||
const result = await Promise.all([ |
|||
getAsyncAddress({ type: 'external', index: v + index }), |
|||
getAsyncAddress({ type: 'internal', index: v + index }), |
|||
]) |
|||
return [...results, result] |
|||
}), |
|||
Promise.resolve([]), |
|||
) |
|||
.then(async results => { |
|||
const addresses = results.reduce((result, v) => [...result, ...v], []) |
|||
const listAddresses = addresses.map(a => a.address) |
|||
|
|||
allAddresses = [...new Set([...allAddresses, ...listAddresses])] |
|||
|
|||
let txs = [] |
|||
|
|||
const operationsOpts = { coin_type: coinTypeForId(currencyId) } |
|||
|
|||
try { |
|||
txs = await ledger.getTransactions(listAddresses, operationsOpts) |
|||
txs = txs.filter(t => !allTxsHash.includes(t.hash)).reverse() |
|||
} catch (e) { |
|||
console.log('getOperations', e) // eslint-disable-line no-console
|
|||
} |
|||
|
|||
const hasOperations = txs.length > 0 |
|||
|
|||
if (hasOperations) { |
|||
const newOperations = txs.map(computeOperation(allAddresses, accountId)) |
|||
|
|||
const txHashs = operations.map(t => t.id) |
|||
|
|||
balance = newOperations |
|||
.filter(t => !txHashs.includes(t.id)) |
|||
.reduce((result, v) => result + v.amount, balance) |
|||
|
|||
lastAddress = getLastAddress(addresses, txs[0]) |
|||
|
|||
operations = uniqBy([...operations, ...newOperations], t => t.id) |
|||
|
|||
onProgress({ |
|||
balance, |
|||
operations, |
|||
balanceByDay: getBalanceByDay(operations), |
|||
}) |
|||
|
|||
return nextPath(index + (GAP_LIMIT_ADDRESSES - 1)) |
|||
} |
|||
|
|||
const { type, ...nextAddress } = |
|||
lastAddress !== null |
|||
? getAddress({ |
|||
type: 'external', |
|||
index: lastAddress.index + 1, |
|||
}) |
|||
: getAddress({ type: 'external', index: 0 }) |
|||
|
|||
// TODO: in the future, it should be done with the libc call
|
|||
const { |
|||
data: { height: blockHeight }, |
|||
} = await axios.get('https://api.ledgerwallet.com/blockchain/v2/btc_testnet/blocks/current') |
|||
|
|||
const account = { |
|||
...nextAddress, |
|||
currencyId, |
|||
addresses: operations.length > 0 ? allAddresses : [], |
|||
balance, |
|||
balanceByDay: getBalanceByDay(operations), |
|||
rootPath, |
|||
operations, |
|||
lastSyncDate: new Date(), |
|||
blockHeight, |
|||
} |
|||
|
|||
onProgress({ |
|||
...account, |
|||
finish: true, |
|||
}) |
|||
|
|||
return account |
|||
}) |
|||
|
|||
if (allAddresses.length === 0 && currentIndex > 0) { |
|||
for (let i = currentIndex; i--; ) { |
|||
allAddresses = [ |
|||
...allAddresses, |
|||
getAddress({ type: 'internal', index: i }).address, |
|||
getAddress({ type: 'external', index: i }).address, |
|||
] |
|||
} |
|||
} |
|||
|
|||
return nextPath(currentIndex) |
|||
} |
|||
|
|||
export function getHDNode({ xpub58, network }: { xpub58: string, network: Object }) { |
|||
return bitcoin.HDNode.fromBase58(xpub58, network) |
|||
} |
|||
@ -1,34 +1,7 @@ |
|||
// @flow
|
|||
|
|||
import { getAccount, getHDNode, networks } from 'helpers/btc' |
|||
|
|||
const network = networks[1] |
|||
|
|||
function syncAccount({ id, operations, ...currentAccount }) { |
|||
const hdnode = getHDNode({ xpub58: id, network }) |
|||
const allTxsHash = operations.map(t => t.id) |
|||
return getAccount({ hdnode, network, allTxsHash, segwit: true, ...currentAccount }).then( |
|||
account => ({ |
|||
id, |
|||
...account, |
|||
}), |
|||
) |
|||
} |
|||
|
|||
export default (send: Function) => ({ |
|||
all: async ({ accounts }: { accounts: Array<Object> }) => { |
|||
send('accounts.sync.progress', null, { kill: false }) |
|||
|
|||
try { |
|||
await Promise.all( |
|||
accounts.map(a => |
|||
syncAccount(a).then(account => send('account.sync.success', account, { kill: false })), |
|||
), |
|||
) |
|||
|
|||
send('accounts.sync.success') |
|||
} catch (err) { |
|||
send('accounts.sync.fail', err.stack || err) |
|||
} |
|||
all: () => { |
|||
setTimeout(() => send('accounts.sync.success'), 5e3) |
|||
}, |
|||
}) |
|||
|
|||
Loading…
Reference in new issue