ref-ing - N messengers

Author: Pavel Loginov

.gitignore

@@ -7,4 +7,6 @@ recap
 state.txt
 *.tar.gz
 
+notes.md
+
 recap.yaml

README.md

@@ -4,6 +4,8 @@ _Mainly, this bot is for using Sber's free tokens, which is why everything is in
 
 Telegram бот для создания краткого содержания из голосовых сообщений и видео-кружочков.
 
+Теперь поддерживает и другие мессенджеры через плагины.
+
 Повтор попыток через очередь, ротация токенов Сбера, резевные модели в OpenRouter.
 
 Коротко: Go, Sber Speech, OpenRouter, Telegram, ffmpeg.
@@ -16,7 +18,7 @@ Telegram бот для создания краткого содержания и
 
 В основе лежит **канал taskQueue** через который проходят все операции:
 
-1. **Receive** → Telegram сообщения попадают в taskQueue
+1. **Receive** → Telegram (и другие) сообщения попадают в taskQueue
 2. **Download** → Загрузка файла из Telegram + конвертация видео в аудио (если нужно)
 3. **STT** → Speech-to-Text преобразование
 4. **Summarize** → OpenRouter создает краткое содержание
@@ -27,6 +29,8 @@ Telegram бот для создания краткого содержания и
 ## Настройка
 
 1. Создайте бота в Telegram через [@BotFather](https://t.me/BotFather) и получите токен.
+   
+   *Для других мессенджеров что-то подобное будет.*
 
 2. Получите API ключи:
    - **Sber**: client_secret (может быть парой client_id:secret в base64, либо укажите client_id отдельно)
@@ -66,14 +70,18 @@ tar xfz recap_x64.tar.gz
 
 Бот будет отвечать на голосовые сообщения и видео-кружочки в группе, как и в личных сообщениях.
 
+*Для других мессенджеров настройки групп могут отличаться. Но часто нужны права на чтение сообщений.*
+
 ## Структура проекта
 
 - `main.go` - точка входа, настройка и запуск
 - `bot.go` - основная логика бота, обработчики сообщений
 - `config.go` - конфигурация и типы
 - `logger.go` - настройка логирования
 - `worker.go` - обработка задач в воркерах
-- `telegram.go` - работа с Telegram API
+- `messenger.go` - интерфейсы мессенджеров (плагинная архитектура)
+- `messenger_telegram.go` - работа с Telegram API
+- `messenger_*` - прочие мессенджеры
 - `media.go` - конвертация медиафайлов
 - `yandex.go` - интеграция с Yandex SpeechKit
 - `sber.go` - интеграция с Sber SpeechKit
@@ -102,3 +110,5 @@ go run main.go
 ## TODO
  - [ ] ограничения на размер войса в минутах (конфиг)
  - [ ] лимиты по вермению на юзера/группу (в памяти)
+ - [ ] новый слой Bot, отделить мессенджер от текство и промптов
+

bot.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"os"
 	"time"
 
@@ -9,31 +10,35 @@ import (
 )
 
 type Config struct {
-	Recognizer    string           `yaml:"recognizer" default:"sber"`
-	Telegram      struct {
-		Token string `yaml:"token"`
-	} `yaml:"telegram"`
-	Max           struct {
-		Token string `yaml:"token"`
-	} `yaml:"max"`
-	Sber struct {
-		Tokens []struct {
-			Name         string        `yaml:"name"`
-			ClientID     string        `yaml:"client_id"`
-			ClientSecret string        `yaml:"client_secret"`
-			Cooldown     time.Duration `yaml:"cooldown"`
-			Limit        time.Duration `yaml:"limit"`
-		} `yaml:"tokens"`
-	} `yaml:"sber"`
-	Openrouter    ConfigOpenrouter  `yaml:"openrouter"`
-	Prompts       ConfigPrompts     `yaml:"prompts"`
-	Messages      ConfigMessages    `yaml:"messages"`
-	RateLimit     ConfigRateLimit   `yaml:"rate_limit"`
-	NumWorkers    int               `yaml:"num_workers" default:"2"`
-	WaitOnError   time.Duration     `yaml:"wait_on_error" default:"3s"`
-	FFmpegPath    string            `yaml:"ffmpeg_path"`
-	SaveDebugMedia bool             `yaml:"save_debug_media" default:"false"`
-	StateFile     string            `yaml:"state_file"`
+	Recognizer     string               `yaml:"recognizer" default:"sber"`
+	Sber           ConfigSber           `yaml:"sber"`
+	Openrouter     ConfigOpenrouter     `yaml:"openrouter"`
+	Prompts        ConfigPrompts        `yaml:"prompts"`
+	Messages       ConfigMessages       `yaml:"messages"`
+	RateLimit      ConfigRateLimit      `yaml:"rate_limit"`
+	NumWorkers     int                  `yaml:"num_workers" default:"2"`
+	WaitOnError    time.Duration        `yaml:"wait_on_error" default:"3s"`
+	FFmpegPath     string               `yaml:"ffmpeg_path" default:"ffmpeg"`
+	SaveDebugMedia bool                 `yaml:"save_debug_media" default:"false"`
+	StateFile      string               `yaml:"state_file"`
+	Bots           map[string]ConfigBot `yaml:"bots"`
+}
+
+type ConfigBot struct {
+	ID        string        `yaml:"id"`
+	Messenger MessengerType `yaml:"messenger"`
+	Token     string        `yaml:"token"`
+	Prompt    string        `yaml:"prompt"`
+}
+
+type ConfigSber struct {
+	Tokens []struct {
+		Name         string        `yaml:"name"`
+		ClientID     string        `yaml:"client_id"`
+		ClientSecret string        `yaml:"client_secret"`
+		Cooldown     time.Duration `yaml:"cooldown"`
+		Limit        time.Duration `yaml:"limit"`
+	} `yaml:"tokens"`
 }
 
 type ConfigOpenrouter struct {
@@ -46,18 +51,17 @@ type ConfigOpenrouter struct {
 }
 
 type ConfigPrompts struct {
-	Prompts struct {
-		SystemPrompt string `yaml:"system"`
-		UserPrompt   string `yaml:"user"`
-	} `yaml:"prompts"`
+	SystemPrompt string `yaml:"system"`
+	UserPrompt   string `yaml:"user"`
 }
 
 type ConfigMessages struct {
-	Listening      string `yaml:"listening" default:"Listening"`
-	StartMessage   string `yaml:"start" default:"Send me a voice message and I'll transcribe it for you!"`
-	ErrorMessage   string `yaml:"error" default:"An error occurred while processing your message. Please try again."`
-	FailureMessage string `yaml:"failure" default:"Failed to transcribe the audio. Please check the audio quality and try again."`
-	RetryMessage   string `yaml:"retry" default:"Retrying..."`
+	Listening        string `yaml:"listening" default:"Listening"`
+	StartMessage     string `yaml:"start" default:"Send me a voice message and I'll transcribe it for you!"`
+	ErrorMessage     string `yaml:"error" default:"An error occurred while processing your message. Please try again."`
+	FailureMessage   string `yaml:"failure" default:"Failed to transcribe the audio. Please check the audio quality and try again."`
+	RetryMessage     string `yaml:"retry" default:"Retrying..."`
+	RateLimitMessage string `yaml:"rate_limit" default:"Rate limit exceeded. Please try again later."`
 }
 
 type ConfigRateLimit struct {
@@ -81,5 +85,18 @@ func loadConfig(filename string) (*Config, error) {
 		return nil, err
 	}
 
+	// Validate that at least one bot is configured
+	if len(config.Bots) == 0 {
+		return nil, fmt.Errorf("at least one bot must be configured")
+	}
+
+	// Fill ID from map keys if not set
+	for botID, botConfig := range config.Bots {
+		if botConfig.ID == "" {
+			botConfig.ID = botID
+			config.Bots[botID] = botConfig
+		}
+	}
+
 	return &config, nil
 }