73 lines
1.2 KiB
Go
73 lines
1.2 KiB
Go
|
package cli
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
|
||
|
"github.com/make-go-great/color-go"
|
||
|
"github.com/urfave/cli/v2"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
name = "gofimports"
|
||
|
usage = "goimports with my opinionated preferences"
|
||
|
|
||
|
// Inspiration from gofmt flags
|
||
|
flagListName = "list"
|
||
|
flagListUsage = "list files will be changed"
|
||
|
|
||
|
flagWriteName = "write"
|
||
|
flagWriteUsage = "actually write changes to (source) files"
|
||
|
|
||
|
// If both write and diff is exist
|
||
|
// Choose diff
|
||
|
flagDiffName = "diff"
|
||
|
flagDiffUsage = "show diff instead of writing changes"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
flagListAliases = []string{"l"}
|
||
|
flagWriteAliases = []string{"w"}
|
||
|
flagDiffAliases = []string{"d"}
|
||
|
)
|
||
|
|
||
|
type App struct {
|
||
|
cliApp *cli.App
|
||
|
}
|
||
|
|
||
|
func NewApp() *App {
|
||
|
a := &action{}
|
||
|
|
||
|
cliApp := &cli.App{
|
||
|
Name: name,
|
||
|
Usage: usage,
|
||
|
Flags: []cli.Flag{
|
||
|
&cli.BoolFlag{
|
||
|
Name: flagListName,
|
||
|
Usage: flagListUsage,
|
||
|
Aliases: flagListAliases,
|
||
|
},
|
||
|
&cli.BoolFlag{
|
||
|
Name: flagWriteName,
|
||
|
Usage: flagWriteUsage,
|
||
|
Aliases: flagWriteAliases,
|
||
|
},
|
||
|
&cli.BoolFlag{
|
||
|
Name: flagDiffName,
|
||
|
Usage: flagDiffUsage,
|
||
|
Aliases: flagDiffAliases,
|
||
|
},
|
||
|
},
|
||
|
Action: a.Run,
|
||
|
}
|
||
|
|
||
|
return &App{
|
||
|
cliApp: cliApp,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (a *App) Run() {
|
||
|
if err := a.cliApp.Run(os.Args); err != nil {
|
||
|
color.PrintAppError(name, err.Error())
|
||
|
}
|
||
|
}
|