%s
(This message is a poll. Please open WhatsApp to vote.)
", event.TextToHTML(msg.GetName()), strings.Join(optionsListHTML, "")) maxChoices := int(msg.GetSelectableOptionsCount()) if maxChoices <= 0 { maxChoices = len(optionNames) } evtType := event.EventMessage if mc.ExtEvPolls { evtType = event.EventUnstablePollStart } return &bridgev2.ConvertedMessagePart{ Type: evtType, Content: &event.MessageEventContent{ Body: body, MsgType: event.MsgText, Format: event.FormatHTML, FormattedBody: formattedBody, }, Extra: map[string]any{ // Custom metadata "fi.mau.whatsapp.poll": map[string]any{ "option_names": optionNames, "selectable_options_count": msg.GetSelectableOptionsCount(), }, // Legacy extensible events "org.matrix.msc1767.message": []map[string]any{ {"mimetype": "text/html", "body": formattedBody}, {"mimetype": "text/plain", "body": body}, }, "org.matrix.msc3381.poll.start": map[string]any{ "kind": "org.matrix.msc3381.poll.disclosed", "max_selections": maxChoices, "question": map[string]any{ "org.matrix.msc1767.text": msg.GetName(), }, "answers": msc3381Answers, }, }, }, msg.GetContextInfo() } func rerouteMessageKey(ctx context.Context, chat, sender types.JID, groupLIDAddressing bool) types.JID { if store := getClient(ctx).Store; store != nil && chat.Server == types.DefaultUserServer && sender.Server == types.HiddenUserServer { senderPN, _ := store.LIDs.GetPNForLID(ctx, sender) zerolog.Ctx(ctx).Debug(). Stringer("orig_participant", sender). Stringer("rerouted_participant", senderPN). Msg("Rerouting message key (PN recipient in LID DM)") if !senderPN.IsEmpty() { return senderPN } } else if store != nil && chat.Server == types.GroupServer && sender.Server == types.DefaultUserServer && groupLIDAddressing { senderLID, _ := store.LIDs.GetLIDForPN(ctx, sender) zerolog.Ctx(ctx).Debug(). Stringer("orig_participant", sender). Stringer("rerouted_participant", senderLID). Msg("Rerouting message key (PN recipient in LID group)") if !senderLID.IsEmpty() { return senderLID } } return sender } func KeyToMessageID(ctx context.Context, client *whatsmeow.Client, chat, sender types.JID, key *waCommon.MessageKey) networkid.MessageID { groupLIDAddressing := sender.Server == types.HiddenUserServer sender = sender.ToNonAD() var err error if !key.GetFromMe() { if key.GetParticipant() != "" { sender, err = types.ParseJID(key.GetParticipant()) if err != nil { // TODO log somehow? return "" } if sender.Server == types.LegacyUserServer { sender.Server = types.DefaultUserServer } } else if chat.Server == types.DefaultUserServer || chat.Server == types.BotServer { if sender.User == client.Store.GetJID().User || sender.User == client.Store.GetLID().User { // Message key is not from the sender, but message sender (containing key) is me, // so message key sender is the other user in the DM sender = chat } else { // Message key is not from the sender, but message sender (containing key) is not me, // so message key sender is me sender = client.Store.GetJID().ToNonAD() } } else { zerolog.Ctx(ctx).Warn(). Stringer("chat", chat). Stringer("sender", sender). Any("key", key). Msg("Failed to get message ID from key") return "" } } remoteJID, err := types.ParseJID(key.GetRemoteJID()) if err == nil && !remoteJID.IsEmpty() { // TODO use remote jid in other cases? if remoteJID.Server == types.GroupServer { chat = remoteJID } } sender = rerouteMessageKey( context.WithValue(ctx, contextKeyClient, client), chat, sender, groupLIDAddressing, ) return waid.MakeMessageID(chat, sender, key.GetID()) } var failedPollUpdatePart = &bridgev2.ConvertedMessagePart{ Type: event.EventUnstablePollResponse, Content: &event.MessageEventContent{}, DontBridge: true, } func (mc *MessageConverter) convertPollUpdateMessage(ctx context.Context, info *types.MessageInfo, msg *waE2E.PollUpdateMessage) (*bridgev2.ConvertedMessagePart, *waE2E.ContextInfo) { log := zerolog.Ctx(ctx) pollMessageID := KeyToMessageID(ctx, getClient(ctx), info.Chat, info.Sender, msg.PollCreationMessageKey) pollMessage, err := mc.Bridge.DB.Message.GetPartByID(ctx, getPortal(ctx).Receiver, pollMessageID, "") if err != nil { log.Err(err).Msg("Failed to get poll update target message") return failedPollUpdatePart, nil } else if pollMessage == nil { log.Warn().Str("target_message_id", string(pollMessageID)).Msg("Poll update target message not found") return failedPollUpdatePart, nil } vote, err := getClient(ctx).DecryptPollVote(ctx, &events.Message{ Info: *info, Message: &waE2E.Message{PollUpdateMessage: msg}, }) if err != nil { log.Err(err).Msg("Failed to decrypt vote message") return failedPollUpdatePart, nil } selectedHashes := make([]string, len(vote.GetSelectedOptions())) if pollMessage.Metadata.(*waid.MessageMetadata).IsMatrixPoll { mappedAnswers, err := mc.DB.PollOption.GetIDs(ctx, pollMessage.MXID, vote.GetSelectedOptions()) if err != nil { log.Err(err).Msg("Failed to get poll option IDs") return failedPollUpdatePart, nil } for i, opt := range vote.GetSelectedOptions() { if len(opt) != 32 { log.Warn().Int("hash_len", len(opt)).Msg("Unexpected option hash length in vote") continue } var ok bool selectedHashes[i], ok = mappedAnswers[[32]byte(opt)] if !ok { log.Warn().Hex("option_hash", opt).Msg("Didn't find ID for option in vote") } } } else { for i, opt := range vote.GetSelectedOptions() { selectedHashes[i] = hex.EncodeToString(opt) } } return &bridgev2.ConvertedMessagePart{ Type: event.EventUnstablePollResponse, Content: &event.MessageEventContent{ RelatesTo: &event.RelatesTo{ Type: event.RelReference, EventID: pollMessage.MXID, }, }, Extra: map[string]any{ "org.matrix.msc3381.poll.response": map[string]any{ "answers": selectedHashes, }, }, }, nil }