thrempp/component/threema/receive.go

package threema

import (
	"errors"
	"fmt"
	"strconv"

	"github.com/bdlm/log"
	"github.com/o3ma/o3"
	"gosrc.io/xmpp"
)

func (a *Account) receiver(out chan<- xmpp.Packet) {
	for receivedMessage := range a.receive {
		if receivedMessage.Err != nil {
			log.Warnf("Error Receiving Message: %s\n", receivedMessage.Err)
			xMSG := xmpp.NewMessage("chat", "", a.XMPP.String(), "", "en")
			xMSG.Body = fmt.Sprintf("error on decoding message:\n%v", receivedMessage.Err)
			out <- xMSG
			continue
		}
		sender := receivedMessage.Msg.Sender().String()
		if string(a.TID) == sender {
			continue
		}
		if p, err := a.receiving(receivedMessage.Msg); err != nil {
			xMSG := xmpp.NewMessage("chat", sender, a.XMPP.String(), "", "en")
			xMSG.Body = fmt.Sprintf("error on decoding message: %s\n%v", err, receivedMessage.Msg.Serialize())
			out <- xMSG
		} else if p != nil {
			out <- p
		}
	}
}

func requestExtensions(xMSG *xmpp.Message) {
	xMSG.Extensions = append(xMSG.Extensions, xmpp.ReceiptRequest{})
	xMSG.Extensions = append(xMSG.Extensions, xmpp.Markable{})
	xMSG.Extensions = append(xMSG.Extensions, xmpp.StateActive{})
}

func (a *Account) receiving(receivedMessage o3.Message) (xmpp.Packet, error) {
	logger := log.WithFields(map[string]interface{}{
		"from": receivedMessage.Sender().String(),
		"to":   a.XMPP.String(),
	})
	switch msg := receivedMessage.(type) {
	case o3.TextMessage:
		sender := msg.Sender().String()
		xMSG := xmpp.NewMessage("chat", sender, a.XMPP.String(), strconv.FormatUint(msg.ID(), 10), "en")
		xMSG.Body = msg.Text()
		requestExtensions(&xMSG)
		logger.WithField("text", xMSG.Body).Debug("send text")
		return xMSG, nil

	case o3.AudioMessage:
		if a.threema.httpUploadPath == "" {
			return nil, errors.New("no place to store files at transport configurated")
		}
		data, err := msg.GetAudioData(a.Session)
		if err != nil {
			logger.Warnf("unable to read data from message: %s", err)
			return nil, err
		}
		xMSG, err := a.FileToXMPP(msg.Sender().String(), msg.ID(), "mp3", data)
		if err != nil {
			logger.Warnf("unable to create data from message: %s", err)
			return nil, err
		}
		xMSG.Type = "chat"
		requestExtensions(&xMSG)
		logger.WithField("url", xMSG.Body).Debug("send audio")
		return xMSG, nil

	case o3.ImageMessage:
		if a.threema.httpUploadPath == "" {
			return nil, errors.New("no place to store files at transport configurated")
		}
		data, err := msg.GetImageData(a.Session)
		if err != nil {
			logger.Warnf("unable to read data from message: %s", err)
			return nil, err
		}
		xMSG, err := a.FileToXMPP(msg.Sender().String(), msg.ID(), "jpg", data)
		if err != nil {
			logger.Warnf("unable to create data from message: %s", err)
			return nil, err
		}
		xMSG.Type = "chat"
		requestExtensions(&xMSG)
		logger.WithField("url", xMSG.Body).Debug("send image")
		return xMSG, nil

	case o3.DeliveryReceiptMessage:
		msgID := msg.MsgID()
		xMSG := xmpp.NewMessage("chat", msg.Sender().String(), a.XMPP.String(), "", "en")
		state := ""

		if msg.Status() == o3.MSGDELIVERED {
			state = "delivered"
			if id, ok := a.deliveredMSG[msgID]; ok {
				xMSG.Extensions = append(xMSG.Extensions, xmpp.ReceiptReceived{ID: id})
				xMSG.Extensions = append(xMSG.Extensions, xmpp.MarkReceived{ID: id})
				delete(a.deliveredMSG, msgID)
			} else {
				logger.Warnf("found not id in cache to announce received on xmpp side")
			}
		}
		if msg.Status() == o3.MSGREAD {
			state = "displayed"
			if id, ok := a.readedMSG[msgID]; ok {
				xMSG.Extensions = append(xMSG.Extensions, xmpp.MarkDisplayed{ID: id})
				delete(a.readedMSG, msgID)
			} else {
				logger.Warnf("found not id in cache to announce readed on xmpp side")
			}
		}

		if len(xMSG.Extensions) > 0 {
			logger.WithField("state", state).Debug("send state")
			return xMSG, nil
		}
		return nil, nil
	case o3.TypingNotificationMessage:
		xMSG := xmpp.NewMessage("chat", msg.Sender().String(), a.XMPP.String(), strconv.FormatUint(msg.ID(), 10), "en")
		if msg.OnOff != 0 {
			logger.Debug("composing")
			xMSG.Extensions = append(xMSG.Extensions, xmpp.StateComposing{})
		} else {
			logger.Debug("inactive")
			xMSG.Extensions = append(xMSG.Extensions, xmpp.StateInactive{})
		}
		return xMSG, nil
	}
	return nil, errors.New("not known data format")
}