move gotd fork into repo. (#111)
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
This commit is contained in:
@@ -0,0 +1,150 @@
|
||||
package entity
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"sort"
|
||||
"strings"
|
||||
"unicode"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
// SortEntities sorts entities as TDLib does it.
|
||||
func SortEntities(entity []tg.MessageEntityClass) {
|
||||
sort.Sort(entitySorter(entity))
|
||||
}
|
||||
|
||||
type entitySorter []tg.MessageEntityClass
|
||||
|
||||
func (e entitySorter) Len() int {
|
||||
return len(e)
|
||||
}
|
||||
|
||||
func (e entitySorter) Less(i, j int) bool {
|
||||
a, b := e[i], e[j]
|
||||
return a.GetOffset() < b.GetOffset() ||
|
||||
a.GetLength() > b.GetLength()
|
||||
}
|
||||
|
||||
func (e entitySorter) Swap(i, j int) {
|
||||
e[i], e[j] = e[j], e[i]
|
||||
}
|
||||
|
||||
// setLength sets Length field of entity.
|
||||
func setLength(index, value int, slice []tg.MessageEntityClass) {
|
||||
reflect.ValueOf(&slice[index]).
|
||||
Elem().Elem().Elem().
|
||||
FieldByName("Length").
|
||||
SetInt(int64(value))
|
||||
}
|
||||
|
||||
// fixEntities trims space, if needed and fixes entities offsets.
|
||||
func (b *Builder) fixEntities(msg string, entities []tg.MessageEntityClass) (string, []tg.MessageEntityClass) {
|
||||
// If there are no entities or last text block does not have entities,
|
||||
// so we just return built message.
|
||||
if len(b.lengths) == 0 || b.lastFormatIndex >= len(entities) {
|
||||
return msg, entities
|
||||
}
|
||||
|
||||
// Since Telegram client does not handle space after formatted message
|
||||
// we should compute length of the last block to trim it.
|
||||
// Get first entity of last text block.
|
||||
entity := b.lengths[len(b.lengths)-1]
|
||||
offset := entity.offset
|
||||
length := entity.length
|
||||
// Get last text block.
|
||||
lastBlock := msg[offset:]
|
||||
// Trim this block.
|
||||
trimmed := strings.TrimRightFunc(lastBlock, unicode.IsSpace)
|
||||
|
||||
// If there are a difference, we should change length of the all entities.
|
||||
if length >= len(lastBlock) && len(trimmed) != len(lastBlock) {
|
||||
length := ComputeLength(trimmed)
|
||||
for idx := range entities[b.lastFormatIndex:] {
|
||||
setLength(idx, length, entities[b.lastFormatIndex:])
|
||||
}
|
||||
|
||||
msg = msg[:offset+len(trimmed)]
|
||||
}
|
||||
|
||||
return msg, entities
|
||||
}
|
||||
|
||||
// Raw returns raw result and resets builder without fixing spaces.
|
||||
func (b *Builder) Raw() (string, []tg.MessageEntityClass) {
|
||||
msg := b.message.String()
|
||||
entities := b.entities
|
||||
b.Reset()
|
||||
return msg, entities
|
||||
}
|
||||
|
||||
// Complete returns build result and resets builder.
|
||||
func (b *Builder) Complete() (string, []tg.MessageEntityClass) {
|
||||
msg, entities := b.Raw()
|
||||
defer SortEntities(entities)
|
||||
|
||||
return b.fixEntities(msg, entities)
|
||||
}
|
||||
|
||||
// ShrinkPreCode merges following <pre> and <code> entities, if needed.
|
||||
//
|
||||
// This function is used by formatters to be compliant with TDLib.
|
||||
func (b *Builder) ShrinkPreCode() {
|
||||
b.entities = shrinkPreCode(b.entities)
|
||||
}
|
||||
|
||||
// equalRange compares ranges of given entities.
|
||||
func equalRange(a, b tg.MessageEntityClass) bool {
|
||||
return a.GetLength() == b.GetLength() && a.GetOffset() == b.GetOffset()
|
||||
}
|
||||
|
||||
// shrinkPreCode merges following <pre> and <code> entities, if needed.
|
||||
func shrinkPreCode(entities []tg.MessageEntityClass) []tg.MessageEntityClass {
|
||||
for i, j := 0, len(entities)-1; i < j; i, j = i+1, j-1 {
|
||||
entities[i], entities[j] = entities[j], entities[i]
|
||||
}
|
||||
|
||||
filter := func(keep func(prev, cur tg.MessageEntityClass) bool) []tg.MessageEntityClass {
|
||||
n := 0
|
||||
for i, val := range entities {
|
||||
if i == 0 || keep(entities[i-1], val) {
|
||||
entities[n] = val
|
||||
n++
|
||||
}
|
||||
}
|
||||
return entities[:n]
|
||||
}
|
||||
|
||||
isPreCode := func(class tg.MessageEntityClass) bool {
|
||||
typeID := class.TypeID()
|
||||
return typeID == tg.MessageEntityCodeTypeID || typeID == tg.MessageEntityPreTypeID
|
||||
}
|
||||
|
||||
hasLang := func(class tg.MessageEntityClass) bool {
|
||||
pre, ok := class.(*tg.MessageEntityPre)
|
||||
return ok && pre.Language != ""
|
||||
}
|
||||
|
||||
resetLang := func(class tg.MessageEntityClass) {
|
||||
pre, ok := class.(*tg.MessageEntityPre)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
pre.Language = ""
|
||||
}
|
||||
|
||||
return filter(func(prev, cur tg.MessageEntityClass) bool {
|
||||
if !isPreCode(prev) ||
|
||||
!isPreCode(cur) ||
|
||||
prev.TypeID() == cur.TypeID() {
|
||||
// Keep if not is Pre/Code entities or if they are same.
|
||||
return true
|
||||
}
|
||||
if !equalRange(prev, cur) {
|
||||
resetLang(prev)
|
||||
resetLang(cur)
|
||||
return true
|
||||
}
|
||||
return !hasLang(prev)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user