117 lines
4.6 KiB
JavaScript
117 lines
4.6 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
function getType(param, flexible) {
|
|
if (param.type === "address" || param.type === "string") {
|
|
return "string";
|
|
}
|
|
if (param.type.substring(0, 5) === "bytes") {
|
|
if (flexible) {
|
|
return "string | ethers.utils.BytesLike";
|
|
}
|
|
return "string";
|
|
}
|
|
var match = param.type.match(/^(u?int)([0-9]+)$/);
|
|
if (match) {
|
|
if (flexible) {
|
|
return "ethers.BigNumberish";
|
|
}
|
|
if (parseInt(match[2]) < 53) {
|
|
return 'number';
|
|
}
|
|
return 'ethers.BigNumber';
|
|
}
|
|
if (param.type === "array") {
|
|
return "Array<" + getType(param.arrayChildren) + ">";
|
|
}
|
|
if (param.type === "tuple") {
|
|
var struct = param.components.map(function (p, i) { return (p.name || "p_" + i) + ": " + getType(p, flexible); });
|
|
return "{ " + struct.join(", ") + " }";
|
|
}
|
|
throw new Error("unknown type");
|
|
return null;
|
|
}
|
|
exports.header = "import { ethers } from \"ethers\";\n\n";
|
|
function generate(contract, bytecode) {
|
|
var lines = [];
|
|
lines.push("export class " + contract.name + " extends ethers.Contract {");
|
|
lines.push("");
|
|
lines.push(" constructor(addressOrName: string, providerOrSigner: ethers.Signer | ethers.providers.Provider) {");
|
|
lines.push(" super(addressOrName, new.target.ABI(), providerOrSigner)");
|
|
lines.push(" }");
|
|
lines.push("");
|
|
lines.push(" connect(providerOrSigner: ethers.Signer | ethers.providers.Provider): " + contract.name + " {");
|
|
lines.push(" return new (<{ new(...args: any[]): " + contract.name + " }>(this.constructor))(this.address, providerOrSigner)");
|
|
lines.push(" }");
|
|
lines.push("");
|
|
lines.push(" attach(addressOrName: string): " + contract.name + " {");
|
|
lines.push(" return new (<{ new(...args: any[]): " + contract.name + " }>(this.constructor))(addressOrName, this.signer || this.provider)");
|
|
lines.push(" }");
|
|
var _loop_1 = function (signature) {
|
|
if (signature.indexOf('(') === -1) {
|
|
return "continue";
|
|
}
|
|
var fragment = contract.interface.functions[signature];
|
|
console.log(fragment);
|
|
var output_1 = "Promise<ethers.providers.TransactionResponse>";
|
|
var overrides = "ethers.CallOverrides";
|
|
if (fragment.constant == false) {
|
|
if (fragment.payable) {
|
|
overrides = "ethers.PayableOverrides";
|
|
}
|
|
else {
|
|
overrides = "ethers.Overrides";
|
|
}
|
|
}
|
|
else if (fragment.outputs.length > 0) {
|
|
if (fragment.outputs.length === 1) {
|
|
output_1 = "Promise<" + getType(fragment.outputs[0]) + ">";
|
|
}
|
|
else {
|
|
throw new Error('not implemented yet');
|
|
}
|
|
}
|
|
var inputs = [];
|
|
var passed = [];
|
|
fragment.inputs.forEach(function (input, index) {
|
|
var name = (input.name || ("p_" + index));
|
|
var type = getType(input, true);
|
|
inputs.push(name + ": " + type);
|
|
passed.push(name);
|
|
});
|
|
inputs.push("_overrides?: " + overrides);
|
|
passed.push("_overrides");
|
|
lines.push("");
|
|
lines.push(" " + fragment.name + "(" + inputs.join(', ') + "): " + output_1 + " {");
|
|
lines.push(" return this.functions[\"" + signature + "\"](" + passed.join(", ") + ");");
|
|
lines.push(" }");
|
|
};
|
|
for (var signature in contract.interface.functions) {
|
|
_loop_1(signature);
|
|
}
|
|
lines.push("");
|
|
lines.push(" static factory(signer?: ethers.Signer): ethers.ContractFactory {");
|
|
lines.push(" return new ethers.ContractFactory(" + contract.name + ".ABI(), " + contract.name + ".bytecode(), signer);");
|
|
lines.push(" }");
|
|
lines.push("");
|
|
lines.push(" static bytecode(): string {");
|
|
if (bytecode == null) {
|
|
lines.push(' return ethers.errors.throwError("no bytecode provided during generation", ethers.errors.UNSUPPORTED_OPERATION, { operation: "contract.bytecode" });');
|
|
}
|
|
else {
|
|
lines.push(' return "' + bytecode + '";');
|
|
}
|
|
lines.push(" }");
|
|
lines.push("");
|
|
lines.push(" static ABI(): Array<string> {");
|
|
lines.push(" return [");
|
|
contract.interface.fragments.forEach(function (fragment) {
|
|
lines.push(" \"" + fragment.format(true) + "\",");
|
|
});
|
|
lines.push(" ];");
|
|
lines.push(" }");
|
|
lines.push("}");
|
|
var output = lines.join("\n") + "\n";
|
|
return output;
|
|
}
|
|
exports.generate = generate;
|