140 lines
5.9 KiB
JavaScript
140 lines
5.9 KiB
JavaScript
"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, decryptJsonWalletSync, 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", "[REDACTED]");
|
|
}
|
|
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", "[REDACTED]");
|
|
}
|
|
}
|
|
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 fromEncryptedJsonSync(json, password) {
|
|
return new Wallet(decryptJsonWalletSync(json, password));
|
|
}
|
|
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);
|
|
}
|