sol2uml 2.2.1 → 2.2.2

package/README.md

@@ -137,9 +137,11 @@ Options:
 Usage: sol2uml flatten <contractAddress> [options]
 
 In order for the merged code to compile, the following is done:
-1. File imports are commented out.
-2. "SPDX-License-Identifier" is renamed to "SPDX--License-Identifier".
-3. Contract dependencies are analysed so the files are merged in an order that will compile.
+1. pragma solidity is set using the compiler of the verified contract.
+2. All pragma solidity lines in the source files are commented out.
+3. File imports are commented out.
+4. "SPDX-License-Identifier" is renamed to "SPDX--License-Identifier".
+5. Contract dependencies are analysed so the files are merged in an order that will compile.
 
 Merges verified source files for a contract from a Blockchain explorer into one local file.
 

package/lib/parserEtherscan.d.ts

@@ -42,6 +42,7 @@ export declare class EtherscanParser {
             filename: string;
         }[];
         contractName: string;
+        compilerVersion: string;
     }>;
 }
 export {};

package/lib/parserEtherscan.js

@@ -8,6 +8,7 @@ const axios_1 = __importDefault(require("axios"));
 const parser_1 = require("@solidity-parser/parser");
 const converterAST2Classes_1 = require("./converterAST2Classes");
 const filterClasses_1 = require("./filterClasses");
+const regEx_1 = require("./utils/regEx");
 require('axios-debug-log');
 const debug = require('debug')('sol2uml');
 exports.networks = [
@@ -123,7 +124,7 @@ class EtherscanParser {
      * @return Promise string of Solidity code
      */
     async getSolidityCode(contractAddress) {
-        const { files, contractName } = await this.getSourceCode(contractAddress);
+        const { files, contractName, compilerVersion } = await this.getSourceCode(contractAddress);
         // Parse the UmlClasses from the Solidity code in each file
         let umlClasses = [];
         for (const file of files) {
@@ -141,7 +142,8 @@ class EtherscanParser {
         const nonDependentFiles = files.filter((f) => !dependentFilenames.includes(f.filename));
         const nonDependentFilenames = nonDependentFiles.map((f) => f.filename);
         debug(`Failed to find dependencies to files: ${nonDependentFilenames}`);
-        let solidityCode = '';
+        const solidityVersion = (0, regEx_1.parseSolidityVersion)(compilerVersion);
+        let solidityCode = `pragma solidity = ${solidityVersion};\n`;
         // output non dependent code before the dependent files just in case sol2uml missed some dependencies
         const filenames = [...nonDependentFilenames, ...dependentFilenames];
         // For each filename
@@ -150,11 +152,13 @@ class EtherscanParser {
             const file = files.find((f) => f.filename === filename);
             if (!file)
                 throw Error(`Failed to find file with filename "${filename}"`);
+            // comment out any pragma solidity lines as its set from the compiler version
+            const removedPragmaSolidity = file.code.replace(/(\s)(pragma\s+solidity.*;)/gm, '$1/* $2 */');
             // comment out any import statements
             // match whitespace before import
             // and characters after import up to ;
             // replace all in file and match across multiple lines
-            const removedImports = file.code.replace(/(\s)(import.*;)/gm, '$1/* $2 */');
+            const removedImports = removedPragmaSolidity.replace(/(\s)(import.*;)/gm, '$1/* $2 */');
             // Rename SPDX-License-Identifier to SPDX--License-Identifier so the merged file will compile
             const removedSPDX = removedImports.replace(/SPDX-/, 'SPDX--');
             solidityCode += removedSPDX;
@@ -241,6 +245,7 @@ class EtherscanParser {
             return {
                 files: results.flat(1),
                 contractName: response.data.result[0].ContractName,
+                compilerVersion: response.data.result[0].CompilerVersion,
             };
         }
         catch (err) {

package/lib/sol2uml.js

@@ -156,9 +156,11 @@ program
     .usage(`<contractAddress> [options]
 
 In order for the merged code to compile, the following is done:
-1. File imports are commented out.
-2. "SPDX-License-Identifier" is renamed to "SPDX--License-Identifier".
-3. Contract dependencies are analysed so the files are merged in an order that will compile.`)
+1. pragma solidity is set using the compiler of the verified contract.
+2. All pragma solidity lines in the source files are commented out.
+3. File imports are commented out.
+4. "SPDX-License-Identifier" is renamed to "SPDX--License-Identifier".
+5. Contract dependencies are analysed so the files are merged in an order that will compile.`)
     .argument('<contractAddress>', 'Contract address in hexadecimal format with a 0x prefix.')
     .action(async (contractAddress, options, command) => {
     try {

package/lib/utils/regEx.d.ts

@@ -1 +1,2 @@
 export declare const isAddress: (input: string) => boolean;
+export declare const parseSolidityVersion: (compilerVersion: string) => string;

package/lib/utils/regEx.js

@@ -1,8 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isAddress = void 0;
+exports.parseSolidityVersion = exports.isAddress = void 0;
 const isAddress = (input) => {
     return input.match(/^0x([A-Fa-f0-9]{40})$/) !== null;
 };
 exports.isAddress = isAddress;
+const parseSolidityVersion = (compilerVersion) => {
+    const result = compilerVersion.match(`v(\\d+.\\d+.\\d+)`);
+    if (result[1]) {
+        return result[1];
+    }
+    throw Error(`Failed to parse compiler version ${compilerVersion}`);
+};
+exports.parseSolidityVersion = parseSolidityVersion;
 //# sourceMappingURL=regEx.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sol2uml",
-  "version": "2.2.1",
+  "version": "2.2.2",
   "description": "Solidity contract visualisation tool.",
   "main": "./lib/index.js",
   "types": "./lib/index.d.ts",