Add converting base58 to hex

.gitignore

@@ -1,3 +1,5 @@
+go/idos-extensions
+
 **/vue/node_modules
 **/vue/dist
 **/release/
@@ -41,4 +43,4 @@ dist/*
 .run
 
 # go
-vendor/
+vendor/

README.md

@@ -28,6 +28,7 @@ The IDOS extension requires two configuration variables: `ETH_RPC_URL` and `NEAR
 export ETH_RPC_URL=xxx
 export NEAR_RPC_URL=xxx
 ```
+By default, `ETH_RPC_URL` is `http://127.0.0.1:8545/`. If you're using macOS and want to connect the chain running in your host machine (i.e. hardhat running locally in development environment), you should use `http://host.docker.internal:8545/` instead to access host machine port from the container.
 
 
 ### Initialization
@@ -50,7 +51,7 @@ action get_credential ($id) public {
     $can_access = grants_eth.has_grants(@caller, $id);
     SELECT CASE WHEN $can_access != 1
     THEN ERROR('caller does not have access') END;
-    
+
     SELECT content
     FROM credentials
     WHERE id = $id;
@@ -69,7 +70,7 @@ interface IKwilWhitelist {
       string dataId;
       uint256 lockedUntil;
     }
-    
+
     function grants_for(address grantee, string dataId)
         public view
         returns(Grant[] memory)

docker-compose.yml

@@ -6,7 +6,7 @@ services:
     ports:
       - 50055:50055
     environment:
-      - ETH_RPC_URL=
+      - ETH_RPC_URL=http://host.docker.internal:8545/
       - NEAR_RPC_URL=
       - PORT=50055
     restart: on-failure

go/extension/chains/ethereum/ethereum.go

@@ -3,6 +3,7 @@ package ethereum
 import (
 	"context"
 	"fmt"
+	"regexp"
 
 	"github.com/idos-network/idos-extensions/extension/chains"
 	"github.com/idos-network/idos-extensions/extension/chains/ethereum/registry"
@@ -64,6 +65,17 @@ func (b *Backend) GrantsFor(ctx context.Context, contract, addr, resource string
 	return grants, nil
 }
 
+var publicKeyRegex = regexp.MustCompile("^(:?0x)?[0-9a-fA-F]{130}$")
+
+// Extracted this to be able to test without having to create an RPC connection.
+func isValidPublicKey(publicKey string) bool {
+	return publicKeyRegex.MatchString(publicKey)
+}
+
+func (b *Backend) IsValidPublicKey(publicKey string) bool {
+	return isValidPublicKey(publicKey)
+}
+
 type driver struct{}
 
 func (d *driver) New(url string) (chains.ChainBackend, error) {

go/extension/chains/ethereum/ethereum_test.go

@@ -0,0 +1,41 @@
+package ethereum
+
+import (
+	"testing"
+)
+
+func Test_isValidPublicKey(t *testing.T) {
+	tests := []struct {
+		publicKey string
+		want      bool
+	}{
+		{
+			"",
+			false,
+		},
+		{
+			"some random string",
+			false,
+		},
+		{
+			// Address; not a public key
+			"0x9E660ba85118b722147BBaf04ED697C95549dF03",
+			false,
+		},
+		{
+			"0408cf359417716c8c4dd03ab0c3b243b383599cb05c1b276b326c92a8f4b2b4acdcbdd98e9443f8bfc370b40e80f677142dab8cffd348a22fdf4b68ab61c7d78f",
+			true,
+		},
+		{
+			"0x0408cf359417716c8c4dd03ab0c3b243b383599cb05c1b276b326c92a8f4b2b4acdcbdd98e9443f8bfc370b40e80f677142dab8cffd348a22fdf4b68ab61c7d78f",
+			true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.publicKey, func(t *testing.T) {
+			if got := isValidPublicKey(tt.publicKey); got != tt.want {
+				t.Errorf("isValidPublicKey() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}

go/extension/chains/interface.go

@@ -25,6 +25,7 @@ type GrantChecker interface {
 type ChainBackend interface {
 	GrantChecker
 	Height(context.Context) (uint64, error)
+	IsValidPublicKey(public_key string) bool
 	// TODO: Close(), for ws connections
 }
 

go/extension/chains/near/near.go

@@ -11,6 +11,7 @@ import (
 	"github.com/eteu-technologies/near-api-go/pkg/client"
 	"github.com/eteu-technologies/near-api-go/pkg/client/block"
 	"github.com/idos-network/idos-extensions/extension/chains"
+	"github.com/mr-tron/base58"
 )
 
 // isNearAcct checks if the string is a valid near account name. This is either
@@ -124,6 +125,35 @@ func (nb *Backend) GrantsFor(ctx context.Context, registry, acct, resource strin
 	return grants, nil
 }
 
+const NEAR_KEY_LENGTH = 32
+
+func isValidPublicKey(publicKey string) (bool, error) {
+	pieces := strings.SplitN(publicKey, ":", 2)
+	if len(pieces) != 2 {
+		return false, fmt.Errorf("unrecognized format for NEAR public key: %s", publicKey)
+	}
+	if pieces[0] != "ed25519" {
+		return false, fmt.Errorf("unsupported NEAR public key type: %s", pieces[0])
+	}
+
+	binaryPayload, err := base58.Decode(pieces[1])
+	if err != nil {
+		return false, fmt.Errorf("unable to decode %s: %w", pieces[1], err)
+	}
+
+	binaryPayloadLen := len(binaryPayload)
+	if binaryPayloadLen != NEAR_KEY_LENGTH {
+		return false, fmt.Errorf("wrong binary length: was expecting %d, got %d", NEAR_KEY_LENGTH, binaryPayloadLen)
+	}
+
+	return true, nil
+}
+
+func (b *Backend) IsValidPublicKey(publicKey string) bool {
+	result, _ := isValidPublicKey(publicKey)
+	return result
+}
+
 type driver struct{}
 
 func (d *driver) New(url string) (chains.ChainBackend, error) {

go/extension/chains/near/near_test.go

@@ -145,3 +145,42 @@ func Test_isNearAcct(t *testing.T) {
 		})
 	}
 }
+
+func Test_isValidPublicKey(t *testing.T) {
+	tests := []struct {
+		publicKey string
+		want      bool
+	}{
+		{
+			"",
+			false,
+		},
+		{
+			"blah.near",
+			false,
+		},
+		{
+			"derp:herp",
+			false,
+		},
+		{
+			"ed25519:INVALID",
+			false,
+		},
+		{
+			"ed25519:7dLLbzqc6kgGAC6smmJUUh9xqxH9habnLhptauA",
+			false,
+		},
+		{
+			"ed25519:7dLLbzqc6kgGAC6smmJUUh9xqxH9habnLhptauAymmUJ",
+			true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.publicKey, func(t *testing.T) {
+			if res, _ := isValidPublicKey(tt.publicKey); res != tt.want {
+				t.Errorf("isNearAcct() = %v, want %v", res, tt.want)
+			}
+		})
+	}
+}

go/extension/extension.go

@@ -2,14 +2,18 @@ package extension
 
 import (
 	"context"
+	"encoding/hex"
 	"errors"
 	"fmt"
 	"log"
+	"regexp"
 
 	"github.com/idos-network/idos-extensions/extension/chains"
 
 	"github.com/kwilteam/kwil-extensions/server"
 	"github.com/kwilteam/kwil-extensions/types"
+
+	"github.com/mr-tron/base58"
 )
 
 const (
@@ -61,8 +65,11 @@ func (e *FractalExt) BuildServer(logger *log.Logger) (*server.ExtensionServer, e
 		}).
 		WithMethods(
 			map[string]server.MethodFunc{
-				"get_block_height": server.WithOutputsCheck(e.BlockHeight, 1),
-				"has_grants":       server.WithInputsCheck(server.WithOutputsCheck(e.GrantsFor, 1), 2),
+				"get_block_height":               server.WithOutputsCheck(e.BlockHeight, 1),
+				"has_grants":                     server.WithInputsCheck(server.WithOutputsCheck(e.GrantsFor, 1), 2),
+				"implicit_address_to_public_key": server.WithInputsCheck(server.WithOutputsCheck(e.ImplicitAddressToPublicKey, 1), 1),
+				"determine_wallet_type":          server.WithInputsCheck(server.WithOutputsCheck(e.DetermineWalletType, 1), 1),
+				"is_valid_public_key":            server.WithInputsCheck(server.WithOutputsCheck(e.IsValidPublicKey, 1), 1),
 			}).
 		Build()
 }
@@ -141,6 +148,61 @@ func (e *FractalExt) GrantsFor(ctx *types.ExecutionContext, values ...*types.Sca
 	return encodeScalarValues(exist)
 }
 
+func (e *FractalExt) ImplicitAddressToPublicKey(ctx *types.ExecutionContext, values ...*types.ScalarValue) ([]*types.ScalarValue, error) {
+	inputHex, err := values[0].String()
+	if err != nil {
+		return nil, fmt.Errorf("convert value to string failed: %w", err)
+	}
+	binaryString, _ := hex.DecodeString(inputHex)
+	base58 := base58.Encode(binaryString)
+	var public_key string
+	if len(inputHex) != 64 || base58 == "" {
+		public_key = ""
+	} else {
+		public_key = fmt.Sprintf("ed25519:%s", base58)
+	}
+
+	return encodeScalarValues(public_key)
+}
+
+func (e *FractalExt) IsValidPublicKey(ctx *types.ExecutionContext, values ...*types.ScalarValue) ([]*types.ScalarValue, error) {
+	be, _, err := e.chainBackend(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	public_key, err := values[0].String()
+	if err != nil {
+		return nil, fmt.Errorf("convert value to string failed: %w", err)
+	}
+
+	var result uint
+	if be.IsValidPublicKey(public_key) {
+		result = 1
+	} else {
+		result = 0
+	}
+	return encodeScalarValues(result)
+}
+
+// This has very dumb logic: eth address returns EVM type, and NEAR returns otherwise.
+// TODO: make the logic more detailed and return error is the address is neither EVM no NEAR.
+func (e *FractalExt) DetermineWalletType(ctx *types.ExecutionContext, values ...*types.ScalarValue) ([]*types.ScalarValue, error) {
+	address, err := values[0].String()
+	if err != nil {
+		return nil, fmt.Errorf("convert value to string failed: %w", err)
+	}
+	re := regexp.MustCompile("^0x[0-9a-fA-F]{40}$")
+	var wallet_type string
+	if re.MatchString(address) {
+		wallet_type = "EVM"
+	} else {
+		wallet_type = "NEAR"
+	}
+
+	return encodeScalarValues(wallet_type)
+}
+
 // initialize checks that the meta data includes all required fields and applies
 // any default values.
 func initialize(ctx context.Context, metadata map[string]string) (map[string]string, error) {

go/go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/eteu-technologies/near-api-go v0.0.2-0.20221130095932-ef78f2789388
 	github.com/ethereum/go-ethereum v1.12.0
 	github.com/kwilteam/kwil-extensions v0.0.0-20230727040522-1cfd930226b7
+	github.com/mr-tron/base58 v1.2.0
 )
 
 require (
@@ -25,18 +26,17 @@ require (
 	github.com/gorilla/websocket v1.4.2 // indirect
 	github.com/grpc-ecosystem/go-grpc-middleware/v2 v2.0.0-rc.5 // indirect
 	github.com/holiman/uint256 v1.2.2-0.20230321075855-87b91420868c // indirect
-	github.com/mr-tron/base58 v1.2.0 // indirect
 	github.com/shirou/gopsutil v3.21.4-0.20210419000835-c7a38de76ee5+incompatible // indirect
 	github.com/shopspring/decimal v1.3.1 // indirect
 	github.com/tklauser/go-sysconf v0.3.5 // indirect
 	github.com/tklauser/numcpus v0.2.2 // indirect
-	golang.org/x/crypto v0.11.0 // indirect
+	golang.org/x/crypto v0.17.0 // indirect
 	golang.org/x/net v0.12.0 // indirect
 	golang.org/x/sync v0.2.0 // indirect
-	golang.org/x/sys v0.10.0 // indirect
-	golang.org/x/text v0.11.0 // indirect
+	golang.org/x/sys v0.15.0 // indirect
+	golang.org/x/text v0.14.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20230710151506-e685fd7b542b // indirect
-	google.golang.org/grpc v1.56.2 // indirect
+	google.golang.org/grpc v1.56.3 // indirect
 	google.golang.org/protobuf v1.31.0 // indirect
 	gopkg.in/natefinch/npipe.v2 v2.0.0-20160621034901-c1b8fa8bdcce // indirect
 )