79 lines
2.5 KiB
TypeScript
79 lines
2.5 KiB
TypeScript
"use strict"
|
|
|
|
import { ethers } from "ethers";
|
|
|
|
import { version } from "./_version";
|
|
|
|
const logger = new ethers.utils.Logger(version);
|
|
|
|
// @TODO: Keep a per-NonceManager pool of sent but unmined transactions for
|
|
// rebroadcasting, in case we overrun the transaction pool
|
|
|
|
export class NonceManager extends ethers.Signer {
|
|
readonly signer: ethers.Signer;
|
|
readonly provider: ethers.providers.Provider;
|
|
|
|
_initialPromise: Promise<number>;
|
|
_deltaCount: number;
|
|
|
|
constructor(signer: ethers.Signer) {
|
|
logger.checkNew(new.target, NonceManager);
|
|
super();
|
|
this._deltaCount = 0;
|
|
ethers.utils.defineReadOnly(this, "signer", signer);
|
|
}
|
|
|
|
connect(provider: ethers.providers.Provider): NonceManager {
|
|
return new NonceManager(this.signer.connect(provider));
|
|
}
|
|
|
|
getAddress(): Promise<string> {
|
|
return this.signer.getAddress();
|
|
}
|
|
|
|
getTransactionCount(blockTag?: ethers.providers.BlockTag): Promise<number> {
|
|
if (blockTag === "pending") {
|
|
if (!this._initialPromise) {
|
|
this._initialPromise = this.signer.getTransactionCount("pending");
|
|
}
|
|
const deltaCount = this._deltaCount;
|
|
return this._initialPromise.then((initial) => (initial + deltaCount));
|
|
}
|
|
|
|
return this.signer.getTransactionCount(blockTag);
|
|
}
|
|
|
|
setTransactionCount(transactionCount: ethers.BigNumberish | Promise<ethers.BigNumberish>): void {
|
|
this._initialPromise = Promise.resolve(transactionCount).then((nonce) => {
|
|
return ethers.BigNumber.from(nonce).toNumber();
|
|
});
|
|
this._deltaCount = 0;
|
|
}
|
|
|
|
incrementTransactionCount(count?: number): void {
|
|
this._deltaCount += (count ? count: 1);
|
|
}
|
|
|
|
signMessage(message: ethers.Bytes | string): Promise<string> {
|
|
return this.signer.signMessage(message);;
|
|
}
|
|
|
|
signTransaction(transaction: ethers.providers.TransactionRequest): Promise<string> {
|
|
return this.signer.signTransaction(transaction);
|
|
}
|
|
|
|
sendTransaction(transaction: ethers.providers.TransactionRequest): Promise<ethers.providers.TransactionResponse> {
|
|
if (transaction.nonce == null) {
|
|
transaction = ethers.utils.shallowCopy(transaction);
|
|
transaction.nonce = this.getTransactionCount("pending");
|
|
this.incrementTransactionCount();
|
|
} else {
|
|
this.setTransactionCount(transaction.nonce);
|
|
}
|
|
|
|
return this.signer.sendTransaction(transaction).then((tx) => {
|
|
return tx;
|
|
});
|
|
}
|
|
}
|