feat: init with converting cbz files

cmd/convert_cbz_command.go

@@ -0,0 +1,120 @@
+package cmd
+
+import (
+	"CBZOptimizer/cbz"
+	"CBZOptimizer/converter"
+	"CBZOptimizer/converter/constant"
+	"fmt"
+	"github.com/spf13/cobra"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+)
+
+func init() {
+	command := &cobra.Command{
+		Use:   "convert",
+		Short: "Convert CBZ files using a specified converter",
+		RunE:  ConvertCbzCommand,
+		Args:  cobra.ExactArgs(1),
+	}
+	command.Flags().Uint8P("quality", "q", 85, "Quality for conversion (0-100)")
+	command.Flags().IntP("parallelism", "n", 2, "Number of chapters to convert in parallel")
+	command.Flags().BoolP("override", "o", false, "Override the original CBZ files")
+	AddCommand(command)
+}
+
+func ConvertCbzCommand(cmd *cobra.Command, args []string) error {
+	path := args[0]
+	if path == "" {
+		return fmt.Errorf("path is required")
+	}
+
+	quality, err := cmd.Flags().GetUint8("quality")
+	if err != nil {
+		return fmt.Errorf("invalid quality value")
+	}
+
+	override, err := cmd.Flags().GetBool("override")
+	if err != nil {
+		return fmt.Errorf("invalid quality value")
+	}
+
+	parallelism, err := cmd.Flags().GetInt("parallelism")
+	if err != nil || parallelism < 1 {
+		return fmt.Errorf("invalid parallelism value")
+	}
+
+	chapterConverter, err := converter.Get(constant.ImageFormatWebP)
+	if err != nil {
+		return fmt.Errorf("failed to get chapterConverter: %v", err)
+	}
+	// Channel to manage the files to process
+	fileChan := make(chan string)
+
+	// WaitGroup to wait for all goroutines to finish
+	var wg sync.WaitGroup
+
+	// Start worker goroutines
+	for i := 0; i < parallelism; i++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			for path := range fileChan {
+				fmt.Printf("Processing file: %s\n", path)
+
+				// Load the chapter
+				chapter, err := cbz.LoadChapter(path)
+				if err != nil {
+					fmt.Printf("Failed to load chapter: %v\n", err)
+					continue
+				}
+
+				// Convert the chapter
+				convertedChapter, err := chapterConverter.ConvertChapter(chapter, quality, func(msg string) {
+					fmt.Println(msg)
+				})
+				if err != nil {
+					fmt.Printf("Failed to convert chapter: %v\n", err)
+					continue
+				}
+
+				// Write the converted chapter back to a CBZ file
+				outputPath := path
+				if !override {
+					outputPath = strings.TrimSuffix(path, ".cbz") + "_converted.cbz"
+				}
+				err = cbz.WriteChapterToCBZ(convertedChapter, outputPath)
+				if err != nil {
+					fmt.Printf("Failed to write converted chapter: %v\n", err)
+					continue
+				}
+
+				fmt.Printf("Converted file written to: %s\n", outputPath)
+			}
+		}()
+	}
+
+	// Walk the path and send files to the channel
+	err = filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if !info.IsDir() && strings.HasSuffix(strings.ToLower(info.Name()), ".cbz") {
+			fileChan <- path
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return fmt.Errorf("error walking the path: %w", err)
+	}
+
+	close(fileChan) // Close the channel to signal workers to stop
+	wg.Wait()       // Wait for all workers to finish
+
+	return nil
+}

cmd/rootcmd.go

@@ -0,0 +1,23 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"os"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "cbzconverter",
+	Short: "Convert CBZ files using a specified converter",
+}
+
+// Execute executes the root command.
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
+func AddCommand(cmd *cobra.Command) {
+	rootCmd.AddCommand(cmd)
+}