115 lines
3.4 KiB
JavaScript
115 lines
3.4 KiB
JavaScript
import {ethers} from "ethers";
|
|
import {getAbi} from "./abi.js";
|
|
import {getProvider, getSigner} from "./blockchain.js";
|
|
import {vaults} from './cache.js';
|
|
import {chainInfo, VAULT_INIT_CODE_HASH} from "./misc.js";
|
|
|
|
// Vault
|
|
// address owner
|
|
// balances { tokenAddress: amount }
|
|
// recentOrders []
|
|
|
|
const vaultAbi = await getAbi('Vault')
|
|
const factoryAbi = await getAbi('Factory')
|
|
|
|
|
|
export function vaultAddress(chainId, owner, num=0) {
|
|
try {
|
|
const salt = ethers.solidityPackedKeccak256(['address','uint8'],[owner,num])
|
|
const result = ethers.getCreate2Address(chainInfo[chainId].factory, salt, VAULT_INIT_CODE_HASH)
|
|
// console.log('vaultAddress', chainId, owner, num, chainInfo[chainId].factory, salt, VAULT_INIT_CODE_HASH, result)
|
|
return result
|
|
}
|
|
catch (e) {
|
|
console.error('Could not generate vault address',chainId,owner,num)
|
|
return null
|
|
}
|
|
}
|
|
|
|
|
|
export async function loginAddress(socket, chainId, address) {
|
|
if( socket.user_room !== undefined)
|
|
socket.leave(socket.user_room)
|
|
if( address === undefined ) {
|
|
socket.emit('vaults', [])
|
|
}
|
|
else {
|
|
socket.user_room = `${chainId}|${address}`
|
|
socket.join(socket.user_room)
|
|
const found = []
|
|
for( let num = 0; ; num++ ) {
|
|
const vault = vaultAddress(chainId, address, num)
|
|
if(await vaults.contains(chainId, vault))
|
|
found.push(vault)
|
|
else
|
|
break
|
|
}
|
|
socket.emit('vaults', found)
|
|
}
|
|
}
|
|
|
|
|
|
export async function ensureVault(socket, chainId, owner, num) {
|
|
console.log('ensureVault', chainId, owner, num)
|
|
const address = vaultAddress(chainId, owner, num)
|
|
if (!await vaults.contains(chainId,address)) {
|
|
try {
|
|
const vault = await createVault(chainId, owner, num)
|
|
console.log(vault)
|
|
if (vault !== null)
|
|
socket.emit('vaults', [vault])
|
|
else
|
|
console.error('got null vault for chainId', chainId)
|
|
} catch {
|
|
}
|
|
}
|
|
else
|
|
console.log('ensureVault', owner, 'exists:', address)
|
|
}
|
|
|
|
|
|
async function createVault(chainId, owner, num) {
|
|
console.log('createVault', chainId, owner, num )
|
|
const signer = getSigner(chainId);
|
|
const deployer = new ethers.Contract(chainInfo[chainId].factory, factoryAbi, signer)
|
|
const addr = vaultAddress(chainId, owner, num)
|
|
try {
|
|
const tx = await deployer.deployVault(owner, num)
|
|
// console.log(`deploying vault for ${owner} #${num} with tx ${tx.hash}`)
|
|
await tx.wait()
|
|
}
|
|
catch (e) {
|
|
const vault = new ethers.Contract(addr, vaultAbi, getProvider(chainId))
|
|
try {
|
|
const ver = await vault.version()
|
|
console.log(`vault already deployed at ${addr} with version ${ver}`)
|
|
}
|
|
catch (e2) {
|
|
console.error('could not deploy vault:',e)
|
|
return null
|
|
}
|
|
}
|
|
return addr
|
|
}
|
|
|
|
|
|
export async function watchErc20Transfer(provider, db, event) {
|
|
console.log('Transfer', event)
|
|
const [from, to, amount] = event.args
|
|
let vault = vaults[from]
|
|
let delta = 0
|
|
if (vault !== undefined)
|
|
delta = -amount
|
|
else {
|
|
vault = vaults[to]
|
|
if (vault !== undefined)
|
|
delta = amount
|
|
}
|
|
if (vault) {
|
|
let oldBalance = vault.balances[event.address]
|
|
if (oldBalance === undefined) {
|
|
|
|
}
|
|
}
|
|
}
|