Merge pull request #675 from pavelpikta/feature/tg-bot

feat: add new telegram bot commands

Author: YouROK

README.md

@@ -176,7 +176,7 @@ On FreeBSD (TrueNAS/FreeNAS) you can use this plugin: <https://github.com/filka9
 - `--pubipv6 PUBIPV6`, `-6 PUBIPV6` - set public IPv6 addr
 - `--searchwa`, `-s` - allow search without authentication
 - `--maxsize MAXSIZE`, `-m MAXSIZE` - max allowed stream size (in Bytes)
-- `--tg TGTOKEN`, `-T TGTOKEN` - telegram bot token
+- `--tg TGTOKEN`, `-T TGTOKEN` - [Telegram bot](server/tgbot/README.md) token
 - `--fuse FUSEPATH`, `-f FUSEPATH` - fuse mount path
 - `--webdav` - enable web dav
 - `--proxyurl PROXYURL` - set proxy URL for BitTorrent traffic (http, socks4, socks5, socks5h), example: socks5h://user:password@example.com:2080

server/server.go

@@ -70,7 +70,9 @@ func Start() {
 	settings.IP = settings.Args.IP
 
 	if settings.Args.TGToken != "" {
-		tgbot.Start(settings.Args.TGToken)
+		if err := tgbot.Start(settings.Args.TGToken); err != nil {
+			log.TLogln("tg bot start failed", err)
+		}
 	}
 	web.Start()
 }

server/tgbot/README.md

@@ -0,0 +1,158 @@
+# TorrServer Telegram Bot
+
+[![GitHub License](https://img.shields.io/github/license/YouROK/TorrServer)](https://github.com/YouROK/TorrServer/blob/master/LICENSE)
+[![TorrServer Integrated](https://img.shields.io/badge/TorrServer-integrated-blue)](https://github.com/YouROK/TorrServer)
+
+## Introduction
+
+Telegram bot for managing [TorrServer](https://github.com/YouROK/TorrServer) — add torrents, stream, search, and control the server directly from Telegram.
+
+## Features
+
+- Torrent management — add, remove, drop, list via magnet, hash, or `torrs://`
+- Export & import — magnets list; import multiple from text
+- Streaming — playback links, M3U playlists, preload
+- Search — RuTor and Torznab with one-click add
+- Inline mode — `@botname` in any chat: list torrents or search
+- Status & snake — real-time status, cache visualization
+- File operations — browse files, download to Telegram
+- FFprobe — media metadata via `/ffp`
+- Localization — Russian and English
+- Admin — shutdown, settings, presets (whitelist users only)
+
+## Getting Started
+
+### Enable the Bot
+
+Start TorrServer with a Telegram bot token:
+
+```bash
+TorrServer --tg YOUR_BOT_TOKEN
+```
+
+Or use `-T`:
+
+```bash
+TorrServer -T YOUR_BOT_TOKEN
+```
+
+Create a bot via [@BotFather](https://t.me/BotFather) to get the token.
+
+### Configuration
+
+Config file `tg.cfg` (JSON) in the TorrServer data directory:
+
+| Field      | Description |
+|------------|-------------|
+| `HostTG`   | Telegram API URL (default: `https://api.telegram.org`) |
+| `HostWeb`  | Base URL for stream links (auto-detected if empty) |
+| `WhiteIds` | Allowed user IDs (empty = allow all) |
+| `BlackIds` | Blocked user IDs |
+
+Example:
+
+```json
+{
+  "HostTG": "https://api.telegram.org",
+  "HostWeb": "http://192.168.1.100:8090",
+  "WhiteIds": [123456789],
+  "BlackIds": []
+}
+```
+
+## Commands
+
+### Core
+
+| Command | Description |
+|---------|-------------|
+| `/help`, `/start`, `/id` | Help and user ID |
+| `/list [compact]` | List torrents with buttons |
+| `/add <link>` | Add torrent (magnet, hash, torrs://) |
+| `/clear` | Remove all (with confirmation) |
+| `/hash [N]` | Show info hashes |
+
+### Management
+
+| Command | Description |
+|---------|-------------|
+| `/remove <hash\|N>` | Remove torrent |
+| `/drop <hash\|N>` | Disconnect (keep in DB) |
+| `/set <hash\|N> <title>` | Set title |
+| `/status [hash\|N]` | Status with refresh/stop |
+| `/cache <hash\|N>` | Cache stats |
+| `/preload <hash\|N> <index>` | Preload file |
+
+### Links & Playback
+
+| Command | Description |
+|---------|-------------|
+| `/link`, `/play` | Stream URL |
+| `/m3u`, `/m3uall` | M3U playlist |
+
+### Search
+
+| Command | Description |
+|---------|-------------|
+| `/search <query>` | RuTor + Torznab (all sources) |
+| `/rutor <query>` | RuTor only |
+| `/torznab <query> [index]` | Torznab indexers |
+
+### Other
+
+| Command | Description |
+|---------|-------------|
+| `/export`, `/import` | Export/import magnets |
+| `/categories` | List categories |
+| `/server`, `/stats`, `/stat` | Server info |
+| `/viewed` | Viewed files |
+| `/ffp <hash\|N> <id> [json]` | FFprobe metadata |
+| `/speedtest [size]` | Download test (1–100 MB) |
+| `/snake [hash\|N] [cols] [rows]` | Cache visualization |
+| `/lang [RU\|EN]` | Language |
+
+### Admin Only
+
+| Command | Description |
+|---------|-------------|
+| `/shutdown` | Shut down server |
+| `/settings` | Interactive settings menu (sub-pages: Search, Network, Other, Cache, Paths, Storage) |
+| `/preset <name>` | Apply named preset: `performance`, `storage`, `streaming`, `low`, `default` |
+| `/preset <key> <value> ...` | Apply key-value pairs: `cache 256`, `preload 50`, `conn 100`, etc. |
+
+**Preset examples:**
+- `/preset performance` — max cache, high preload, no limits
+- `/preset cache 256 preload 50` — set cache 256 MB and preload 50%
+- `/preset cache 512 conn 100 down 0 up 0` — multiple values
+
+**Preset keys:** `cache`, `preload`, `readahead`, `conn`, `timeout`, `port`, `down`, `up`, `retr`, `responsive`, `cachedrop`
+
+## Inline Mode
+
+Type `@YourBotName` in any chat:
+
+- **Empty, "list", or "play"** — torrents with play links
+- **2+ characters** — search RuTor + Torznab
+
+## Text Input
+
+Paste as plain message to add torrent:
+
+- `magnet:?xt=urn:btih:...`
+- `torrs://...`
+- 40-char info hash
+
+Reply to file list with `2-12` to download files 2–12 to Telegram.
+
+## Security
+
+- **Whitelist** — restrict to specific user IDs
+- **Blacklist** — block user IDs
+- **Admin** — when whitelist is used, admin = whitelisted users
+- **Settings** — sensitive values masked in `/settings`
+
+## Dependencies
+
+- [telebot v4](https://gopkg.in/telebot.v4) — Telegram Bot API
+- [go-humanize](https://github.com/dustin/go-humanize)
+- [go-ffprobe](https://gopkg.in/vansante/go-ffprobe.v2)

server/tgbot/add.go

@@ -2,51 +2,53 @@ package tgbot
 
 import (
 	"errors"
+	"fmt"
+	"io"
 	"strings"
 
+	"github.com/anacrolix/torrent"
 	tele "gopkg.in/telebot.v4"
 	"server/log"
 	set "server/settings"
 	"server/torr"
 	"server/web/api/utils"
 )
 
-func addTorrent(c tele.Context, link string) error {
-	msg, err := c.Bot().Send(c.Sender(), "Подключение к торренту...")
-	if err != nil {
-		return err
-	}
-	log.TLogln("tg add torrent", link)
-	link = strings.ReplaceAll(link, "&", "&")
-	torrSpec, err := utils.ParseLink(link)
+func addTorrentFromSpec(c tele.Context, torrSpec *torrent.TorrentSpec, displayLabel string) error {
+	msg, err := c.Bot().Send(c.Sender(), tr(c.Sender().ID, "connecting"))
 	if err != nil {
-		log.TLogln("tg error parse link:", err)
 		return err
 	}
 
 	tor, err := torr.AddTorrent(torrSpec, "", "", "", "")
-
-	if tor.Data != "" && set.BTsets.EnableDebug {
-		log.TLogln("torrent data:", tor.Data)
+	if err != nil {
+		log.TLogln("tg add err", err)
+		_, _ = c.Bot().Edit(msg, fmt.Sprintf(tr(c.Sender().ID, "add_error"), err.Error()))
+		return err
 	}
-	if tor.Category != "" && set.BTsets.EnableDebug {
-		log.TLogln("torrent category:", tor.Category)
+	if tor == nil {
+		_, _ = c.Bot().Edit(msg, tr(c.Sender().ID, "add_not_created"))
+		return errors.New("torrent not created")
 	}
 
-	if err != nil {
-		log.TLogln("tg error add torrent:", err)
-		c.Bot().Edit(msg, "Ошибка при подключении: "+err.Error())
-		return err
+	if set.BTsets != nil && set.BTsets.EnableDebug {
+		if tor.Data != "" {
+			log.TLogln("tg add data", logSafeStr(tor.Data, 60))
+		}
+		if tor.Category != "" {
+			log.TLogln("tg add category", logSafeStr(tor.Category, 40))
+		}
 	}
 
+	_, _ = c.Bot().Edit(msg, tr(c.Sender().ID, "add_getting_meta"))
 	if !tor.GotInfo() {
-		log.TLogln("tg error add torrent: timeout connection get torrent info")
-		c.Bot().Edit(msg, "Ошибка при добаваления торрента: timeout connection get torrent info")
+		log.TLogln("tg add err", "timeout get torrent info")
+		_, _ = c.Bot().Edit(msg, tr(c.Sender().ID, "add_timeout"))
 		return errors.New("timeout connection get torrent info")
 	}
 
 	if tor.Title == "" {
-		tor.Title = torrSpec.DisplayName // prefer dn over name
+		tor.Title = torrSpec.DisplayName
 		tor.Title = strings.ReplaceAll(tor.Title, "rutor.info", "")
 		tor.Title = strings.ReplaceAll(tor.Title, "_", " ")
 		tor.Title = strings.Trim(tor.Title, " ")
@@ -57,7 +59,71 @@ func addTorrent(c tele.Context, link string) error {
 
 	torr.SaveTorrentToDB(tor)
 
-	c.Bot().Edit(msg, "Торрент добавлен:\n<code>"+link+"</code>")
+	if len(displayLabel) > 80 {
+		displayLabel = displayLabel[:77] + "..."
+	}
+	_, _ = c.Bot().Edit(msg, fmt.Sprintf(tr(c.Sender().ID, "add_success"), displayLabel))
 
 	return nil
 }
+
+func addTorrent(c tele.Context, link string) error {
+	log.TLogln("tg add torrent", logHashOrTruncate(link))
+	link = strings.ReplaceAll(link, "&amp;", "&")
+	var torrSpec *torrent.TorrentSpec
+	var err error
+	if strings.HasPrefix(strings.ToLower(link), "torrs://") {
+		torrSpec, _, err = utils.ParseTorrsHash(link)
+	} else {
+		torrSpec, err = utils.ParseLink(link)
+	}
+	if err != nil {
+		log.TLogln("tg add parse err", err)
+		return err
+	}
+	return addTorrentFromSpec(c, torrSpec, link)
+}
+
+func addTorrentFromDocument(c tele.Context, doc *tele.Document) error {
+	if doc == nil || doc.FileID == "" {
+		return errors.New("no document")
+	}
+	reader, err := c.Bot().File(&doc.File)
+	if err != nil {
+		log.TLogln("tg add document getfile err", err)
+		return err
+	}
+	defer func() { _ = reader.Close() }()
+	data, err := io.ReadAll(reader)
+	if err != nil {
+		log.TLogln("tg add document read err", err)
+		return err
+	}
+	torrSpec, err := utils.ParseFromBytes(data)
+	if err != nil {
+		log.TLogln("tg add document parse err", err)
+		return err
+	}
+	displayLabel := doc.FileName
+	if displayLabel == "" {
+		displayLabel = ".torrent"
+	}
+	return addTorrentFromSpec(c, torrSpec, displayLabel)
+}
+
+func cmdAdd(c tele.Context) error {
+	uid := c.Sender().ID
+	args := c.Args()
+	if len(args) == 0 {
+		return c.Send(tr(uid, "add_usage"))
+	}
+	link := strings.TrimSpace(strings.Join(args, " "))
+	if link == "" {
+		return c.Send(tr(uid, "add_no_link"))
+	}
+	err := addTorrent(c, link)
+	if err != nil {
+		return err
+	}
+	return list(c)
+}

server/tgbot/admin_common.go

@@ -0,0 +1,17 @@
+package tgbot
+
+import (
+	"server/tgbot/config"
+)
+
+func isAdmin(userID int64) bool {
+	if len(config.Cfg.WhiteIds) == 0 {
+		return false
+	}
+	for _, id := range config.Cfg.WhiteIds {
+		if id == userID {
+			return true
+		}
+	}
+	return false
+}