135 lines
3.5 KiB
Go
135 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/umlock-music/cli/algo/common"
|
|
_ "github.com/umlock-music/cli/algo/kgm"
|
|
_ "github.com/umlock-music/cli/algo/kwm"
|
|
_ "github.com/umlock-music/cli/algo/ncm"
|
|
_ "github.com/umlock-music/cli/algo/qmc"
|
|
_ "github.com/umlock-music/cli/algo/tm"
|
|
_ "github.com/umlock-music/cli/algo/xm"
|
|
"github.com/umlock-music/cli/internal/logging"
|
|
"github.com/urfave/cli/v2"
|
|
"go.uber.org/zap"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
app := cli.App{
|
|
Name: "Unlock Music CLI",
|
|
HelpName: "um",
|
|
Usage: "Unlock your encrypted music file https://github.com/unlock-music/cli",
|
|
Version: "v0.0.1",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{Name: "input", Aliases: []string{"i"}, Usage: "path to input file or dir", Required: true},
|
|
&cli.StringFlag{Name: "output", Aliases: []string{"o"}, Usage: "path to output dir", Required: true},
|
|
},
|
|
Action: appMain,
|
|
Copyright: "Copyright (c) 2020 Unlock Music https://github.com/unlock-music/cli/blob/master/LICENSE",
|
|
HideHelpCommand: true,
|
|
UsageText: "um -i /path/to/input -o /path/to/output/dir",
|
|
}
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func appMain(c *cli.Context) error {
|
|
input := c.String("input")
|
|
output := c.String("output")
|
|
inputStat, err := os.Stat(input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
outputStat, err := os.Stat(output)
|
|
if err != nil {
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
err = os.MkdirAll(output, 0755)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else if !outputStat.IsDir() {
|
|
return errors.New("output should be a writable directory")
|
|
}
|
|
|
|
if inputStat.IsDir() {
|
|
return dealDirectory(input, output)
|
|
} else {
|
|
ext := strings.TrimLeft(filepath.Ext(inputStat.Name()), ".")
|
|
allDec := common.GetDecoder(ext)
|
|
if len(allDec) == 0 {
|
|
logging.Log().Fatal("skipping while no suitable decoder")
|
|
}
|
|
return tryDecFile(input, output, allDec)
|
|
}
|
|
|
|
}
|
|
func dealDirectory(inputDir string, outputDir string) error {
|
|
items, err := os.ReadDir(inputDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, item := range items {
|
|
if item.IsDir() {
|
|
continue
|
|
}
|
|
ext := strings.TrimLeft(filepath.Ext(item.Name()), ".")
|
|
allDec := common.GetDecoder(ext)
|
|
if len(allDec) == 0 {
|
|
logging.Log().Info("skipping while no suitable decoder", zap.String("file", item.Name()))
|
|
continue
|
|
}
|
|
|
|
err := tryDecFile(filepath.Join(inputDir, item.Name()), outputDir, allDec)
|
|
if err != nil {
|
|
logging.Log().Error("conversion failed", zap.String("source", item.Name()))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func tryDecFile(inputFile string, outputDir string, allDec []common.NewDecoderFunc) error {
|
|
file, err := os.ReadFile(inputFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var dec common.Decoder
|
|
for _, decFunc := range allDec {
|
|
dec = decFunc(file)
|
|
if err := dec.Validate(); err == nil {
|
|
break
|
|
}
|
|
logging.Log().Warn("try decode failed", zap.Error(err))
|
|
dec = nil
|
|
}
|
|
if dec == nil {
|
|
return errors.New("no any decoder can resolve the file")
|
|
}
|
|
if err := dec.Decode(); err != nil {
|
|
return errors.New("failed while decoding: " + err.Error())
|
|
}
|
|
|
|
outExt := dec.GetAudioExt()
|
|
if outExt == "" {
|
|
outExt = "mp3"
|
|
}
|
|
filenameOnly := strings.TrimSuffix(filepath.Base(inputFile), filepath.Ext(inputFile))
|
|
|
|
outPath := filepath.Join(outputDir, filenameOnly+"."+outExt)
|
|
err = os.WriteFile(outPath, dec.GetAudioData(), 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logging.Log().Info("successfully converted",
|
|
zap.String("source", inputFile), zap.String("destination", outPath))
|
|
return nil
|
|
}
|