Add pubkey back to wallet template

We may need this later, to pass into the VM template
spacemeshos · Oct 8, 2024 · 2e94bda · 2e94bda
1 parent e134441
commit 2e94bda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/vm/templates/wallet/wallet.go b/vm/templates/wallet/wallet.go
@@ -8,13 +8,15 @@ import (
 
 // New returns Wallet instance with SpawnArguments.
 func New(args *SpawnArguments) *Wallet {
-	return &Wallet{}
+	return &Wallet{PublicKey: args.PublicKey}
 }
 
 //go:generate scalegen
 
 // Wallet is a single-key wallet.
-type Wallet struct{}
+type Wallet struct {
+	PublicKey core.PublicKey
+}
 
 // MaxSpend returns amount specified in the SpendArguments for Spend method.
 func (s *Wallet) MaxSpend(args any) (uint64, error) {