fix(unfinished): store video metadata for single video item

Author: xdagiz

internal/app/update.go

@@ -119,7 +119,8 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			FormatID:           msg.FormatID,
 			IsAudioTab:         msg.IsAudioTab,
 			ABR:                msg.ABR,
-			Title:              msg.SelectedVideo.Title(),
+			Title:              m.Download.SelectedVideo.Title(),
+			Videos:             []types.VideoItem{m.Download.SelectedVideo},
 			Options:            m.Search.DownloadOptions,
 			CookiesFromBrowser: m.Search.CookiesFromBrowser,
 			Cookies:            m.Search.Cookies,
@@ -195,11 +196,21 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		}
 
 		m.Download.SelectedVideo = types.VideoItem{VideoTitle: resumeTitle}
+		if len(resumeVideos) > 0 {
+			m.Download.SelectedVideo = resumeVideos[0]
+		} else if resumeTitle != "" {
+			m.Download.SelectedVideo = types.VideoItem{
+				ID:         msg.URL,
+				VideoTitle: resumeTitle,
+			}
+		}
 		req := types.DownloadRequest{
 			URL:                msg.URL,
 			FormatID:           resumeFormatID,
 			IsAudioTab:         false,
 			ABR:                0,
+			Title:              m.Download.SelectedVideo.Title(),
+			Videos:             []types.VideoItem{m.Download.SelectedVideo},
 			Options:            m.Search.DownloadOptions,
 			CookiesFromBrowser: m.Search.CookiesFromBrowser,
 			Cookies:            m.Search.Cookies,

internal/app/update_queue_test.go

@@ -382,3 +382,53 @@ func TestModelUpdateRetryCurrentQueueItemClearsError(t *testing.T) {
 		t.Fatalf("m.Download.QueueError = %q, want empty", m.Download.QueueError)
 	}
 }
+
+func TestModelUpdateStartResumeDownloadUsesVideoInfoFromUnfinishedItem(t *testing.T) {
+	m := newQueueTestModel(t)
+
+	updated, cmd := m.Update(types.StartResumeDownloadMsg{
+		URL:      "https://www.youtube.com/watch?v=abc123",
+		FormatID: "best",
+		Title:    "Fallback Title",
+		Videos: []types.VideoItem{
+			{
+				ID:         "https://www.youtube.com/watch?v=abc123",
+				VideoTitle: "Real Video Title",
+				Channel:    "Real Channel",
+				Duration:   120,
+			},
+		},
+	})
+	m = updated.(*Model)
+
+	if cmd == nil {
+		t.Fatalf("expected non-nil download command")
+	}
+	if m.Download.SelectedVideo.VideoTitle != "Real Video Title" {
+		t.Fatalf("SelectedVideo.VideoTitle = %q, want %q", m.Download.SelectedVideo.VideoTitle, "Real Video Title")
+	}
+	if m.Download.SelectedVideo.Channel != "Real Channel" {
+		t.Fatalf("SelectedVideo.Channel = %q, want %q", m.Download.SelectedVideo.Channel, "Real Channel")
+	}
+}
+
+func TestModelUpdateStartResumeDownloadFallbacksToTitleAndURL(t *testing.T) {
+	m := newQueueTestModel(t)
+
+	updated, cmd := m.Update(types.StartResumeDownloadMsg{
+		URL:      "https://www.youtube.com/watch?v=xyz789",
+		FormatID: "best",
+		Title:    "Stored Title",
+	})
+	m = updated.(*Model)
+
+	if cmd == nil {
+		t.Fatalf("expected non-nil download command")
+	}
+	if m.Download.SelectedVideo.VideoTitle != "Stored Title" {
+		t.Fatalf("SelectedVideo.VideoTitle = %q, want %q", m.Download.SelectedVideo.VideoTitle, "Stored Title")
+	}
+	if m.Download.SelectedVideo.ID != "https://www.youtube.com/watch?v=xyz789" {
+		t.Fatalf("SelectedVideo.ID = %q, want URL", m.Download.SelectedVideo.ID)
+	}
+}

internal/utils/download.go

@@ -19,6 +19,11 @@ import (
 
 func StartDownload(dm *DownloadManager, program *tea.Program, req types.DownloadRequest) tea.Cmd {
 	return tea.Cmd(func() tea.Msg {
+		videos := req.Videos
+		if len(videos) == 0 && req.Title != "" {
+			videos = []types.VideoItem{{ID: req.URL, VideoTitle: req.Title}}
+		}
+
 		key := req.UnfinishedKey
 		if key == "" {
 			key = req.URL
@@ -33,7 +38,7 @@ func StartDownload(dm *DownloadManager, program *tea.Program, req types.Download
 			Title:     title,
 			Desc:      req.UnfinishedDesc,
 			URLs:      req.URLs,
-			Videos:    req.Videos,
+			Videos:    videos,
 			Timestamp: time.Now(),
 		}
 

internal/utils/download_integration_test.go

@@ -96,6 +96,17 @@ func setupUnfinishedFilePath(t *testing.T) {
 	})
 }
 
+func setupDownloadConfigDir(t *testing.T) {
+	t.Helper()
+
+	orig := config.GetConfigDir
+	dir := filepath.Join(t.TempDir(), "config")
+	config.GetConfigDir = func() string { return dir }
+	t.Cleanup(func() {
+		config.GetConfigDir = orig
+	})
+}
+
 func waitForManagerReady(t *testing.T, dm *DownloadManager) {
 	t.Helper()
 
@@ -284,3 +295,80 @@ func TestDoDownload_CancelSendsCancelled(t *testing.T) {
 		t.Fatalf("expected cancellation error, got: %q", last.Err)
 	}
 }
+
+func TestStartDownload_PersistsSingleVideoInfoForResume(t *testing.T) {
+	setupUnfinishedFilePath(t)
+	setupDownloadConfigDir(t)
+
+	m, p := runCollectorProgram(t)
+	_ = m
+	dm := NewDownloadManager()
+
+	cmd := StartDownload(dm, p, types.DownloadRequest{
+		URL:      "https://www.youtube.com/watch?v=abc123",
+		FormatID: "best",
+		Title:    "Saved Title",
+	})
+	if cmd == nil {
+		t.Fatalf("expected non-nil start command")
+	}
+
+	msg := cmd()
+	if msg != nil {
+		t.Fatalf("start command msg = %T, want nil", msg)
+	}
+
+	entry := GetUnfinishedByURL("https://www.youtube.com/watch?v=abc123")
+	if entry == nil {
+		t.Fatalf("expected unfinished entry to exist")
+	}
+	if len(entry.Videos) != 1 {
+		t.Fatalf("entry.Videos len = %d, want 1", len(entry.Videos))
+	}
+	if entry.Videos[0].ID != "https://www.youtube.com/watch?v=abc123" {
+		t.Fatalf("entry.Videos[0].ID = %q, want URL", entry.Videos[0].ID)
+	}
+	if entry.Videos[0].VideoTitle != "Saved Title" {
+		t.Fatalf("entry.Videos[0].VideoTitle = %q, want %q", entry.Videos[0].VideoTitle, "Saved Title")
+	}
+}
+
+func TestStartDownload_PersistsFullSingleVideoMetadata(t *testing.T) {
+	setupUnfinishedFilePath(t)
+	setupDownloadConfigDir(t)
+
+	_, p := runCollectorProgram(t)
+	dm := NewDownloadManager()
+
+	video := types.VideoItem{
+		ID:         "https://www.youtube.com/watch?v=meta123",
+		VideoTitle: "Meta Title",
+		Desc:       "Meta Description",
+		Views:      12345,
+		Duration:   321,
+		Channel:    "Meta Channel",
+	}
+
+	cmd := StartDownload(dm, p, types.DownloadRequest{
+		URL:      video.ID,
+		FormatID: "best",
+		Title:    video.Title(),
+		Videos:   []types.VideoItem{video},
+	})
+	if cmd == nil {
+		t.Fatalf("expected non-nil start command")
+	}
+	_ = cmd()
+
+	entry := GetUnfinishedByURL(video.ID)
+	if entry == nil {
+		t.Fatalf("expected unfinished entry to exist")
+	}
+	if len(entry.Videos) != 1 {
+		t.Fatalf("entry.Videos len = %d, want 1", len(entry.Videos))
+	}
+	got := entry.Videos[0]
+	if got.Desc != "Meta Description" || got.Views != 12345 || got.Duration != 321 || got.Channel != "Meta Channel" {
+		t.Fatalf("video metadata not preserved: %+v", got)
+	}
+}