135 lines
3.3 KiB
TypeScript
135 lines
3.3 KiB
TypeScript
import { Job, Queue } from 'bull';
|
|
import { BigNumber } from 'ethers';
|
|
import { TxManager } from 'tx-manager';
|
|
|
|
import { Injectable } from '@nestjs/common';
|
|
import { ConfigService } from '@nestjs/config';
|
|
import { InjectQueue, Process, Processor } from '@nestjs/bull';
|
|
|
|
import { toWei } from '@/utilities';
|
|
import { getGasPrice } from '@/services';
|
|
import { getTornadoPool } from '@/contracts';
|
|
|
|
import { BaseProcessor } from './base.processor';
|
|
|
|
export interface Withdrawal {
|
|
args: string[];
|
|
proof: string;
|
|
amount: string;
|
|
txHash: string;
|
|
status: string;
|
|
contract: string;
|
|
confirmations: number;
|
|
}
|
|
|
|
@Injectable()
|
|
@Processor('withdrawal')
|
|
export class WithdrawalProcessor extends BaseProcessor<Withdrawal> {
|
|
constructor(
|
|
@InjectQueue('withdrawal') public withdrawalQueue: Queue,
|
|
private configService: ConfigService,
|
|
) {
|
|
super();
|
|
this.queueName = 'withdrawal';
|
|
this.queue = withdrawalQueue;
|
|
}
|
|
|
|
@Process()
|
|
async processWithdrawals(job: Job<Withdrawal>) {
|
|
try {
|
|
await job.isActive();
|
|
|
|
const { args, amount } = job.data;
|
|
|
|
await this.checkFee({ fee: args[4], amount });
|
|
await this.submitTx(job);
|
|
} catch (err) {
|
|
await job.moveToFailed(err, true);
|
|
}
|
|
}
|
|
|
|
async submitTx(job: Job<Withdrawal>) {
|
|
const txManager = new TxManager(this.configService.get('txManager'));
|
|
|
|
const prepareTx = await this.prepareTransaction(job.data);
|
|
const tx = await txManager.createTx(prepareTx);
|
|
|
|
try {
|
|
const receipt = await tx
|
|
.send()
|
|
.on('transactionHash', async (txHash: string) => {
|
|
job.data.txHash = txHash;
|
|
job.data.status = 'SENT';
|
|
|
|
await job.update(job.data);
|
|
})
|
|
.on('mined', async () => {
|
|
job.data.status = 'MINED';
|
|
|
|
await job.update(job.data);
|
|
})
|
|
.on('confirmations', async (confirmations) => {
|
|
job.data.confirmations = confirmations;
|
|
|
|
await job.update(job.data);
|
|
});
|
|
|
|
if (receipt.status === 1) {
|
|
await job.isCompleted();
|
|
|
|
job.data.status = 'SENT';
|
|
|
|
await job.update(job.data);
|
|
} else {
|
|
throw new Error('Submitted transaction failed');
|
|
}
|
|
} catch (e) {
|
|
throw new Error(`Revert by smart contract ${e.message}`);
|
|
}
|
|
}
|
|
|
|
async prepareTransaction({ proof, args, amount }) {
|
|
const contract = getTornadoPool(1);
|
|
|
|
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
|
|
// @ts-ignore
|
|
const data = contract.interface.encodeFunctionData('transaction', [
|
|
proof,
|
|
...args,
|
|
]);
|
|
|
|
const gasLimit = this.configService.get<number>('gasLimits');
|
|
|
|
return {
|
|
data,
|
|
gasLimit,
|
|
value: amount,
|
|
to: contract.address,
|
|
};
|
|
}
|
|
|
|
async checkFee({ fee, amount }) {
|
|
const gasLimit = this.configService.get<number>('gasLimits');
|
|
|
|
const { fast } = await getGasPrice(1);
|
|
|
|
const expense = BigNumber.from(toWei(fast.toString(), 'gwei')).mul(
|
|
gasLimit,
|
|
);
|
|
|
|
const serviceFee = this.configService.get<number>('fee');
|
|
|
|
const feePercent = BigNumber.from(toWei(amount))
|
|
.mul(toWei(serviceFee.toString()))
|
|
.div(100);
|
|
|
|
const desiredFee = expense.add(feePercent);
|
|
|
|
if (fee.lt(desiredFee)) {
|
|
throw new Error(
|
|
'Provided fee is not enough. Probably it is a Gas Price spike, try to resubmit.',
|
|
);
|
|
}
|
|
}
|
|
}
|