feat: add user client

userclient/auth.go

@@ -0,0 +1,80 @@
+package userclient
+
+import (
+	"strings"
+
+	"github.com/celestix/gotgproto"
+	"github.com/charmbracelet/huh"
+	"github.com/charmbracelet/log"
+	"github.com/fatih/color"
+)
+
+type termialAuthConversator struct{}
+
+func (t *termialAuthConversator) AskPhoneNumber() (string, error) {
+	phone := ""
+	err := huh.NewInput().Title("Your Phone Number").
+		Placeholder("+44 123456").
+		Prompt("> ").
+		Value(&phone).
+		WithTheme(huh.ThemeCatppuccin()).
+		Run()
+
+	if err != nil {
+		return "", err
+	}
+
+	log.Info("Sending code to your phone number...")
+
+	return strings.TrimSpace(phone), nil
+}
+
+func (t *termialAuthConversator) AskCode() (string, error) {
+	code := ""
+	err := huh.NewInput().Title("Your Code").
+		Placeholder("123456").
+		Value(&code).
+		Prompt("> ").
+		WithTheme(huh.ThemeCatppuccin()).
+		Run()
+
+	if err != nil {
+		return "", err
+	}
+
+	return strings.TrimSpace(code), nil
+}
+
+func (t *termialAuthConversator) AskPassword() (string, error) {
+	pwd := ""
+
+	err := huh.NewInput().Title("Your 2FA Password").
+		EchoMode(huh.EchoModePassword).
+		Value(&pwd).
+		Prompt("> ").
+		WithTheme(huh.ThemeCatppuccin()).
+		Run()
+	if err != nil {
+		return "", err
+	}
+
+	return strings.TrimSpace(pwd), nil
+}
+
+func (t *termialAuthConversator) AuthStatus(authStatus gotgproto.AuthStatus) {
+	switch authStatus.Event {
+	case gotgproto.AuthStatusPhoneRetrial:
+		color.Red("The phone number you just entered seems to be incorrect,")
+		color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
+		color.Red("Please try again....")
+	case gotgproto.AuthStatusPasswordRetrial:
+		color.Red("The 2FA password you just entered seems to be incorrect,")
+		color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
+		color.Red("Please try again....")
+	case gotgproto.AuthStatusPhoneCodeRetrial:
+		color.Red("The OTP you just entered seems to be incorrect,")
+		color.Red("Attempts Left: %d", authStatus.AttemptsLeft)
+		color.Red("Please try again....")
+	default:
+	}
+}

userclient/middlewares/middlewares.go

@@ -0,0 +1,29 @@
+package middlewares
+
+import (
+	"context"
+	"time"
+
+	"github.com/cenkalti/backoff/v4"
+	"github.com/gotd/contrib/middleware/floodwait"
+	"github.com/gotd/td/telegram"
+	"github.com/krau/SaveAny-Bot/userclient/middlewares/recovery"
+	"github.com/krau/SaveAny-Bot/userclient/middlewares/retry"
+)
+
+func NewDefaultMiddlewares(ctx context.Context, timeout time.Duration) []telegram.Middleware {
+	return []telegram.Middleware{
+		recovery.New(ctx, newBackoff(timeout)),
+		retry.New(5),
+		floodwait.NewSimpleWaiter(),
+	}
+}
+
+func newBackoff(timeout time.Duration) backoff.BackOff {
+	b := backoff.NewExponentialBackOff()
+
+	b.Multiplier = 1.1
+	b.MaxElapsedTime = timeout
+	b.MaxInterval = 10 * time.Second
+	return b
+}

userclient/middlewares/recovery/recovery.go

@@ -0,0 +1,61 @@
+package recovery
+
+import (
+	"context"
+	"time"
+
+	"github.com/cenkalti/backoff/v4"
+	"github.com/go-faster/errors"
+	"github.com/gotd/td/bin"
+	"github.com/gotd/td/telegram"
+	"github.com/gotd/td/tg"
+	"github.com/gotd/td/tgerr"
+	"github.com/krau/SaveAny-Bot/common"
+)
+
+type recovery struct {
+	ctx     context.Context
+	backoff backoff.BackOff
+}
+
+func New(ctx context.Context, backoff backoff.BackOff) telegram.Middleware {
+	return &recovery{
+		ctx:     ctx,
+		backoff: backoff,
+	}
+}
+
+func (r *recovery) Handle(next tg.Invoker) telegram.InvokeFunc {
+	return func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+
+		return backoff.RetryNotify(func() error {
+			if err := next.Invoke(ctx, input, output); err != nil {
+				if r.shouldRecover(ctx, err) {
+					return errors.Wrap(err, "recover")
+				}
+
+				return backoff.Permanent(err)
+			}
+
+			return nil
+		}, r.backoff, func(err error, duration time.Duration) {
+			common.Log.Debug("Wait for connection recovery", "error", err, "duration", duration)
+		})
+	}
+}
+
+func (r *recovery) shouldRecover(ctx context.Context, err error) bool {
+	// context in recovery is used to stop recovery process by external os signal, otherwise we will wait till max retries when user press ctrl+c
+	select {
+	case <-r.ctx.Done():
+		return false
+	case <-ctx.Done():
+		return false
+	default:
+	}
+
+	// we try recover when encountered any error that is not telegram business error
+	_, ok := tgerr.As(err)
+
+	return !ok
+}

userclient/middlewares/retry/retry.go

@@ -0,0 +1,56 @@
+package retry
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/go-faster/errors"
+	"github.com/gotd/td/bin"
+	"github.com/gotd/td/telegram"
+	"github.com/gotd/td/tg"
+	"github.com/gotd/td/tgerr"
+	"github.com/krau/SaveAny-Bot/common"
+)
+
+var internalErrors = []string{
+	"Timedout", // #373
+	"No workers running",
+	"RPC_CALL_FAIL",
+	"RPC_MCGET_FAIL",
+	"WORKER_BUSY_TOO_LONG_RETRY", // #462
+	"memory limit exit",          // #504
+}
+
+type retry struct {
+	max    int
+	errors []string
+}
+
+func (r retry) Handle(next tg.Invoker) telegram.InvokeFunc {
+	return func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+		retries := 0
+
+		for retries < r.max {
+			if err := next.Invoke(ctx, input, output); err != nil {
+				if tgerr.Is(err, r.errors...) {
+					common.Log.Debug("retry middleware", "retries", retries, "error", err)
+					retries++
+					continue
+				}
+				return errors.Wrap(err, "retry middleware skip")
+			}
+
+			return nil
+		}
+
+		return fmt.Errorf("retry limit reached after %d attempts", r.max)
+	}
+}
+
+// New returns middleware that retries request if it fails with one of provided errors.
+func New(max int, errors ...string) telegram.Middleware {
+	return retry{
+		max:    max,
+		errors: append(errors, internalErrors...), // #373
+	}
+}

userclient/userclient.go

@@ -0,0 +1,74 @@
+package userclient
+
+import (
+	"context"
+	"time"
+
+	"github.com/celestix/gotgproto"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/celestix/gotgproto/sessionMaker"
+	"github.com/glebarez/sqlite"
+	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/userclient/middlewares"
+)
+
+var UC *gotgproto.Client
+var ectx *ext.Context
+
+func GetCtx() *ext.Context {
+	if ectx != nil {
+		return ectx
+	}
+	ectx = UC.CreateContext()
+	return ectx
+}
+
+func Login(ctx context.Context) (*gotgproto.Client, error) {
+	common.Log.Debug("Logging in as user client")
+	if UC != nil {
+		return UC, nil
+	}
+	res := make(chan struct {
+		client *gotgproto.Client
+		err    error
+	})
+	go func() {
+		tclient, err := gotgproto.NewClient(
+			config.Cfg.Telegram.AppID,
+			config.Cfg.Telegram.AppHash,
+			gotgproto.ClientTypePhone(""),
+			&gotgproto.ClientOpts{
+				Session:         sessionMaker.SqlSession(sqlite.Open(config.Cfg.Telegram.Userbot.Session)),
+				AuthConversator: &termialAuthConversator{},
+				// Context:          ctx,
+				DisableCopyright: true,
+				Middlewares:      middlewares.NewDefaultMiddlewares(ctx, 5*time.Minute),
+			},
+		)
+		if err != nil {
+			res <- struct {
+				client *gotgproto.Client
+				err    error
+			}{nil, err}
+		}
+		res <- struct {
+			client *gotgproto.Client
+			err    error
+		}(struct {
+			client *gotgproto.Client
+			err    error
+		}{tclient, nil})
+	}()
+
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	case r := <-res:
+		if r.err != nil {
+			return nil, r.err
+		}
+		UC = r.client
+		return UC, nil
+	}
+}