@graphprotocol/graph-cli
Version:
CLI for building for and deploying to The Graph
136 lines (135 loc) • 5.55 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const path_1 = __importDefault(require("path"));
const fs_extra_1 = __importDefault(require("fs-extra"));
const immutable_1 = __importDefault(require("immutable"));
const abi_1 = __importDefault(require("./codegen/abi"));
const TUPLE_ARRAY_PATTERN = /^tuple\[([0-9]*)\]$/;
const TUPLE_MATRIX_PATTERN = /^tuple\[([0-9]*)\]\[([0-9]*)\]$/;
const buildOldSignatureParameter = (input) => {
return input.get('type') === 'tuple'
? `(${input
.get('components')
.map((component) => buildSignatureParameter(component))
.join(',')})`
: String(input.get('type'));
};
const buildSignatureParameter = (input) => {
if (input.get('type') === 'tuple') {
return `(${input.get('indexed') ? 'indexed ' : ''}${input
.get('components')
.map((component) => buildSignatureParameter(component))
.join(',')})`;
}
if (input.get('type').match(TUPLE_ARRAY_PATTERN)) {
const length = input.get('type').match(TUPLE_ARRAY_PATTERN)[1];
return `(${input.get('indexed') ? 'indexed ' : ''}${input
.get('components')
.map((component) => buildSignatureParameter(component))
.join(',')})[${length || ''}]`;
}
if (input.get('type').match(TUPLE_MATRIX_PATTERN)) {
const length1 = input.get('type').match(TUPLE_MATRIX_PATTERN)[1];
const length2 = input.get('type').match(TUPLE_MATRIX_PATTERN)[2];
return `(${input.get('indexed') ? 'indexed ' : ''}${input
.get('components')
.map((component) => buildSignatureParameter(component))
.join(',')})[${length1 || ''}][${length2 || ''}]`;
}
return `${input.get('indexed') ? 'indexed ' : ''}${input.get('type')}`;
};
class ABI {
constructor(name, file, data) {
this.name = name;
this.file = file;
this.data = data;
this.name = name;
this.file = file;
this.data = data;
}
codeGenerator() {
return new abi_1.default(this);
}
static oldEventSignature(event) {
return `${event.get('name')}(${event
.get('inputs', [])
.map(buildOldSignatureParameter)
.join(',')})`;
}
static eventSignature(event) {
return `${event.get('name')}(${event
.get('inputs', [])
.map(buildSignatureParameter)
.join(',')})`;
}
/**
* For the ABI of a function, returns a string function signature compatible
* with the Rust `ethabi` library. It is of the form
*
* <function>([<input-type-1>, ...])[:(<output-type-1,...)]
*
* A few examples for a function called `example`:
*
* - No inputs or outputs: `example()`
* - One input and output: `example(uint256):(bool)`
* - Multiple inputs and outputs: `example(uint256,(string,bytes32)):(bool,uint256)`
*/
functionSignature(fn) {
const inputs = fn.get('inputs', []).map(buildSignatureParameter).join(',');
const outputs = fn.get('outputs', []).map(buildSignatureParameter).join(',');
return `${fn.get('name')}(${inputs})${outputs.length > 0 ? `:(${outputs})` : ''}`;
}
oldEventSignatures() {
return this.data
.filter((entry) => entry.get('type') === 'event')
.map(ABI.oldEventSignature);
}
eventSignatures() {
return this.data.filter((entry) => entry.get('type') === 'event').map(ABI.eventSignature);
}
callFunctions() {
// An entry is a function if its type is not set or if it is one of
// 'constructor', 'function' or 'fallback'
const functionTypes = immutable_1.default.Set(['constructor', 'function', 'fallback']);
const functions = this.data.filter((entry) => !entry.has('type') || functionTypes.includes(entry.get('type')));
// A function is a call function if it is nonpayable, payable or
// not constant
const mutabilityTypes = immutable_1.default.Set(['nonpayable', 'payable']);
return functions.filter((entry) => mutabilityTypes.includes(entry.get('stateMutability')) || entry.get('constant') === false);
}
callFunctionSignatures() {
return this.callFunctions()
.filter((entry) => entry.get('type') !== 'constructor')
.map((entry) => {
const name = entry.get('name', '<default>');
const inputs = entry
.get('inputs', immutable_1.default.List())
.map((input) => buildSignatureParameter(input));
return `${name}(${inputs.join(',')})`;
});
}
static normalized(json) {
if (Array.isArray(json)) {
return json;
}
if (json.abi !== undefined) {
return json.abi;
}
if (json.compilerOutput?.abi) {
return json.compilerOutput.abi;
}
return undefined;
}
static load(name, file) {
const data = JSON.parse(fs_extra_1.default.readFileSync(file).toString());
const abi = ABI.normalized(data);
if (abi === null || abi === undefined) {
throw Error(`No valid ABI in file: ${path_1.default.relative(process.cwd(), file)}`);
}
return new ABI(name, file, immutable_1.default.fromJS(abi));
}
}
exports.default = ABI;