starter/telegram_bot/telegram_bot.go

package telegram_bot

import (
	"context"
	"errors"
	"github.com/ManyakRus/starter/contextmain"
	"github.com/ManyakRus/starter/log"
	"github.com/ManyakRus/starter/micro"
	"github.com/ManyakRus/starter/stopapp"
	botapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
	"os"
	"sync"
)

// SettingsINI - структура для хранения всех нужных переменных окружения
type SettingsINI struct {
	API_KEY string
}

// Settings хранит все нужные переменные окружения
var Settings SettingsINI

// Client - клиент подклюенный к Telegram
var Client *botapi.BotAPI

// mutexReconnect - защита от многопоточности Reconnect()
var mutexReconnect = &sync.Mutex{}

// NeedReconnect - флаг необходимости переподключения
var NeedReconnect bool

// Connect - подключается к Telegram, или паника при ошибке
func Connect() {

	if Settings.API_KEY == "" {
		FillSettings()
	}

	err := Connect_err()
	LogInfo_Connected(err)

}

// Connect_err - подключается к Telegram, возвращает ошибку
func Connect_err() error {
	var err error

	Client, err = botapi.NewBotAPI(Settings.API_KEY)
	if err != nil {
		return err
	}

	Client.Debug = false

	//log.Printf("Authorized on account %s", bot.Self.UserName)
	//
	//u := botapi.NewUpdate(0)
	//u.Timeout = 60
	//
	//updates := bot.GetUpdatesChan(u)
	//var Users []User
	//Users = make([]User, 0)
	//
	//for update := range updates {
	//	if update.Message == nil { // ignore any non-Message Updates
	//		continue
	//	}
	//
	//	//log.Printf("[%s] %s", update.Message.From.UserName, update.Message.Text)
	//
	//	TextFromUser := update.Message.Text
	//
	//	Text := "Not understand. Fill name or phone number. "
	//	if valid.IsInt(TextFromUser) == true {
	//		Users = FindUser_by_Phone(TextFromUser)
	//		if len(Users) == 0 {
	//			Users = FindUser_by_CellPhone(TextFromUser)
	//		}
	//	} else if HaveAt(TextFromUser) == true {
	//		Users = FindUser_by_Email(TextFromUser)
	//	} else if valid.IsInt(TextFromUser) == false && HaveNumbers(TextFromUser) == false {
	//		Users = FindUser_by_Name(TextFromUser)
	//		if len(Users) == 0 {
	//			Users = FindUser_by_Post(TextFromUser)
	//		}
	//	} else {
	//		Users = FindUser_by_Adress(TextFromUser)
	//	}
	//
	//	if len(Users) > 0 {
	//		Text = ""
	//		for _, User1 := range Users {
	//			Text = Text + User1.String() + "\n"
	//		}
	//	}
	//
	//	if len(Text) > 2000 {
	//		Text = Text[0:2000]
	//		Text = Text + "\n" + "..."
	//		Text = Text + "\n" + "..."
	//		Text = Text + "\n" + "..."
	//	}
	//
	//	msg := botapi.NewMessage(update.Message.Chat.ID, Text)
	//	msg.ReplyToMessageID = update.Message.MessageID
	//	//msg.Entities = append(msg.Entities, )
	//	//append(msg.Entities, )
	//
	//	bot.Send(msg)
	//}

	return err
}

// Reconnect повторное подключение к Telegram, если оно отключено
// или полная остановка программы
func Reconnect(err error) {
	mutexReconnect.Lock()
	defer mutexReconnect.Unlock()

	if err == nil {
		return
	}

	if errors.Is(err, context.Canceled) {
		return
	}

	if Client == nil {
		log.Warn("Reconnect()")
		err := Connect_err()
		if err != nil {
			log.Error("error: ", err)
		}
		return
	}

	sError := err.Error()
	if sError == "Conn closed" {
		micro.Pause(1000)
		log.Warn("Reconnect()")
		err := Connect_err()
		if err != nil {
			log.Error("error: ", err)
		}
		return
	}

	////остановим программу т.к. она не должна работать при неработающеё БД
	//log.Error("STOP app. Error: ", err)
	//stopapp.StopApp()

}

// CloseConnection - закрытие соединения с Telegram
func CloseConnection() {
	if Client == nil {
		return
	}

	err := CloseConnection_err()
	if err != nil {
		log.Error("Postgres gorm CloseConnection() error: ", err)
	} else {
		log.Info("Postgres gorm connection closed")
	}

	return
}

// CloseConnection - закрытие соединения с Telegram, возвращает ошибку
func CloseConnection_err() error {
	var err error
	if Client == nil {
		return err
	}

	Client.StopReceivingUpdates()
	Client = nil

	return err
}

// WaitStop - ожидает отмену глобального контекста
func WaitStop() {

	select {
	case <-contextmain.GetContext().Done():
		log.Warn("Context app is canceled. telegram_bot")
	}

	//
	stopapp.WaitTotalMessagesSendingNow("telegram_bot")

	//
	CloseConnection()

	stopapp.GetWaitGroup_Main().Done()
}

// Start_ctx - необходимые процедуры для подключения к серверу Telegram
// Свой контекст и WaitGroup нужны для остановки работы сервиса Graceful shutdown
// Для тех кто пользуется этим репозиторием для старта и останова сервиса можно просто StartDB()
func Start_ctx(ctx *context.Context, WaitGroup *sync.WaitGroup) error {
	var err error

	//запомним к себе контекст
	contextmain.Ctx = ctx
	if ctx == nil {
		contextmain.GetContext()
	}

	//запомним к себе WaitGroup
	stopapp.SetWaitGroup_Main(WaitGroup)
	if WaitGroup == nil {
		stopapp.StartWaitStop()
	}

	//
	err = Connect_err()
	if err != nil {
		return err
	}

	stopapp.GetWaitGroup_Main().Add(1)
	go WaitStop()

	return err
}

// Start - делает соединение с Telegram, отключение и др.
func Start() {
	err := Connect_err()
	if err != nil {
		log.Panic("telegram_bot Start() error: ", err)
	}

	stopapp.GetWaitGroup_Main().Add(1)
	go WaitStop()

}

// FillSettings загружает переменные окружения в структуру из файла или из переменных окружения
func FillSettings() {
	Settings = SettingsINI{}

	// заполним из переменных оуружения
	Settings.API_KEY = os.Getenv("API_KEY")
	if Settings.API_KEY == "" {
		log.Panicln("Need fill API_KEY ! in os.ENV ")
	}
}

// GetConnection - возвращает соединение к нужной базе данных
func GetConnection() *botapi.BotAPI {
	if Client == nil {
		Connect()
	}

	return Client
}

// LogInfo_Connected - выводит сообщение в Лог, или паника при ошибке
func LogInfo_Connected(err error) {
	if err != nil {
		log.Panicf("Telegram bot not connected with API KEY: %s, error: %w", Settings.API_KEY, err)
	} else {
		log.Info("Telegram bot Connected. With API KEY: %s", Settings.API_KEY)
	}

}