get jetton master for wallet in intentions

Author: erokhinav

pkg/api/event_handlers.go

@@ -186,7 +186,7 @@ func (h *Handler) GetEvent(ctx context.Context, params oas.GetEventParams) (*oas
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}
-	result := bath.EnrichWithIntentions(trace, actions)
+	result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 	event, err := h.toEvent(ctx, trace, result, params.AcceptLanguage)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
@@ -236,7 +236,7 @@ func (h *Handler) GetAccountEvents(ctx context.Context, params oas.GetAccountEve
 			continue
 			//return nil, toError(http.StatusInternalServerError, err)
 		}
-		result := bath.EnrichWithIntentions(trace, actions)
+		result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 		e, err := h.toAccountEvent(ctx, account.ID, trace, result, params.AcceptLanguage, params.SubjectOnly.Value)
 		if err != nil {
 			events = append(events, h.toUnknownAccountEvent(account.ID, traceID))
@@ -263,7 +263,7 @@ func (h *Handler) GetAccountEvents(ctx context.Context, params oas.GetAccountEve
 			if err != nil {
 				return nil, toError(http.StatusInternalServerError, err)
 			}
-			result := bath.EnrichWithIntentions(trace, actions)
+			result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 			event, err := h.toAccountEvent(ctx, account.ID, trace, result, params.AcceptLanguage, params.SubjectOnly.Value)
 			if err != nil {
 				return nil, toError(http.StatusInternalServerError, err)
@@ -314,7 +314,7 @@ func (h *Handler) GetAccountEvent(ctx context.Context, params oas.GetAccountEven
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}
-	result := bath.EnrichWithIntentions(trace, actions)
+	result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 	event, err := h.toAccountEvent(ctx, account.ID, trace, result, params.AcceptLanguage, params.SubjectOnly.Value)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
@@ -379,7 +379,7 @@ func (h *Handler) EmulateMessageToAccountEvent(ctx context.Context, request *oas
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}
-	result := bath.EnrichWithIntentions(trace, actions)
+	result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 	event, err := h.toAccountEvent(ctx, account.ID, trace, result, params.AcceptLanguage, false)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
@@ -431,7 +431,7 @@ func (h *Handler) EmulateMessageToEvent(ctx context.Context, request *oas.Emulat
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}
-	result := bath.EnrichWithIntentions(trace, actions)
+	result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 	event, err := h.toEvent(ctx, trace, result, params.AcceptLanguage)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
@@ -608,7 +608,7 @@ func (h *Handler) EmulateMessageToWallet(ctx context.Context, request *oas.Emula
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)
 	}
-	result := bath.EnrichWithIntentions(trace, actions)
+	result := bath.EnrichWithIntentions(ctx, trace, actions, h.storage)
 	event, err := h.toAccountEvent(ctx, *walletAddress, trace, result, params.AcceptLanguage, true)
 	if err != nil {
 		return nil, toError(http.StatusInternalServerError, err)

pkg/bath/bath_test.go

@@ -448,7 +448,7 @@ func TestFindActions(t *testing.T) {
 				WithStraws(straws),
 				WithInformationSource(source))
 			require.Nil(t, err)
-			actionsList = EnrichWithIntentions(trace, actionsList)
+			actionsList = EnrichWithIntentions(context.Background(), trace, actionsList, source)
 			results := result{
 				Actions: actionsList.Actions,
 			}

pkg/bath/intentions.go

@@ -1,6 +1,8 @@
 package bath
 
 import (
+	"context"
+	"github.com/labstack/gommon/log"
 	"github.com/tonkeeper/opentonapi/internal/g"
 	"github.com/tonkeeper/opentonapi/pkg/core"
 	"github.com/tonkeeper/tongo"
@@ -17,14 +19,14 @@ type OutMessage struct {
 	tx             *core.Transaction
 }
 
-func EnrichWithIntentions(trace *core.Trace, actions *ActionsList) *ActionsList {
+func EnrichWithIntentions(ctx context.Context, trace *core.Trace, actions *ActionsList, source core.InformationSource) *ActionsList {
 	outMessages, inMsgCount := extractIntentions(trace)
 	if len(outMessages) <= inMsgCount {
 		return actions
 	}
 	outMessages = removeMatchedIntentions(trace, &outMessages)
 	for _, outMsg := range outMessages {
-		newAction := createActionFromMessage(outMsg)
+		newAction := createActionFromMessage(ctx, outMsg, source)
 		added := false
 		for i, action := range actions.Actions {
 			if slices.Contains(action.BaseTransactions, outMsg.tx.Hash) {
@@ -200,7 +202,7 @@ func getOutMessages(transaction *core.Transaction) []OutMessage {
 	return messages
 }
 
-func createActionFromMessage(msgOut OutMessage) Action {
+func createActionFromMessage(ctx context.Context, msgOut OutMessage, source core.InformationSource) Action {
 	var action Action
 	switch body := msgOut.body.(type) {
 	case abi.TextCommentMsgBody:
@@ -258,11 +260,20 @@ func createActionFromMessage(msgOut OutMessage) Action {
 		if msgOut.tx != nil {
 			sender = &msgOut.tx.Account
 		}
+		var jetton tongo.AccountID
+		masters, err := source.JettonMastersForWallets(ctx, []tongo.AccountID{sendersWallet})
+		if err != nil {
+			log.Warn("error getting jetton master: ", err)
+		}
+		if value, ok := masters[sendersWallet]; ok {
+			jetton = value
+		}
 		action = Action{Type: JettonTransfer, JettonTransfer: &JettonTransferAction{
 			Recipient:     recipient,
 			Sender:        sender,
 			Amount:        body.Amount,
 			SendersWallet: sendersWallet,
+			Jetton:        jetton,
 		}}
 	default:
 		dest := parseAccount(msgOut.messageRelaxed.MessageInternal.Dest)