Package subcommand code
This commit is contained in:
37
cmd/alias/alias.go
Normal file
37
cmd/alias/alias.go
Normal file
@ -0,0 +1,37 @@
|
||||
package alias
|
||||
|
||||
import (
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var AliasCmd = &cobra.Command{
|
||||
Use: "alias",
|
||||
Aliases: []string{"aliases", "a"},
|
||||
Short: "Manage project aliases",
|
||||
Long: util.AliasCmdLong,
|
||||
// Just re-use the hooks for project
|
||||
PersistentPreRun: util.InitProjects,
|
||||
PersistentPostRun: util.PostProjectCmd,
|
||||
}
|
||||
|
||||
var utils *util.Utils
|
||||
|
||||
func aliasCmdPreRun(cmd *cobra.Command, args []string) {
|
||||
utils = util.MustFromCtx(cmd.Context())
|
||||
util.InitProjects(cmd, args)
|
||||
}
|
||||
|
||||
func mustHaveAliases(cmd *cobra.Command, args []string) {
|
||||
utils := util.MustFromCtx(cmd.Context())
|
||||
|
||||
if len(utils.Cache().Aliases) == 0 {
|
||||
utils.Logger().Fatal("No aliases set, nothing to " + cmd.Name())
|
||||
}
|
||||
}
|
||||
|
||||
func init() {
|
||||
AliasCmd.AddCommand(aliasAddCmd)
|
||||
AliasCmd.AddCommand(aliasListCmd)
|
||||
AliasCmd.AddCommand(aliasDeleteCmd)
|
||||
}
|
||||
106
cmd/alias/alias_add.go
Normal file
106
cmd/alias/alias_add.go
Normal file
@ -0,0 +1,106 @@
|
||||
package alias
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/pterm/pterm"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/spf13/viper"
|
||||
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/cache"
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
||||
)
|
||||
|
||||
var aliasAddCmd = &cobra.Command{
|
||||
Use: "add",
|
||||
Aliases: []string{"set", "a", "s"},
|
||||
Short: "Add a project alias",
|
||||
Args: cobra.ArbitraryArgs,
|
||||
Long: util.AliasAddCmdLong,
|
||||
Run: runAddAliasCmd,
|
||||
}
|
||||
|
||||
func runAddAliasCmd(cmd *cobra.Command, args []string) {
|
||||
var project *projects.Project
|
||||
|
||||
// Check by flag
|
||||
if projectID := viper.GetInt(util.ViperAliasAddPID); projectID > 0 {
|
||||
utils.Logger().Debug(fmt.Sprintf("Adding for inbound project ID %d", projectID))
|
||||
project = utils.Cache().GetProjectByID(projectID)
|
||||
}
|
||||
|
||||
// Check by arg
|
||||
if len(args) > 0 {
|
||||
project = utils.FzfFindProject(&util.FzfProjectOpts{Ctx: cmd.Context(), Search: utils.SearchStringFromArgs(args)})
|
||||
}
|
||||
|
||||
// Collect by fzf
|
||||
if project == nil {
|
||||
var err error
|
||||
project, err = utils.FzfProject(&util.FzfProjectOpts{Ctx: cmd.Context()})
|
||||
if err != nil || project == nil {
|
||||
utils.Logger().Fatal("No project to alias, nothing to do", utils.Logger().Args("error", err))
|
||||
}
|
||||
}
|
||||
|
||||
addNewAliases(project.ID)
|
||||
}
|
||||
|
||||
func addNewAliases(projectID int) {
|
||||
project := utils.Cache().GetProjectByID(projectID)
|
||||
if project == nil {
|
||||
utils.Logger().Error("Failed to find project to alias", utils.Logger().Args("projectID", projectID))
|
||||
return
|
||||
}
|
||||
|
||||
// Collect the aliases
|
||||
aliases := promptAliasesForProject(project)
|
||||
|
||||
// Add aliases
|
||||
for _, a := range aliases {
|
||||
a = strings.Trim(a, " '\"%<>|`")
|
||||
if a == "" {
|
||||
continue
|
||||
}
|
||||
if err := utils.Cache().AddAlias(a, project.ID, project.Remote); err != nil {
|
||||
utils.Logger().Debug("Skipping alias add", utils.Logger().Args(
|
||||
"error", err,
|
||||
"alias", a,
|
||||
))
|
||||
} else {
|
||||
utils.Logger().Info("Successfully added alias to project", utils.Logger().Args(
|
||||
"project", project.String(),
|
||||
"alias", a,
|
||||
))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func promptAliasesForProject(p *projects.Project) []string {
|
||||
aliases := utils.Cache().GetProjectAliases(p)
|
||||
if len(aliases) > 0 {
|
||||
utils.Logger().Info("Adding aliases to project", utils.Logger().Args(
|
||||
"project", p.String(),
|
||||
"existingAliases", cache.ProjectAliasesString(aliases),
|
||||
))
|
||||
} else {
|
||||
pterm.Info.Printfln("Adding aliases to %s", p.Name)
|
||||
}
|
||||
|
||||
response, _ := pterm.DefaultInteractiveTextInput.
|
||||
WithMultiLine(false).
|
||||
WithDefaultValue(p.Path + " ").
|
||||
Show("Enter aliases separated by space")
|
||||
|
||||
return strings.Split(response, " ")
|
||||
}
|
||||
|
||||
func init() {
|
||||
aliasAddCmd.PersistentFlags().Int(util.FlagProjectID, 0, "Specify a project by ID")
|
||||
|
||||
aliasAddCmd.RegisterFlagCompletionFunc(util.FlagProjectID, util.ValidProjectIdFunc)
|
||||
|
||||
viper.BindPFlag(util.ViperAliasAddPID, aliasAddCmd.Flag(util.FlagProjectID))
|
||||
}
|
||||
86
cmd/alias/alias_delete.go
Normal file
86
cmd/alias/alias_delete.go
Normal file
@ -0,0 +1,86 @@
|
||||
package alias
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
||||
"github.com/pterm/pterm"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
var aliasDeleteCmd = &cobra.Command{
|
||||
Use: "delete [fuzzy project or alias]",
|
||||
Aliases: []string{"rm", "del", "d"},
|
||||
Short: "Delete a project alias",
|
||||
Long: util.AliasDeleteCmdLong,
|
||||
PreRun: mustHaveAliases,
|
||||
Run: runDeleteAliasCmd,
|
||||
}
|
||||
|
||||
func runDeleteAliasCmd(cmd *cobra.Command, args []string) {
|
||||
var project *projects.Project
|
||||
var err error
|
||||
|
||||
fzfOpts := &util.FzfProjectOpts{
|
||||
Ctx: cmd.Context(),
|
||||
MustHaveAlias: true,
|
||||
}
|
||||
|
||||
if len(args) > 0 {
|
||||
fzfOpts.Search = utils.SearchStringFromArgs(args)
|
||||
project = utils.FzfFindProject(fzfOpts)
|
||||
} else {
|
||||
project, err = utils.FzfProject(fzfOpts)
|
||||
}
|
||||
|
||||
if project == nil || err != nil {
|
||||
utils.Logger().Fatal("Failed to find project to delete aliases from", utils.Logger().Args(
|
||||
"error", err,
|
||||
))
|
||||
}
|
||||
|
||||
aliasStrings := utils.Cache().GetProjectAliasStrings(project)
|
||||
|
||||
deletionCandidates, err := pterm.DefaultInteractiveMultiselect.
|
||||
WithOptions(aliasStrings).
|
||||
Show()
|
||||
|
||||
if err != nil || len(deletionCandidates) < 1 {
|
||||
utils.Logger().Fatal("Failed to find project to delete aliases from", utils.Logger().Args(
|
||||
"error", err,
|
||||
))
|
||||
}
|
||||
|
||||
for _, a := range deletionCandidates {
|
||||
|
||||
confirm, _ := pterm.DefaultInteractiveConfirm.
|
||||
WithDefaultText(fmt.Sprintf("Really delete %s -> %s?",
|
||||
a, project.String())).
|
||||
WithConfirmText("y").
|
||||
Show()
|
||||
|
||||
if !confirm {
|
||||
utils.Logger().Warn("Alias deletion cancelled")
|
||||
continue
|
||||
}
|
||||
|
||||
utils.Logger().Info("Deleting alias", utils.Logger().Args(
|
||||
"project", project.String(),
|
||||
"alias", a,
|
||||
))
|
||||
|
||||
utils.Cache().DeleteAlias(utils.Cache().GetAliasByName(a))
|
||||
}
|
||||
|
||||
fmt.Println(utils.Cache().ProjectString(project))
|
||||
}
|
||||
|
||||
func init() {
|
||||
aliasDeleteCmd.PersistentFlags().Int("projectID", 0, "Specify a project by ID")
|
||||
|
||||
aliasDeleteCmd.RegisterFlagCompletionFunc("projectID", util.ValidProjectIdFunc)
|
||||
|
||||
viper.BindPFlag("alias.delete.projectID", aliasDeleteCmd.Flag("projectID"))
|
||||
}
|
||||
34
cmd/alias/alias_list.go
Normal file
34
cmd/alias/alias_list.go
Normal file
@ -0,0 +1,34 @@
|
||||
package alias
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||
"github.com/pterm/pterm"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
// listCmd represents the list command
|
||||
var aliasListCmd = &cobra.Command{
|
||||
Use: "list",
|
||||
Aliases: []string{"dump", "show", "ls", "ll", "l"},
|
||||
Short: "List Aliases",
|
||||
PreRun: mustHaveAliases,
|
||||
Long: util.AliasListCmdLong,
|
||||
Run: runListAliasCmd,
|
||||
}
|
||||
|
||||
func runListAliasCmd(cmd *cobra.Command, args []string) {
|
||||
remotes := viper.GetStringSlice(FlagRemote)
|
||||
pterm.DefaultBox.
|
||||
WithLeftPadding(5).WithRightPadding(5).
|
||||
WithBoxStyle(&pterm.Style{pterm.FgLightBlue}).
|
||||
WithTitle(pterm.Bold.Sprint(pterm.LightGreen("Aliases by Project"))).
|
||||
Print("\n" + projectCache.AliasesByProjectString(remotes...))
|
||||
fmt.Print("\n\n")
|
||||
}
|
||||
|
||||
func init() {
|
||||
aliasCmd.AddCommand(aliasListCmd)
|
||||
}
|
||||
Reference in New Issue
Block a user