Added process-management to zipSources Func

Author: larsn03

src/internal/file_operations_compress.go

@@ -2,67 +2,128 @@ package internal
 
 import (
 	"archive/zip"
+	"fmt"
 	"io"
+	"log/slog"
 	"os"
 	"path/filepath"
 	"strings"
+
+	"github.com/charmbracelet/bubbles/progress"
+	"github.com/lithammer/shortuuid"
+	"github.com/yorukot/superfile/src/config/icon"
+	"github.com/yorukot/superfile/src/internal/common"
 )
 
 func zipSources(sources []string, target string) error {
+	id := shortuuid.New()
+	prog := progress.New()
+	prog.PercentageStyle = common.FooterStyle
+	var err error = nil
+
+	for _, src := range sources {
+		if _, err := os.Stat(src); os.IsNotExist(err) {
+			return fmt.Errorf("source path does not exist: %s", src)
+		}
+	}
+
+	totalFiles := 0
+	for _, src := range sources {
+		count, err := countFiles(src)
+		if err != nil {
+			slog.Error("Error while zip file count files ", "error", err)
+		}
+		totalFiles += count
+	}
+	p := process{
+		name:     "zip files",
+		progress: prog,
+		state:    inOperation,
+		total:    totalFiles,
+		done:     0,
+	}
+	message := channelMessage{
+		messageID:       id,
+		messageType:     sendProcess,
+		processNewState: p,
+	}
+
+	_, err = os.Stat(target)
+	if err == nil {
+		p.name = icon.CompressFile + icon.Space + "File already exist"
+		message.processNewState = p
+		channel <- message
+		return nil
+	}
+
 	f, err := os.Create(target)
 	if err != nil {
 		return err
 	}
-
 	defer f.Close()
 	writer := zip.NewWriter(f)
 	defer writer.Close()
 
 	for _, src := range sources {
 		srcParentDir := filepath.Dir(src)
 		err = filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
+			p.name = icon.CompressFile + icon.Space + filepath.Base(path)
+			if len(channel) < 5 {
+				message.processNewState = p
+				channel <- message
+			}
 			if err != nil {
 				return err
 			}
-
 			relPath, err := filepath.Rel(srcParentDir, path)
 			if err != nil {
 				return err
 			}
-
 			header, err := zip.FileInfoHeader(info)
 			if err != nil {
 				return err
 			}
-
 			header.Method = zip.Deflate
 			header.Name = relPath
 			if info.IsDir() {
 				header.Name += "/"
 			}
-
-			hw, err := writer.CreateHeader(header)
+			headerWriter, err := writer.CreateHeader(header)
 			if err != nil {
 				return err
 			}
-
 			if info.IsDir() {
 				return nil
 			}
-
 			file, err := os.Open(path)
 			if err != nil {
 				return err
 			}
-
 			defer file.Close()
-			_, err = io.Copy(hw, file)
-			return err
+			_, err = io.Copy(headerWriter, file)
+			if err != nil {
+				return err
+			}
+			p.done++
+			if len(channel) < 5 {
+				message.processNewState = p
+				channel <- message
+			}
+			return nil
 		})
 		if err != nil {
+			slog.Error("Error while zip file", "error", err)
+			p.state = failure
+			message.processNewState = p
+			channel <- message
 			return err
 		}
 	}
+
+	p.state = successful
+	p.done = totalFiles
+	message.processNewState = p
+	channel <- message
 	return nil
 }