contracts/flatten.js at master · Akemona/contracts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
/*

the flattened file is used for manual verification of polgyon/etherscan contracts

*/
var fs = require("fs");
const util = require('util');
const exec = util.promisify(require('child_process').exec);


async function commentCircular() {
    fs.readdir('./contracts', function(err, files) {

        for (let i=0; i < files.length; i++) {

            fs.readFile('./contracts/'+files[i], 'utf8', function (err,data) {
              if (err) {
                return console.log(err);
              }

              var result = data.replace(/(.+) \/\/#circular/g, '//#circular $1');
              fs.writeFile('./contracts/'+files[i], result, 'utf8', function (err) {
                  if (err) return console.log(err);
              });

            });

        }

    });
}


async function uncommentCircular() {
    fs.readdir('./contracts', function(err, files) {

        for (let i=0; i < files.length; i++) {

            fs.readFile('./contracts/'+files[i], 'utf8', function (err,data) {
              if (err) {
                return console.log(err);
              }

              var result = data.replace(/\/\/#circular +(.*)/g, '$1 //#circular');
              fs.writeFile('./contracts/'+files[i], result, 'utf8', function (err) {
                  if (err) return console.log(err);
              });

            });

        }

    });
}

async function removeSPDX() {
    fs.readFile('./flattened/AkemonaCrowdsale.sol', 'utf8', function (err,data) {
        if (err) {
          return console.log(err);
        }

        var license = "// SPDX-License-Identifier: Business Source License 1.1";

        var result = license + "\r\n" + data.replace(/\/\/ SPDX-License-Identifier: MIT/g, '');
        fs.writeFile('./flattened/AkemonaCrowdsale.sol', result, 'utf8', function (err) {
            if (err) return console.log(err);
        });

    });
}

async function flatten() {
    console.log('flattening contracts for etherscan verification');
    console.log('commenting out circular dependencies');
    await commentCircular();
    console.log('generating flat file');
    const { stdout, stderr } = await exec('truffle-flattener contracts/AkemonaCrowdsale.sol > flattened/AkemonaCrowdsale.sol');
    if (stderr) console.log(stderr);
    // verification fails if theres multiple licenses
    console.log('removing SPDX licenses');
    removeSPDX();
    await uncommentCircular();
    console.log('flat file completed');
}
flatten();