96 lines
3.6 KiB
TypeScript
96 lines
3.6 KiB
TypeScript
import { getAddress, resolveAddress } from "../address/index.js";
|
|
import { hashMessage, TypedDataEncoder } from "../hash/index.js";
|
|
import { AbstractSigner } from "../providers/index.js";
|
|
import { computeAddress, Transaction } from "../transaction/index.js";
|
|
import {
|
|
defineProperties, resolveProperties, throwArgumentError, throwError
|
|
} from "../utils/index.js";
|
|
|
|
import type { SigningKey } from "../crypto/index.js";
|
|
import type { TypedDataDomain, TypedDataField } from "../hash/index.js";
|
|
import type { Provider, TransactionRequest } from "../providers/index.js";
|
|
import type { TransactionLike } from "../transaction/index.js";
|
|
|
|
export class BaseWallet extends AbstractSigner {
|
|
readonly address!: string;
|
|
|
|
readonly #signingKey: SigningKey;
|
|
|
|
constructor(privateKey: SigningKey, provider?: null | Provider) {
|
|
super(provider);
|
|
this.#signingKey = privateKey;
|
|
|
|
const address = computeAddress(this.signingKey.publicKey);
|
|
defineProperties<BaseWallet>(this, { address });
|
|
}
|
|
|
|
// Store these in getters to reduce visibility in console.log
|
|
get signingKey(): SigningKey { return this.#signingKey; }
|
|
get privateKey(): string { return this.signingKey.privateKey; }
|
|
|
|
async getAddress(): Promise<string> { return this.address; }
|
|
|
|
connect(provider: null | Provider): BaseWallet {
|
|
return new BaseWallet(this.#signingKey, provider);
|
|
}
|
|
|
|
async signTransaction(tx: TransactionRequest): Promise<string> {
|
|
|
|
// Replace any Addressable or ENS name with an address
|
|
const { to, from } = await resolveProperties({
|
|
to: (tx.to ? resolveAddress(tx.to, this.provider): undefined),
|
|
from: (tx.from ? resolveAddress(tx.from, this.provider): undefined)
|
|
});
|
|
|
|
if (to != null) { tx.to = to; }
|
|
if (from != null) { tx.from = from; }
|
|
|
|
if (tx.from != null) {
|
|
if (getAddress(<string>(tx.from)) !== this.address) {
|
|
throwArgumentError("transaction from address mismatch", "tx.from", tx.from);
|
|
}
|
|
delete tx.from;
|
|
}
|
|
|
|
// Build the transaction
|
|
const btx = Transaction.from(<TransactionLike<string>>tx);
|
|
btx.signature = this.signingKey.sign(btx.unsignedHash);
|
|
|
|
return btx.serialized;
|
|
}
|
|
|
|
async signMessage(message: string | Uint8Array): Promise<string> {
|
|
return this.signingKey.sign(hashMessage(message)).serialized;
|
|
}
|
|
|
|
// @TODO: Add a secialized signTx and signTyped sync that enforces
|
|
// all parameters are known?
|
|
signMessageSync(message: string | Uint8Array): string {
|
|
return this.signingKey.sign(hashMessage(message)).serialized;
|
|
}
|
|
|
|
async signTypedData(domain: TypedDataDomain, types: Record<string, Array<TypedDataField>>, value: Record<string, any>): Promise<string> {
|
|
|
|
// Populate any ENS names
|
|
const populated = await TypedDataEncoder.resolveNames(domain, types, value, async (name: string) => {
|
|
if (this.provider == null) {
|
|
return throwError("cannot resolve ENS names without a provider", "UNSUPPORTED_OPERATION", {
|
|
operation: "resolveName",
|
|
info: { name }
|
|
});
|
|
}
|
|
|
|
const address = await this.provider.resolveName(name);
|
|
if (address == null) {
|
|
return throwError("unconfigured ENS name", "UNCONFIGURED_NAME", {
|
|
value: name
|
|
});
|
|
}
|
|
|
|
return address;
|
|
});
|
|
|
|
return this.signingKey.sign(TypedDataEncoder.hash(populated.domain, types, populated.value)).serialized;
|
|
}
|
|
}
|