@dfns/lib-tron 0.1.0-beta.6

package/index.d.ts

@@ -0,0 +1,18 @@
+import { DfnsApiClient } from '@dfns/sdk';
+import { SignedTransaction, Transaction } from '@tronweb3/tronwallet-abstract-adapter';
+export type DfnsWalletOptions = {
+    walletId: string;
+    dfnsClient: DfnsApiClient;
+    maxRetries?: number;
+    retryInterval?: number;
+};
+export declare class DfnsWallet {
+    private options;
+    private tronAddress;
+    private constructor();
+    static init(options: DfnsWalletOptions): Promise<DfnsWallet>;
+    get address(): string;
+    waitForSignature(signatureId: string): Promise<string>;
+    signTransaction(transaction: Transaction): Promise<SignedTransaction>;
+    signMessage(message: string): Promise<string>;
+}

package/index.js

@@ -0,0 +1,102 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DfnsWallet = void 0;
+const Wallets_1 = require("@dfns/sdk/codegen/datamodel/Wallets");
+const elliptic = __importStar(require("elliptic"));
+const TronWeb = require('tronweb');
+const sleep = (interval = 0) => new Promise((resolve) => setTimeout(resolve, interval));
+class DfnsWallet {
+    constructor(options, tronAddress) {
+        var _a, _b;
+        this.options = {
+            ...options,
+            maxRetries: (_a = options.maxRetries) !== null && _a !== void 0 ? _a : 3,
+            retryInterval: (_b = options.retryInterval) !== null && _b !== void 0 ? _b : 1000,
+        };
+        this.tronAddress = tronAddress;
+    }
+    static async init(options) {
+        var _a, _b;
+        const { walletId, dfnsClient } = options;
+        const res = await dfnsClient.wallets.getWallet({ walletId });
+        if (!res.signingKey || res.signingKey.scheme !== Wallets_1.KeyScheme.ECDSA || res.signingKey.curve !== Wallets_1.KeyCurve.secp256k1) {
+            throw new Error(`wallet ${walletId} has incompatible scheme (${(_a = res.signingKey) === null || _a === void 0 ? void 0 : _a.scheme}) or curve (${(_b = res.signingKey) === null || _b === void 0 ? void 0 : _b.curve})`);
+        }
+        // Decompressed the key to derive the address
+        const ec = new elliptic.ec('secp256k1');
+        const keyAsArray = ec.keyFromPublic(res.signingKey.publicKey, 'hex').getPublic(false, 'array');
+        const tronAddress = TronWeb.utils.crypto.getBase58CheckAddress(TronWeb.utils.crypto.computeAddress(keyAsArray));
+        return new DfnsWallet(options, tronAddress);
+    }
+    get address() {
+        return this.tronAddress;
+    }
+    async waitForSignature(signatureId) {
+        const { walletId, dfnsClient, retryInterval } = this.options;
+        let maxRetries = this.options.maxRetries;
+        while (maxRetries > 0) {
+            await sleep(retryInterval);
+            const res = await dfnsClient.wallets.getSignature({ walletId, signatureId });
+            if (res.status === Wallets_1.SignatureStatus.Signed) {
+                if (!res.signature)
+                    break;
+                const r = res.signature.r.substring(2);
+                const s = res.signature.s.substring(2);
+                const v = (res.signature.recid ? 0x1c : 0x1b).toString(16);
+                return r + s + v;
+            }
+            else if (res.status === Wallets_1.SignatureStatus.Failed) {
+                console.log("signature failed: %s", res);
+                break;
+            }
+            maxRetries -= 1;
+        }
+        throw new Error(`signature ${signatureId} not available`);
+    }
+    async signTransaction(transaction) {
+        const { walletId, dfnsClient } = this.options;
+        const res = await dfnsClient.wallets.generateSignature({
+            walletId,
+            body: { kind: Wallets_1.SignatureKind.Hash, hash: transaction.txID },
+        });
+        const signature = await this.waitForSignature(res.id);
+        const signedTransaction = {
+            ...transaction,
+            signature: [signature],
+        };
+        return signedTransaction;
+    }
+    async signMessage(message) {
+        const { walletId, dfnsClient } = this.options;
+        let messageHash = TronWeb.utils.message.hashMessage(message);
+        const res = await dfnsClient.wallets.generateSignature({
+            walletId,
+            body: { kind: Wallets_1.SignatureKind.Hash, hash: messageHash },
+        });
+        return this.waitForSignature(res.id);
+    }
+}
+exports.DfnsWallet = DfnsWallet;

package/package.json

@@ -0,0 +1,17 @@
+{
+  "name": "@dfns/lib-tron",
+  "version": "0.1.0-beta.6",
+  "dependencies": {
+    "@tronweb3/tronwallet-abstract-adapter": "^1.1.5",
+    "buffer": "6.0.3",
+    "cross-fetch": "3.1.6",
+    "elliptic": "6.5.4",
+    "tronweb": "^5.3.0",
+    "uuid": "9.0.0"
+  },
+  "peerDependencies": {
+    "@dfns/sdk": "0.1.0-beta.6"
+  },
+  "main": "./index.js",
+  "types": "./index.d.ts"
+}