matrixfmt: text formatting Matrix -> TG

Signed-off-by: Sumner Evans <sumner.evans@automattic.com>
2024-08-01 11:42:58 -06:00
parent 882582456e
commit e8b5d286dc
4 changed files with 704 additions and 145 deletions
@@ -0,0 +1,103 @@
+// mautrix-telegram - A Matrix-Telegram puppeting bridge.
+// Copyright (C) 2024 Sumner Evans
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package matrixfmt
+
+import (
+	"context"
+
+	"github.com/gotd/td/tg"
+	"maunium.net/go/mautrix/event"
+
+	"go.mau.fi/mautrix-telegram/pkg/connector/ids"
+	"go.mau.fi/mautrix-telegram/pkg/connector/telegramfmt"
+)
+
+func toTelegramEntity(br telegramfmt.BodyRange) tg.MessageEntityClass {
+	switch val := br.Value.(type) {
+	case telegramfmt.Mention:
+		userID, _ := ids.ParseUserID(val.UserID)
+		return &tg.MessageEntityMentionName{
+			Offset: br.Start,
+			Length: br.Length,
+			UserID: userID,
+		}
+	case telegramfmt.Style:
+		switch val.Type {
+		case telegramfmt.StyleBold:
+			return &tg.MessageEntityBold{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleItalic:
+			return &tg.MessageEntityItalic{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleUnderline:
+			return &tg.MessageEntityUnderline{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleStrikethrough:
+			return &tg.MessageEntityStrike{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleBlockquote:
+			return &tg.MessageEntityBlockquote{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleCode:
+			return &tg.MessageEntityCode{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StylePre:
+			return &tg.MessageEntityPre{Offset: br.Start, Length: br.Length, Language: val.Language}
+		case telegramfmt.StyleEmail:
+			return &tg.MessageEntityEmail{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleTextURL:
+			return &tg.MessageEntityTextURL{Offset: br.Start, Length: br.Length, URL: val.URL}
+		case telegramfmt.StyleURL:
+			return &tg.MessageEntityURL{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleBotCommand:
+			return &tg.MessageEntityBotCommand{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleHashtag:
+			return &tg.MessageEntityHashtag{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleCashtag:
+			return &tg.MessageEntityCashtag{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StylePhone:
+			return &tg.MessageEntityPhone{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleSpoiler:
+			return &tg.MessageEntitySpoiler{Offset: br.Start, Length: br.Length}
+		case telegramfmt.StyleBankCard:
+			return &tg.MessageEntityBankCard{Offset: br.Start, Length: br.Length}
+		default:
+			panic("unsupported style type")
+		}
+	default:
+		panic("unknown body range value")
+	}
+}
+
+func Parse(ctx context.Context, parser *HTMLParser, content *event.MessageEventContent) (string, []tg.MessageEntityClass) {
+	if content.MsgType.IsMedia() && content.FileName == "" {
+		// The body is the filename.
+		return "", nil
+	}
+
+	if content.Format != event.FormatHTML {
+		return content.Body, nil
+	}
+	parseCtx := NewContext(ctx)
+	parseCtx.AllowedMentions = content.Mentions
+	parsed := parser.Parse(content.FormattedBody, parseCtx)
+	if parsed == nil {
+		return "", nil
+	}
+	var bodyRanges []tg.MessageEntityClass
+	if len(parsed.Entities) > 0 {
+		bodyRanges = make([]tg.MessageEntityClass, len(parsed.Entities))
+		for i, ent := range parsed.Entities {
+			bodyRanges[i] = toTelegramEntity(ent)
+		}
+	}
+	return parsed.String.String(), bodyRanges
+}