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,2 @@
|
||||
// Package unpack contains some updates result unpacking helpers.
|
||||
package unpack
|
||||
@@ -0,0 +1,80 @@
|
||||
package unpack
|
||||
|
||||
import (
|
||||
"github.com/go-faster/errors"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/internal/upconv"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
func extractClass(u tg.UpdateClass) (tg.MessageClass, bool) {
|
||||
switch v := u.(type) {
|
||||
case *tg.UpdateNewMessage:
|
||||
return v.Message, true
|
||||
case *tg.UpdateNewChannelMessage:
|
||||
return v.Message, true
|
||||
default:
|
||||
return nil, false
|
||||
}
|
||||
}
|
||||
|
||||
// MessageClass tries to unpack sent message and returns it as MessageClass.
|
||||
func MessageClass(u tg.UpdatesClass, err error) (tg.MessageClass, error) {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var updates []tg.UpdateClass
|
||||
switch v := u.(type) {
|
||||
case *tg.UpdateShortMessage:
|
||||
short := upconv.ShortMessage(v)
|
||||
updates = []tg.UpdateClass{short.Update}
|
||||
case *tg.UpdateShortChatMessage:
|
||||
short := upconv.ShortChatMessage(v)
|
||||
updates = []tg.UpdateClass{short.Update}
|
||||
case *tg.UpdateShortSentMessage:
|
||||
short := upconv.ShortSentMessage(v)
|
||||
updates = []tg.UpdateClass{short.Update}
|
||||
case *tg.UpdateShort:
|
||||
updates = []tg.UpdateClass{v.Update}
|
||||
case *tg.UpdatesCombined:
|
||||
updates = v.GetUpdates()
|
||||
case *tg.Updates:
|
||||
updates = v.GetUpdates()
|
||||
default:
|
||||
return nil, errors.Errorf("unexpected type %T", u)
|
||||
}
|
||||
|
||||
for _, update := range updates {
|
||||
if msg, ok := extractClass(update); ok {
|
||||
return msg, nil
|
||||
}
|
||||
}
|
||||
|
||||
return nil, errors.Errorf("bad updates result %+v", updates)
|
||||
}
|
||||
|
||||
// Message tries to unpack sent message and returns it as Message.
|
||||
func Message(u tg.UpdatesClass, err error) (*tg.Message, error) {
|
||||
msg, err := MessageClass(u, err)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
m, ok := msg.(*tg.Message)
|
||||
if !ok {
|
||||
return nil, errors.Errorf("unexpected type %T", msg)
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
||||
// MessageID tries to unpack sent message and returns message id.
|
||||
func MessageID(u tg.UpdatesClass, err error) (int, error) {
|
||||
msg, err := MessageClass(u, err)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return msg.GetID(), nil
|
||||
}
|
||||
@@ -0,0 +1,44 @@
|
||||
package unpack_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/unpack"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
func unpackMessage(ctx context.Context) error {
|
||||
client, err := telegram.ClientFromEnvironment(telegram.Options{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return client.Run(ctx, func(ctx context.Context) error {
|
||||
sender := message.NewSender(tg.NewClient(client))
|
||||
|
||||
msg, err := unpack.Message(sender.Resolve("@durovschat").Dice(ctx))
|
||||
// Sends dice "🎲" to the @durovschat.
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Println("Sent message ID:", msg.ID)
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func ExampleMessage() {
|
||||
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
|
||||
defer cancel()
|
||||
|
||||
if err := unpackMessage(ctx); err != nil {
|
||||
_, _ = fmt.Fprintf(os.Stderr, "%+v\n", err)
|
||||
os.Exit(2)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,111 @@
|
||||
package unpack
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
func TestMessage(t *testing.T) {
|
||||
testMsg := &tg.Message{
|
||||
ID: 10,
|
||||
Message: "Golang is always going to do some approximation of the right thing.",
|
||||
}
|
||||
testErr := errors.New("женой накормили толпу")
|
||||
|
||||
type args struct {
|
||||
u tg.UpdatesClass
|
||||
err error
|
||||
}
|
||||
tests := []struct {
|
||||
name string
|
||||
input args
|
||||
output *tg.Message
|
||||
isErr bool
|
||||
}{
|
||||
{"Good", args{
|
||||
u: &tg.Updates{
|
||||
Updates: []tg.UpdateClass{
|
||||
&tg.UpdateNewMessage{
|
||||
Message: testMsg,
|
||||
},
|
||||
},
|
||||
},
|
||||
err: nil,
|
||||
}, testMsg, false},
|
||||
{"GoodChannel", args{
|
||||
u: &tg.Updates{
|
||||
Updates: []tg.UpdateClass{
|
||||
&tg.UpdateNewChannelMessage{
|
||||
Message: testMsg,
|
||||
},
|
||||
},
|
||||
},
|
||||
err: nil,
|
||||
}, testMsg, false},
|
||||
{"Short", args{
|
||||
u: &tg.UpdateShort{
|
||||
Update: &tg.UpdateNewMessage{
|
||||
Message: testMsg,
|
||||
},
|
||||
},
|
||||
err: nil,
|
||||
}, testMsg, false},
|
||||
{"ShortSent", args{
|
||||
u: &tg.UpdateShortSentMessage{
|
||||
Out: testMsg.Out,
|
||||
ID: testMsg.ID,
|
||||
Date: testMsg.Date,
|
||||
Media: testMsg.Media,
|
||||
Entities: testMsg.Entities,
|
||||
TTLPeriod: testMsg.TTLPeriod,
|
||||
},
|
||||
err: nil,
|
||||
}, testMsg, false},
|
||||
{"ShortChat", args{
|
||||
u: &tg.UpdateShortChatMessage{
|
||||
Out: testMsg.Out,
|
||||
Mentioned: testMsg.Mentioned,
|
||||
MediaUnread: testMsg.MediaUnread,
|
||||
Silent: testMsg.Silent,
|
||||
ID: testMsg.ID,
|
||||
Message: testMsg.Message,
|
||||
Date: testMsg.Date,
|
||||
FwdFrom: testMsg.FwdFrom,
|
||||
ViaBotID: testMsg.ViaBotID,
|
||||
ReplyTo: testMsg.ReplyTo,
|
||||
Entities: testMsg.Entities,
|
||||
TTLPeriod: testMsg.TTLPeriod,
|
||||
},
|
||||
err: nil,
|
||||
}, testMsg, false},
|
||||
{"ExternalError", args{
|
||||
u: nil,
|
||||
err: testErr,
|
||||
}, nil, true},
|
||||
{"BadUnpack", args{
|
||||
u: &tg.UpdateShort{
|
||||
Update: &tg.UpdateConfig{},
|
||||
},
|
||||
err: nil,
|
||||
}, nil, true},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
a := require.New(t)
|
||||
|
||||
r, err := Message(test.input.u, test.input.err)
|
||||
if test.isErr {
|
||||
a.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
a.NoError(err)
|
||||
a.Equal(testMsg.ID, r.ID)
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user