Generate boilerplate or no code deploy common contracts #77

abi_wrapper.go

@@ -5,7 +5,7 @@ import (
 	"os"
 	"strings"
 
-	abis "github.com/gochain-io/web3/abi"
+	"github.com/gochain-io/web3/assets"
 
 	"github.com/gochain-io/gochain/v3/accounts/abi"
 )
@@ -34,5 +34,5 @@ func readAbi(reader io.Reader) (*abi.ABI, error) {
 }
 
 var bundledContracts = map[string]string{
-	"erc20":  abis.ERC20,
-	"erc721": abis.ERC721}
+	"erc20":  assets.ERC20,
+	"erc721": assets.ERC721}

abi/erc20.go → assets/erc20.go

@@ -1,5 +1,32 @@
-package abis
+package assets
 
+import (
+	"math/big"
+)
+
+type Erc20Params struct {
+	Symbol      string
+	TokenName   string
+	Decimals    int
+	TotalSupply *big.Int
+}
+
+const ERC20_CAPPED_PAUSABLE_TEMPLATE = `pragma solidity ^0.5.2;
+
+import "./lib/oz/contracts/token/ERC20/ERC20Pausable.sol";
+import "./lib/oz/contracts/token/ERC20/ERC20Capped.sol";
+import "./lib/oz/contracts/token/ERC20/ERC20Detailed.sol";
+
+contract CappedToken is ERC20Detailed, ERC20Capped, ERC20Pausable {
+
+    // ------------------------------------------------------------------------
+    // Constructor
+    // ------------------------------------------------------------------------
+    constructor() 
+    ERC20Detailed("{{.TokenName}}", "{{.Symbol}}", {{.Decimals}}) 
+    ERC20Capped({{.TotalSupply}})
+    public {}
+}`
 const ERC20 = `[
 	{
 		"constant": true,

abi/erc721.go → assets/erc721.go

@@ -1,4 +1,4 @@
-package abis
+package assets
 
 const ERC721 = `[
 	{

cmd/web3/main.go

@@ -5,12 +5,15 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"html/template"
 	"io/ioutil"
 	"log"
 	"math/big"
 	"os"
+	"os/exec"
 	"os/signal"
 	"sort"
+	"strconv"
 	"strings"
 	"syscall"
 	"time"
@@ -20,6 +23,7 @@ import (
 	"github.com/gochain-io/gochain/v3/common/hexutil"
 	"github.com/gochain-io/gochain/v3/core/types"
 	"github.com/gochain-io/web3"
+	"github.com/gochain-io/web3/assets"
 	"github.com/urfave/cli"
 )
 
@@ -360,6 +364,19 @@ func main() {
 				}
 			},
 		},
+		{
+			Name:  "generate",
+			Usage: "Generate a contract",
+			Subcommands: []cli.Command{
+				{
+					Name:  "erc20",
+					Usage: "Generate a erc20 contract(use the following list of arguments after the command - SYMBOL TOKEN_NAME DECIMALS TOTAL_SUPPLY)",
+					Action: func(c *cli.Context) {
+						GenerateContract(ctx, "erc20", c.Args())
+					},
+				},
+			},
+		},
 	}
 	app.Run(os.Args)
 }
@@ -871,6 +888,54 @@ func Send(ctx context.Context, rpcURL, privateKey, toAddress, amount string) {
 	fmt.Println("Transaction address:", tx.Hash.Hex())
 }
 
+func GenerateContract(ctx context.Context, contractType string, args []string) {
+	if _, err := os.Stat("lib/oz"); os.IsNotExist(err) {
+		cmd := exec.Command("git", "clone", "--depth", "1", "--branch", "master", "https://github.com/OpenZeppelin/openzeppelin-solidity", "lib/oz")
+		log.Printf("Cloning OpenZeppeling repo...")
+		cmd.Stdout = os.Stdout
+		cmd.Stderr = os.Stderr
+		err := cmd.Run()
+		if err != nil {
+			log.Fatalf("Cloning finished with error: %v", err)
+		}
+	}
+	if contractType == "erc20" {
+		if len(args) != 4 {
+			log.Fatalln("wrong amount of the parameters, the following parameters are mandatory SYMBOL TOKEN_NAME DECIMALS TOTAL_SUPPLY")
+		}
+		decimals, err := strconv.Atoi(args[2])
+		if err != nil {
+			log.Fatalln("Cannot parse DECIMALS", err)
+		}
+		totalSupply, ok := new(big.Int).SetString(args[3], 10)
+		if !ok {
+			log.Fatalln("Cannot parse total supply")
+		}
+		totalSupply.Mul(totalSupply, new(big.Int).Exp(big.NewInt(10), big.NewInt(int64(decimals)), nil))
+		params := assets.Erc20Params{
+			Symbol:      args[0],
+			TokenName:   args[1],
+			Decimals:    decimals,
+			TotalSupply: totalSupply,
+		}
+		tmpl, err := template.New("contract").Parse(assets.ERC20_CAPPED_PAUSABLE_TEMPLATE)
+		if err != nil {
+			log.Fatalf("Cannot parse the template: %v", err)
+		}
+		f, err := os.Create(params.TokenName + ".sol")
+		if err != nil {
+			log.Fatalf("Cannot create the file: %v", err)
+			return
+		}
+		err = tmpl.Execute(f, params)
+		if err != nil {
+			log.Fatalf("Cannot execute the template: %v", err)
+			return
+		}
+		fmt.Println("The sample contract has been successfully written to", params.TokenName+".sol", "file")
+	}
+}
+
 func printReceiptDetails(r *web3.Receipt, myabi *abi.ABI) {
 	var logs []web3.Event
 	var err error