v1.5.0: UI rework, multi-progress tracking, performance optimizations

go_backend/progress.go

@@ -1,10 +1,11 @@
 package gobackend
 
 import (
+	"encoding/json"
 	"sync"
 )
 
-// DownloadProgress represents current download progress
+// DownloadProgress represents current download progress (legacy single download)
 type DownloadProgress struct {
 	CurrentFile   string  `json:"current_file"`
 	Progress      float64 `json:"progress"`
@@ -14,20 +15,128 @@ type DownloadProgress struct {
 	IsDownloading bool    `json:"is_downloading"`
 }
 
+// ItemProgress represents progress for a single download item
+type ItemProgress struct {
+	ItemID        string  `json:"item_id"`
+	BytesTotal    int64   `json:"bytes_total"`
+	BytesReceived int64   `json:"bytes_received"`
+	Progress      float64 `json:"progress"` // 0.0 to 1.0
+	IsDownloading bool    `json:"is_downloading"`
+}
+
+// MultiProgress holds progress for multiple concurrent downloads
+type MultiProgress struct {
+	Items map[string]*ItemProgress `json:"items"`
+}
+
 var (
 	currentProgress DownloadProgress
 	progressMu      sync.RWMutex
 	downloadDir     string
 	downloadDirMu   sync.RWMutex
+	
+	// Multi-download progress tracking
+	multiProgress = MultiProgress{Items: make(map[string]*ItemProgress)}
+	multiMu       sync.RWMutex
 )
 
-// getProgress returns current download progress
+// getProgress returns current download progress (legacy)
 func getProgress() DownloadProgress {
 	progressMu.RLock()
 	defer progressMu.RUnlock()
 	return currentProgress
 }
 
+// GetMultiProgress returns progress for all active downloads as JSON
+func GetMultiProgress() string {
+	multiMu.RLock()
+	defer multiMu.RUnlock()
+	
+	jsonBytes, err := json.Marshal(multiProgress)
+	if err != nil {
+		return "{\"items\":{}}"
+	}
+	return string(jsonBytes)
+}
+
+// GetItemProgress returns progress for a specific item as JSON
+func GetItemProgress(itemID string) string {
+	multiMu.RLock()
+	defer multiMu.RUnlock()
+	
+	if item, ok := multiProgress.Items[itemID]; ok {
+		jsonBytes, _ := json.Marshal(item)
+		return string(jsonBytes)
+	}
+	return "{}"
+}
+
+// StartItemProgress initializes progress tracking for an item
+func StartItemProgress(itemID string) {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	multiProgress.Items[itemID] = &ItemProgress{
+		ItemID:        itemID,
+		BytesTotal:    0,
+		BytesReceived: 0,
+		Progress:      0,
+		IsDownloading: true,
+	}
+}
+
+// SetItemBytesTotal sets total bytes for an item
+func SetItemBytesTotal(itemID string, total int64) {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	if item, ok := multiProgress.Items[itemID]; ok {
+		item.BytesTotal = total
+	}
+}
+
+// SetItemBytesReceived sets bytes received for an item
+func SetItemBytesReceived(itemID string, received int64) {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	if item, ok := multiProgress.Items[itemID]; ok {
+		item.BytesReceived = received
+		if item.BytesTotal > 0 {
+			item.Progress = float64(received) / float64(item.BytesTotal)
+		}
+	}
+}
+
+// CompleteItemProgress marks an item as complete
+func CompleteItemProgress(itemID string) {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	if item, ok := multiProgress.Items[itemID]; ok {
+		item.Progress = 1.0
+		item.IsDownloading = false
+	}
+}
+
+// RemoveItemProgress removes progress tracking for an item
+func RemoveItemProgress(itemID string) {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	delete(multiProgress.Items, itemID)
+}
+
+// ClearAllItemProgress clears all item progress
+func ClearAllItemProgress() {
+	multiMu.Lock()
+	defer multiMu.Unlock()
+	
+	multiProgress.Items = make(map[string]*ItemProgress)
+}
+
+// Legacy functions for backward compatibility
+
 // SetDownloadProgress sets the current download progress (MB downloaded)
 func SetDownloadProgress(mbDownloaded float64) {
 	progressMu.Lock()
@@ -47,7 +156,6 @@ func SetDownloadSpeed(speedMBps float64) {
 func SetCurrentFile(filename string) {
 	progressMu.Lock()
 	defer progressMu.Unlock()
-	// Reset progress for new file
 	currentProgress.BytesReceived = 0
 	currentProgress.BytesTotal = 0
 	currentProgress.Progress = 0
@@ -101,7 +209,7 @@ func SetBytesReceived(received int64) {
 	}
 }
 
-// ProgressWriter wraps io.Writer to track download progress
+// ProgressWriter wraps io.Writer to track download progress (legacy single)
 type ProgressWriter struct {
 	writer  interface{ Write([]byte) (int, error) }
 	total   int64
@@ -110,7 +218,6 @@ type ProgressWriter struct {
 
 // NewProgressWriter creates a new progress writer wrapping an io.Writer
 func NewProgressWriter(w interface{ Write([]byte) (int, error) }) *ProgressWriter {
-	// Reset bytes received when starting new download
 	SetBytesReceived(0)
 	return &ProgressWriter{
 		writer:  w,
@@ -135,3 +242,30 @@ func (pw *ProgressWriter) Write(p []byte) (int, error) {
 func (pw *ProgressWriter) GetTotal() int64 {
 	return pw.total
 }
+
+// ItemProgressWriter wraps io.Writer to track download progress for a specific item
+type ItemProgressWriter struct {
+	writer  interface{ Write([]byte) (int, error) }
+	itemID  string
+	current int64
+}
+
+// NewItemProgressWriter creates a new progress writer for a specific item
+func NewItemProgressWriter(w interface{ Write([]byte) (int, error) }, itemID string) *ItemProgressWriter {
+	return &ItemProgressWriter{
+		writer:  w,
+		itemID:  itemID,
+		current: 0,
+	}
+}
+
+// Write implements io.Writer
+func (pw *ItemProgressWriter) Write(p []byte) (int, error) {
+	n, err := pw.writer.Write(p)
+	if err != nil {
+		return n, err
+	}
+	pw.current += int64(n)
+	SetItemBytesReceived(pw.itemID, pw.current)
+	return n, nil
+}