107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"ncmdump/ncmcrypt"
|
|
"ncmdump/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()
|
|
if err != nil {
|
|
utils.ErrorPrintfln("Processing '%s' failed: '%s'", filePath, err.Error())
|
|
return err
|
|
}
|
|
if dump {
|
|
metadata, _ := currentFile.FixMetadata()
|
|
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
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|