woop/api/service/staking/service.go

package staking

import (
	"crypto/ecdsa"
	"math/big"
	"strings"
	"time"

	"github.com/ethereum/go-ethereum/accounts/abi"
	"github.com/harmony-one/harmony/contracts"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/log"
	"github.com/ethereum/go-ethereum/params"
	"github.com/ethereum/go-ethereum/rlp"
	protobuf "github.com/golang/protobuf/proto"
	proto "github.com/harmony-one/harmony/api/client/service/proto"
	proto_common "github.com/harmony-one/harmony/api/proto"
	"github.com/harmony-one/harmony/api/proto/message"
	msg_pb "github.com/harmony-one/harmony/api/proto/message"
	"github.com/harmony-one/harmony/core"
	"github.com/harmony-one/harmony/core/types"
	"github.com/harmony-one/harmony/internal/utils"
	contract_constants "github.com/harmony-one/harmony/internal/utils/contract"
	"github.com/harmony-one/harmony/p2p"
	"github.com/harmony-one/harmony/p2p/host"
)

const (
	// WaitTime is the delay time for resending staking transaction if the previous transaction did not get approved.
	WaitTime = 5 * time.Second
	// StakingContractAddress is the staking deployed contract address
	StakingContractAddress = "TODO(minhdoan): Create a PR to generate staking contract address"
	// StakingAmount is the amount of stake to put
	StakingAmount = 10
)

// State is the state of staking service.
type State byte

// Service is the staking service.
// Service requires private key here which is not a right design.
// In stead in the right design, the end-user who runs mining needs to provide signed tx to this service.
type Service struct {
	host          p2p.Host
	stopChan      chan struct{}
	stoppedChan   chan struct{}
	accountKey    *ecdsa.PrivateKey
	blsAddress    [20]byte
	stakingAmount int64
	state         State
	beaconChain   *core.BlockChain
	messageChan   chan *msg_pb.Message
}

// New returns staking service.
func New(host p2p.Host, accountKey *ecdsa.PrivateKey, beaconChain *core.BlockChain, blsAddress [20]byte) *Service {
	return &Service{
		host:          host,
		stopChan:      make(chan struct{}),
		stoppedChan:   make(chan struct{}),
		accountKey:    accountKey,
		blsAddress:    blsAddress,
		stakingAmount: StakingAmount,
		beaconChain:   beaconChain,
	}
}

// StartService starts staking service.
func (s *Service) StartService() {
	log.Info("Start Staking Service")
	s.Run()
}

// Run runs staking.
func (s *Service) Run() {
	tick := time.NewTicker(WaitTime)
	go func() {
		defer close(s.stoppedChan)
		// Do service first time and after that doing it every 5 minutes.
		// The reason we have to do it in every x minutes because of beacon chain syncing.
		s.DoService()
		for {
			select {
			case <-tick.C:
				if s.IsStaked() {
					return
				}
				s.DoService()
				return
			case <-s.stopChan:
				return
			}
		}
	}()
}

// IsStaked checks if the txn gets accepted and approved in the beacon chain.
func (s *Service) IsStaked() bool {
	return false
}

// DoService does staking.
func (s *Service) DoService() {
	utils.GetLogInstance().Info("Trying to send a staking transaction.")

	if s.beaconChain == nil {
		utils.GetLogInstance().Info("Can not send a staking transaction because of nil beacon chain.")
		return
	}

	if msg := s.createStakingMessage(); msg != nil {
		s.host.SendMessageToGroups([]p2p.GroupID{p2p.GroupIDBeacon}, host.ConstructP2pMessage(byte(17), msg))
	} else {
		utils.GetLogInstance().Error("Can not create staking transaction")
	}
}

func (s *Service) getStakingInfo() *proto.StakingContractInfoResponse {
	address := crypto.PubkeyToAddress(s.accountKey.PublicKey)
	state, err := s.beaconChain.State()
	if err != nil {
		utils.GetLogInstance().Error("error to get beacon chain state when getting staking info")
		return nil
	}
	balance := state.GetBalance(address)
	if balance == common.Big0 {
		utils.GetLogInstance().Error("account balance empty when getting staking info")
		return nil
	}
	nonce := state.GetNonce(address)
	if nonce == 0 {
		utils.GetLogInstance().Error("nonce zero when getting staking info")
		return nil
	}
	return &proto.StakingContractInfoResponse{
		ContractAddress: StakingContractAddress,
		Balance:         balance.Bytes(),
		Nonce:           nonce,
	}
}

func (s *Service) getFakeStakingInfo() *proto.StakingContractInfoResponse {
	balance := big.NewInt(params.Ether)
	nonce := uint64(0) // TODO: make it a incrementing field

	priKey := contract_constants.GenesisBeaconAccountPriKey
	contractAddress := crypto.PubkeyToAddress(priKey.PublicKey)

	stakingContractAddress := crypto.CreateAddress(contractAddress, uint64(nonce))
	return &proto.StakingContractInfoResponse{
		ContractAddress: stakingContractAddress.Hex(),
		Balance:         balance.Bytes(),
		Nonce:           nonce,
	}
}

// Constructs the staking message
func constructStakingMessage(ts types.Transactions) []byte {
	tsBytes, err := rlp.EncodeToBytes(ts)
	if err == nil {
		msg := &message.Message{
			Type: message.MessageType_NEWNODE_BEACON_STAKING,
			Request: &message.Message_Staking{
				Staking: &message.StakingRequest{
					Transaction: tsBytes,
					NodeId:      "",
				},
			},
		}
		if data, err := protobuf.Marshal(msg); err == nil {
			return data
		}
	}
	utils.GetLogInstance().Error("Error when creating staking message", "error", err)
	return nil
}

func (s *Service) createRawStakingMessage() []byte {
	// TODO(minhdoan): Enable getStakingInfo back after testing.
	stakingInfo := s.getFakeStakingInfo()
	toAddress := common.HexToAddress(stakingInfo.ContractAddress)

	abi, err := abi.JSON(strings.NewReader(contracts.StakeLockContractABI))
	if err != nil {
		utils.GetLogInstance().Error("Failed to generate staking contract's ABI", "error", err)
	}
	// TODO: the bls address should be signed by the bls private key
	bytesData, err := abi.Pack("lock", s.blsAddress)
	if err != nil {
		utils.GetLogInstance().Error("Failed to generate ABI function bytes data", "error", err)
	}

	tx := types.NewTransaction(
		stakingInfo.Nonce,
		toAddress,
		0,
		big.NewInt(s.stakingAmount),
		params.TxGas*10,
		nil,
		bytesData,
	)

	if signedTx, err := types.SignTx(tx, types.HomesteadSigner{}, s.accountKey); err == nil {
		ts := types.Transactions{signedTx}
		return constructStakingMessage(ts)
	}
	return nil
}

func (s *Service) createStakingMessage() []byte {
	if msg := s.createRawStakingMessage(); msg != nil {
		return proto_common.ConstructStakingMessage(msg)
	}
	return nil
}

// StopService stops staking service.
func (s *Service) StopService() {
	utils.GetLogInstance().Info("Stopping staking service.")
	s.stopChan <- struct{}{}
	<-s.stoppedChan
	utils.GetLogInstance().Info("Role conversion stopped.")
}

// NotifyService notify service
func (s *Service) NotifyService(params map[string]interface{}) {
	return
}

// SetMessageChan sets up message channel to service.
func (s *Service) SetMessageChan(messageChan chan *msg_pb.Message) {
	s.messageChan = messageChan
}