ktogame/util/ktoFunction.go

package util

import (
	"context"
	"encoding/hex"
	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/logs"
	"github.com/ethereum/go-ethereum/accounts/abi"
	"github.com/ethereum/go-ethereum/accounts/abi/bind"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/ethclient"
	utilAddress "github.com/korthochain/korthochain/pkg/address"
	"github.com/korthochain/korthochain/pkg/block"
	pb "github.com/korthochain/korthochain/pkg/server/grpcserver/message"
	"github.com/korthochain/korthochain/pkg/transaction"
	"github.com/shopspring/decimal"
	"google.golang.org/grpc"
	"os"
	"server_fhl/dbUtil"
	"server_fhl/enums"
	"server_fhl/models"
	"strings"
	"time"
)

var ktoClient pb.GreeterClient
var conn *ethclient.Client
var ktoRpc = beego.AppConfig.String("hostHttp_mit_kto")
var ethRpc = beego.AppConfig.String("miner_eth")
var IdWork *Snowflake
var kx int64

func init(){
	fmt.Println("a3")
	IdWork, _ = NewSnowflake(0)
	kc, err := grpc.Dial(ktoRpc,grpc.WithInsecure(),grpc.WithBlock())
	if err!=nil {
		logs.Info(err)
		os.Exit(1)
	}
	ktoClient = pb.NewGreeterClient(kc)
	utilAddress.SetNetWork("mainnet")
	conn,err = ethclient.Dial(ethRpc)
	if err!= nil {
		fmt.Printf("Failed to connect to etherum client: %v\n",err)
	}
	go getNewBlock()
	go listenKto()
	go cheakWithdrawal()
	//go fenfa()
}

func listenKto(){
	time.Sleep(time.Second*2)
	kxredis := dbUtil.GetValue(enums.USE_FHL_BLOCK,1)
	if kxredis.(int64)==0 {
		nb := dbUtil.GetValue(enums.MAX_FHL_BLOCK,1)
		kx = nb.(int64)
	}else{
		kx = kxredis.(int64)
	}
	//kx =62203394
	for {
		nb := dbUtil.GetValue(enums.MAX_FHL_BLOCK,1)
		fmt.Println("当前=",kx,"最新=",nb.(int64),"相差=",nb.(int64)-kx)
		if nb.(int64)<kx+1 {
			time.Sleep(time.Second*10)
			continue
		}
		func(){
			reqBlock := new(pb.ReqBlockByNumber)
			reqBlock.Height = uint64(kx)
			bl, err := ktoClient.GetBlockByNum(context.Background(), reqBlock)
			if err!=nil || bl.Code !=0{
				fmt.Println("获取交易block错误=",err)
				time.Sleep(time.Second*5)
				return
			}
			blc, errs := block.Deserialize(bl.Data)
			if errs!=nil {
				time.Sleep(time.Second*5)
				fmt.Println("获取交易blc错误=",errs)
			}
			for _, v := range blc.Transactions {
				evm, errLog := transaction.DecodeEvmData(v.Input)
				if errLog!=nil {
					continue
				}
				if len(evm.Logs)!=1 {
					continue
				}
				for _, l := range evm.Logs {

					th :=l.Topics[0].Hex()
					method:=dbUtil.TopsMap[th]
					addr := l.Address.String()
					if method==enums.NEWDEPOSIT  && addr==enums.USDTMINTCONTRACT{
						var ev models.Ihot
						abi, err := abi.JSON(strings.NewReader(FhlMetaData.ABI))
						if err != nil {
							logs.Error("abi err1=", err)
						}
						err = abi.UnpackIntoInterface(&ev, method, l.Data)
						if err != nil {
							logs.Error("abi err2=", err)
						}
						fmt.Printf("data=%+v\n ", ev)
						usdrido(ev.User.String(),ev.Referrer.String(),v.HashToString(),decimal.NewFromBigInt(ev.Amount,-11),decimal.NewFromBigInt(ev.Price,-18),kx)
					}else if method==enums.WITHDRAWAL  && addr==enums.WITHDRAWCONTRACT{
						var ev models.Withdraw
						abi, err := abi.JSON(strings.NewReader(FhlMetaData.ABI))
						if err != nil {
							logs.Error("abi err1=", err)
						}
						err = abi.UnpackIntoInterface(&ev, method, l.Data)
						if err != nil {
							logs.Error("abi err2=", err)
						}
						fmt.Printf("data=%+v\n ", ev)
						withdrawal(kx,"0x"+v.HashToString(),ev.User.String(),decimal.NewFromBigInt(ev.Amount,-18))
					}
				}
			}
		}()
		kx++
		dbUtil.SetValue(enums.USE_FHL_BLOCK,kx)
		continue
		//return
	}
}

func getNewBlock(){
	for  {
		number := new(pb.ReqMaxBlockHeight)
		num, err := ktoClient.GetMaxBlockHeight(context.Background(), number)
		if err!=nil {
			logs.Info("断开重连2",err)
			time.Sleep(time.Second * 10)
			continue
		}else{
			dbUtil.SetValue(enums.MAX_FHL_BLOCK, int64(num.MaxHeight))
		}
		time.Sleep(time.Second * 10)
	}
}

func usdrido(addr,bindAddr,hash string,da,price decimal.Decimal,block int64 ){
	fmt.Println("==进来==",addr)
	//当前用户
	var user models.UserAddr

	//上级
	var bindUser models.UserAddr
	dbUtil.Engine.Id(bindAddr).Get(&bindUser)

	//上上级
	var bossUser models.UserAddr
	bb, _ := dbUtil.Engine.Where("id_flag=?", bindUser.Pid).Get(&bossUser)

	//原始上上级
	var realBossUser models.UserAddr
	var rb bool
	if bindUser.Pid!=bindUser.RealPid {
		rb, _ = dbUtil.Engine.Where("id_flag=?", bindUser.RealPid).Get(&realBossUser)
	}

	//上上上级
	var pbossUser models.UserAddr
	pbb, _ := dbUtil.Engine.Where("id_flag=?", bossUser.Pid).Get(&pbossUser)

	//上上上上级
	var ppbossUser models.UserAddr
	ppbb, _ := dbUtil.Engine.Where("id_flag=?", pbossUser.Pid).Get(&ppbossUser)

	var com models.CommonInfo
	dbUtil.Engine.Get(&com)
	//1.处理用户
	var msxSort models.UserAddr
	dbUtil.Engine.Desc("id_flag").Get(&msxSort) //最大序号
	user.Addr = addr
	user.IdFlag = msxSort.IdFlag+1
	//上级是boss，不转移
	if bindUser.Level>0 || bindUser.Pid==-1{
		user.Pid = bindUser.IdFlag
		user.RealPid = bindUser.IdFlag
		user.PidFlag = bindUser.PidFlag+","+fmt.Sprint(bindUser.IdFlag)
	}else{
		user.Pid = bossUser.IdFlag
		user.RealPid = bindUser.IdFlag
		user.PidFlag = bossUser.PidFlag+","+fmt.Sprint(bossUser.IdFlag)
	}
	user.Msg = "暂无惩罚"
	user.BadTime =NowTimeString()
	user.CreateTime =NowTimeString()
	user.Amount,_ = da.Float64()
	user.Price,_= price.Float64()
	user.Block = block
	user.Hash = hash
	//-----------处理用户结束--------------

	nt := time.Now().Unix()
	var sbbu,sbbo,sbpb,sbpbp models.SuccessBill
	//2.处理用户的直推
	var isUp bool
	if bindUser.Pid!=-1 {
		te, _ := time.ParseInLocation("2006-01-02 15:04:05", bindUser.BadTime,time.Local)
		sbbu = models.SuccessBill{IdWork.Generate(),bindUser.Addr,0,1,0,"",addr,1,NowTimeString()}
		if bindUser.Level>0 && te.Unix()<nt { //在处罚中
			sbbu.Amount,_ = da.Mul(decimal.NewFromFloat(0.3)).Float64()
			bindUser.AmountLv,_= decimal.NewFromFloat(bindUser.AmountLv).Add(decimal.NewFromFloat(0.3)).Float64()
		}else{
			sbbu.Amount,_ = da.Mul(decimal.NewFromFloat(0.5)).Float64()
			bindUser.AmountLv,_= decimal.NewFromFloat(bindUser.AmountLv).Add(decimal.NewFromFloat(0.5)).Float64()
		}
		bindUser.UseAmount,_=decimal.NewFromFloat(bindUser.UseAmount).Add(decimal.NewFromFloat(sbbu.Amount)).Float64()
		bindUser.Profit1,_ = decimal.NewFromFloat(bindUser.Profit1).Add(decimal.NewFromFloat(sbbu.Amount)).Float64()
		if bindUser.RemovePeople==0 {
			bindUser.RemovePeople = bindUser.RemovePeople+1
		}else if bindUser.RemovePeople==1{
			isUp = true
			bindUser.RemovePeople = bindUser.RemovePeople+1
			if bindUser.Level<3 {
				bindUser.Level = 1
			}
			format := time.Now().AddDate(0, 0, 7).Format("2006-01-02 15:04:05")
			bindUser.BadTime = format
			bindUser.Msg = "当给上级boss转移的2位员工发生直推时，将重置惩罚时间"
		}else{
			bindUser.People = bindUser.People+1
		}
	}else{
		mul,_ := da.Mul(decimal.NewFromFloat(0.5)).Float64()
		sbbu = models.SuccessBill{IdWork.Generate(),bindUser.Addr,mul,1,0,"",addr,1,NowTimeString()}
		bindUser.Profit1,_ = decimal.NewFromFloat(bindUser.Profit1).Add(decimal.NewFromFloat(mul)).Float64()
		bindUser.UseAmount,_=decimal.NewFromFloat(bindUser.UseAmount).Add(decimal.NewFromFloat(mul)).Float64()
	}

	//-----------处理上级结束--------------

	//3.处理上上级
	if bb {
		te, _ := time.ParseInLocation("2006-01-02 15:04:05", bossUser.BadTime,time.Local)
		sbbo = models.SuccessBill{IdWork.Generate(),bossUser.Addr,0,1,0,"",addr,2,NowTimeString()}
		if bossUser.Pid!=-1 {
			if bossUser.Level>0 && te.Unix()<nt { //在处罚中
				sbbo.Amount,_ = da.Mul(decimal.NewFromFloat(0.1)).Float64()
				bossUser.AmountLv,_= decimal.NewFromFloat(bossUser.AmountLv).Add(decimal.NewFromFloat(0.1)).Float64()
			}else{
				sbbo.Amount,_ = da.Mul(decimal.NewFromFloat(0.3)).Float64()
				bossUser.AmountLv,_= decimal.NewFromFloat(bossUser.AmountLv).Add(decimal.NewFromFloat(0.3)).Float64()
			}
			bossUser.Profit2,_ = decimal.NewFromFloat(bossUser.Profit2).Add(decimal.NewFromFloat(sbbo.Amount)).Float64()
			bossUser.UseAmount,_=decimal.NewFromFloat(bossUser.UseAmount).Add(decimal.NewFromFloat(sbbo.Amount)).Float64()
			if isUp {
				if bb {
					if bossUser.TeamBoss>0 && bossUser.TeamBoss<3 {
						bossUser.Level = bossUser.TeamBoss+1
					}
					bossUser.TeamBoss = bossUser.TeamBoss+1
				}
			}
		}else{
			f,_:= da.Mul(decimal.NewFromFloat(0.3)).Float64()
			sbbo.Amount = f
			bossUser.UseAmount,_=decimal.NewFromFloat(bossUser.UseAmount).Add(decimal.NewFromFloat(f)).Float64()
			bossUser.Profit2,_ = decimal.NewFromFloat(bossUser.Profit2).Add(decimal.NewFromFloat(f)).Float64()
		}
	}

	//-----------处理上上级结束--------------

	//4.处理原始上上级
	if rb {
		var rpUser []models.UserAddr
		dbUtil.Engine.Where("real_pid=? and pid!=real_pid",bindUser.RealPid).Find(&rpUser)
		if realBossUser.RemovePeople == 2 {
			e1:= rpUser[0].RemovePeople
			e2:= rpUser[1].RemovePeople
			if bindUser.IdFlag==rpUser[0].IdFlag {
				e1 = e1+1
			}
			if bindUser.IdFlag==rpUser[1].IdFlag {
				e2 = e2+1
			}
			if e1+e2>1 && e1!=0 && e2!=0{
				if e1>0 && e2>0 && (e1<2 || e2<2){
					format := time.Now().AddDate(0, 0, 7).Format("2006-01-02 15:04:05")
					realBossUser.BadTime = format
					realBossUser.Msg = "当给上级boss转移的2位员工都成为BOSS时，您将永久解除惩罚"
				}else if e1>1 && e2>1{
					realBossUser.BadTime = "2099-01-01 08:00:00"
					realBossUser.Msg = "您已永久解除惩罚"
				}
			}
		}
	}
	//-----------处理原始上上级结束--------------

	//5.处理上上上级
	if pbb {
		te, _ := time.ParseInLocation("2006-01-02 15:04:05", pbossUser.BadTime,time.Local)
		sbpb = models.SuccessBill{IdWork.Generate(),pbossUser.Addr,0,1,0,"",addr,3,NowTimeString()}
		if pbossUser.Pid!=-1{
			if pbb {
				ru1:= da.Mul(decimal.NewFromFloat(0.04))
				if pbossUser.Level>0 && te.Unix()<nt { //在处罚中
					pbossUser.LockAmount2,_=decimal.NewFromFloat(pbossUser.LockAmount2).Add(ru1).Float64()
				}else{
					sbpb.Amount,_ = ru1.Float64()
					pbossUser.AmountLv,_= decimal.NewFromFloat(pbossUser.AmountLv).Add(decimal.NewFromFloat(0.04)).Float64()
				}
				pbossUser.UseAmount,_=decimal.NewFromFloat(pbossUser.UseAmount).Add(decimal.NewFromFloat(sbpb.Amount)).Float64()
				pbossUser.Profit3,_ = decimal.NewFromFloat(pbossUser.Profit3).Add(decimal.NewFromFloat(sbpb.Amount)).Float64()
			}
		}else{
			mul,_ := da.Mul(decimal.NewFromFloat(0.04)).Float64()
			sbpb.Amount = mul
			pbossUser.UseAmount,_=decimal.NewFromFloat(pbossUser.UseAmount).Add(decimal.NewFromFloat(mul)).Float64()
			pbossUser.Profit3,_ = decimal.NewFromFloat(pbossUser.Profit3).Add(decimal.NewFromFloat(mul)).Float64()
		}
	}


	//5.处理上上上上级
	if ppbb {
		te, _ := time.ParseInLocation("2006-01-02 15:04:05", ppbossUser.BadTime,time.Local)
		sbpbp = models.SuccessBill{IdWork.Generate(),ppbossUser.Addr,0,1,0,"",addr,4,NowTimeString()}
		ru1:= da.Mul(decimal.NewFromFloat(0.06))
		if ppbossUser.Pid!=-1{
			if ppbb {
				if ppbossUser.Level>0 && te.Unix()<nt { //在处罚中
					ppbossUser.LockAmount2,_=decimal.NewFromFloat(ppbossUser.LockAmount2).Add(ru1).Float64()
				}else{
					sbpbp.Amount,_ = ru1.Float64()
					ppbossUser.AmountLv,_= decimal.NewFromFloat(ppbossUser.AmountLv).Add(decimal.NewFromFloat(0.06)).Float64()
				}
				ppbossUser.UseAmount,_=decimal.NewFromFloat(ppbossUser.UseAmount).Add(decimal.NewFromFloat(sbpbp.Amount)).Float64()
				ppbossUser.Profit3,_ = decimal.NewFromFloat(ppbossUser.Profit3).Add(decimal.NewFromFloat(sbpbp.Amount)).Float64()
			}
		}else{
			sbpbp.Amount,_ = ru1.Float64()
			ppbossUser.UseAmount,_=decimal.NewFromFloat(ppbossUser.UseAmount).Add(ru1).Float64()
			ppbossUser.Profit3,_ = decimal.NewFromFloat(ppbossUser.Profit3).Add(ru1).Float64()
		}
	}

	dbUtil.Engine.Insert(&user)
	dbUtil.Engine.ID(bindUser.Addr).Update(&bindUser)

	if bb {
		dbUtil.Engine.ID(bossUser.Addr).Update(&bossUser)
	}
	if rb {
		dbUtil.Engine.ID(realBossUser.Addr).Update(&realBossUser)
	}

	if pbb {
		dbUtil.Engine.ID(pbossUser.Addr).Update(&pbossUser)
	}
	if ppbb {
		dbUtil.Engine.ID(ppbossUser.Addr).Update(&ppbossUser)
	}
	var suc decimal.Decimal
	if sbbu.Amount>0 {
		suc = suc.Add(decimal.NewFromFloat(sbbu.Amount))
		dbUtil.Engine.Insert(&sbbu)
	}

	if sbbo.Amount>0 {
		suc = suc.Add(decimal.NewFromFloat(sbbo.Amount))
		dbUtil.Engine.Insert(&sbbo)
	}

	if sbpb.Amount>0 {
		suc = suc.Add(decimal.NewFromFloat(sbpb.Amount))
		dbUtil.Engine.Insert(&sbpb)
	}

	if sbpbp.Amount>0 {
		suc = suc.Add(decimal.NewFromFloat(sbpbp.Amount))
		dbUtil.Engine.Insert(&sbpbp)
	}
	sub := da.Sub(suc)
	com.Amount,_=decimal.NewFromFloat(com.Amount).Add(sub).Float64()
	dbUtil.Engine.ID(com.Id).Update(&com)
}

func Sign(amountStr,addr,timeStr string)string{
	keccakA := crypto.Keccak256([]byte(amountStr),[]byte(enums.DIGIT))
	keccakS := crypto.Keccak256(keccakA, []byte(timeStr),[]byte(addr))
	return "0x"+hex.EncodeToString(keccakS)
}

//提币
func withdrawal(kx int64,hash,addr string,amount decimal.Decimal){
	var ua models.UserAddr
	b, _ := dbUtil.Engine.Id(addr).Get(&ua)
	if !b {
		return
	}
	f,_:= amount.Float64()
	var wb models.WithdrawalBill
	bw, _ := dbUtil.Engine.Where("addr=? and status=0 and hash=?", addr, hash).Get(&wb)
	if !bw {
		bill := models.WithdrawalBill{IdWork.Generate(), addr, f, 1, kx,kx, hash, NowTimeString()}
		dbUtil.Engine.Insert(&bill)
		ua.UseAmount,_ = decimal.NewFromFloat(ua.UseAmount).Sub(amount).Float64()
		dbUtil.Engine.ID(ua.Addr).Cols("use_amount").Update(&ua)
		return
	}
	wb.Status = 1
	wb.SucBlock = kx
	dbUtil.Engine.ID(wb.Id).Cols("status,suc_block").Update(&wb)
}

func cheakWithdrawal(){
	for {
		h, _ := time.ParseDuration("-1200s")
		format := time.Now().AddDate(0, 0, 0).Add(h).Format("2006-01-02 15:04:05")
		var wb []models.WithdrawalBill
		dbUtil.Engine.Where("status=0 AND create_block<? AND create_time<?",kx-int64(30),format).Find(&wb)
		for _, v := range wb {
			req1 := new(pb.ReqTxByHash)
			req1.Hash=v.Hash
			data, err := ktoClient.GetTxByHash(context.Background(), req1)
			if err!=nil {
				logs.Error("addr=",v.Addr,"block=",kx,"err=",err)
				continue
			}
			tx, err := transaction.DeserializeFinishedTransaction(data.Data)
			if err!=nil {
				v.Status = -1
				dbUtil.Engine.ID(v.Id).Cols("status").Update(&v)
				var ua models.UserAddr
				dbUtil.Engine.Id(v.Addr).Get(&ua)
				ua.UseAmount,_ = decimal.NewFromFloat(ua.UseAmount).Add( decimal.NewFromFloat(v.Amount)).Float64()
				dbUtil.Engine.ID(ua.Addr).Cols("use_amount").Update(&ua)
			}else{
				v.Status = 1
				v.SucBlock = int64(tx.BlockNum)
				v.Hash = tx.HashToString()
				dbUtil.Engine.ID(v.Id).Cols("status,suc_block,hash").Update(&v)
			}
		}
		time.Sleep(time.Second*300)
	}
}

func getAmountOut()decimal.Decimal{
	ic, err := NewRouter(common.HexToAddress(enums.AMOUNTCONTRACT), conn)
	if err!=nil {
		logs.Error("err=",err)
		return decimal.Zero
	}
	amountIn:=decimal.New(1,11).BigInt()
	addrs :=make([]common.Address,0)
	gdlAddress := common.HexToAddress(enums.KTOCONTRACT)
	usdtAddress := common.HexToAddress(enums.USDT)
	addrs = append(addrs,gdlAddress,usdtAddress)
	b,err := ic.GetAmountsOut(&bind.CallOpts{}, amountIn, addrs)
	if err!=nil {
		logs.Error("err=",err)
		return decimal.Zero
	}
	return decimal.NewFromBigInt(b[1],-18)
}