@instadapp/interop-x 0.0.0-dev.051cd4e

package/src/net/protocol/dial/BaseDialProtocol.ts

@@ -0,0 +1,104 @@
+import pipe from 'it-pipe';
+import Libp2p from 'libp2p';
+import PeerId from 'peer-id';
+import { asyncCallWithTimeout } from '@/utils';
+import wait from 'waait';
+
+export class BaseDialProtocol<TRequest extends any, TResponse extends any> {
+    protected libp2p: Libp2p;
+    protected protocol: string;
+    protected timeout = 20000;
+
+    constructor(libp2p: Libp2p, protocol: string) {
+        this.libp2p = libp2p
+        this.protocol = protocol
+
+        this.libp2p.handle(this.protocol, this.handle.bind(this))
+    }
+
+    encode(data: TRequest) {
+        return JSON.stringify(data)
+    }
+
+    encodeResponse(data: TResponse): String | Buffer {
+        return JSON.stringify(data)
+    }
+
+    decode(data: Buffer) {
+        return JSON.parse(data.toString())
+    }
+
+    decodeResponse(data: Buffer): any {
+        return JSON.parse(data.toString())
+    }
+
+    async response(data: TRequest): Promise<TResponse> {
+        return data as TResponse;
+    }
+
+    async handle({ connection, stream }) {
+        const instance = this;
+
+        try {
+            await pipe(stream, async function (source) {
+                for await (const message of source) {
+                    const response = instance.encodeResponse(
+                        await instance.response(
+                            instance.decode(message)
+                        )
+                    )
+
+                    await pipe(
+                        [response],
+                        stream
+                    );
+                }
+            });
+        } catch (err) {
+            console.error(err);
+        }
+    }
+
+    async send(data: TRequest, peerId: string) {
+        return await asyncCallWithTimeout<TResponse>(new Promise(async (resolve, reject) => {
+            try {
+                let connection = this.libp2p.connectionManager.get(PeerId.createFromB58String(peerId))
+
+                if (!connection) {
+
+                    await wait(5000)
+
+                    connection = this.libp2p.connectionManager.get(PeerId.createFromB58String(peerId))
+
+
+                    if (!connection) {
+
+                        await wait(3000)
+
+                        connection = this.libp2p.connectionManager.get(PeerId.createFromB58String(peerId))
+
+                        if (!connection) {
+                            throw new Error('No connection available')
+                        }
+                    }
+                }
+
+
+
+                const { stream } = await connection.newStream([this.protocol])
+
+                const instance = this;
+
+                pipe([this.encode(data)], stream, async function (source) {
+                    for await (const message of source) {
+                        resolve(instance.decodeResponse(message))
+                        stream.close()
+                    }
+                });
+            } catch (error) {
+                reject(error)
+            }
+        }), this.timeout)
+
+    }
+}

package/src/net/protocol/dial/SignatureDialProtocol.ts

@@ -0,0 +1,71 @@
+import { BaseDialProtocol } from "./BaseDialProtocol";
+import wait from "waait";
+import config from "@/config";
+import { Transaction } from "@/db";
+import { buildDataForTransaction, signGnosisSafeTx } from "@/utils";
+import { addresses } from "@/constants";
+import { ChainId } from "@/types";
+
+export interface ISignatureRequest {
+    type: 'source' | 'target' ,
+    transactionHash: string
+    safeTxGas: string
+    safeNonce: string
+}
+export interface ISignatureResponse {
+    signer: string,
+    data?: string | null,
+    error?: string | null,
+}
+export class SignatureDialProtocol extends BaseDialProtocol<ISignatureRequest, ISignatureResponse> {
+    protected timeout = 30000;
+
+    constructor(libp2p) {
+        super(libp2p, '/interop-x/signatures')
+    }
+
+    async response(data: ISignatureRequest): Promise<ISignatureResponse> {
+        const signer = config.wallet;
+
+        let transaction: Transaction | null;
+        let maxTimeout = 20000;
+
+        do {
+            transaction = await Transaction.findOne({ where: { transactionHash: data.transactionHash } })
+
+            if (!transaction) {
+                await wait(1000);
+                maxTimeout -= 1000;
+            }
+        } while (!transaction && maxTimeout > 0)
+
+        if (!transaction) {
+            return {
+                signer: signer.address,
+                data: null,
+                error: 'Event not found'
+            };
+        }
+
+        console.log("signing:", {
+            to: addresses[transaction.targetChainId].multisend,
+            data: await buildDataForTransaction(transaction, data.type),
+            chainId: transaction.targetChainId as ChainId,
+            safeTxGas: data.safeTxGas,
+            nonce: data.safeNonce,
+        });
+
+        const signedData = await signGnosisSafeTx({
+            to: addresses[transaction.targetChainId].multisend,
+            data: await buildDataForTransaction(transaction, data.type),
+            chainId: transaction.targetChainId as ChainId,
+            safeTxGas: data.safeTxGas,
+            nonce: data.safeNonce,
+        }, { signer });
+
+        return {
+            signer: signer.address,
+            data: signedData
+        }
+    }
+}

package/src/net/protocol/index.ts

@@ -0,0 +1,182 @@
+import { EventEmitter } from "stream";
+import Libp2p from 'libp2p';
+import { bufferToInt, rlp } from 'ethereumjs-util'
+import { SignatureDialProtocol, ISignatureRequest, ISignatureResponse } from "./dial/SignatureDialProtocol";
+import { IPeerInfo, peerPool } from "..";
+import config from "@/config";
+import { Event } from "@/types";
+import { Transaction } from "@/db";
+
+export interface ProtocolOptions {
+    /* Handshake timeout in ms (default: 8000) */
+    timeout?: number
+}
+
+
+export type Message = {
+    name: string
+    code: number
+    payload?: any
+    response?: number
+    encode: Function
+    decode: Function
+}
+
+interface BaseMessageEvent {
+    peerId: string
+    name: string
+    code: number
+    data: any
+}
+
+
+interface PeerInfoEvent extends BaseMessageEvent {
+    data: Omit<IPeerInfo, 'id' | 'updated' | 'idle' | 'pooled'>
+}
+
+interface TransactionStatusEvent extends BaseMessageEvent {
+    data: Pick<Transaction, 'transactionHash' | 'sourceStatus' | 'sourceTransactionHash' | 'sourceErrors' | 'targetStatus' | 'targetTransactionHash' | 'targetErrors' | 'status'>
+}
+
+declare interface Protocol {
+    on(event: 'TransactionStatus', listener: (payload: TransactionStatusEvent) => void): this;
+    on(event: 'PeerInfo', listener: (payload: PeerInfoEvent) => void): this;
+    on(event: string, listener: (payload: BaseMessageEvent) => void): this;
+}
+
+class Protocol extends EventEmitter {
+    private libp2p: Libp2p;
+    private topic: string;
+    private protocolMessages: Message[] = [
+        {
+            name: 'PeerInfo',
+            code: 0x01,
+            encode: (info: Pick<IPeerInfo, 'publicAddress'>) => [
+                Buffer.from(info.publicAddress),
+            ],
+            decode: ([publicAddress]: [Buffer]) => ({
+                publicAddress: publicAddress.toString(),
+            }),
+        },
+        {
+            name: 'TransactionStatus',
+            code: 0x02,
+            encode: (transaction: Transaction) => [
+                Buffer.from(transaction.transactionHash),
+
+                Buffer.from(transaction.sourceStatus),
+                Buffer.from(transaction.sourceTransactionHash),
+                transaction.sourceErrors ? transaction.sourceErrors.map((e) => Buffer.from(e)) : [],
+
+                Buffer.from(transaction.targetStatus),
+                Buffer.from(transaction.targetTransactionHash),
+                transaction.targetErrors ? transaction.targetErrors.map((e) => Buffer.from(e)) : [],
+
+                Buffer.from(transaction.status),
+            ],
+            decode: ([transactionHash, sourceStatus, sourceTransactionHash, sourceErrors, targetStatus, targetTransactionHash, targetErrors, status]: [Buffer, Buffer, Buffer, Buffer[], Buffer, Buffer, Buffer[], Buffer]) => ({
+                transactionHash: transactionHash.toString(),
+
+                sourceStatus: sourceStatus.toString(),
+                sourceTransactionHash: sourceTransactionHash.toString(),
+                sourceErrors: sourceErrors.map((e) => e.toString()),
+
+                targetStatus: targetStatus.toString(),
+                targetTransactionHash: targetTransactionHash.toString(),
+                targetErrors: targetErrors.map((e) => e.toString()),
+
+                status: status.toString(),
+            }),
+        },
+    ];
+    private signature: SignatureDialProtocol;
+
+
+    start({ libp2p, topic = null, }) {
+        this.libp2p = libp2p
+        this.topic = topic || 'itnerop-x-protocol'
+
+        if (this.libp2p.isStarted()) this.init()
+
+        this.on('PeerInfo', (payload) => {
+
+            config.events.emit(Event.PEER_CONNECTED, {
+                id: payload.peerId,
+                publicAddress: payload.data.publicAddress,
+                pooled: false,
+                updated: new Date(),
+            })
+        })
+
+        this.signature = new SignatureDialProtocol(this.libp2p);
+    }
+
+
+    init() {
+        this.libp2p.pubsub.subscribe(this.topic)
+
+        this.libp2p.pubsub.on(this.topic, (message) => {
+            try {
+                const [codeBuf, payload]: any = rlp.decode(message.data);
+
+                const code = bufferToInt(codeBuf);
+
+                const protocolMessage = this.protocolMessages.find((m) => m.code === code);
+                if (!protocolMessage) {
+                    return;
+                }
+                const decoded = protocolMessage.decode(payload);
+                this.emit(
+                    protocolMessage.name,
+                    {
+                        peerId: message.from,
+                        name: protocolMessage.name,
+                        code: protocolMessage.code,
+                        data: decoded
+                    }
+                )
+
+                this.emit('all', {
+                    peerId: message.from,
+                    name: protocolMessage.name,
+                    code: protocolMessage.code,
+                    data: decoded,
+                })
+            } catch (err) {
+                console.error(err)
+            }
+        })
+    }
+
+
+    public sendPeerInfo(data: Pick<IPeerInfo, 'publicAddress'>) {
+        const message = this.protocolMessages.find((m) => m.name === 'PeerInfo')!
+
+        const encoded = rlp.encode([message.code, message.encode(data)]);
+
+        this.libp2p.pubsub.publish(this.topic, encoded)
+    }
+
+    public sendTransaction(transaction: Transaction) {
+        const message = this.protocolMessages.find((m) => m.name === 'TransactionStatus')!
+
+        const encoded = rlp.encode([message.code, message.encode(transaction)]);
+
+        this.libp2p.pubsub.publish(this.topic, encoded)
+    }
+
+    async requestSignatures(data: ISignatureRequest, peerIds?: string[]) {
+        try {
+            peerIds = peerIds || peerPool.activePeerIds;
+            const promises = peerIds.map((peerId) => this.signature.send(data, peerId))
+            return (await Promise.allSettled(promises))
+                .map((p) => p.status === 'fulfilled' ? p.value : null)
+                .filter(Boolean) as ISignatureResponse[]
+        } catch (error) {
+            console.log(error);
+            return []
+        }
+    }
+}
+
+export const protocol = new Protocol();

package/src/tasks/AutoUpdateTask.ts

@@ -0,0 +1,54 @@
+import { BaseTask } from "./BaseTask";
+import Logger from '@/logger';
+import packageJson from '../../package.json'
+import { http } from "@/utils";
+import { spawn } from 'child_process';
+import config from "@/config";
+const currentVersion = packageJson.version;
+
+class AutoUpdateTask extends BaseTask {
+    pollIntervalMs: number = 60 * 1000
+
+    constructor() {
+        super({
+            logger: new Logger("AutoUpdateTask"),
+        })
+    }
+
+    prePollHandler(): boolean {
+        return config.autoUpdate && !config.isLeadNode();
+    }
+
+    async pollHandler() {
+
+        const { data } = await http.get('https://registry.npmjs.org/@instadapp/interop-x')
+
+        const version = data['dist-tags'].latest
+
+        if (version === currentVersion) {
+            return;
+        }
+
+        this.logger.warn(`New version ${version} available.`)
+
+
+        const update = spawn('npm', ['-g', 'install', '@instadapp/interop-x']);
+
+        update.on("close", () => {
+            this.logger.warn(`Installed version ${version}`)
+            this.logger.warn(`Restarting...`)
+
+            const restart = spawn(process.argv[0], process.argv.slice(1), {
+                cwd: process.cwd(),
+                detached: true,
+                stdio: "inherit"
+            });
+
+            restart.on("close", () => {
+                process.exit()
+            })
+        })
+    }
+}
+
+export default AutoUpdateTask;

package/src/tasks/BaseTask.ts

@@ -0,0 +1,75 @@
+import config from "@/config";
+import EventEmitter from "events";
+import wait from "waait";
+import Logger from "@/logger";
+
+export interface IBaseTask {
+    pollCheck(): Promise<void>
+    prePollHandler(): boolean
+    pollHandler(): Promise<void>
+    postPollHandler(): Promise<void>
+
+    start(): Promise<void>
+    stop(): Promise<void>
+}
+
+
+export class BaseTask extends EventEmitter implements IBaseTask {
+    logger: Logger
+    started: boolean = false
+    pollIntervalMs: number = 10 * 1000
+    leadNodeOnly: boolean = false
+
+    public constructor({ logger }: { logger?: Logger }) {
+        super()
+
+        this.logger = logger ?? new Logger('BaseTask')
+    }
+
+    async pollCheck() {
+        while (true) {
+            if (!this.started) {
+                return
+            }
+            try {
+                const shouldPoll = this.prePollHandler()
+                if (shouldPoll) {
+                    await this.pollHandler()
+                }
+            } catch (err) {
+                this.logger.error(`poll check error:\n${err.message}\ntrace: ${err.stack}`)
+            }
+
+            await this.postPollHandler()
+        }
+    }
+
+    prePollHandler(): boolean {
+        if (!this.leadNodeOnly) {
+            return true
+        }
+
+        return config.isLeadNode()
+    }
+
+    async pollHandler() {
+        this.logger.warn('pollHandler not implemented')
+    }
+
+    async postPollHandler() {
+        await wait(this.pollIntervalMs)
+    }
+
+    async start() {
+        this.started = true
+        try {
+            await this.pollCheck()
+        } catch (err) {
+            this.logger.error(`base task error: ${err.message}\ntrace: ${err.stack}`)
+        }
+    }
+
+    async stop() {
+        this.started = false
+    }
+}