signer helper (#29)

flashbots · Oct 30, 2024 · f6b894b · f6b894b
1 parent fc623ca
commit f6b894b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/signature/signature.go b/signature/signature.go
@@ -87,6 +87,22 @@ func NewSigner(privateKey *ecdsa.PrivateKey) Signer {
 	}
 }
 
+// NewSignerFromHexPrivateKey creates new signer from 0x-prefixed hex-encoded private key
+func NewSignerFromHexPrivateKey(hexPrivateKey string) (*Signer, error) {
+	privateKeyBytes, err := hexutil.Decode(hexPrivateKey)
+	if err != nil {
+		return nil, err
+	}
+
+	privateKey, err := crypto.ToECDSA(privateKeyBytes)
+	if err != nil {
+		return nil, err
+	}
+
+	signer := NewSigner(privateKey)
+	return &signer, nil
+}
+
 func NewRandomSigner() (*Signer, error) {
 	privateKey, err := crypto.GenerateKey()
 	if err != nil {

diff --git a/signature/signature_test.go b/signature/signature_test.go
@@ -147,14 +147,9 @@ func TestSignatureCreateCompareToCastAndEthers(t *testing.T) {
 	// This purposefully uses the already highly compromised keypair from the go-ethereum book:
 	// https://goethereumbook.org/transfer-eth/
 	// privateKey = fad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19
-	privateKeyBytes, err := hexutil.Decode("0xfad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
+	signer, err := signature.NewSignerFromHexPrivateKey("0xfad9c8855b740a0b7ed4c221dbad0f33a83a49cad6b3fe8d5817ac83d38b6a19")
 	require.NoError(t, err)
 
-	privateKey, err := crypto.ToECDSA(privateKeyBytes)
-	require.NoError(t, err)
-
-	signer := signature.NewSigner(privateKey)
-
 	address := signer.Address()
 	body := []byte("Hello")