Add P-Chain address derivation support

cmd/server/main.go

@@ -111,17 +111,27 @@ func main() {
 		cfg.NetworkName = networkName
 	}
 
-	network := &types.NetworkIdentifier{
+	networkP := &types.NetworkIdentifier{
+		Blockchain: service.BlockchainName,
+		Network:    cfg.NetworkName,
+		SubNetworkIdentifier: &types.SubNetworkIdentifier{
+			Network: mapper.PChainNetworkIdentifier,
+		},
+	}
+	networkC := &types.NetworkIdentifier{
 		Blockchain: service.BlockchainName,
 		Network:    cfg.NetworkName,
 	}
 
 	asserter, err := asserter.NewServer(
-		mapper.OperationTypes,               // supported operation types
-		true,                                // historical balance lookup
-		[]*types.NetworkIdentifier{network}, // supported networks
-		[]string{},                          // call methods
-		false,                               // mempool coins
+		mapper.OperationTypes, // supported operation types
+		true,                  // historical balance lookup
+		[]*types.NetworkIdentifier{ // supported networks
+			networkP,
+			networkC,
+		},
+		[]string{}, // call methods
+		false,      // mempool coins
 	)
 	if err != nil {
 		log.Fatal("server asserter init error:", err)
@@ -130,7 +140,7 @@ func main() {
 	serviceConfig := &service.Config{
 		Mode:               cfg.Mode,
 		ChainID:            big.NewInt(cfg.ChainID),
-		NetworkID:          network,
+		NetworkID:          networkC,
 		GenesisBlockHash:   cfg.GenesisBlockHash,
 		AvaxAssetID:        assetID,
 		AP5Activation:      AP5Activation,

mapper/helper.go

@@ -1,6 +1,17 @@
 package mapper
 
-import "strings"
+import (
+	"errors"
+	"strings"
+
+	"github.com/ava-labs/avalanchego/utils/constants"
+	"github.com/coinbase/rosetta-sdk-go/types"
+)
+
+var (
+	errUnsupportedChain   = errors.New("unsupported chain")
+	errUnsupportedNetwork = errors.New("unsupported network")
+)
 
 // EqualFoldContains checks if the array contains the string regardless of casing
 func EqualFoldContains(arr []string, str string) bool {
@@ -11,3 +22,30 @@ func EqualFoldContains(arr []string, str string) bool {
 	}
 	return false
 }
+
+// IsPChain checks network identifier to make sure sub-network identifier set to "P"
+func IsPChain(networkIdentifier *types.NetworkIdentifier) bool {
+	return networkIdentifier != nil &&
+		networkIdentifier.SubNetworkIdentifier != nil &&
+		networkIdentifier.SubNetworkIdentifier.Network == PChainNetworkIdentifier
+}
+
+// GetAliasAndHRP fetches chain id alias and hrp for address formatting.
+// Right now only P chain id alias is supported
+func GetAliasAndHRP(networkIdentifier *types.NetworkIdentifier) (string, string, error) {
+	if !IsPChain(networkIdentifier) {
+		return "", "", errUnsupportedChain
+	}
+
+	var hrp string
+	switch networkIdentifier.Network {
+	case FujiNetwork:
+		hrp = constants.GetHRP(constants.FujiID)
+	case MainnetNetwork:
+		hrp = constants.GetHRP(constants.MainnetID)
+	default:
+		return "", "", errUnsupportedNetwork
+	}
+
+	return PChainNetworkIdentifier, hrp, nil
+}

mapper/types.go

@@ -2,21 +2,24 @@ package mapper
 
 import (
 	"github.com/ava-labs/coreth/params"
-	"github.com/ethereum/go-ethereum/common"
-
 	"github.com/coinbase/rosetta-sdk-go/types"
+	"github.com/ethereum/go-ethereum/common"
 )
 
 const (
 	MainnetChainID = 43114
 	MainnetAssetID = "FvwEAhmxKfeiG8SnEvq42hc6whRyY3EFYAvebMqDNDGCgxN5Z"
+	MainnetNetwork = "Mainnet"
 
 	FujiChainID = 43113
 	FujiAssetID = "U8iRqJoiJm8xZHAacmvYyZVwqQx6uDNtQeP3CQ6fcgQk3JqnK"
+	FujiNetwork = "Fuji"
 
 	ContractAddressMetadata  = "contractAddress"
 	IndexTransferredMetadata = "indexTransferred"
 
+	PChainNetworkIdentifier = "P"
+
 	OpCall          = "CALL"
 	OpFee           = "FEE"
 	OpCreate        = "CREATE"

service/chain/p/client.go

@@ -0,0 +1,46 @@
+package p
+
+import (
+	"context"
+
+	"github.com/ava-labs/avalanche-rosetta/mapper"
+	"github.com/ava-labs/avalanchego/utils/crypto"
+	"github.com/ava-labs/avalanchego/utils/formatting/address"
+	"github.com/coinbase/rosetta-sdk-go/types"
+)
+
+type Client struct {
+	fac crypto.FactorySECP256K1R
+}
+
+func NewClient() *Client {
+	return &Client{
+		fac: crypto.FactorySECP256K1R{},
+	}
+}
+
+func (c *Client) DeriveAddress(
+	ctx context.Context,
+	req *types.ConstructionDeriveRequest,
+) (*types.ConstructionDeriveResponse, error) {
+	pub, err := c.fac.ToPublicKey(req.PublicKey.Bytes)
+	if err != nil {
+		return nil, err
+	}
+
+	chainIDAlias, hrp, getErr := mapper.GetAliasAndHRP(req.NetworkIdentifier)
+	if getErr != nil {
+		return nil, getErr
+	}
+
+	addr, err := address.Format(chainIDAlias, hrp, pub.Address().Bytes())
+	if err != nil {
+		return nil, err
+	}
+
+	return &types.ConstructionDeriveResponse{
+		AccountIdentifier: &types.AccountIdentifier{
+			Address: addr,
+		},
+	}, nil
+}

service/helper.go

@@ -6,11 +6,11 @@ import (
 	"math/big"
 	"strings"
 
-	"github.com/ava-labs/avalanche-rosetta/client"
-	"github.com/coinbase/rosetta-sdk-go/types"
-
 	ethtypes "github.com/ava-labs/coreth/core/types"
+	"github.com/coinbase/rosetta-sdk-go/types"
 	ethcommon "github.com/ethereum/go-ethereum/common"
+
+	"github.com/ava-labs/avalanche-rosetta/client"
 )
 
 const (

service/service_construction.go

@@ -6,20 +6,20 @@ import (
 	"fmt"
 	"math/big"
 
+	ethtypes "github.com/ava-labs/coreth/core/types"
+	"github.com/ava-labs/coreth/interfaces"
 	"github.com/coinbase/rosetta-sdk-go/parser"
 	"github.com/coinbase/rosetta-sdk-go/server"
 	"github.com/coinbase/rosetta-sdk-go/types"
 	"github.com/coinbase/rosetta-sdk-go/utils"
-	"golang.org/x/crypto/sha3"
-
-	ethtypes "github.com/ava-labs/coreth/core/types"
-	"github.com/ava-labs/coreth/interfaces"
 	ethcommon "github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/common/hexutil"
+	ethcrypto "github.com/ethereum/go-ethereum/crypto"
+	"golang.org/x/crypto/sha3"
 
 	"github.com/ava-labs/avalanche-rosetta/client"
 	"github.com/ava-labs/avalanche-rosetta/mapper"
-	"github.com/ethereum/go-ethereum/common/hexutil"
-	ethcrypto "github.com/ethereum/go-ethereum/crypto"
+	"github.com/ava-labs/avalanche-rosetta/service/chain/p"
 )
 
 const (
@@ -33,13 +33,15 @@ const (
 type ConstructionService struct {
 	config *Config
 	client client.Client
+	p      *p.Client
 }
 
 // NewConstructionService returns a new construction servicer
 func NewConstructionService(config *Config, client client.Client) server.ConstructionAPIServicer {
 	return &ConstructionService{
 		config: config,
 		client: client,
+		p:      p.NewClient(),
 	}
 }
 
@@ -226,6 +228,15 @@ func (s ConstructionService) ConstructionDerive(
 		return nil, wrapError(errInvalidInput, "public key is not provided")
 	}
 
+	if mapper.IsPChain(req.NetworkIdentifier) {
+		res, err := s.p.DeriveAddress(ctx, req)
+		if err != nil {
+			return nil, wrapError(errInternalError, "p chain address derivation failed")
+		}
+
+		return res, nil
+	}
+
 	key, err := ethcrypto.DecompressPubkey(req.PublicKey.Bytes)
 	if err != nil {
 		return nil, wrapError(errInvalidInput, err)

service/service_construction_test.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/ava-labs/avalanche-rosetta/mapper"
 	mocks "github.com/ava-labs/avalanche-rosetta/mocks/client"
+	"github.com/ava-labs/avalanche-rosetta/service/chain/p"
 	"github.com/ava-labs/coreth/interfaces"
 
 	"github.com/coinbase/rosetta-sdk-go/types"
@@ -238,7 +239,9 @@ func TestContructionHash(t *testing.T) {
 }
 
 func TestConstructionDerive(t *testing.T) {
-	service := ConstructionService{}
+	service := ConstructionService{
+		p: p.NewClient(),
+	}
 
 	t.Run("no public key", func(t *testing.T) {
 		resp, err := service.ConstructionDerive(
@@ -285,6 +288,33 @@ func TestConstructionDerive(t *testing.T) {
 			resp.AccountIdentifier.Address,
 		)
 	})
+
+	t.Run("p-chain address", func(t *testing.T) {
+		src := "02e0d4392cfa224d4be19db416b3cf62e90fb2b7015e7b62a95c8cb490514943f6"
+		b, _ := hex.DecodeString(src)
+
+		resp, err := service.ConstructionDerive(
+			context.Background(),
+			&types.ConstructionDeriveRequest{
+				NetworkIdentifier: &types.NetworkIdentifier{
+					Network: mapper.FujiNetwork,
+					SubNetworkIdentifier: &types.SubNetworkIdentifier{
+						Network: mapper.PChainNetworkIdentifier,
+					},
+				},
+				PublicKey: &types.PublicKey{
+					Bytes:     b,
+					CurveType: types.Secp256k1,
+				},
+			},
+		)
+		assert.Nil(t, err)
+		assert.Equal(
+			t,
+			"P-fuji15f9g0h5xkr5cp47n6u3qxj6yjtzzzrdr23a3tl",
+			resp.AccountIdentifier.Address,
+		)
+	})
 }
 
 func forceMarshalMap(t *testing.T, i interface{}) map[string]interface{} {