dione/ethclient/wallet.go

74 lines
1.9 KiB
Go

package ethclient
import (
"context"
"crypto/ecdsa"
"math/big"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/crypto"
"github.com/sirupsen/logrus"
)
// // Balance returns the balance of the given ethereum address.
func (c *EthereumClient) Balance(ctx context.Context, address string) (*big.Int, error) {
ethereumAddress := common.HexToAddress(address)
value, err := c.client.BalanceAt(ctx, ethereumAddress, nil)
if err != nil {
return nil, err
}
return value, nil
}
func (c *EthereumClient) SendTransaction(ctx context.Context, private_key, to string, amount int64) string {
privateKey, err := crypto.HexToECDSA(private_key)
if err != nil {
logrus.Fatal("Failed to parse private key", err)
}
publicKey := privateKey.Public()
publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
if !ok {
logrus.Fatal("Cannot assert type: publicKey is not of type *ecdsa.PublicKey", err)
}
fromAddress := crypto.PubkeyToAddress(*publicKeyECDSA)
nonce, err := c.client.PendingNonceAt(ctx, fromAddress)
if err != nil {
logrus.Fatal("Failed to generate wallet nonce value", err)
}
value := big.NewInt(amount)
gasLimit := uint64(21000) // in units
gasPrice, err := c.client.SuggestGasPrice(ctx)
if err != nil {
logrus.Fatal("Failed to suggest new gas price", err)
}
toAddress := common.HexToAddress(to)
var data []byte
tx := types.NewTransaction(nonce, toAddress, value, gasLimit, gasPrice, data)
chainID, err := c.client.NetworkID(ctx)
if err != nil {
logrus.Fatal("Failed to get network ID", err)
}
signedTx, err := types.SignTx(tx, types.NewEIP155Signer(chainID), privateKey)
if err != nil {
logrus.Fatal("Failed to sign transaction", err)
}
err = c.client.SendTransaction(ctx, signedTx)
if err != nil {
logrus.Fatal("Failed to send signed transaction", err)
}
TxHash := signedTx.Hash().Hex()
logrus.Info("Transaction sent: %s", TxHash)
return TxHash
}