Close #10 Drag to Decrypt

This commit is contained in:
Emmm Monster 2021-05-16 16:59:49 +08:00
parent 1760737121
commit c71ad9cc79
No known key found for this signature in database
GPG Key ID: C98279C83FB50DB9

View File

@ -12,7 +12,6 @@ import (
"github.com/unlock-music/cli/internal/logging" "github.com/unlock-music/cli/internal/logging"
"github.com/urfave/cli/v2" "github.com/urfave/cli/v2"
"go.uber.org/zap" "go.uber.org/zap"
"log"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
@ -27,9 +26,10 @@ func main() {
Usage: "Unlock your encrypted music file https://github.com/unlock-music/cli", Usage: "Unlock your encrypted music file https://github.com/unlock-music/cli",
Version: AppVersion, Version: AppVersion,
Flags: []cli.Flag{ Flags: []cli.Flag{
&cli.StringFlag{Name: "input", Aliases: []string{"i"}, Usage: "path to input file or dir", Required: true}, &cli.StringFlag{Name: "input", Aliases: []string{"i"}, Usage: "path to input file or dir", Required: false},
&cli.StringFlag{Name: "output", Aliases: []string{"o"}, Usage: "path to output dir", Required: true}, &cli.StringFlag{Name: "output", Aliases: []string{"o"}, Usage: "path to output dir", Required: false},
}, },
Action: appMain, Action: appMain,
Copyright: "Copyright (c) 2020 Unlock Music https://github.com/unlock-music/cli/blob/master/LICENSE", Copyright: "Copyright (c) 2020 Unlock Music https://github.com/unlock-music/cli/blob/master/LICENSE",
HideHelpCommand: true, HideHelpCommand: true,
@ -37,13 +37,25 @@ func main() {
} }
err := app.Run(os.Args) err := app.Run(os.Args)
if err != nil { if err != nil {
log.Fatal(err) logging.Log().Fatal("run app failed", zap.Error(err))
} }
} }
func appMain(c *cli.Context) error { func appMain(c *cli.Context) error {
input := c.String("input") input := c.String("input")
if input == "" && c.Args().Present() {
input = c.Args().Get(c.Args().Len() - 1)
}
output := c.String("output") output := c.String("output")
if output == "" {
var err error
output, err = os.Getwd()
if err != nil {
return err
}
}
inputStat, err := os.Stat(input) inputStat, err := os.Stat(input)
if err != nil { if err != nil {
return err return err