Add state init to jetton transfer

tonkeeper · Sep 27, 2024 · fa4e5f1 · fa4e5f1
1 parent 3a23954
commit fa4e5f1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/contract/jetton/jetton.go b/contract/jetton/jetton.go
@@ -36,6 +36,7 @@ type TransferMessage struct {
 	ForwardTonAmount    tlb.Grams
 	ForwardPayload      *boc.Cell
 	CustomPayload       *boc.Cell
+	StateInit           *tlb.StateInit
 }
 
 func (tm TransferMessage) ToInternal() (tlb.Message, uint8, error) {
@@ -73,6 +74,14 @@ func (tm TransferMessage) ToInternal() (tlb.Message, uint8, error) {
 		Mode:    wallet.DefaultMessageMode,
 		Body:    c,
 	}
+	if tm.StateInit != nil {
+		if tm.StateInit.Code.Exists {
+			m.Code = &tm.StateInit.Code.Value.Value
+		}
+		if tm.StateInit.Data.Exists {
+			m.Data = &tm.StateInit.Data.Value.Value
+		}
+	}
 	return m.ToInternal()
 }