git.iko.soy/bridge
on c19b2a8 help sign in
bridge/src/lib/starRelease.js
101 lines 3.2 kB history raw
1import { linearSR, conditionalSR } from 'azimuth-js';
3function mapRange(amount, cb) {
4 return [...Array(amount).keys()].map(cb);
5}
7// Preferentially withdraw from conditional contracts and use linear
8// to mop up the rest
9export function generateWithdrawTxs(contracts, batchLimits, amount, to) {
10 const [left, ...csrTxs] = generateConditionalWithdrawTxs(
11 contracts,
12 batchLimits,
13 amount,
14 to
15 );
16 const lsrTxs = generateLinearWithdrawTxs(contracts, left, to);
17 return [...csrTxs, ...lsrTxs];
20/**
21 * Recurse over batchLimits to generate a valid number of withdraw txs.
22 * amount may exceed the number of withdraw txs available in the consolidated SR contract.
23 *
24 * @param {object} contracts - The azimuth contracts object
25 * @param {array} batchLimits - An array of the maximum withdraw limit for each batch
26 * @param {amount} amount - The number of txs to be generated
27 * @param {string} to - The addresss to withdraw to
28 * @return {array} An array with the number of transactions missing as the head
29 * (i.e. requested amount exceeded available), and the generated transactions
30 * as the rest of the list
31 */
32function generateConditionalWithdrawTxs(
33 contracts,
34 batchLimits,
35 amount,
36 to,
37 offset = 0
38) {
39 // Successfully withdrawn all requested, none missing
40 if (amount <= 0) {
41 return [0];
42 }
43 // Return missing amount in head of list
44 if (batchLimits.length === 0) {
45 return [amount];
46 }
47 const [next, ...rest] = batchLimits;
48 const currentAmount = Math.min(amount, next);
49 return [
50 ...generateConditionalWithdrawTxs(
51 contracts,
52 rest,
53 amount - currentAmount,
54 to,
55 offset + 1
56 ),
57 ...mapRange(currentAmount, () =>
58 conditionalSR.withdrawTo(contracts, offset, to)
59 ),
60 ];
63function generateLinearWithdrawTxs(contracts, amount, to) {
64 return mapRange(amount, () => linearSR.withdrawTo(contracts, to));
67export async function getConditional(contracts, address) {
68 const [commitment, batches] = await Promise.all([
69 conditionalSR.getCommitment(contracts, address),
70 conditionalSR.getBatches(contracts, address),
71 ]);
73 if (commitment === null || batches === null) {
74 return { total: 0, available: 0, withdrawn: 0, batchLimits: [] };
75 }
77 const { total } = commitment;
78 const withdrawLimits = await Promise.all(
79 batches.map((_, idx) =>
80 conditionalSR.getWithdrawLimit(contracts, address, idx)
81 )
82 );
83 const available = withdrawLimits.reduce((acc, val) => acc + val, 0);
85 const withdrawnAmounts = await conditionalSR.getWithdrawn(contracts, address);
87 const withdrawn = withdrawnAmounts.reduce((acc, val) => acc + val, 0);
88 const batchLimits = withdrawLimits.map(
89 (limit, idx) => limit - withdrawnAmounts[idx]
90 );
92 return { total, available, withdrawn, batchLimits };
95export async function getLinear(contracts, address) {
96 const { amount, withdrawn } = await linearSR.getBatch(contracts, address);
98 const available = // Contract errors on withdrawing from a user that does not have any
99 amount > 0 ? await linearSR.getWithdrawLimit(contracts, address) : 0;
100 return { available, withdrawn, total: amount };