34 lines
876 B
Go
34 lines
876 B
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/vilmibm/trunkless/cutup"
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.Flags().StringP("cutupdir", "d", "", "directory in which to write phrase files")
|
|
rootCmd.Flags().StringP("srcdir", "s", "", "directory of files to cut up")
|
|
rootCmd.Flags().IntP("workers", "w", 10, "number of workers to use when cutting up")
|
|
|
|
rootCmd.MarkFlagRequired("cutupdir")
|
|
rootCmd.MarkFlagRequired("srcdir")
|
|
|
|
rootCmd.AddCommand(cutupCmd)
|
|
}
|
|
|
|
var cutupCmd = &cobra.Command{
|
|
Use: "cutup",
|
|
Args: cobra.MaximumNArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cutupdir, _ := cmd.Flags().GetString("cutupdir")
|
|
srcdir, _ := cmd.Flags().GetString("srcdir")
|
|
workers, _ := cmd.Flags().GetInt("workers")
|
|
opts := cutup.CutupOpts{
|
|
CutupDir: cutupdir,
|
|
SrcDir: srcdir,
|
|
NumWorkers: workers,
|
|
}
|
|
return cutup.Cutup(opts)
|
|
},
|
|
}
|