core/service/telegram_svc.go

package service

import (
	"fmt"
	"github.com/gofiber/fiber/v2"
	"github.com/rs/xid"
	"path/filepath"
	"penahub.gitlab.yandexcloud.net/backend/quiz/core/clients/telegram"
	"penahub.gitlab.yandexcloud.net/backend/tdlib/client"
)

type Message struct {
	Type string `json:"type"`
	Data string `json:"data"`
}

func (s *Service) GetPoolTgAccounts(ctx *fiber.Ctx) error {
	return nil
}

// todo save to slice and save to db for rewoke
func (s *Service) AddingTgAccount(ctx *fiber.Ctx) error {
	var req telegram.AuthTgUserReq
	if err := ctx.BodyParser(&req); err != nil {
		return ctx.Status(fiber.StatusBadRequest).SendString("Invalid request data")
	}
	authorizer := client.ClientAuthorizerr()
	authorizer.TdlibParameters <- &client.SetTdlibParametersRequest{
		UseTestDc:           false,
		DatabaseDirectory:   filepath.Join(".tdlib", "database"),
		FilesDirectory:      filepath.Join(".tdlib", "files"),
		UseFileDatabase:     false,
		UseChatInfoDatabase: false,
		UseMessageDatabase:  false,
		UseSecretChats:      false,
		ApiId:               req.ApiID,
		ApiHash:             req.ApiHash,
		SystemLanguageCode:  "en",
		DeviceModel:         "Server",
		SystemVersion:       "1.0.0",
		ApplicationVersion:  "1.0.0",
	}

	_, err := client.SetLogVerbosityLevel(&client.SetLogVerbosityLevelRequest{
		NewVerbosityLevel: 1,
	})
	if err != nil {
		return ctx.Status(fiber.StatusInternalServerError).SendString(err.Error())
	}

	var tdlibClient *client.Client

	// todo проверить как завершается
	go func() {
		tdlibClient, err = client.NewClient(authorizer)
		if err != nil {
			fmt.Println("new client failed", err)
			return
		}
	}()

	for {
		state, ok := <-authorizer.State
		if !ok {
			return ctx.SendStatus(fiber.StatusOK)
		}
		fmt.Println("currnet state:", state)
		switch state.AuthorizationStateType() {
		case client.TypeAuthorizationStateWaitPhoneNumber:
			authorizer.PhoneNumber <- req.PhoneNumber

		case client.TypeAuthorizationStateWaitCode:
			id := xid.New()
			s.telegramClient.AddedToMap(telegram.WaitingClient{
				TdLibClient: tdlibClient,
				PreviousReq: req,
				Authorizer:  authorizer,
			}, id.String())
			return ctx.Status(fiber.StatusOK).JSON(id.String())

		case client.TypeAuthorizationStateLoggingOut, client.TypeAuthorizationStateClosing, client.TypeAuthorizationStateClosed:
			return ctx.Status(fiber.StatusForbidden).SendString("auth failed")
		}
	}
}

func (s *Service) SettingTgCode(ctx *fiber.Ctx) error {
	var req struct {
		Code string `json:"code"`
		ID   string `json:"id"`
	}
	if err := ctx.BodyParser(&req); err != nil {
		return ctx.Status(fiber.StatusBadRequest).SendString("Invalid request data")
	}

	data, ok := s.telegramClient.GetFromMap(req.ID)
	if !ok {
		return ctx.Status(fiber.StatusBadRequest).SendString("Invalid id, don't have data")
	}
	data.Authorizer.Code <- req.Code
	for {
		state, ok := <-data.Authorizer.State
		if !ok {
			return ctx.SendStatus(fiber.StatusOK)
		}
		fmt.Println("currnet state:", state)
		switch state.AuthorizationStateType() {
		case client.TypeAuthorizationStateReady:
			return ctx.SendStatus(fiber.StatusOK)
		case client.TypeAuthorizationStateWaitPassword:
			data.Authorizer.Password <- data.PreviousReq.Password
		case client.TypeAuthorizationStateLoggingOut, client.TypeAuthorizationStateClosing, client.TypeAuthorizationStateClosed:
			return ctx.Status(fiber.StatusForbidden).SendString("auth failed")
		}
	}
}

func (s *Service) DeleteTgAccountByID(ctx *fiber.Ctx) error {
	return nil
}