Добавить пересылку коротких YouTube-роликов в MAX как видео.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-09 15:42:05 +02:00
parent c6091fb581
commit 84380e9798
5 changed files with 264 additions and 5 deletions
@@ -0,0 +1,138 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"regexp"
+	"strconv"
+	"strings"
+)
+
+const (
+	maxYouTubeDurationSec = 5 * 60
+	maxYouTubeFileBytes   = 250 * 1024 * 1024 // лимит MAX для video
+)
+
+var youtubeURLRe = regexp.MustCompile(`(?i)https?://(?:(?:www|m)\.)?(?:youtube\.com/(?:watch\?[^ \t\n]*v=|shorts/|embed/|live/)|youtu\.be/)[^\s<>"']+`)
+
+func extractYouTubeURL(text string) string {
+	m := youtubeURLRe.FindString(text)
+	if m == "" {
+		return ""
+	}
+	return strings.TrimRight(m, ".,);]!?")
+}
+
+type ytDLPInfo struct {
+	Duration float64 `json:"duration"`
+	Title    string  `json:"title"`
+	ID       string  `json:"id"`
+}
+
+func getYouTubeInfo(videoURL string) (*ytDLPInfo, error) {
+	cmd := exec.Command("yt-dlp", "--dump-json", "--no-playlist", "--no-download", videoURL)
+	out, err := cmd.Output()
+	if err != nil {
+		if ee, ok := err.(*exec.ExitError); ok {
+			return nil, fmt.Errorf("yt-dlp info: %v: %s", err, string(ee.Stderr))
+		}
+		return nil, fmt.Errorf("yt-dlp не найден или ошибка: %w", err)
+	}
+	var info ytDLPInfo
+	if err := json.Unmarshal(out, &info); err != nil {
+		return nil, err
+	}
+	return &info, nil
+}
+
+// downloadYouTubeIfShort скачивает ролик, если он короче 5 минут и уложится в лимит MAX.
+// Возвращает путь к временному файлу (caller должен удалить) или ошибку.
+func downloadYouTubeIfShort(videoURL string) (filePath string, title string, err error) {
+	info, err := getYouTubeInfo(videoURL)
+	if err != nil {
+		return "", "", err
+	}
+	if info.Duration <= 0 {
+		return "", "", fmt.Errorf("не удалось определить длительность")
+	}
+	if info.Duration > float64(maxYouTubeDurationSec) {
+		return "", "", fmt.Errorf("видео длиннее 5 минут (%.0f с)", info.Duration)
+	}
+
+	tmpDir, err := os.MkdirTemp("", "bridge-yt-*")
+	if err != nil {
+		return "", "", err
+	}
+	outTmpl := filepath.Join(tmpDir, "video.%(ext)s")
+
+	// mp4 предпочтительнее для MAX; ограничение размера 250 МБ
+	cmd := exec.Command(
+		"yt-dlp",
+		"--no-playlist",
+		"--max-filesize", strconv.Itoa(maxYouTubeFileBytes),
+		"-f", "bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4]/bv*+ba/b",
+		"--merge-output-format", "mp4",
+		"-o", outTmpl,
+		videoURL,
+	)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Run(); err != nil {
+		_ = os.RemoveAll(tmpDir)
+		return "", "", fmt.Errorf("yt-dlp download: %w", err)
+	}
+
+	entries, err := os.ReadDir(tmpDir)
+	if err != nil || len(entries) == 0 {
+		_ = os.RemoveAll(tmpDir)
+		return "", "", fmt.Errorf("файл после скачивания не найден")
+	}
+	filePath = filepath.Join(tmpDir, entries[0].Name())
+	st, err := os.Stat(filePath)
+	if err != nil {
+		_ = os.RemoveAll(tmpDir)
+		return "", "", err
+	}
+	if st.Size() > maxYouTubeFileBytes {
+		_ = os.RemoveAll(tmpDir)
+		return "", "", fmt.Errorf("файл больше лимита MAX 250 МБ (%d)", st.Size())
+	}
+	log.Printf("[YouTube] скачано: %s (%.0f с, %d байт)", info.Title, info.Duration, st.Size())
+	return filePath, info.Title, nil
+}
+
+func cleanupYouTubeFile(filePath string) {
+	if filePath == "" {
+		return
+	}
+	_ = os.RemoveAll(filepath.Dir(filePath))
+}
+
+// SendToMAXSmart: при YouTube-ссылке и ролике < 5 мин отправляет видео-вложение, иначе текст (со ссылкой).
+func (b *Bridge) SendToMAXSmart(chatID int64, text string) error {
+	ytURL := extractYouTubeURL(text)
+	if ytURL == "" {
+		return b.SendToMAX(chatID, text)
+	}
+
+	filePath, title, err := downloadYouTubeIfShort(ytURL)
+	if err != nil {
+		log.Printf("[YouTube→MAX] не удалось скачать (%s), отправляю ссылкой: %v", ytURL, err)
+		return b.SendToMAX(chatID, text)
+	}
+	defer cleanupYouTubeFile(filePath)
+
+	caption := text
+	if title != "" && !strings.Contains(text, title) {
+		caption = text + "\n" + title
+	}
+	if err := b.SendVideoFileToMAX(chatID, filePath, caption); err != nil {
+		log.Printf("[YouTube→MAX] ошибка загрузки видео, отправляю ссылкой: %v", err)
+		return b.SendToMAX(chatID, text)
+	}
+	return nil
+}