Go语言打造以太坊智能合约测试框架（level2）

传送门: 柏链项目学院

---

第二课 智能合约自动化编译

前期内容回顾

以前咱们的介绍的是如何经过solc编译智能合约，而且调用智能合约，本节咱们继续实践，将智能合约的代码自动化编译以及abi文件生成搞定。

咱们须要掌握什么技能呢？

• go语言调用命令行
• toml配置文件处理
• awk工具的使用

go调用命令行

go调用命令行，咱们使用exec包

• LookPath 能够判断一个可执行程序是否存在
• Command 建立一个命令行
• cmd.Run() 运行命令行，也可使用Start()模式，能够去接收管道信息来获得程序返回结果
• 若是是一个shell脚本，那么能够用/bin/bash来启动

toml配置文件处理

TOML的全称是Tom's Obvious, Minimal Language，由于它的做者是GitHub联合创始人Tom Preston-Werner。TOML 的目标是成为一个极简的配置文件格式。TOML 被设计成能够无歧义地被映射为哈希表，从而被多种语言解析。

toml学习教程
在使用的时候，记得要安装toml第三方包。

go get -u github.com/BurntSushi/toml

以后能够根据咱们的须要，来编写配置文件，配置文件的目的仍然是为了让程序运行更灵活，而不该该成为咱们的负担！

awk工具使用

awk实际上是一个语言，unix平台上处理文本的一种语言，其名称得自于它的创始人 Alfred Aho 、Peter Weinberger 和 Brian Kernighan 姓氏的首个字母。该语言的能力十分强大，能够支持字符串处理，打印等操做，工程中对于文本处理要求比较高的环节多会使用awk进行操做。

awk功能举例：

1. factory.txt 是一个工厂内目前产品库存状况，若是数量低于75，须要从新下订单，如何处理？

yekaideMacBook-Pro:awk yk$ cat factory.txt 
ProdA 70
ProdB 85
ProdC 74

示例以下：

awk '{if ($2 < 75) printf("%s reorder\n",$0);if ($2 >= 75) print $0}' factory.txt

1. 查看本系统中shell是bash的用户名，并打印

cat /etc/passwd |grep bash|awk -F ":" '{print $1}'

1. awk处理合约的go文件，将abi信息截取处理存储到文本当中

awk '/const.+ABI = .+/{print substr($4,2,length($4)-2) }' pdbank.go > pdbank.abi

编写自动编译功能

main.go

package main

import (
    "fmt"
    "os"
)

func Usage() {
    fmt.Printf("%s 1  -- compiler code\n", os.Args[0])
    fmt.Printf("%s 2  -- build test code\n", os.Args[0])
}

func main() {
    if len(os.Args) < 2 {
        Usage()
        os.Exit(0)
    }
    if os.Args[1] == "1" {
        CompilerRun()
    } else if os.Args[1] == "1" {
        //build test code
    } else {
        Usage()
        os.Exit(0)
    }

}

接下来开始作填空题，也就是如何编译，咱们先来实现。先编写扫描目录的代码，获取指定目录的sol文件，而后自动化的造成编译命令，送到命令行执行。

扫描指定目录的sol文件

func CompilerRun() error {
    infos, err := ioutil.ReadDir("sol")
    if err != nil {
        fmt.Println("failed to readdir ", err)
        return err
    }
    for _, v := range infos {

        //后4位位.sol
        strNameRune := []rune(v.Name())
        strfix := string(strNameRune[len(strNameRune)-4:])
        if strfix == ".sol" && !v.IsDir() {
            fmt.Println(v.IsDir(), v.Name(), v.Size(), "ok")
            err = CompilerOnece("sol", v.Name(), "contracts")
            if err != nil {
                fmt.Println("call ompilerOnece err", err)
                break
            }
        }
    }
    return err
}

编译函数

//编译一个智能合约
func CompilerOnece(solPath, solName, targetPath string) error {

    //xxx.sol - > xxx.go
    goName := strings.Replace(solName, ".sol", ".go", -1)

    cmd := exec.Command("abigen", "-sol", solPath+"/"+solName, "-pkg", targetPath, "-out", targetPath+"/"+goName)
    return cmd.Run()
}

构建abi函数，咱们须要先用awk实现一个shell脚本，用来处理go文件的abi信息。

func BuildAbi(goCodeName string) error {
    abiName := strings.Replace(goCodeName, ".go", ".abi", -1)
    cmd := exec.Command("/bin/bash", "abi.sh", goCodeName, abiName)
    err := cmd.Run()
    fmt.Println("run BuildAbi ok!!", err)
    return nil
}

abi.sh

filename=$1
targetfile=$2
awk '/const.+ABI = .+/{print substr($4,2,length($4)-2) }' $filename > $targetfile

统一调用处理

func ParseRun() {
    solfiles, err := ParseDir("sol")
    fmt.Println(solfiles, err)
    for _, solfile := range solfiles {
        fmt.Println(solfile)
        codeName, err := Compiler(solfile, "sol", "contracts")
        if err != nil {
            fmt.Println("failed to complie code", err)
            return
        }
        err = BuildAbi(codeName)
        if err != nil {
            fmt.Println("failed to build abi", err)
            return
        }
    }
}

这样咱们的基础工做完成了，可是代码不够完美，咱们须要将部分写死的变量用配置文件来设置，因此再加入toml处理配置文件的部分。

添加config.tomls

[version]
auth = "yekai"
company = "pdj"
buildday = "2019-01-01"
ver = "1.0.0" # 版本

[common]
solidityPath = "sol"
goPath = "contracts"
abiSH = "./abi.sh"

添加config.go

package main

import (
    "log"

    "github.com/BurntSushi/toml"
)

type ServerConfig struct {
    Version VersionInfo
    Common  CommonInfo
}

type VersionInfo struct {
    Auth     string
    Company  string
    BuildDay string
    Ver      string
}

type CommonInfo struct {
    SolidityPath string //智能合约原路径
    GoPath       string //输出go代码路径
    AbiSH        string //处理abi的shell脚本路径
}

var ServConf ServerConfig

func init() {
    getConfig()
}

func getConfig() {
    var servConf ServerConfig
    _, err := toml.DecodeFile("config.toml", &servConf)
    if err != nil {
        log.Panic("faild to decodefile ", err)
    }
    ServConf = servConf
    //fmt.Println(servConf)
}

接下来替换原来的代码部分

package main

import (
    "fmt"
    "io/ioutil"
    "os/exec"
    "strings"
)

//编译一个智能合约
func CompilerOnece(solPath, solName, targetPath string) error {

    //xxx.sol - > xxx.go
    goName := strings.Replace(solName, ".sol", ".go", -1)

    cmd := exec.Command("abigen", "-sol", solPath+"/"+solName, "-pkg", targetPath, "-out", targetPath+"/"+goName)
    return cmd.Run()
}

//构造abi
func BuildAbi(codePath string) error {
    //"contracts/pdbank.go"
    //xxx.sol - > xxx.abi
    abiName := strings.Replace(codePath, ".sol", ".abi", -1)
    goName := strings.Replace(codePath, ".sol", ".go", -1)

    cmd := exec.Command(ServConf.Common.AbiSH, goName, abiName)
    return cmd.Run()
}

//扫描目录，得到所有的文件
func CompilerRun() error {
    infos, err := ioutil.ReadDir(ServConf.Common.SolidityPath)
    if err != nil {
        fmt.Println("failed to readdir ", err)
        return err
    }
    for _, v := range infos {

        //后4位位.sol
        strNameRune := []rune(v.Name())
        strfix := string(strNameRune[len(strNameRune)-4:])
        if strfix == ".sol" && !v.IsDir() {
            fmt.Println(v.IsDir(), v.Name(), v.Size(), "ok")
            err = CompilerOnece(ServConf.Common.SolidityPath, v.Name(), ServConf.Common.GoPath)
            if err != nil {
                fmt.Println("call ompilerOnece err", err)
                break
            }
            //建立abi
            err = BuildAbi(ServConf.Common.GoPath + "/" + v.Name())
            if err != nil {
                fmt.Println("call BuildAbi err", err)
                break
            }
        }
    }
    return err
}

---