netrunner/utils/utils.go

package utils

import (
	"encoding/json"
	"errors"
	"fmt"
	"io/fs"
	"os"
	"time"

	"github.com/luxfi/ids"
	"github.com/luxfi/log"
	rpcb "github.com/luxfi/netrunner/rpcpb"
	"github.com/luxfi/netrunner/ux"
	"github.com/luxfi/node/staking"
)

const (
	genesisNetworkIDKey = "networkID"
	dirTimestampFormat  = "20060102_150405"
)

func ToNodeID(stakingKey, stakingCert []byte) (ids.NodeID, error) {
	cert, err := staking.LoadTLSCertFromBytes(stakingKey, stakingCert)
	if err != nil {
		return ids.EmptyNodeID, err
	}
	stakingCertificate := &ids.Certificate{
		Raw:       cert.Leaf.Raw,
		PublicKey: cert.Leaf.PublicKey,
	}
	nodeID := ids.NodeIDFromCert(stakingCertificate)
	return nodeID, nil
}

// Returns the network ID in the given genesis
func NetworkIDFromGenesis(genesis []byte) (uint32, error) {
	genesisMap := map[string]interface{}{}
	if err := json.Unmarshal(genesis, &genesisMap); err != nil {
		return 0, fmt.Errorf("couldn't unmarshal genesis: %w", err)
	}
	networkIDIntf, ok := genesisMap[genesisNetworkIDKey]
	if !ok {
		return 0, fmt.Errorf("couldn't find key %q in genesis", genesisNetworkIDKey)
	}
	networkID, ok := networkIDIntf.(float64)
	if !ok {
		return 0, fmt.Errorf("expected float64 but got %T", networkIDIntf)
	}
	return uint32(networkID), nil
}

var (
	ErrInvalidExecPath = errors.New("lux exec is invalid")
	ErrNotExists       = errors.New("lux exec not exists")
	ErrNotExistsPlugin = errors.New("plugin exec not exists")
)

func CheckExecPath(exec string) error {
	if exec == "" {
		return ErrInvalidExecPath
	}
	_, err := os.Stat(exec)
	if err != nil {
		if errors.Is(err, fs.ErrNotExist) {
			return ErrNotExists
		}
		return fmt.Errorf("failed to stat exec %q (%w)", exec, err)
	}
	return nil
}

func CheckPluginPath(pluginExec string) error {
	var err error
	if _, err = os.Stat(pluginExec); err != nil {
		if errors.Is(err, fs.ErrNotExist) {
			return ErrNotExistsPlugin
		}
		return fmt.Errorf("failed to stat plugin exec %q (%w)", pluginExec, err)
	}

	return nil
}

func VMID(vmName string) (ids.ID, error) {
	if len(vmName) > 32 {
		return ids.Empty, fmt.Errorf("VM name must be <= 32 bytes, found %d", len(vmName))
	}
	b := make([]byte, 32)
	copy(b, []byte(vmName))
	return ids.ToID(b)
}

func MkDirWithTimestamp(dirPrefix string) (string, error) {
	currentTime := time.Now().Format(dirTimestampFormat)
	dirName := dirPrefix + "_" + currentTime
	return dirName, os.MkdirAll(dirName, os.ModePerm)
}

func VerifyChainHasCorrectParticipants(
	logger log.Logger,
	chainParticipants []string,
	cluster *rpcb.ClusterInfo,
	chainID string,
) bool {
	if cluster != nil {
		participatingNodeNames := cluster.Chains[chainID].GetChainParticipants().GetNodeNames()

		var nodeIsInList bool
		// Check that all chain validators are equal to the node IDs added as participant in chain creation
		for _, node := range chainParticipants {
			nodeIsInList = false
			for _, chainValidator := range participatingNodeNames {
				if chainValidator == node {
					nodeIsInList = true
					break
				}
			}
			if !nodeIsInList {
				ux.Print(logger, "%s", log.Red.Wrap(fmt.Sprintf("VerifyChainHasCorrectParticipants: %#v", cluster)))
				ux.Print(logger, "%s", log.Red.Wrap(fmt.Sprintf("VerifyChainHasCorrectParticipants: node not in list chain %q node %q %v %v", chainID, node, chainParticipants, participatingNodeNames)))
				return false
			}
		}
		return true
	} else {
		ux.Print(logger, "%s", log.Red.Wrap("VerifyChainHasCorrectParticipants: cluster is nil"))
	}
	return false
}