wallet.go

package evmc

import (
	"crypto/ecdsa"
	"errors"
	"math/big"

	"github.com/ethereum/go-ethereum/common/hexutil"
	"github.com/ethereum/go-ethereum/core/types"
	"github.com/ethereum/go-ethereum/crypto"
)

type Wallet struct {
	pk      *ecdsa.PrivateKey
	address string
}

func NewWallet(privateKey string) (*Wallet, error) {
	if privateKey[:2] == "0x" {
		privateKey = privateKey[2:]
	}
	pk, err := crypto.HexToECDSA(privateKey)
	if err != nil {
		return nil, err
	}
	return &Wallet{
		pk:      pk,
		address: crypto.PubkeyToAddress(pk.PublicKey).Hex(),
	}, nil
}

func (w *Wallet) SignTx(sendingTx *SendingTx, chainID uint64) (hash, rawTx string, err error) {
	if chainID == 0 {
		return "", "", errors.New("chainID is zero")
	}
	gethTx := types.NewTx(sendingTx.txData)
	signedTx, err := types.SignTx(gethTx, types.LatestSignerForChainID(new(big.Int).SetUint64(chainID)), w.pk)
	if err != nil {
		return "", "", err
	}
	raw, err := signedTx.MarshalBinary()
	if err != nil {
		return "", "", err
	}
	hash = signedTx.Hash().Hex()
	rawTx = hexutil.Encode(raw)
	return
}

func (w *Wallet) Address() string {
	return w.address
}