feat: soporte para marca de agua de imagen/logo y posiciones customizadas

internal/ffmpeg/ffmpeg.go

@@ -224,36 +224,88 @@ func SanitizeSettings(s protocol.VideoSettings) protocol.VideoSettings {
 	return s
 }
 
+// wmTextPosition devuelve las expresiones x/y de drawtext para la posición elegida
+// ("tr" arriba-derecha, "tl" arriba-izquierda, "br" abajo-derecha, "bl" abajo-izquierda).
+func wmTextPosition(pos string) (x, y string) {
+	switch strings.ToLower(strings.TrimSpace(pos)) {
+	case "tl":
+		return "10", "10"
+	case "br":
+		return "w-tw-10", "h-th-10"
+	case "bl":
+		return "10", "h-th-10"
+	default: // "tr" y cualquier valor vacío/desconocido
+		return "w-tw-10", "10"
+	}
+}
+
+// wmOverlayExpr devuelve la expresión del filtro overlay para la marca de agua de imagen.
+func wmOverlayExpr(pos string) string {
+	switch strings.ToLower(strings.TrimSpace(pos)) {
+	case "tl":
+		return "10:10"
+	case "br":
+		return "W-w-10:H-h-10"
+	case "bl":
+		return "10:H-h-10"
+	default:
+		return "W-w-10:10"
+	}
+}
+
+// Compress ejecuta la compresión usando FFmpeg y reporta el progreso en vivo.
+// wmImagePath es la ruta local a un logo de marca de agua (opcional); si se
+// indica, tiene prioridad sobre la marca de agua de texto.
 func Compress(
 	ctx context.Context,
 	inputPath, outputPath string,
 	settings protocol.VideoSettings,
+	wmImagePath string,
 	totalDuration float64,
 	onProgress func(p protocol.TaskProgress),
 ) error {
 	settings = SanitizeSettings(settings)
 
-	var vfParts []string
-	vfParts = append(vfParts, BuildScaleFilter(settings.Resolution))
-
-	if settings.Watermark != "" {
-		fontPath := findFontFile()
-		if fontPath != "" {
-			vfParts = append(vfParts, fmt.Sprintf("drawtext=fontfile='%s':text='%s':x=w-tw-10:y=10:fontsize=20:fontcolor=white:box=1:boxcolor=black@0.4:boxborderw=4", escapeFontPath(fontPath), settings.Watermark))
-		} else {
-			vfParts = append(vfParts, fmt.Sprintf("drawtext=font='Sans':text='%s':x=w-tw-10:y=10:fontsize=20:fontcolor=white", settings.Watermark))
-		}
-	}
-
-	vfFilter := strings.Join(vfParts, ",")
-
 	args := []string{
 		"-y",
 		"-i", inputPath,
 	}
 
-	// Mapear video y audio
-	args = append(args, "-map", "0:v:0", "-map", "0:a?")
+	useImageWM := wmImagePath != ""
+	if useImageWM {
+		args = append(args, "-i", wmImagePath)
+	}
+
+	if useImageWM {
+		// Cadena con filter_complex: escala/pad del video principal, logo
+		// escalado a máx. 150px de ancho y overlay en la posición elegida.
+		chain := "[0:v]" + BuildScaleFilter(settings.Resolution) + "[vbase];" +
+			"[1:v]scale='if(gt(iw,150),150,iw)':-1[wm];" +
+			"[vbase][wm]overlay=" + wmOverlayExpr(settings.WatermarkPos) + "[v]"
+		args = append(args,
+			"-filter_complex", chain,
+			"-map", "[v]", "-map", "0:a?",
+		)
+	} else {
+		var vfParts []string
+		vfParts = append(vfParts, BuildScaleFilter(settings.Resolution))
+
+		if settings.Watermark != "" {
+			x, y := wmTextPosition(settings.WatermarkPos)
+			fontPath := findFontFile()
+			if fontPath != "" {
+				vfParts = append(vfParts, fmt.Sprintf("drawtext=fontfile='%s':text='%s':x=%s:y=%s:fontsize=20:fontcolor=white:box=1:boxcolor=black@0.4:boxborderw=4", escapeFontPath(fontPath), settings.Watermark, x, y))
+			} else {
+				vfParts = append(vfParts, fmt.Sprintf("drawtext=font='Sans':text='%s':x=%s:y=%s:fontsize=20:fontcolor=white", settings.Watermark, x, y))
+			}
+		}
+
+		vfFilter := strings.Join(vfParts, ",")
+		args = append(args, "-map", "0:v:0", "-map", "0:a?")
+		if vfFilter != "" {
+			args = append(args, "-vf", vfFilter)
+		}
+	}
 
 	// Preservar subtítulos de texto si existen en el contenedor original (mkv/mp4)
 	textSubStreams := GetTextSubtitleStreams(ctx, inputPath)
@@ -264,10 +316,6 @@ func Compress(
 		args = append(args, "-c:s", "mov_text")
 	}
 
-	if vfFilter != "" {
-		args = append(args, "-vf", vfFilter)
-	}
-
 	codec := settings.Codec
 	if codec == "" {
 		codec = "libx264"

internal/protocol/messages.go

@@ -24,6 +24,13 @@ type VideoSettings struct {
 	Preset       string `json:"preset"`
 	Codec        string `json:"codec"`
 	Watermark    string `json:"watermark"`
+	// WatermarkPos: posición de la marca de agua ("tr", "tl", "br", "bl"; vacío = tr)
+	WatermarkPos string `json:"watermark_pos,omitempty"`
+	// WatermarkImageFileID: referencia al logo en Telegram (modo imagen;
+	// si está presente, tiene prioridad sobre el texto)
+	WatermarkImageFileID string `json:"watermark_image_file_id,omitempty"`
+	// ThumbnailFileID: referencia a la miniatura personalizada en Telegram
+	ThumbnailFileID string `json:"thumbnail_file_id,omitempty"`
 }
 
 // WSMessage es el contenedor genérico para los mensajes que viajan por WebSocket
@@ -62,6 +69,9 @@ type TaskAssign struct {
 	DownloadURL string        `json:"download_url"`
 	UploadURL   string        `json:"upload_url"`
 	Settings    VideoSettings `json:"settings"`
+	// WatermarkImageURL: de dónde descargar el logo si la marca de agua es imagen
+	// (vacío si la tarea no usa marca de imagen)
+	WatermarkImageURL string `json:"watermark_image_url,omitempty"`
 }
 
 // TaskProgress es el reporte en vivo que el Worker envía mientras FFmpeg procesa

main.go

@@ -454,6 +454,19 @@ func (w *Worker) executeTask(parentCtx context.Context, task protocol.TaskAssign
 		originalSize = inStat.Size()
 	}
 
+	// 1.5. Descargar el logo de marca de agua si la tarea incluye uno
+	wmImagePath := ""
+	if task.WatermarkImageURL != "" {
+		candidate := filepath.Join(w.workDir, fmt.Sprintf("%s_wm.png", task.TaskID))
+		if wmErr := w.downloadSimpleFile(taskCtx, w.resolveURL(task.WatermarkImageURL), candidate); wmErr != nil {
+			log.Printf("[%s] No se pudo descargar el logo (%v); se usará la marca de texto si existe", task.TaskID, wmErr)
+			_ = os.Remove(candidate)
+		} else {
+			wmImagePath = candidate
+			defer func() { _ = os.Remove(candidate) }()
+		}
+	}
+
 	// 2. Obtener duración del video con ffprobe
 	duration, err := ffmpeg.GetDuration(taskCtx, inputPath)
 	if err != nil {
@@ -465,7 +478,7 @@ func (w *Worker) executeTask(parentCtx context.Context, task protocol.TaskAssign
 	log.Printf("[%s] Iniciando compresión con FFmpeg (Duración: %.1fs)...", task.TaskID, duration)
 	startTime := time.Now()
 
-	err = ffmpeg.Compress(taskCtx, inputPath, outputPath, task.Settings, duration, func(p protocol.TaskProgress) {
+	err = ffmpeg.Compress(taskCtx, inputPath, outputPath, task.Settings, wmImagePath, duration, func(p protocol.TaskProgress) {
 		p.TaskID = task.TaskID
 		_ = w.sendMessage(protocol.TypeTaskProgress, p)
 	})
@@ -560,6 +573,35 @@ func (w *Worker) executeTask(parentCtx context.Context, task protocol.TaskAssign
 	log.Printf("[%s] ✅ Tarea finalizada y confirmada.", task.TaskID)
 }
 
+// downloadSimpleFile descarga un archivo pequeño del Manager (logo de marca de agua).
+func (w *Worker) downloadSimpleFile(ctx context.Context, rawURL, destPath string) error {
+	req, err := http.NewRequestWithContext(ctx, "GET", rawURL, nil)
+	if err != nil {
+		return err
+	}
+	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36")
+	req.Header.Set("X-Worker-Secret", w.secret)
+
+	resp, err := httpClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("status %d", resp.StatusCode)
+	}
+
+	out, err := os.Create(destPath)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	_, err = io.Copy(out, resp.Body)
+	return err
+}
+
 func (w *Worker) reportFailed(taskID, errMsg string) {
 	_ = w.sendMessage(protocol.TypeTaskFailed, protocol.TaskFailed{
 		TaskID: taskID,