Merge pull request #1 from Aryagorjipour/Add-Watch-Command

feat: Add the watch command

internal/manager/manager.go

@@ -1,9 +1,14 @@
 package manager
 
 import (
+	"bufio"
 	"fmt"
-	"github.com/Aryagorjipour/smart-file-downloader/internal/task"
+	"os"
+	"strings"
 	"sync"
+	"time"
+
+	"github.com/Aryagorjipour/smart-file-downloader/internal/task"
 )
 
 // DownloadManager manages download tasks
@@ -92,3 +97,36 @@ func (dm *DownloadManager) GetTasks() map[int]*task.DownloadTask {
 	}
 	return copyTasks
 }
+
+func (mgr *DownloadManager) Watch() {
+	reader := bufio.NewReader(os.Stdin)
+	ticker := time.NewTicker(1 * time.Second)
+	defer ticker.Stop()
+
+	done := make(chan bool)
+
+	go func() {
+		for {
+			select {
+			case <-ticker.C:
+				mgr.ListDownloads()
+			case <-done:
+				return
+			}
+		}
+	}()
+
+	for {
+		input, err := reader.ReadString('\n')
+		if err != nil {
+			fmt.Println("Error reading input:", err)
+			continue
+		}
+		if strings.TrimSpace(input) == "q" {
+			break
+		}
+	}
+
+	done <- true
+	fmt.Println("Exited watch mode.")
+}

internal/ui/ui.go

@@ -3,11 +3,11 @@ package ui
 import (
 	"bufio"
 	"fmt"
-	"github.com/Aryagorjipour/smart-file-downloader/internal/manager"
 	"os"
 	"strconv"
 	"strings"
-	"time"
+
+	"github.com/Aryagorjipour/smart-file-downloader/internal/manager"
 )
 
 // Start starts the UI
@@ -16,23 +16,15 @@ func Start(mgr *manager.DownloadManager) {
 	fmt.Println("Concurrent File Downloader")
 	fmt.Printf("Downloads will be saved to: %s\n", mgr.DownloadDir)
 
-	// Start a goroutine to display progress periodically
-	go func() {
-		for len(mgr.Tasks) > 0 {
-			time.Sleep(2 * time.Second)
-			mgr.ListDownloads()
-			fmt.Println("------")
-		}
-	}()
-
 	for {
 		fmt.Println("Commands:")
 		fmt.Println("1. add <URL>")
 		fmt.Println("2. list")
 		fmt.Println("3. pause <ID>")
 		fmt.Println("4. resume <ID>")
 		fmt.Println("5. cancel <ID>")
-		fmt.Println("6. exit")
+		fmt.Println("6. watch")
+		fmt.Println("0. exit")
 		fmt.Print("Enter command: ")
 		input, err := reader.ReadString('\n')
 		if err != nil {
@@ -89,6 +81,8 @@ func Start(mgr *manager.DownloadManager) {
 				continue
 			}
 			mgr.CancelDownload(id)
+		case "watch":
+			mgr.Watch()
 		case "exit":
 			fmt.Println("Exiting...")
 			os.Exit(0)