97 lines
2.5 KiB
TypeScript
97 lines
2.5 KiB
TypeScript
|
'use strict';
|
||
|
|
|
||
|
|
import fs from "fs";
|
||
|
|
import { dirname, resolve } from "path";
|
||
|
|
|
||
|
|
import { ethers } from "ethers";
|
||
|
|
|
||
|
|
let _solc: any = null;
|
||
|
|
function getSolc(): any {
|
||
|
|
if (!_solc) {
|
||
|
|
_solc = require("solc");
|
||
|
|
}
|
||
|
|
return _solc;
|
||
|
|
}
|
||
|
|
|
||
|
|
export interface ContractCode {
|
||
|
|
interface: ethers.utils.Interface;
|
||
|
|
name: string;
|
||
|
|
bytecode?: string;
|
||
|
|
runtime?: string
|
||
|
|
};
|
||
|
|
|
||
|
|
export type CompilerOptions = {
|
||
|
|
filename?: string;
|
||
|
|
basedir?: string;
|
||
|
|
optimize?: boolean;
|
||
|
|
throwWarnings?: boolean;
|
||
|
|
};
|
||
|
|
|
||
|
|
export function compile(source: string, options?: CompilerOptions): Array<ContractCode> {
|
||
|
|
options = ethers.utils.shallowCopy(options || { });
|
||
|
|
|
||
|
|
if (options.filename && !options.basedir) {
|
||
|
|
options.basedir = dirname(options.filename);
|
||
|
|
}
|
||
|
|
if (!options.filename) { options.filename = "_contract.sol"; }
|
||
|
|
if (!options.basedir) { options.basedir = "."; }
|
||
|
|
|
||
|
|
let sources: { [ filename: string]: { content: string } } = { };
|
||
|
|
sources[options.filename] = { content: source };
|
||
|
|
|
||
|
|
let input: any = {
|
||
|
|
language: "Solidity",
|
||
|
|
sources: sources,
|
||
|
|
settings: {
|
||
|
|
outputSelection: {
|
||
|
|
"*": {
|
||
|
|
"*": [ "*" ]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
if (options.optimize) {
|
||
|
|
input.settings.optimizer = {
|
||
|
|
enabled: true,
|
||
|
|
runs: 200
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
let findImport = (filename: string): { contents?: string, error?: string } => {
|
||
|
|
try {
|
||
|
|
return {
|
||
|
|
contents: fs.readFileSync(resolve(options.basedir, options.filename)).toString()
|
||
|
|
};
|
||
|
|
} catch (error) {
|
||
|
|
return { error: error.message }
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
let output = JSON.parse(getSolc().compile(JSON.stringify(input), findImport));
|
||
|
|
|
||
|
|
let errors = (output.errors || []).filter((x: any) => (x.severity === "error" || options.throwWarnings)).map((x: any) => x.formattedMessage);
|
||
|
|
if (errors.length) {
|
||
|
|
let error = new Error("compilation error");
|
||
|
|
(<any>error).errors = errors;
|
||
|
|
throw error;
|
||
|
|
}
|
||
|
|
|
||
|
|
let result: Array<ContractCode> = [];
|
||
|
|
for (let filename in output.contracts) {
|
||
|
|
for (let name in output.contracts[filename]) {
|
||
|
|
let contract = output.contracts[filename][name];
|
||
|
|
|
||
|
|
result.push({
|
||
|
|
name: name,
|
||
|
|
interface: new ethers.utils.Interface(contract.abi),
|
||
|
|
bytecode: "0x" + contract.evm.bytecode.object,
|
||
|
|
runtime: "0x" + contract.evm.deployedBytecode.object
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|