package wallet

import (
	"bytes"
	"encoding/hex"
	"fmt"
	"sync"

	"github.com/33cn/chain33/common"
	"github.com/33cn/chain33/common/address"
	"github.com/33cn/chain33/common/crypto"
	"github.com/33cn/chain33/common/db"
	"github.com/33cn/chain33/common/log/log15"
	"github.com/33cn/chain33/types"
	wcom "github.com/33cn/chain33/wallet/common"
	zt "github.com/33cn/plugin/plugin/dapp/zksync/types"
	"github.com/consensys/gnark-crypto/ecc/bn254/fr/mimc"
	"github.com/consensys/gnark-crypto/ecc/bn254/twistededwards/eddsa"
)

var (
	bizlog = log15.New("module", "wallet.zksync")

	maxTxNumPerBlock int64 = types.MaxTxsPerBlock
)

func init() {
	wcom.RegisterPolicy(zt.Zksync, New())
}

func New() wcom.WalletBizPolicy {
	return &zksyncPolicy{
		mtx: &sync.Mutex{},
	}
}

type zksyncPolicy struct {
	mtx           *sync.Mutex
	walletOperate wcom.WalletOperate
}

func (policy *zksyncPolicy) setWalletOperate(walletBiz wcom.WalletOperate) {
	policy.mtx.Lock()
	defer policy.mtx.Unlock()
	policy.walletOperate = walletBiz
}

func (policy *zksyncPolicy) getWalletOperate() wcom.WalletOperate {
	policy.mtx.Lock()
	defer policy.mtx.Unlock()
	return policy.walletOperate
}

func (policy *zksyncPolicy) Init(walletOperate wcom.WalletOperate, sub []byte) {
	policy.setWalletOperate(walletOperate)
}

func (policy *zksyncPolicy) OnCreateNewAccount(acc *types.Account) {
}

func (policy *zksyncPolicy) OnImportPrivateKey(acc *types.Account) {
}

func (policy *zksyncPolicy) OnAddBlockFinish(block *types.BlockDetail) {

}

func (policy *zksyncPolicy) OnDeleteBlockFinish(block *types.BlockDetail) {

}

func (policy *zksyncPolicy) OnClose() {

}

func (policy *zksyncPolicy) OnSetQueueClient() {
}

func (policy *zksyncPolicy) OnWalletLocked() {
}

func (policy *zksyncPolicy) OnWalletUnlocked(WalletUnLock *types.WalletUnLock) {
}

func (policy *zksyncPolicy) Call(funName string, in types.Message) (ret types.Message, err error) {
	err = types.ErrNotSupport
	return
}

func (policy *zksyncPolicy) SignTransaction(key crypto.PrivKey, req *types.ReqSignRawTx) (needSysSign bool, signtxhex string, err error) {
	needSysSign = false
	bytesVal, err := common.FromHex(req.GetTxHex())
	if err != nil {
		bizlog.Error("SignTransaction", "common.FromHex error", err)
		return
	}
	tx := new(types.Transaction)
	if err = types.Decode(bytesVal, tx); err != nil {
		bizlog.Error("SignTransaction", "Decode Transaction error", err)
		return
	}

	action := new(zt.ZksyncAction)
	if err = types.Decode(tx.Payload, action); err != nil {
		return
	}

	privateKey, err := eddsa.GenerateKey(bytes.NewReader(key.Bytes()))

	if err != nil {
		bizlog.Error("SignTransaction", "eddsa.GenerateKey error", err)
		return
	}

	var msg *zt.ZkMsg
	var signInfo *zt.ZkSignature
	switch action.GetTy() {
	case zt.TyDepositAction:
		deposit := action.GetDeposit()
		msg = GetDepositMsg(deposit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		deposit.Signature = signInfo
	case zt.TyWithdrawAction:
		withDraw := action.GetWithdraw()
		msg = GetWithdrawMsg(withDraw)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		withDraw.Signature = signInfo
	case zt.TyContractToTreeAction:
		contractToLeaf := action.GetContractToTree()
		msg = GetContractToTreeMsg(contractToLeaf)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		contractToLeaf.Signature = signInfo
	case zt.TyTreeToContractAction:
		leafToContract := action.GetTreeToContract()
		msg = GetTreeToContractMsg(leafToContract)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		leafToContract.Signature = signInfo
	case zt.TyTransferAction:
		transfer := action.GetTransfer()
		msg = GetTransferMsg(transfer)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		transfer.Signature = signInfo
	case zt.TyTransferToNewAction:
		transferToNew := action.GetTransferToNew()
		msg = GetTransferToNewMsg(transferToNew)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		transferToNew.Signature = signInfo
	case zt.TyForceExitAction:
		forceQuit := action.GetForceExit()
		msg = GetForceExitMsg(forceQuit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		forceQuit.Signature = signInfo
	case zt.TySetPubKeyAction:
		setPubKey := action.GetSetPubKey()

		pubKey := &zt.ZkPubKey{
			X: privateKey.PublicKey.A.X.String(),
			Y: privateKey.PublicKey.A.Y.String(),
		}
		setPubKey.PubKey = pubKey
		msg = GetSetPubKeyMsg(setPubKey)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		setPubKey.Signature = signInfo
	case zt.TyFullExitAction:
		forceQuit := action.GetFullExit()
		msg = GetFullExitMsg(forceQuit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		forceQuit.Signature = signInfo
	}

	tx.Payload = types.Encode(action)
	tx.Sign(int32(policy.getWalletOperate().GetSignType()), key)
	signtxhex = hex.EncodeToString(types.Encode(tx))
	return
}

func SignTx(msg *zt.ZkMsg, privateKey eddsa.PrivateKey) (*zt.ZkSignature, error) {
	signInfo, err := privateKey.Sign(GetMsgHash(msg), mimc.NewMiMC(zt.ZkMimcHashSeed))
	if err != nil {
		bizlog.Error("SignTransaction", "privateKey.Sign error", err)
		return nil, err
	}
	pubKey := &zt.ZkPubKey{
		X: privateKey.PublicKey.A.X.String(),
		Y: privateKey.PublicKey.A.Y.String(),
	}
	sign := &zt.ZkSignature{
		PubKey:   pubKey,
		SignInfo: hex.EncodeToString(signInfo),
		Msg:      msg,
	}
	return sign, nil
}

func (policy *zksyncPolicy) OnAddBlockTx(block *types.BlockDetail, tx *types.Transaction, index int32, dbbatch db.Batch) *types.WalletTxDetail {
	txdetail := &types.WalletTxDetail{}

	blockheight := block.Block.Height*maxTxNumPerBlock + int64(index)
	heightstr := fmt.Sprintf("%018d", blockheight)
	key := wcom.CalcTxKey(heightstr)
	txdetail.Tx = tx
	txdetail.Height = block.Block.Height
	txdetail.Index = int64(index)
	txdetail.Receipt = block.Receipts[index]
	txdetail.Blocktime = block.Block.BlockTime

	txdetail.ActionName = tx.ActionName()
	txdetail.Amount, _ = tx.Amount()
	txdetail.Txhash = tx.Hash()

	pubkey := block.Block.Txs[index].Signature.GetPubkey()
	txdetail.Fromaddr = address.PubKeyToAddr(address.DefaultID, pubkey)

	txdetailbyte := types.Encode(txdetail)
	dbbatch.Set(key, txdetailbyte)

	return nil
}

func (policy *zksyncPolicy) OnDeleteBlockTx(block *types.BlockDetail, tx *types.Transaction, index int32, dbbatch db.Batch) *types.WalletTxDetail {
	blockheight := block.Block.Height*maxTxNumPerBlock + int64(index)
	heightstr := fmt.Sprintf("%018d", blockheight)
	dbbatch.Delete(wcom.CalcTxKey(heightstr))

	return nil
}
package wallet

import (
	"bytes"
	"encoding/hex"
	"fmt"
	"sync"

	"github.com/33cn/chain33/common"
	"github.com/33cn/chain33/common/address"
	"github.com/33cn/chain33/common/crypto"
	"github.com/33cn/chain33/common/db"
	"github.com/33cn/chain33/common/log/log15"
	"github.com/33cn/chain33/types"
	wcom "github.com/33cn/chain33/wallet/common"
	zt "github.com/33cn/plugin/plugin/dapp/zksync/types"
	"github.com/consensys/gnark-crypto/ecc/bn254/fr/mimc"
	"github.com/consensys/gnark-crypto/ecc/bn254/twistededwards/eddsa"
)

var (
	bizlog = log15.New("module", "wallet.zksync")
	// maxTxNumPerBlock 单个区块最大数
	maxTxNumPerBlock int64 = types.MaxTxsPerBlock
)

func init() {
	wcom.RegisterPolicy(zt.Zksync, New())
}

// New 创建一盒钱包业务策略
func New() wcom.WalletBizPolicy {
	return &zksyncPolicy{
		mtx: &sync.Mutex{},
	}
}

type zksyncPolicy struct {
	mtx           *sync.Mutex
	walletOperate wcom.WalletOperate
}

func (policy *zksyncPolicy) setWalletOperate(walletBiz wcom.WalletOperate) {
	policy.mtx.Lock()
	defer policy.mtx.Unlock()
	policy.walletOperate = walletBiz
}

func (policy *zksyncPolicy) getWalletOperate() wcom.WalletOperate {
	policy.mtx.Lock()
	defer policy.mtx.Unlock()
	return policy.walletOperate
}

// Init 初始化处理
func (policy *zksyncPolicy) Init(walletOperate wcom.WalletOperate, sub []byte) {
	policy.setWalletOperate(walletOperate)
}

// OnCreateNewAccount 在账号创建时做一些处理
func (policy *zksyncPolicy) OnCreateNewAccount(acc *types.Account) {
}

// OnImportPrivateKey 在私钥导入时做一些处理
func (policy *zksyncPolicy) OnImportPrivateKey(acc *types.Account) {
}

// OnAddBlockFinish 在区块被添加成功时做一些处理
func (policy *zksyncPolicy) OnAddBlockFinish(block *types.BlockDetail) {

}

// OnDeleteBlockFinish 在区块被删除成功时做一些处理
func (policy *zksyncPolicy) OnDeleteBlockFinish(block *types.BlockDetail) {

}

// OnClose 在钱包关闭时做一些处理
func (policy *zksyncPolicy) OnClose() {

}

// OnSetQueueClient 在钱包消息队列初始化时做一些处理
func (policy *zksyncPolicy) OnSetQueueClient() {
}

// OnWalletLocked 在钱包加锁时做一些处理
func (policy *zksyncPolicy) OnWalletLocked() {
}

// OnWalletUnlocked 在钱包解锁时做一些处理
func (policy *zksyncPolicy) OnWalletUnlocked(WalletUnLock *types.WalletUnLock) {
}

// Call 调用隐私的方法
func (policy *zksyncPolicy) Call(funName string, in types.Message) (ret types.Message, err error) {
	err = types.ErrNotSupport
	return
}

// SignTransaction 对zksync交易进行签名
func (policy *zksyncPolicy) SignTransaction(key crypto.PrivKey, req *types.ReqSignRawTx) (needSysSign bool, signtxhex string, err error) {
	needSysSign = false
	bytesVal, err := common.FromHex(req.GetTxHex())
	if err != nil {
		bizlog.Error("SignTransaction", "common.FromHex error", err)
		return
	}
	tx := new(types.Transaction)
	if err = types.Decode(bytesVal, tx); err != nil {
		bizlog.Error("SignTransaction", "Decode Transaction error", err)
		return
	}

	action := new(zt.ZksyncAction)
	if err = types.Decode(tx.Payload, action); err != nil {
		return
	}

	privateKey, err := eddsa.GenerateKey(bytes.NewReader(key.Bytes()))

	if err != nil {
		bizlog.Error("SignTransaction", "eddsa.GenerateKey error", err)
		return
	}

	var msg *zt.ZkMsg
	var signInfo *zt.ZkSignature
	switch action.GetTy() {
	case zt.TyDepositAction:
		deposit := action.GetDeposit()
		msg = GetDepositMsg(deposit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		deposit.Signature = signInfo
	case zt.TyWithdrawAction:
		withDraw := action.GetWithdraw()
		msg = GetWithdrawMsg(withDraw)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		withDraw.Signature = signInfo
	case zt.TyContractToTreeAction:
		contractToLeaf := action.GetContractToTree()
		msg = GetContractToTreeMsg(contractToLeaf)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		contractToLeaf.Signature = signInfo
	case zt.TyTreeToContractAction:
		leafToContract := action.GetTreeToContract()
		msg = GetTreeToContractMsg(leafToContract)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		leafToContract.Signature = signInfo
	case zt.TyTransferAction:
		transfer := action.GetTransfer()
		msg = GetTransferMsg(transfer)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		transfer.Signature = signInfo
	case zt.TyTransferToNewAction:
		transferToNew := action.GetTransferToNew()
		msg = GetTransferToNewMsg(transferToNew)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		transferToNew.Signature = signInfo
	case zt.TyForceExitAction:
		forceQuit := action.GetForceExit()
		msg = GetForceExitMsg(forceQuit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		forceQuit.Signature = signInfo
	case zt.TySetPubKeyAction:
		setPubKey := action.GetSetPubKey()
		//如果是添加公钥的操作，则默认设置这里生成的公钥 todo:要是未来修改可以自定义公钥，这里需要删除
		pubKey := &zt.ZkPubKey{
			X: privateKey.PublicKey.A.X.String(),
			Y: privateKey.PublicKey.A.Y.String(),
		}
		setPubKey.PubKey = pubKey
		msg = GetSetPubKeyMsg(setPubKey)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		setPubKey.Signature = signInfo
	case zt.TyFullExitAction:
		forceQuit := action.GetFullExit()
		msg = GetFullExitMsg(forceQuit)
		signInfo, err = SignTx(msg, privateKey)
		if err != nil {
			bizlog.Error("SignTransaction", "eddsa.signTx error", err)
			return
		}
		forceQuit.Signature = signInfo
	}

	tx.Payload = types.Encode(action)
	tx.Sign(int32(policy.getWalletOperate().GetSignType()), key)
	signtxhex = hex.EncodeToString(types.Encode(tx))
	return
}

func SignTx(msg *zt.ZkMsg, privateKey eddsa.PrivateKey) (*zt.ZkSignature, error) {
	signInfo, err := privateKey.Sign(GetMsgHash(msg), mimc.NewMiMC(zt.ZkMimcHashSeed))
	if err != nil {
		bizlog.Error("SignTransaction", "privateKey.Sign error", err)
		return nil, err
	}
	pubKey := &zt.ZkPubKey{
		X: privateKey.PublicKey.A.X.String(),
		Y: privateKey.PublicKey.A.Y.String(),
	}
	sign := &zt.ZkSignature{
		PubKey:   pubKey,
		SignInfo: hex.EncodeToString(signInfo),
		Msg:      msg,
	}
	return sign, nil
}

// OnAddBlockTx 响应区块交易添加的处理
func (policy *zksyncPolicy) OnAddBlockTx(block *types.BlockDetail, tx *types.Transaction, index int32, dbbatch db.Batch) *types.WalletTxDetail {
	txdetail := &types.WalletTxDetail{}

	blockheight := block.Block.Height*maxTxNumPerBlock + int64(index)
	heightstr := fmt.Sprintf("%018d", blockheight)
	key := wcom.CalcTxKey(heightstr)
	txdetail.Tx = tx
	txdetail.Height = block.Block.Height
	txdetail.Index = int64(index)
	txdetail.Receipt = block.Receipts[index]
	txdetail.Blocktime = block.Block.BlockTime

	txdetail.ActionName = tx.ActionName()
	txdetail.Amount, _ = tx.Amount()
	txdetail.Txhash = tx.Hash()

	pubkey := block.Block.Txs[index].Signature.GetPubkey()
	txdetail.Fromaddr = address.PubKeyToAddr(address.DefaultID, pubkey)

	txdetailbyte := types.Encode(txdetail)
	dbbatch.Set(key, txdetailbyte)

	return nil
}

// OnDeleteBlockTx 响应删除区块交易的处理
func (policy *zksyncPolicy) OnDeleteBlockTx(block *types.BlockDetail, tx *types.Transaction, index int32, dbbatch db.Batch) *types.WalletTxDetail {
	blockheight := block.Block.Height*maxTxNumPerBlock + int64(index)
	heightstr := fmt.Sprintf("%018d", blockheight)
	dbbatch.Delete(wcom.CalcTxKey(heightstr))
	// 自己处理掉所有事务，不需要外部处理了
	return nil
}
