github.com/turingchain2020/turingchain@v1.1.21/cmd/autotest/types/utils.go

// Copyright Turing Corp. 2018 All Rights Reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package types

import (
	"encoding/json"
	"errors"
	"fmt"
	"os/exec"
	"strconv"
	"strings"

	"github.com/turingchain2020/turingchain/common/log/log15"
)

//FloatDiff const
const FloatDiff = 0.00001

//AutoTestLogFormat customize log15 log format
func AutoTestLogFormat() log15.Format {

	logfmt := log15.LogfmtFormat()

	return log15.FormatFunc(func(r *log15.Record) []byte {

		if r.Msg == "PrettyJsonLogFormat" && len(r.Ctx) == 4 {

			b, ok := r.Ctx[3].([]byte)
			if ok {
				//return raw json data directly
				return b
			}
		}

		return logfmt.Format(r)
	})

}

//RunTuringchainCli invoke turingchain client
func RunTuringchainCli(para []string) (string, error) {

	rawOut, err := exec.Command(CliCmd, para[0:]...).CombinedOutput()

	strOut := string(rawOut)

	return strOut, err
}

//IsBalanceEqualFloat according to the accuracy of coins balance
func IsBalanceEqualFloat(f1 float64, f2 float64) bool {

	if (f2-f1 < FloatDiff) && (f1-f2 < FloatDiff) {
		return true
	}

	return false
}

func checkTxHashValid(txHash string) bool {

	return len(txHash) == 66 && strings.HasPrefix(txHash, "0x")
}

//SendTxCommand excute
func SendTxCommand(cmd string) (string, bool) {

	output, err := RunTuringchainCli(strings.Fields(cmd))
	if err != nil {
		return err.Error(), false
	} else if len(output) == 67 {
		output = output[0 : len(output)-1]
	} else {

		//check if privacy transaction
		var jsonMap map[string]interface{}
		err = json.Unmarshal([]byte(output), &jsonMap)
		if err != nil {
			return output, false
		}
		if hash, ok := jsonMap["hash"].(string); ok {
			output = hash
		}
	}

	return output, checkTxHashValid(output)
}

//SendPrivacyTxCommand 隐私交易执行回执哈希为json格式，需要解析
func SendPrivacyTxCommand(cmd string) (string, bool) {

	output, err := RunTuringchainCli(strings.Fields(cmd))

	if err != nil {
		return err.Error(), false
	}

	if strings.Contains(output, "Err") {
		return output, false
	}

	var jsonMap map[string]interface{}
	err = json.Unmarshal([]byte(output), &jsonMap)
	if err != nil {
		return output, false
	}
	output = jsonMap["hash"].(string)

	return output, checkTxHashValid(output)
}

//GetTxRecpTyname get tx query -s TxHash
func GetTxRecpTyname(txInfo map[string]interface{}) (tyname string, bSuccess bool) {

	tyname = txInfo["receipt"].(map[string]interface{})["tyName"].(string)

	bSuccess = false

	if tyname == "ExecOk" {
		bSuccess = true
	}

	return tyname, bSuccess
}

//GetTxInfo get tx receipt with tx hash code if exist
func GetTxInfo(txHash string) (string, bool) {

	bReady := false
	txInfo, err := RunTuringchainCli(strings.Fields(fmt.Sprintf("tx query -s %s", txHash)))

	if err == nil && txInfo != "tx not exist\n" {

		bReady = true
	} else if err != nil {

		txInfo = err.Error()
	}

	return txInfo, bReady
}

//CheckBalanceDeltaWithAddr diff balance
func CheckBalanceDeltaWithAddr(log map[string]interface{}, addr string, delta float64) bool {

	logAddr := log["current"].(map[string]interface{})["addr"].(string)
	prev, _ := strconv.ParseFloat(log["prev"].(map[string]interface{})["balance"].(string), 64)
	curr, _ := strconv.ParseFloat(log["current"].(map[string]interface{})["balance"].(string), 64)

	logDelta := (curr - prev) / 1e8

	return (logAddr == addr) && (IsBalanceEqualFloat(logDelta, delta))
}

//CheckFrozenDeltaWithAddr check
func CheckFrozenDeltaWithAddr(log map[string]interface{}, addr string, delta float64) bool {

	logAddr := log["current"].(map[string]interface{})["addr"].(string)
	prev, _ := strconv.ParseFloat(log["prev"].(map[string]interface{})["frozen"].(string), 64)
	curr, _ := strconv.ParseFloat(log["current"].(map[string]interface{})["frozen"].(string), 64)

	logDelta := (curr - prev) / 1e8

	return (logAddr == addr) && (IsBalanceEqualFloat(logDelta, delta))
}

//CalcTxUtxoAmount calculate total amount in tx in/out utxo set, key = ["keyinput" | "keyoutput"]
func CalcTxUtxoAmount(log map[string]interface{}, key string) float64 {

	if log[key] == nil {
		return 0
	}

	utxoArr := log[key].([]interface{})
	var totalAmount float64

	for i := range utxoArr {

		temp, _ := strconv.ParseFloat(utxoArr[i].(map[string]interface{})["amount"].(string), 64)
		totalAmount += temp
	}

	return totalAmount / 1e8
}

//CalcUtxoAvailAmount calculate available utxo with specific addr and TxHash
func CalcUtxoAvailAmount(addr string, txHash string) (float64, error) {

	outStr, err := RunTuringchainCli(strings.Fields(fmt.Sprintf("privacy showpai -d 1 -a %s", addr)))

	if err != nil {
		return 0, err
	}
	var jsonMap map[string]interface{}
	err = json.Unmarshal([]byte(outStr), &jsonMap)

	if err != nil {
		return 0, err
	}

	var totalAmount float64
	if jsonMap["AvailableDetail"] == nil {
		return 0, nil
	}

	availArr := jsonMap["AvailableDetail"].([]interface{})

	for i := range availArr {

		if availArr[i].(map[string]interface{})["Txhash"].(string) == txHash {

			temp, _ := strconv.ParseFloat(availArr[i].(map[string]interface{})["Amount"].(string), 64)
			totalAmount += temp
		}
	}

	return totalAmount, nil
}

//CalcUtxoSpendAmount calculate spend utxo with specific addr and TxHash
func CalcUtxoSpendAmount(addr string, txHash string) (float64, error) {

	outStr, err := RunTuringchainCli(strings.Fields(fmt.Sprintf("privacy showpas -a %s", addr)))

	if strings.Contains(outStr, "Err") {
		return 0, errors.New(outStr)
	}

	idx := strings.Index(outStr, "\n") + 1

	if err != nil {
		return 0, err
	}
	var jsonArr []interface{}
	err = json.Unmarshal([]byte(outStr[idx:]), &jsonArr)

	if err != nil {
		return 0, err
	}

	var totalAmount float64

	for i := range jsonArr {

		if jsonArr[i].(map[string]interface{})["Txhash"].(string) == txHash[2:] {

			spendArr := jsonArr[i].(map[string]interface{})["Spend"].([]interface{})
			for i := range spendArr {

				temp, _ := strconv.ParseFloat(spendArr[i].(map[string]interface{})["Amount"].(string), 64)
				totalAmount += temp
			}

			break
		}
	}

	return totalAmount, nil
}