login: reimplement login in connector interface

Signed-off-by: Sumner Evans <sumner.evans@automattic.com>

pkg/connector/client.go

@@ -0,0 +1,103 @@
+package connector
+
+import (
+	"context"
+	"errors"
+
+	"github.com/gotd/td/telegram"
+	"github.com/rs/zerolog"
+	"maunium.net/go/mautrix/bridgev2"
+	"maunium.net/go/mautrix/bridgev2/database"
+	"maunium.net/go/mautrix/bridgev2/networkid"
+)
+
+type TelegramClient struct {
+	main         *TelegramConnector
+	userLogin    *bridgev2.UserLogin
+	client       *telegram.Client
+	clientCancel context.CancelFunc
+}
+
+var _ bridgev2.NetworkAPI = (*TelegramClient)(nil)
+
+// connectTelegramClient blocks until client is connected, calling Run
+// internally.
+// Technique from: https://github.com/gotd/contrib/blob/master/bg/connect.go
+func connectTelegramClient(ctx context.Context, client *telegram.Client) (context.CancelFunc, error) {
+	ctx, cancel := context.WithCancel(ctx)
+
+	errC := make(chan error, 1)
+	initDone := make(chan struct{})
+	go func() {
+		defer close(errC)
+		errC <- client.Run(ctx, func(ctx context.Context) error {
+			close(initDone)
+			<-ctx.Done()
+			if errors.Is(ctx.Err(), context.Canceled) {
+				return nil
+			}
+			return ctx.Err()
+		})
+	}()
+
+	select {
+	case <-ctx.Done(): // context canceled
+		cancel()
+		return func() {}, ctx.Err()
+	case err := <-errC: // startup timeout
+		cancel()
+		return func() {}, err
+	case <-initDone: // init done
+	}
+
+	return cancel, nil
+}
+
+func (t *TelegramClient) Connect(ctx context.Context) (err error) {
+	t.clientCancel, err = connectTelegramClient(ctx, t.client)
+	return
+}
+
+func (t *TelegramClient) GetChatInfo(ctx context.Context, portal *bridgev2.Portal) (*bridgev2.PortalInfo, error) {
+	panic("unimplemented getchatinfo")
+}
+
+func (t *TelegramClient) GetUserInfo(ctx context.Context, ghost *bridgev2.Ghost) (*bridgev2.UserInfo, error) {
+	panic("unimplemented getuserinfo")
+}
+
+func (t *TelegramClient) HandleMatrixEdit(ctx context.Context, msg *bridgev2.MatrixEdit) error {
+	panic("unimplemented edit")
+}
+
+func (t *TelegramClient) HandleMatrixMessage(ctx context.Context, msg *bridgev2.MatrixMessage) (message *database.Message, err error) {
+	panic("unimplemented message")
+}
+
+func (t *TelegramClient) HandleMatrixMessageRemove(ctx context.Context, msg *bridgev2.MatrixMessageRemove) error {
+	panic("unimplemented remove")
+}
+
+func (t *TelegramClient) HandleMatrixReaction(ctx context.Context, msg *bridgev2.MatrixReaction) (emojiID networkid.EmojiID, err error) {
+	panic("unimplemented reaction")
+}
+
+func (t *TelegramClient) HandleMatrixReactionRemove(ctx context.Context, msg *bridgev2.MatrixReactionRemove) error {
+	panic("unimplemented reaction remove")
+}
+
+func (t *TelegramClient) IsLoggedIn() bool {
+	_, err := t.client.Self(context.TODO())
+	return err == nil
+}
+
+func (t *TelegramClient) IsThisUser(ctx context.Context, userID networkid.UserID) bool {
+	panic("unimplemented istheiruser")
+}
+
+func (t *TelegramClient) LogoutRemote(ctx context.Context) {
+	_, err := t.client.API().AuthLogOut(ctx)
+	if err != nil {
+		zerolog.Ctx(ctx).Err(err).Msg("failed to logout on Telegram")
+	}
+}