Added creation of taproot addresses

Author: shadowy-pycoder

bmt/main.go

@@ -622,7 +622,7 @@ func (k *PrivateKey) ToWif(uncompressed bool) (*string, error) {
 //
 // If both parameters are nil generates a random wallet
 func CreateNewWallet(raw *big.Int, wif *string) (*Wallet, error) {
-	var nestedAddress, nativeAddress string
+	var nestedAddress, nativeAddress, taprootAddress string
 	privKey, err := NewPrivateKey(raw, wif)
 	if err != nil {
 		return nil, err
@@ -636,13 +636,15 @@ func CreateNewWallet(raw *big.Int, wif *string) (*Wallet, error) {
 	if !privKey.Uncompressed {
 		nestedAddress = createNestedSegwit(pubKey)
 		nativeAddress = createNativeSegwit(pubKey)
+		taprootAddress = createTaproot(createTweakedPubKey(rawPubKey))
 	}
 	return &Wallet{PrivKey: privKey,
 		RawPubKey: rawPubKey,
 		PubKey:    hex.EncodeToString(pubKey),
 		Legacy:    legacyAddress,
 		Nested:    nestedAddress,
-		Native:    nativeAddress}, nil
+		Native:    nativeAddress,
+		Taproot:   taprootAddress}, nil
 }
 
 type Wallet struct {
@@ -652,6 +654,7 @@ type Wallet struct {
 	Legacy    string
 	Nested    string
 	Native    string
+	Taproot   string
 }
 
 // String returns a formatted string representation of the Wallet.
@@ -670,7 +673,8 @@ Public Key (HEX Compressed): %s
 Legacy Address: %s
 Nested SegWit Address: %s
 Native SegWit Address: %s
-`, w.PrivKey.Raw, *w.PrivKey.Wif, w.RawPubKey, w.PubKey, w.Legacy, w.Nested, w.Native)
+Taproot Address: %s
+`, w.PrivKey.Raw, *w.PrivKey.Wif, w.RawPubKey, w.PubKey, w.Legacy, w.Nested, w.Native, w.Taproot)
 }
 
 // NewInt converts a hexadecimal string to a big.Int pointer.
@@ -832,6 +836,32 @@ func createPubKey(rawPubKey *Point, uncompressed bool) []byte {
 	return buf[:33]
 }
 
+func calculateTweak(rawPubKey *Point) *big.Int {
+	var tweak big.Int
+	buf := make([]byte, 32)
+	h1 := sha256.New()
+	h2 := sha256.New()
+	h1.Write([]byte("TapTweak"))
+	h2.Write(joinBytes([][]byte{h1.Sum(nil), h1.Sum(nil), rawPubKey.X.FillBytes(buf)}...))
+	tweak.SetBytes(h2.Sum(nil))
+	return &tweak
+}
+
+func createTweakedPubKey(rawPubKey *Point) []byte {
+	var q JacobianPoint
+	tweak := calculateTweak(rawPubKey)
+	p := &Point{X: new(big.Int).Set(rawPubKey.X), Y: new(big.Int).Set(rawPubKey.Y)}
+	if IsOdd(p.Y) {
+		p.Y.Sub(Secp256k1.PCurve, p.Y)
+	}
+	q.Add(p.ToJacobian(), q.Mul(tweak, nil))
+	qa := q.ToAffine()
+	if IsOdd(qa.Y) {
+		qa.Y.Sub(Secp256k1.PCurve, qa.Y)
+	}
+	return createPubKey(qa, false)[1:]
+}
+
 // checkSum calculates the checksum of the input byte slice using DoubleSHA256 and returns the first 4 bytes.
 //
 // Parameters:
@@ -893,6 +923,21 @@ func createNativeSegwit(pubKey []byte) string {
 	return addr
 }
 
+func createTaproot(pubKey []byte) string {
+	converted, err := bech32.ConvertBits(pubKey, 8, 5, true)
+	if err != nil {
+		panic(err)
+	}
+	combined := make([]byte, len(converted)+1)
+	combined[0] = byte(1)
+	copy(combined[1:], converted)
+	addr, err := bech32.EncodeM("bc", combined)
+	if err != nil {
+		panic(err)
+	}
+	return addr
+}
+
 // varInt generates a variable-length integer in bytes based on the input length.
 //
 // Parameters:

bmt/main_test.go

@@ -79,10 +79,11 @@ func TestCreateNewWalletFromRawPrivateKey(t *testing.T) {
 					X: NewInt("e4ab48ce61667f57bb9ca0f31b9bca94981303004d3802b1c11faa9343a820ba"),
 					Y: NewInt("17d5acd0ade3141bdccab4689a82151e638a70a56c1b4d38788e77c76e414dd3"),
 				},
-				PubKey: "03e4ab48ce61667f57bb9ca0f31b9bca94981303004d3802b1c11faa9343a820ba",
-				Legacy: "1LBdJkrB4nkWkLLdKSBAD2CPRrpSaYQW51",
-				Nested: "39RBCbgESnqjHpapfvqVPfLCPGrvGKxpbC",
-				Native: "bc1q6fkdkwv5p6mxdhyj2gxhgu2tp5np53g6g8u0t0",
+				PubKey:  "03e4ab48ce61667f57bb9ca0f31b9bca94981303004d3802b1c11faa9343a820ba",
+				Legacy:  "1LBdJkrB4nkWkLLdKSBAD2CPRrpSaYQW51",
+				Nested:  "39RBCbgESnqjHpapfvqVPfLCPGrvGKxpbC",
+				Native:  "bc1q6fkdkwv5p6mxdhyj2gxhgu2tp5np53g6g8u0t0",
+				Taproot: "bc1puzp40apyn38h3nqhzzxk6tmxxwvd0jcegnk4k88qvxntstsfhyvq5nr3us",
 			},
 		},
 		{
@@ -97,10 +98,11 @@ func TestCreateNewWalletFromRawPrivateKey(t *testing.T) {
 					X: NewInt("e483d1df60d0c9e16035672dfb92d7ddac6858b5233d33bb04996ae6a23f0149"),
 					Y: NewInt("d0fb3cafc082957fdd8934a9fa2b5fbf2fee701ca595da7bb79a95f0063a13c5"),
 				},
-				PubKey: "03e483d1df60d0c9e16035672dfb92d7ddac6858b5233d33bb04996ae6a23f0149",
-				Legacy: "13KXYTfq5FRQj6T3t6ds6FYqCDymYMwdmC",
-				Nested: "34GeUW8iPUkWdfQSWFvumAt5RGtxhp3KCG",
-				Native: "bc1qr9cncqd8v5j9uv86pfprqlkuh7htp5uwu6x56m",
+				PubKey:  "03e483d1df60d0c9e16035672dfb92d7ddac6858b5233d33bb04996ae6a23f0149",
+				Legacy:  "13KXYTfq5FRQj6T3t6ds6FYqCDymYMwdmC",
+				Nested:  "34GeUW8iPUkWdfQSWFvumAt5RGtxhp3KCG",
+				Native:  "bc1qr9cncqd8v5j9uv86pfprqlkuh7htp5uwu6x56m",
+				Taproot: "bc1ph4zg3xz7z0kvmdp9nne6kpyc8yk9rsweepa3g5se40zkqa7rvwvslkl8fd",
 			},
 		},
 	}
@@ -140,10 +142,11 @@ func TestCreateNewWalletFromWifPrivateKey(t *testing.T) {
 					X: NewInt("66fde8efb47bfe53be63cccd068996f3a1c9172f1c2f6bc345dff6c589daefc5"),
 					Y: NewInt("379ab4aa30d3247cb6b7c327067bd8199ff7bec32205ef043d3c9239ec6765b7"),
 				},
-				PubKey: "0366fde8efb47bfe53be63cccd068996f3a1c9172f1c2f6bc345dff6c589daefc5",
-				Legacy: "14VXJudstcvqpx56BPQFSYJ1K39KGc3twM",
-				Nested: "32vwEU1Esvp1AywrYgrnziHCaigLTn5AKN",
-				Native: "bc1qyex4xghj4rldkysse2hdpt0arphh78xhp28ze2",
+				PubKey:  "0366fde8efb47bfe53be63cccd068996f3a1c9172f1c2f6bc345dff6c589daefc5",
+				Legacy:  "14VXJudstcvqpx56BPQFSYJ1K39KGc3twM",
+				Nested:  "32vwEU1Esvp1AywrYgrnziHCaigLTn5AKN",
+				Native:  "bc1qyex4xghj4rldkysse2hdpt0arphh78xhp28ze2",
+				Taproot: "bc1pss2mar9sv2nktnx8ujqrjd9ld7n0zn9swl7zup5pcl8shq9es4kqlgx6g3",
 			},
 		},
 		{