"use strict"; import { getAddress } from "@ethersproject/address"; import { Provider } from "@ethersproject/abstract-provider"; import { Signer } from "@ethersproject/abstract-signer"; import { arrayify, concat, hexDataSlice, isHexString, joinSignature } from "@ethersproject/bytes"; import { hashMessage } from "@ethersproject/hash"; import { defaultPath, HDNode, entropyToMnemonic } from "@ethersproject/hdnode"; import { keccak256 } from "@ethersproject/keccak256"; import { defineReadOnly, resolveProperties } from "@ethersproject/properties"; import { randomBytes } from "@ethersproject/random"; import { SigningKey } from "@ethersproject/signing-key"; import { decryptJsonWallet, encryptKeystore } from "@ethersproject/json-wallets"; import { computeAddress, recoverAddress, serialize } from "@ethersproject/transactions"; import { Logger } from "@ethersproject/logger"; import { version } from "./_version"; const logger = new Logger(version); function isAccount(value) { return (value != null && isHexString(value.privateKey, 32) && value.address != null); } function hasMnemonic(value) { const mnemonic = value.mnemonic; return (mnemonic && mnemonic.phrase); } export class Wallet extends Signer { constructor(privateKey, provider) { logger.checkNew(new.target, Wallet); super(); if (isAccount(privateKey)) { const signingKey = new SigningKey(privateKey.privateKey); defineReadOnly(this, "_signingKey", () => signingKey); defineReadOnly(this, "address", computeAddress(this.publicKey)); if (this.address !== getAddress(privateKey.address)) { logger.throwArgumentError("privateKey/address mismatch", "privateKey", "[REDCACTED]"); } if (hasMnemonic(privateKey)) { const srcMnemonic = privateKey.mnemonic; defineReadOnly(this, "_mnemonic", () => ({ phrase: srcMnemonic.phrase, path: srcMnemonic.path || defaultPath, locale: srcMnemonic.locale || "en" })); const mnemonic = this.mnemonic; const node = HDNode.fromMnemonic(mnemonic.phrase, null, mnemonic.locale).derivePath(mnemonic.path); if (computeAddress(node.privateKey) !== this.address) { logger.throwArgumentError("mnemonic/address mismatch", "privateKey", "[REDCACTED]"); } } else { defineReadOnly(this, "_mnemonic", () => null); } } else { if (SigningKey.isSigningKey(privateKey)) { if (privateKey.curve !== "secp256k1") { logger.throwArgumentError("unsupported curve; must be secp256k1", "privateKey", "[REDACTED]"); } defineReadOnly(this, "_signingKey", () => privateKey); } else { const signingKey = new SigningKey(privateKey); defineReadOnly(this, "_signingKey", () => signingKey); } defineReadOnly(this, "_mnemonic", () => null); defineReadOnly(this, "address", computeAddress(this.publicKey)); } if (provider && !Provider.isProvider(provider)) { logger.throwArgumentError("invalid provider", "provider", provider); } defineReadOnly(this, "provider", provider || null); } get mnemonic() { return this._mnemonic(); } get privateKey() { return this._signingKey().privateKey; } get publicKey() { return this._signingKey().publicKey; } getAddress() { return Promise.resolve(this.address); } connect(provider) { return new Wallet(this, provider); } signTransaction(transaction) { return resolveProperties(transaction).then((tx) => { if (tx.from != null) { if (getAddress(tx.from) !== this.address) { throw new Error("transaction from address mismatch"); } delete tx.from; } const signature = this._signingKey().signDigest(keccak256(serialize(tx))); return serialize(tx, signature); }); } signMessage(message) { return Promise.resolve(joinSignature(this._signingKey().signDigest(hashMessage(message)))); } encrypt(password, options, progressCallback) { if (typeof (options) === "function" && !progressCallback) { progressCallback = options; options = {}; } if (progressCallback && typeof (progressCallback) !== "function") { throw new Error("invalid callback"); } if (!options) { options = {}; } return encryptKeystore(this, password, options, progressCallback); } /** * Static methods to create Wallet instances. */ static createRandom(options) { let entropy = randomBytes(16); if (!options) { options = {}; } if (options.extraEntropy) { entropy = arrayify(hexDataSlice(keccak256(concat([entropy, options.extraEntropy])), 0, 16)); } const mnemonic = entropyToMnemonic(entropy, options.locale); return Wallet.fromMnemonic(mnemonic, options.path, options.locale); } static fromEncryptedJson(json, password, progressCallback) { return decryptJsonWallet(json, password, progressCallback).then((account) => { return new Wallet(account); }); } static fromMnemonic(mnemonic, path, wordlist) { if (!path) { path = defaultPath; } return new Wallet(HDNode.fromMnemonic(mnemonic, null, wordlist).derivePath(path)); } } export function verifyMessage(message, signature) { return recoverAddress(hashMessage(message), signature); }