package main

import (
	"fmt"
	"github.com/taurusxin/ncmdump-go/ncmcrypt"
	"github.com/taurusxin/ncmdump-go/utils"
	"os"
	"path/filepath"

	flag "github.com/spf13/pflag"
)

func processFile(filePath string) error {
	currentFile, err := ncmcrypt.NewNeteaseCloudMusic(filePath)
	if err != nil {
		utils.ErrorPrintfln("Reading '%s' failed: %s", filePath, err.Error())
		return err
	}
	dump, err := currentFile.Dump(filepath.Dir(filePath))
	if err != nil {
		utils.ErrorPrintfln("Processing '%s' failed: %s", filePath, err.Error())
		return err
	}
	if dump {
		metadata, err := currentFile.FixMetadata(true)
		if !metadata {
			utils.WarningPrintfln("Fix metadata for '%s' failed: %s", filePath, err.Error())
			return err
		}
		utils.DonePrintfln("'%s' -> '%s'", filePath, currentFile.GetDumpFilePath())
	}
	return nil
}

func main() {
	var folderPath string
	showHelp := flag.BoolP("help", "h", false, "Display help message")
	showVersion := flag.BoolP("version", "v", false, "Display version information")
	flag.StringVarP(&folderPath, "dir", "d", "", "Process all files in the directory")
	flag.Parse()

	if len(os.Args) == 1 {
		flag.Usage()
		os.Exit(0)
	}

	if *showHelp {
		flag.Usage()
		os.Exit(0)
	}

	if *showVersion {
		fmt.Println("ncmdump version 1.5.0")
		os.Exit(0)
	}

	if folderPath != "" {
		// check if the folder exists
		info, err := os.Stat(folderPath)
		if err != nil {
			utils.ErrorPrintfln("Unable to access directory: '%s'", folderPath)
			os.Exit(1)
		}

		if !info.IsDir() {
			utils.ErrorPrintfln("Not a directory: %s", folderPath)
			os.Exit(1)
		}

		// dump files in the folder
		files, err := os.ReadDir(folderPath)
		if err != nil {
			utils.ErrorPrintfln("Unable to read directory: '%s'", folderPath)
			os.Exit(1)
		}

		for _, file := range files {
			if file.IsDir() {
				continue
			}

			filePath := filepath.Join(folderPath, file.Name())
			// skip if the extension is not .ncm
			if filePath[len(filePath)-4:] != ".ncm" {
				continue
			}
			err = processFile(filePath)
			if err != nil {
				continue
			}
		}
	} else {
		// dump file from args
		for _, filePath := range flag.Args() {
			// skip if the extension is not .ncm
			if filePath[len(filePath)-4:] != ".ncm" {
				continue
			}
			err := processFile(filePath)
			if err != nil {
				continue
			}
		}
	}

}