fix minor bugs

Author: Magomed

cmd/main.go

@@ -37,7 +37,7 @@ func main() {
 	botAPI.Debug = false
 	slog.Info(fmt.Sprintf("Авторизован на аккаунте %s", botAPI.Self.UserName))
 
-	connDB, err := bot.ConnectPostgresDB() // Подключение к базе данных PostgreSQL
+	connDB, err := bot.ConnectPostgresDB() // Подключение к базе данных Psql
 	if err != nil {
 		os.Exit(1)
 	}
@@ -111,27 +111,3 @@ func initBotAPI() (*tgbotapi.BotAPI, tgbotapi.UpdatesChannel, error) {
 
 	return botAPI, updates, nil
 }
-
-//Таблица users (Пользователи)
-//Поле						Тип данных									Описание
-//id						SERIAL (Primary Key)						Уникальный ID пользователя в базе
-//telegram_id				BIGINT UNIQUE								ID пользователя в Telegram
-//chat_id					BIGINT UNIQUE								ID чата с пользователем
-//referral_code 			VARCHAR(255) UNIQUE							Реферальная ссылка пользователя
-//created_at				TIMESTAMP DEFAULT CURRENT_TIMESTAMP			Время создания записи
-//referred_by				INT											ID пользователя кто его пригласил
-//FOREIGN KEY (referred_by) REFERENCES users(id) ON DELETE SET NULL  	Внешний ключ на приглашённого
-
-//Таблица keys (Ключи доступа)
-//Поле						Тип данных									Описание
-//id						SERIAL (Primary Key)						Уникальный ID ключа
-//user_id					INT (Foreign Key)							Ссылка на пользователя (users.id)
-//key						VARCHAR(255)								Ключ доступа
-//created_at				TIMESTAMP									Время создания ключа
-//expires_at				TIMESTAMP									Время истечения ключа
-//CHECK (expires_at > created_at)         								Проверка, что время истечения ключа больше времени создания
-
-//CREATE INDEX idx_referral_code ON users(referral_code);
-//CREATE INDEX idx_telegram_id ON users(telegram_id);
-//CREATE INDEX idx_referred_by ON users(referred_by); - В ДАЛЁКОМ БУДУШЕМ (ЕСЛИ ПРОЕКТ БУДЕТ ЖИТЬ)
-//

internal/bot/http_request.go

@@ -97,8 +97,6 @@ func (h *HttpRequest) AddedOutlineKeyLimit(keyID string, url string) error {
 	// Формируем полный URL для запроса
 	apiURL := fmt.Sprintf("%s/access-keys/%s/data-limit", url, keyID)
 
-	fmt.Println("ВОТ:", apiURL)
-
 	// Формируем тело запроса с ограничением (например, 10000 байт).
 	requestBody := `{"limit": {"bytes": 1048576}}` // 1 МБ
 
@@ -128,7 +126,7 @@ func (h *HttpRequest) AddedOutlineKeyLimit(keyID string, url string) error {
 		return err
 	}
 
-	slog.Info("Ограничение по ключу %d успешно установлено\n", keyID)
+	slog.Info(fmt.Sprintf("Ограничение по ключу %v успешно установлено\n", keyID))
 	return nil
 }
 

internal/bot/messages.go

@@ -125,9 +125,13 @@ func (m *MessengerBot) GetInfoStart(update tgbotapi.Update) {
 	case err == nil && time.Now().After(expirationTime):
 		// Если ключ истёк
 		respMessage =
-			"⚠️ *Доступ заблокирован* \n\n" +
-				"❌ *Ваш ключ истёк!* \n\n" +
-				"🔄 *Продлите подписку, чтобы продолжить пользоваться сервисом.*"
+			"⚠️ *К сожалению, доступ временно заблокирован.* \n\n" +
+				"❌ *Ваш ключ истёк, и вы больше не можете пользоваться сервисом.* \n\n" +
+				"💡 *Что делать дальше?*\n" +
+				"🔹 Нажмите на «Продлить».\n" +
+				"🔹 Выберите подходящий тариф и оплатите его.\n" +
+				"🔹 Наслаждайтесь безопасным и быстрым интернетом снова! 🚀\n\n" +
+				"🔄 *Продлите подписку прямо сейчас и вернитесь в сеть!*"
 
 	case err == nil:
 		// Если всё в порядке
@@ -448,13 +452,13 @@ func (m *MessengerBot) ManageUserDataAfterPayment(value, userTgID string) {
 		return
 	}
 
-	expirationDate := expiresAt.Format("02.01.2006 15:04")
+	expirationDate := newExpiration.Format("02.01.2006 15:04")
 
 	// Формируем сообщение для пользователя о новом сроке действия
 	text := fmt.Sprintf(
 		"🎉 *Ваш ключ успешно продлён!* 🎉\n\n"+
 			"📅 *Ключ действителен до: * `%s`.\n\n"+
-			"Спасибо, что остаетесь с нами!"+
+			"Спасибо, что остаетесь с нами!\n"+
 			"Мы ценим вашу поддержку! 😊",
 		expirationDate,
 	)
@@ -485,7 +489,7 @@ func (m *MessengerBot) AddReferralSubscriptionDays(userID string, expirationTime
 
 	// Получаем реферала пользователя
 	referralUserID, err := m.repo.GetUserReferral(intUserID)
-	if errors.Is(err, sql.ErrNoRows) {
+	if err != nil || referralUserID == 0 {
 		// Если реферала нет, просто выходим
 		return
 	}
@@ -709,10 +713,20 @@ func (m *MessengerBot) GetConnectStrOrReferral(userTgID int64, referral bool) {
 
 	// Логируем успешную отправку ключа или реферальной ссылки
 	if referral {
-		text = fmt.Sprintf("🎉 *Ваша реферальная ссылка для приглашения:* \n```%s```", keyOrReferral)
+		text = fmt.Sprintf("🎉 *Ваша реферальная ссылка:* \n\n```%s``` \n"+
+			"💡 *Как это работает:*\n\n"+
+			"🔹 Поделитесь ссылкой с друзьями.\n"+
+			"🔹 За *каждый купленный месяц подписки* вашим рефералом вы получаете *+4 бесплатных дня* к своей подписке!\n"+
+			"🔹 Чем больше друзей пригласите, тем больше бесплатных дней получите. 🚀\n\n"+
+			"✨ *Спасибо, что пользуетесь Keeper VPN!*", keyOrReferral)
 	} else {
 		// Если ключ найден, отправляем его пользователю
-		text = fmt.Sprintf("🔑 *Ваш ключ доступа:* \n```%s```", keyOrReferral)
+		text = fmt.Sprintf("🔑 *Ваш ключ доступа:* \n\n```%s``` \n"+
+			"💡 *Как использовать ключ:*\n\n"+
+			"🔹 Установите приложение *Outline*.\n"+
+			"🔹 Скопируйте ключ и вставьте его в приложение.\n"+
+			"🔹 Наслаждайтесь безопасным и быстрым интернетом! 🚀\n\n"+
+			"✨ *Спасибо, что пользуетесь Keeper VPN!*", keyOrReferral)
 	}
 
 	// Создаем сообщение и устанавливаем ParseMode
@@ -785,12 +799,16 @@ func (m *MessengerBot) SetLimitForExpiredKeys() {
 	for {
 		select {
 		case <-ticker.C:
+			slog.Info("Старт проверки истёкших ключей")
+
 			// Идем в базу и вытаскиваем key_id_outline для всех просроченных ключей
 			expiredKeysID, err := m.repo.GetExpirationTimeKeysID()
 			if err != nil {
 				continue
 			}
 
+			slog.Info(fmt.Sprintf("Найдено %v протухших ключей", len(expiredKeysID)))
+
 			// Параллельно делаем HTTP-запросы к Outline Manager и ставим лимиты
 			var wg sync.WaitGroup
 			for _, keyID := range expiredKeysID {

internal/bot/repository.go

@@ -70,14 +70,15 @@ func (r *Repo) GetUserReferral(userID int) (int, error) {
 
 	err := r.db.QueryRow(query, userID).Scan(&referralUserID)
 	if err != nil {
-		if errors.Is(err, sql.ErrNoRows) {
-			slog.Warn(op, slog.String("message", "Нет реферала у пользователя"), slog.Int("userID", userID))
-			return 0, nil // Нет реферала — это не ошибка
-		}
 		slog.Error(op, slog.String("Ошибка запроса к БД", err.Error()))
 		return 0, err
 	}
 
+	if referralUserID == 0 {
+		slog.Warn(op, slog.String("message", "Нет реферала у пользователя"), slog.Int("userID", userID))
+		return 0, nil // Нет реферала — это не ошибка
+	}
+
 	return referralUserID, nil
 }
 
@@ -216,7 +217,7 @@ func (r *Repo) UpdateProcessedKey(keyID int, flag bool) error {
 		return fmt.Errorf("ключ с key_id_outline=%d не найден", keyID)
 	}
 
-	slog.Info(op, "Флажок processed успешно обновлён",
+	slog.Info("Флажок processed успешно обновлён",
 		slog.Int("keyID", keyID),
 		slog.Bool("flag", flag),
 	)

internal/bot/router.go

@@ -4,5 +4,5 @@ import "net/http"
 
 // InitRout - Функция инициализирует маршруты для HTTP-сервера.
 func InitRout(handler *HttpHandler) {
-	http.HandleFunc("/webhook", handler.PaymentWebhook) // TODO: изменить эндпоинт по мере необходимости.
+	http.HandleFunc("/webhook", handler.PaymentWebhook)
 }