upload to github

src/handlers/common.go

@@ -0,0 +1,90 @@
+package handlers
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"beat-harvester/src/models"
+	"beat-harvester/src/utils"
+	"beat-harvester/src/broadcast"
+)
+
+// DownloadPaths holds the directory paths for a download operation
+type DownloadPaths struct {
+	AlbumDir string
+	TempDir  string
+}
+
+// getAlbumName extracts or returns the album name for the download
+func (h *Handler) getAlbumName(req models.DownloadRequest) (string, error) {
+	if req.AlbumName != "" {
+		broadcast.Log(fmt.Sprintf("📝 Using provided album name: %s", req.AlbumName))
+		return req.AlbumName, nil
+	}
+
+	broadcast.Log("📝 Extracting video title...")
+	title, err := h.extractVideoInfo(req.YtLink)
+	if err != nil {
+		broadcast.Log(fmt.Sprintf("❌ Failed to extract title: %v", err))
+		return "", err
+	}
+
+	broadcast.Log(fmt.Sprintf("📝 Extracted title: %s", title))
+	return title, nil
+}
+
+// setupDownloadPaths creates and returns the necessary directory paths
+func (h *Handler) setupDownloadPaths(albumName, outputPath string) DownloadPaths {
+	sanitizedAlbumName := utils.SanitizeFilename(albumName)
+	baseOutputPath := outputPath
+	if baseOutputPath == "" {
+		baseOutputPath = h.config.DefaultOutputPath
+	}
+
+	albumDir := filepath.Join(baseOutputPath, sanitizedAlbumName)
+	tempDir := filepath.Join(albumDir, h.config.TempDir)
+
+	broadcast.Log(fmt.Sprintf("📁 Album directory: %s", albumDir))
+	broadcast.Log(fmt.Sprintf("📁 Temp directory: %s", tempDir))
+
+	// Create directories
+	if err := os.MkdirAll(tempDir, 0755); err != nil {
+		broadcast.Log(fmt.Sprintf("❌ Failed to create directories: %v", err))
+	} else {
+		broadcast.Log("✅ Directories created successfully")
+	}
+
+	return DownloadPaths{
+		AlbumDir: albumDir,
+		TempDir:  tempDir,
+	}
+}
+
+// extractVideoInfo gets video title using yt-dlp
+func (h *Handler) extractVideoInfo(url string) (string, error) {
+	broadcast.Log("🔍 Extracting video information...")
+	
+	cmd := exec.Command("yt-dlp", "--get-title", "--no-warnings", url)
+	
+	// Capture both stdout and stderr for better debugging
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		broadcast.Log(fmt.Sprintf("❌ Failed to get video title: %v", err))
+		if len(output) > 0 {
+			broadcast.Log(fmt.Sprintf("❌ yt-dlp output: %s", string(output)))
+		}
+		return "", err
+	}
+	
+	title := strings.TrimSpace(string(output))
+	if title == "" {
+		broadcast.Log("⚠️  Empty title received, using fallback")
+		return "Unknown", nil
+	}
+	
+	broadcast.Log(fmt.Sprintf("✅ Video title extracted: %s", title))
+	return title, nil
+}