Release v0.1.0

address_generator.go

@@ -0,0 +1,103 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/muun/libwallet"
+)
+
+type signingDetails struct {
+	RedeemScript []byte
+	Address 	 libwallet.MuunAddress
+}
+
+type AddressGenerator struct {
+	addrs   map[string]signingDetails
+	userKey *libwallet.HDPrivateKey
+	muunKey *libwallet.HDPrivateKey
+}
+
+func NewAddressGenerator(userKey, muunKey *libwallet.HDPrivateKey) *AddressGenerator {
+	return &AddressGenerator{
+		addrs:   make(map[string]signingDetails),
+		userKey: userKey,
+		muunKey: muunKey,
+	}
+}
+
+func (g *AddressGenerator) Addresses() map[string]signingDetails {
+	return g.addrs
+}
+
+func (g *AddressGenerator) Generate() {
+	g.generateChangeAddrs()
+	g.generateExternalAddrs()
+	g.generateContactAddrs(100)
+}
+
+func (g *AddressGenerator) generateChangeAddrs() {
+	const changePath = "m/1'/1'/0"
+	changeUserKey, _ := g.userKey.DeriveTo(changePath)
+	changeMuunKey, _ := g.muunKey.DeriveTo(changePath)
+
+	g.deriveTree(changeUserKey, changeMuunKey, 2000, "change")
+}
+
+func (g *AddressGenerator) generateExternalAddrs() {
+	const externalPath = "m/1'/1'/1"
+	externalUserKey, _ := g.userKey.DeriveTo(externalPath)
+	externalMuunKey, _ := g.muunKey.DeriveTo(externalPath)
+
+	g.deriveTree(externalUserKey, externalMuunKey, 2000, "external")
+}
+
+func (g *AddressGenerator) generateContactAddrs(numContacts int64) {
+	const addressPath = "m/1'/1'/2"
+	contactUserKey, _ := g.userKey.DeriveTo(addressPath)
+	contactMuunKey, _ := g.muunKey.DeriveTo(addressPath)
+	for i := int64(0); i <= numContacts; i++ {
+		partialContactUserKey, _ := contactUserKey.DerivedAt(i, false)
+		partialMuunUserKey, _ := contactMuunKey.DerivedAt(i, false)
+
+		branch := fmt.Sprintf("contacts-%v", i)
+		g.deriveTree(partialContactUserKey, partialMuunUserKey, 200, branch)
+	}
+}
+
+func (g *AddressGenerator) deriveTree(rootUserKey, rootMuunKey *libwallet.HDPrivateKey, count int64, name string) {
+
+	for i := int64(0); i <= count; i++ {
+		userKey, err := rootUserKey.DerivedAt(i, false)
+		if err != nil {
+			log.Printf("skipping child %v for %v due to %v", i, name, err)
+			continue
+		}
+		muunKey, err := rootMuunKey.DerivedAt(i, false)
+		if err != nil {
+			log.Printf("skipping child %v for %v due to %v", i, name, err)
+			continue
+		}
+
+		addrV2, err := libwallet.CreateAddressV2(userKey.PublicKey(), muunKey.PublicKey())
+		if err == nil {
+			g.addrs[addrV2.Address()] = signingDetails{
+				RedeemScript: addrV2.(libwallet.RedeemableAddress).RedeemScript(),
+				Address: addrV2,
+			}
+		}  else {
+			log.Printf("failed to generate %v v2 for %v due to %v", name, i, err)
+		}
+
+		addrV3, err := libwallet.CreateAddressV3(userKey.PublicKey(), muunKey.PublicKey())
+		if err == nil {
+			g.addrs[addrV3.Address()] = signingDetails{
+				RedeemScript: addrV3.(libwallet.RedeemableAddress).RedeemScript(),
+				Address: addrV3,
+			}
+		}  else {
+			log.Printf("failed to generate %v v3 for %v due to %v", name, i, err)
+		}
+
+	}
+}