80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/loveuer/esgo2dump/internal/core"
|
||
|
|
"github.com/loveuer/esgo2dump/internal/opt"
|
||
|
|
"github.com/loveuer/esgo2dump/internal/tool"
|
||
|
|
"github.com/loveuer/esgo2dump/pkg/log"
|
||
|
|
"github.com/loveuer/esgo2dump/pkg/model"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func preRun(cmd *cobra.Command, args []string) error {
|
||
|
|
if opt.Cfg.Debug {
|
||
|
|
log.SetLogLevel(log.LogLevelDebug)
|
||
|
|
}
|
||
|
|
|
||
|
|
if opt.Cfg.Args.Version {
|
||
|
|
fmt.Printf("esgo2dump version: %s\n", opt.Version)
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
if opt.Cfg.Debug {
|
||
|
|
tool.TablePrinter(opt.Cfg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// check args
|
||
|
|
if opt.Cfg.Args.Input == "" {
|
||
|
|
return cmd.Help()
|
||
|
|
}
|
||
|
|
|
||
|
|
if opt.Cfg.Args.Limit == 0 || opt.Cfg.Args.Limit > 10000 {
|
||
|
|
return fmt.Errorf("invalid limit(1 - 10000)")
|
||
|
|
}
|
||
|
|
|
||
|
|
if opt.Cfg.Args.Query != "" && opt.Cfg.Args.QueryFile != "" {
|
||
|
|
return fmt.Errorf("cannot specify both query and query_file at the same time")
|
||
|
|
}
|
||
|
|
|
||
|
|
switch opt.Cfg.Args.Type {
|
||
|
|
case "data", "mapping", "setting":
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("unknown type=%s", opt.Cfg.Args.Type)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func run(cmd *cobra.Command, args []string) error {
|
||
|
|
var (
|
||
|
|
err error
|
||
|
|
input model.IO[map[string]any]
|
||
|
|
output model.IO[map[string]any]
|
||
|
|
)
|
||
|
|
|
||
|
|
if input, err = core.NewIO(cmd.Context(), opt.Cfg.Args.Input, model.Input); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if output, err = core.NewIO(cmd.Context(), opt.Cfg.Args.Output, model.Output); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
<-cmd.Context().Done()
|
||
|
|
log.Fatal(cmd.Context().Err().Error())
|
||
|
|
}()
|
||
|
|
|
||
|
|
switch opt.Cfg.Args.Type {
|
||
|
|
case "data":
|
||
|
|
return core.RunData(cmd, input, output)
|
||
|
|
case "mapping":
|
||
|
|
return core.RunMapping(cmd, input, output)
|
||
|
|
case "setting":
|
||
|
|
return core.RunSetting(cmd, input, output)
|
||
|
|
}
|
||
|
|
return fmt.Errorf("unknown args: type = %s", opt.Cfg.Args.Type)
|
||
|
|
}
|