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

pkg/gotd/telegram/auth/qrlogin/qrlogin_test.go

@@ -0,0 +1,187 @@
+package qrlogin
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/gotd/neo"
+
+	"go.mau.fi/mautrix-telegram/pkg/gotd/constant"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/testutil"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/tgmock"
+)
+
+func testQR(t *testing.T, migrate func(ctx context.Context, dcID int) error) (*tgmock.Mock, QR) {
+	mock := tgmock.New(t)
+	return mock, NewQR(tg.NewClient(mock), constant.TestAppID, constant.TestAppHash, Options{
+		Migrate: migrate,
+	})
+}
+
+var testToken = NewToken([]byte("token"), 0)
+
+func TestQR_Export(t *testing.T) {
+	ctx := context.Background()
+	a := require.New(t)
+	mock, qr := testQR(t, nil)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:     constant.TestAppID,
+		APIHash:   constant.TestAppHash,
+		ExceptIDs: []int64{0},
+	}).ThenResult(&tg.AuthLoginToken{
+		Expires: 0,
+		Token:   testToken.token,
+	})
+	result, err := qr.Export(ctx, 0)
+	a.NoError(err)
+	a.Equal(Token{
+		token:   testToken.token,
+		expires: time.Unix(0, 0),
+	}, result)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginTokenMigrateTo{
+		Token: testToken.token,
+	})
+	_, err = qr.Export(ctx)
+	a.Error(err)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenErr(testutil.TestError())
+	_, err = qr.Export(ctx)
+	a.ErrorIs(err, testutil.TestError())
+}
+
+func TestQR_Accept(t *testing.T) {
+	ctx := context.Background()
+	a := require.New(t)
+	mock, qr := testQR(t, nil)
+
+	auth := &tg.Authorization{
+		APIID: 1,
+	}
+	mock.ExpectCall(&tg.AuthAcceptLoginTokenRequest{
+		Token: testToken.token,
+	}).ThenResult(auth)
+	result, err := qr.Accept(ctx, testToken)
+	a.NoError(err)
+	a.Equal(auth, result)
+
+	mock.ExpectCall(&tg.AuthAcceptLoginTokenRequest{
+		Token: testToken.token,
+	}).ThenErr(testutil.TestError())
+	_, err = qr.Accept(ctx, testToken)
+	a.ErrorIs(err, testutil.TestError())
+}
+
+func TestQR_Import(t *testing.T) {
+	ctx := context.Background()
+	a := require.New(t)
+	mock, qr := testQR(t, nil)
+
+	auth := &tg.AuthAuthorization{
+		User: &tg.User{ID: 10},
+	}
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginTokenSuccess{
+		Authorization: auth,
+	})
+	result, err := qr.Import(ctx)
+	a.NoError(err)
+	a.Equal(auth, result)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginTokenMigrateTo{
+		DCID: 1,
+	})
+	_, err = qr.Import(ctx)
+	var mig *MigrationNeededError
+	a.ErrorAs(err, &mig)
+	a.Equal(1, mig.MigrateTo.DCID)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginToken{
+		Token: testToken.token,
+	})
+	_, err = qr.Import(ctx)
+	a.Error(err)
+
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenErr(testutil.TestError())
+	_, err = qr.Import(ctx)
+	a.ErrorIs(err, testutil.TestError())
+}
+
+func TestQR_Auth(t *testing.T) {
+	a := require.New(t)
+	mock := tgmock.New(t)
+	clock := neo.NewTime(time.Now())
+
+	auth := &tg.AuthAuthorization{
+		User: &tg.User{ID: 10},
+	}
+	mock.ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginToken{
+		Expires: int(clock.Now().Add(time.Minute).Unix()),
+		Token:   testToken.token,
+	}).ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginToken{
+		Expires: int(clock.Now().Add(2 * time.Minute).Unix()),
+		Token:   testToken.token,
+	}).ExpectCall(&tg.AuthExportLoginTokenRequest{
+		APIID:   constant.TestAppID,
+		APIHash: constant.TestAppHash,
+	}).ThenResult(&tg.AuthLoginTokenSuccess{
+		Authorization: auth,
+	})
+
+	qr := NewQR(tg.NewClient(mock), constant.TestAppID, constant.TestAppHash, Options{
+		Clock: clock,
+	})
+
+	show := make(chan struct{})
+	done := make(chan error)
+	loggedIn := make(chan struct{})
+	go func() {
+		_, err := qr.Auth(context.Background(), loggedIn, func(ctx context.Context, token Token) error {
+			show <- struct{}{}
+			return nil
+		})
+		done <- err
+	}()
+
+	// Show QR first time.
+	<-show
+
+	// Skip 1 minute, token expires.
+	clock.Travel(time.Minute + 1)
+
+	// Show QR second time.
+	<-show
+
+	// Emulate update, auth done.
+	loggedIn <- struct{}{}
+
+	a.NoError(<-done)
+}