Arsen Musayelyan
076f90bbd7
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
package cliutils
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/AlecAivazis/survey/v2"
|
|
"go.arsenm.dev/logger/log"
|
|
"go.arsenm.dev/lure/internal/config"
|
|
"go.arsenm.dev/lure/internal/db"
|
|
"go.arsenm.dev/lure/internal/pager"
|
|
"go.arsenm.dev/translate"
|
|
)
|
|
|
|
// YesNoPrompt asks the user a yes or no question, using def as the default answer
|
|
func YesNoPrompt(msg string, def bool, t translate.Translator) (bool, error) {
|
|
var answer bool
|
|
err := survey.AskOne(
|
|
&survey.Confirm{
|
|
Message: t.TranslateTo(msg, config.Language),
|
|
Default: def,
|
|
},
|
|
&answer,
|
|
)
|
|
return answer, err
|
|
}
|
|
|
|
// PromptViewScript asks the user if they'd like to see a script,
|
|
// shows it if they answer yes, then asks if they'd still like to
|
|
// continue, and exits if they answer no.
|
|
func PromptViewScript(script, name, style string, t translate.Translator) error {
|
|
view, err := YesNoPrompt(t.TranslateTo("Would you like to view the build script for", config.Language)+" "+name, false, t)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if view {
|
|
err = ShowScript(script, name, style)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cont, err := YesNoPrompt("Would you still like to continue?", false, t)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !cont {
|
|
log.Fatal(t.TranslateTo("User chose not to continue after reading script", config.Language)).Send()
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ShowScript uses the built-in pager to display a script at a
|
|
// given path, in the given syntax highlighting style.
|
|
func ShowScript(path, name, style string) error {
|
|
scriptFl, err := os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer scriptFl.Close()
|
|
|
|
str, err := pager.SyntaxHighlightBash(scriptFl, style)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pgr := pager.New(name, str)
|
|
return pgr.Run()
|
|
}
|
|
|
|
// FlattenPkgs attempts to flatten the a map of slices of packages into a single slice
|
|
// of packages by prompting the user if multiple packages match.
|
|
func FlattenPkgs(found map[string][]db.Package, verb string, t translate.Translator) []db.Package {
|
|
var outPkgs []db.Package
|
|
for _, pkgs := range found {
|
|
if len(pkgs) > 1 {
|
|
choices, err := PkgPrompt(pkgs, verb, t)
|
|
if err != nil {
|
|
log.Fatal("Error prompting for choice of package").Send()
|
|
}
|
|
outPkgs = append(outPkgs, choices...)
|
|
} else if len(pkgs) == 1 {
|
|
outPkgs = append(outPkgs, pkgs[0])
|
|
}
|
|
}
|
|
return outPkgs
|
|
}
|
|
|
|
// PkgPrompt asks the user to choose between multiple packages.
|
|
// The user may choose multiple packages.
|
|
func PkgPrompt(options []db.Package, verb string, t translate.Translator) ([]db.Package, error) {
|
|
names := make([]string, len(options))
|
|
for i, option := range options {
|
|
names[i] = option.Repository + "/" + option.Name + " " + option.Version
|
|
}
|
|
|
|
prompt := &survey.MultiSelect{
|
|
Options: names,
|
|
Message: t.TranslateTo("Choose which package(s) to "+verb, config.Language),
|
|
}
|
|
|
|
var choices []int
|
|
err := survey.AskOne(prompt, &choices)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
out := make([]db.Package, len(choices))
|
|
for i, choiceIndex := range choices {
|
|
out[i] = options[choiceIndex]
|
|
}
|
|
|
|
return out, nil
|
|
}
|