add /warm feature and some improvements for repository

2025-06-23 22:47:07 +03:00
7 changed files with 23 additions and 92 deletions
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # Yandex.Music Downloader Bot

-[![Build Status](https://drone.mrixs.me/api/badges/Mrixs/yamusic-bot/status.svg?branch=dev)](https://drone.mrixs.me/Mrixs/yamusic-bot?branch=dev) [![License](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE)
+[![Build Status](https://drone.mrixs.me/api/badges/Mrixs/yamusic-bot/status.svg?branch=master)](https://drone.mrixs.me/Mrixs/yamusic-bot?branch=master) [![License](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE)

 Удобный и быстрый Telegram-бот для получения аудиофайлов из сервиса Yandex.Music. Работает в inline-режиме, позволяя отправлять музыку в любой чат. Поддерживает поиск, а также ссылки на треки, альбомы и исполнителей.

--- a/cmd/bot/main.go
+++ b/cmd/bot/main.go
@@ -1,5 +1,3 @@
-// yamusic-bot/cmd/bot/main.go
-
 package main

 import (
@@ -17,7 +15,6 @@ import (
 	"gitea.mrixs.me/Mrixs/yamusic-bot/pkg/tagger"
 	"gitea.mrixs.me/Mrixs/yamusic-bot/pkg/yamusic"
 	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
-	"golang.org/x/time/rate"
 )

 func main() {
@@ -57,24 +54,11 @@ func main() {
 	// 4. Инициализация компонентов
 	downloaderComponent := downloader.NewHTTPDownloader()
 	taggerComponent := tagger.NewID3Tagger()
-
-	// Создаем БЫСТРЫЙ лимитер для общих вызовов API (в секунду)
-	fastLimiter := rate.NewLimiter(rate.Limit(cfg.TelegramAPIRateLimit), cfg.TelegramAPIRateLimit)
-	slog.Info("General Telegram API rate limit set", "requests_per_second", cfg.TelegramAPIRateLimit)
-
-	// Создаем МЕДЛЕННЫЙ лимитер для кэш-канала (в минуту)
-	// rate.Limit измеряется в событиях/секунду, поэтому конвертируем
-	// Burst size = 1, чтобы избежать отправки пачки сообщений и долгого ожидания
-	cacheRPS := float64(cfg.TelegramCacheRateLimitPerMinute) / 60.0
-	cacheLimiter := rate.NewLimiter(rate.Limit(cacheRPS), 1)
-	slog.Info("Cache channel Telegram API rate limit set", "requests_per_minute", cfg.TelegramCacheRateLimitPerMinute)
-
-	// Передаем оба лимитера в адаптер
-	telegramClient := bot.NewTelegramClientAdapter(tgAPI, cfg.TelegramCacheChatID, fastLimiter, cacheLimiter)
+	telegramClient := bot.NewTelegramClientAdapter(tgAPI, cfg.TelegramCacheChatID)

 	trackProcessor := processor.NewTrackProcessor(db, yandexClient, downloaderComponent, taggerComponent, telegramClient)

-	adminHandler := admin.NewHandler(db, telegramClient, yandexClient, taggerComponent, startTime)
+	adminHandler := admin.NewHandler(db, telegramClient, yandexClient, startTime)
 	inlineHandler := bot.NewInlineHandler(yandexClient, trackProcessor, telegramClient)

 	// 5. Создание и запуск приложения
--- a/internal/admin/handler.go
+++ b/internal/admin/handler.go
@@ -18,17 +18,15 @@ type Handler struct {
 	storage   interfaces.TrackStorage
 	telegram  interfaces.TelegramClient
 	yandex    interfaces.YandexMusicClient
-	tagger    interfaces.Tagger
 	startTime time.Time
 }

 // NewHandler создает новый обработчик команд администратора.
-func NewHandler(storage interfaces.TrackStorage, telegram interfaces.TelegramClient, yandex interfaces.YandexMusicClient, tagger interfaces.Tagger, startTime time.Time) *Handler {
+func NewHandler(storage interfaces.TrackStorage, telegram interfaces.TelegramClient, yandex interfaces.YandexMusicClient, startTime time.Time) *Handler {
 	return &Handler{
 		storage:   storage,
 		telegram:  telegram,
 		yandex:    yandex,
-		tagger:    tagger,
 		startTime: startTime,
 	}
 }
@@ -167,24 +165,17 @@ func (h *Handler) handleWarmFromDir(ctx context.Context, chatID int64, dirPath s
 				continue
 			}

-			// 2. Читаем метатеги из файла
-			title, artist, err := h.tagger.ReadMetadata(fullPath)
-			if err != nil {
-				slog.Warn("Failed to read metadata from file, using fallback", "path", fullPath, "error", err)
-				title = ""  // Используем ID как заголовок
-				artist = "" // Используем заглушку как исполнителя
-			}
-
-			// 3. Загружаем в Telegram с корректными метаданными
+			// 2. Загружаем в Telegram
+			// Поскольку метатеги уже вшиты, для отображения в кэш-канале можно использовать простые title/performer
 			slog.Debug("Uploading track to cache channel", "track_id", trackID, "path", fullPath)
-			fileID, err := h.telegram.SendAudioToCacheChannel(ctx, fullPath, title, artist)
+			fileID, err := h.telegram.SendAudioToCacheChannel(ctx, fullPath, trackID, "Pre-cached")
 			if err != nil {
 				slog.Error("Failed to upload pre-cached file", "track_id", trackID, "error", err)
 				errorCount++
 				continue
 			}

-			// 4. Сохраняем в БД
+			// 3. Сохраняем в БД
 			err = h.storage.Set(ctx, trackID, fileID)
 			if err != nil {
 				slog.Error("Failed to save pre-cached file to storage", "track_id", trackID, "error", err)
--- a/internal/bot/telegram.go
+++ b/internal/bot/telegram.go
@@ -6,35 +6,24 @@ import (
 	"log/slog"

 	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
-	"golang.org/x/time/rate"
 )

 // TelegramClientAdapter адаптирует библиотеку tgbotapi под наш интерфейс interfaces.TelegramClient.
 type TelegramClientAdapter struct {
 	api         *tgbotapi.BotAPI
 	cacheChatID int64
-	fastLimiter  *rate.Limiter // Для общих быстрых запросов
-	cacheLimiter *rate.Limiter // Для медленных запросов в кэш-канал
 }

 // NewTelegramClientAdapter создает новый адаптер.
-func NewTelegramClientAdapter(api *tgbotapi.BotAPI, cacheChatID int64, fastLimiter, cacheLimiter *rate.Limiter) *TelegramClientAdapter {
+func NewTelegramClientAdapter(api *tgbotapi.BotAPI, cacheChatID int64) *TelegramClientAdapter {
 	return &TelegramClientAdapter{
 		api:         api,
 		cacheChatID: cacheChatID,
-		fastLimiter:  fastLimiter,
-		cacheLimiter: cacheLimiter,
 	}
 }

 // SendAudioToCacheChannel загружает аудиофайл в кэш-канал и возвращает его FileID.
-// ИСПОЛЬЗУЕТ МЕДЛЕННЫЙ ЛИМИТЕР.
 func (t *TelegramClientAdapter) SendAudioToCacheChannel(ctx context.Context, audioPath, title, performer string) (string, error) {
-	// Ждем, пока МЕДЛЕННЫЙ лимитер разрешит выполнить запрос
-	if err := t.cacheLimiter.Wait(ctx); err != nil {
-		return "", err
-	}
-
 	audio := tgbotapi.NewAudio(t.cacheChatID, tgbotapi.FilePath(audioPath))
 	audio.Title = title
 	audio.Performer = performer
@@ -52,17 +41,11 @@ func (t *TelegramClientAdapter) SendAudioToCacheChannel(ctx context.Context, aud
 }

 // AnswerInlineQuery отвечает на inline-запрос.
-// ИСПОЛЬЗУЕТ БЫСТРЫЙ ЛИМИТЕР.
 func (t *TelegramClientAdapter) AnswerInlineQuery(ctx context.Context, queryID string, results []interface{}) error {
-	// Ждем, пока БЫСТРЫЙ лимитер разрешит выполнить запрос
-	if err := t.fastLimiter.Wait(ctx); err != nil {
-		return err
-	}
-
 	inlineConfig := tgbotapi.InlineConfig{
 		InlineQueryID: queryID,
 		Results:       results,
-		CacheTime:     1,
+		CacheTime:     1, // Кэшируем результат на стороне Telegram на 1 секунду
 	}

 	if _, err := t.api.Request(inlineConfig); err != nil {
@@ -72,13 +55,7 @@ func (t *TelegramClientAdapter) AnswerInlineQuery(ctx context.Context, queryID s
 }

 // SendMessage отправляет текстовое сообщение.
-// ИСПОЛЬЗУЕТ БЫСТРЫЙ ЛИМИТЕР.
 func (t *TelegramClientAdapter) SendMessage(ctx context.Context, chatID int64, text string) error {
-	// Ждем, пока БЫСТРЫЙ лимитер разрешит выполнить запрос
-	if err := t.fastLimiter.Wait(ctx); err != nil {
-		return err
-	}
-
 	msg := tgbotapi.NewMessage(chatID, text)
 	if _, err := t.api.Send(msg); err != nil {
 		return fmt.Errorf("failed to send message: %w", err)
--- a/internal/config/config.go
+++ b/internal/config/config.go
@@ -18,9 +18,7 @@ type Config struct {
 	LogLevel            string  `env:"LOG_LEVEL" envDefault:"info"`
 	ProcessorWorkers    int     `env:"PROCESSOR_WORKERS" envDefault:"4"`
 	YandexAPIRateLimit  int     `env:"YANDEX_API_RATE_LIMIT" envDefault:"5"`
-	TelegramAPIRateLimit            int     `env:"TELEGRAM_API_RATE_LIMIT" envDefault:"25"`              // Общий лимит в секунду
-	TelegramCacheRateLimitPerMinute int     `env:"TELEGRAM_CACHE_RATE_LIMIT_PER_MINUTE" envDefault:"20"` // Лимит для кэш-канала в минуту
-	TelegramAdminIDs                []int64 `env:"-"`
+	TelegramAdminIDs    []int64 `env:"-"` // Это поле будет заполнено после парсинга
 }

 // New загружает конфигурацию из переменных окружения и парсит необходимые поля.
--- a/internal/interfaces/interfaces.go
+++ b/internal/interfaces/interfaces.go
@@ -34,7 +34,6 @@ type TelegramClient interface {
 // Tagger определяет методы для работы с метаданными аудиофайлов.
 type Tagger interface {
 	WriteTags(filePath string, coverPath string, info *model.TrackInfo) error
-	ReadMetadata(filePath string) (title, artist string, err error)
 }

 // FileDownloader определяет метод для скачивания файла.
--- a/pkg/tagger/id3.go
+++ b/pkg/tagger/id3.go
@@ -17,24 +17,6 @@ func NewID3Tagger() *ID3Tagger {
 	return &ID3Tagger{}
 }

-// ReadMetadata читает основные метаданные (название, исполнитель) из аудиофайла.
-func (t *ID3Tagger) ReadMetadata(filePath string) (string, string, error) {
-	tag, err := id3v2.Open(filePath, id3v2.Options{Parse: true})
-	if err != nil {
-		return "", "", fmt.Errorf("failed to open mp3 file for reading tags: %w", err)
-	}
-	defer tag.Close()
-
-	title := tag.Title()
-	artist := tag.Artist()
-
-	if title == "" || artist == "" {
-		return "", "", fmt.Errorf("title or artist tag is empty")
-	}
-
-	return title, artist, nil
-}
-
 // WriteTags записывает метаданные и обложку в указанный аудиофайл.
 func (t *ID3Tagger) WriteTags(filePath string, coverPath string, info *model.TrackInfo) error {
 	tag, err := id3v2.Open(filePath, id3v2.Options{Parse: true})