refactor-cmd-pkgs #1
25
cmd/alias.go
25
cmd/alias.go
@ -1,25 +0,0 @@
|
|||||||
package cmd
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
)
|
|
||||||
|
|
||||||
var aliasCmd = &cobra.Command{
|
|
||||||
Use: "alias",
|
|
||||||
Aliases: []string{"aliases", "a"},
|
|
||||||
Short: "Manage project aliases",
|
|
||||||
Long: aliasCmdLong,
|
|
||||||
// Just re-use the hooks for project
|
|
||||||
PersistentPreRun: initProjectCmd,
|
|
||||||
PersistentPostRun: postProjectCmd,
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
rootCmd.AddCommand(aliasCmd)
|
|
||||||
}
|
|
||||||
|
|
||||||
func mustHaveAliases(cmd *cobra.Command, args []string) {
|
|
||||||
if len(projectCache.Aliases) == 0 {
|
|
||||||
plog.Fatal("No aliases set, nothing to " + cmd.Name())
|
|
||||||
}
|
|
||||||
}
|
|
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)
|
||||||
|
}
|
@ -1,4 +1,4 @@
|
|||||||
package cmd
|
package alias
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
@ -8,6 +8,7 @@ import (
|
|||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"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/cache"
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
||||||
)
|
)
|
||||||
@ -17,7 +18,7 @@ var aliasAddCmd = &cobra.Command{
|
|||||||
Aliases: []string{"set", "a", "s"},
|
Aliases: []string{"set", "a", "s"},
|
||||||
Short: "Add a project alias",
|
Short: "Add a project alias",
|
||||||
Args: cobra.ArbitraryArgs,
|
Args: cobra.ArbitraryArgs,
|
||||||
Long: aliasAddCmdLong,
|
Long: util.AliasAddCmdLong,
|
||||||
Run: runAddAliasCmd,
|
Run: runAddAliasCmd,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -25,22 +26,22 @@ func runAddAliasCmd(cmd *cobra.Command, args []string) {
|
|||||||
var project *projects.Project
|
var project *projects.Project
|
||||||
|
|
||||||
// Check by flag
|
// Check by flag
|
||||||
if projectID := viper.GetInt(ViperAliasAddPID); projectID > 0 {
|
if projectID := viper.GetInt(util.ViperAliasAddPID); projectID > 0 {
|
||||||
plog.Debug(fmt.Sprintf("Adding for inbound project ID %d", projectID))
|
utils.Logger().Debug(fmt.Sprintf("Adding for inbound project ID %d", projectID))
|
||||||
project = projectCache.GetProjectByID(projectID)
|
project = utils.Cache().GetProjectByID(projectID)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check by arg
|
// Check by arg
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
project = fzfFindProject(&fzfProjectOpts{Ctx: cmd.Context(), Search: searchStringFromArgs(args)})
|
project = utils.FzfFindProject(&util.FzfProjectOpts{Ctx: cmd.Context(), Search: utils.SearchStringFromArgs(args)})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Collect by fzf
|
// Collect by fzf
|
||||||
if project == nil {
|
if project == nil {
|
||||||
var err error
|
var err error
|
||||||
project, err = fzfProject(&fzfProjectOpts{Ctx: cmd.Context()})
|
project, err = utils.FzfProject(&util.FzfProjectOpts{Ctx: cmd.Context()})
|
||||||
if err != nil || project == nil {
|
if err != nil || project == nil {
|
||||||
plog.Fatal("No project to alias, nothing to do", plog.Args("error", err))
|
utils.Logger().Fatal("No project to alias, nothing to do", utils.Logger().Args("error", err))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -48,9 +49,9 @@ func runAddAliasCmd(cmd *cobra.Command, args []string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func addNewAliases(projectID int) {
|
func addNewAliases(projectID int) {
|
||||||
project := projectCache.GetProjectByID(projectID)
|
project := utils.Cache().GetProjectByID(projectID)
|
||||||
if project == nil {
|
if project == nil {
|
||||||
plog.Error("Failed to find project to alias", plog.Args("projectID", projectID))
|
utils.Logger().Error("Failed to find project to alias", utils.Logger().Args("projectID", projectID))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -63,13 +64,13 @@ func addNewAliases(projectID int) {
|
|||||||
if a == "" {
|
if a == "" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if err := projectCache.AddAlias(a, project.ID, project.Remote); err != nil {
|
if err := utils.Cache().AddAlias(a, project.ID, project.Remote); err != nil {
|
||||||
plog.Debug("Skipping alias add", plog.Args(
|
utils.Logger().Debug("Skipping alias add", utils.Logger().Args(
|
||||||
"error", err,
|
"error", err,
|
||||||
"alias", a,
|
"alias", a,
|
||||||
))
|
))
|
||||||
} else {
|
} else {
|
||||||
plog.Info("Successfully added alias to project", plog.Args(
|
utils.Logger().Info("Successfully added alias to project", utils.Logger().Args(
|
||||||
"project", project.String(),
|
"project", project.String(),
|
||||||
"alias", a,
|
"alias", a,
|
||||||
))
|
))
|
||||||
@ -78,9 +79,9 @@ func addNewAliases(projectID int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func promptAliasesForProject(p *projects.Project) []string {
|
func promptAliasesForProject(p *projects.Project) []string {
|
||||||
aliases := projectCache.GetProjectAliases(p)
|
aliases := utils.Cache().GetProjectAliases(p)
|
||||||
if len(aliases) > 0 {
|
if len(aliases) > 0 {
|
||||||
plog.Info("Adding aliases to project", plog.Args(
|
utils.Logger().Info("Adding aliases to project", utils.Logger().Args(
|
||||||
"project", p.String(),
|
"project", p.String(),
|
||||||
"existingAliases", cache.ProjectAliasesString(aliases),
|
"existingAliases", cache.ProjectAliasesString(aliases),
|
||||||
))
|
))
|
||||||
@ -97,10 +98,9 @@ func promptAliasesForProject(p *projects.Project) []string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
aliasCmd.AddCommand(aliasAddCmd)
|
aliasAddCmd.PersistentFlags().Int(util.FlagProjectID, 0, "Specify a project by ID")
|
||||||
aliasAddCmd.PersistentFlags().Int(FlagProjectID, 0, "Specify a project by ID")
|
|
||||||
|
|
||||||
aliasAddCmd.RegisterFlagCompletionFunc(FlagProjectID, validProjectIdFunc)
|
aliasAddCmd.RegisterFlagCompletionFunc(util.FlagProjectID, util.ValidProjectIdFunc)
|
||||||
|
|
||||||
viper.BindPFlag(ViperAliasAddPID, aliasAddCmd.Flag(FlagProjectID))
|
viper.BindPFlag(util.ViperAliasAddPID, aliasAddCmd.Flag(util.FlagProjectID))
|
||||||
}
|
}
|
@ -1,19 +1,20 @@
|
|||||||
package cmd
|
package alias
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"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/pterm/pterm"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var aliasDeleteCmd = &cobra.Command{
|
var aliasDeleteCmd = &cobra.Command{
|
||||||
Use: "delete [fuzzy project or alias]",
|
Use: "delete [fuzzy project or alias]",
|
||||||
Aliases: []string{"rm", "del", "d"},
|
Aliases: []string{"rm", "del", "d"},
|
||||||
Short: "Delete a project alias",
|
Short: "Delete a project alias",
|
||||||
Long: aliasDeleteCmdLong,
|
Long: util.AliasDeleteCmdLong,
|
||||||
PreRun: mustHaveAliases,
|
PreRun: mustHaveAliases,
|
||||||
Run: runDeleteAliasCmd,
|
Run: runDeleteAliasCmd,
|
||||||
}
|
}
|
||||||
@ -22,32 +23,32 @@ func runDeleteAliasCmd(cmd *cobra.Command, args []string) {
|
|||||||
var project *projects.Project
|
var project *projects.Project
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
fzfOpts := &fzfProjectOpts{
|
fzfOpts := &util.FzfProjectOpts{
|
||||||
Ctx: cmd.Context(),
|
Ctx: cmd.Context(),
|
||||||
MustHaveAlias: true,
|
MustHaveAlias: true,
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
fzfOpts.Search = searchStringFromArgs(args)
|
fzfOpts.Search = utils.SearchStringFromArgs(args)
|
||||||
project = fzfFindProject(fzfOpts)
|
project = utils.FzfFindProject(fzfOpts)
|
||||||
} else {
|
} else {
|
||||||
project, err = fzfProject(fzfOpts)
|
project, err = utils.FzfProject(fzfOpts)
|
||||||
}
|
}
|
||||||
|
|
||||||
if project == nil || err != nil {
|
if project == nil || err != nil {
|
||||||
plog.Fatal("Failed to find project to delete aliases from", plog.Args(
|
utils.Logger().Fatal("Failed to find project to delete aliases from", utils.Logger().Args(
|
||||||
"error", err,
|
"error", err,
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
aliasStrings := projectCache.GetProjectAliasStrings(project)
|
aliasStrings := utils.Cache().GetProjectAliasStrings(project)
|
||||||
|
|
||||||
deletionCandidates, err := pterm.DefaultInteractiveMultiselect.
|
deletionCandidates, err := pterm.DefaultInteractiveMultiselect.
|
||||||
WithOptions(aliasStrings).
|
WithOptions(aliasStrings).
|
||||||
Show()
|
Show()
|
||||||
|
|
||||||
if err != nil || len(deletionCandidates) < 1 {
|
if err != nil || len(deletionCandidates) < 1 {
|
||||||
plog.Fatal("Failed to find project to delete aliases from", plog.Args(
|
utils.Logger().Fatal("Failed to find project to delete aliases from", utils.Logger().Args(
|
||||||
"error", err,
|
"error", err,
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
@ -61,26 +62,25 @@ func runDeleteAliasCmd(cmd *cobra.Command, args []string) {
|
|||||||
Show()
|
Show()
|
||||||
|
|
||||||
if !confirm {
|
if !confirm {
|
||||||
plog.Warn("Alias deletion cancelled")
|
utils.Logger().Warn("Alias deletion cancelled")
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
plog.Info("Deleting alias", plog.Args(
|
utils.Logger().Info("Deleting alias", utils.Logger().Args(
|
||||||
"project", project.String(),
|
"project", project.String(),
|
||||||
"alias", a,
|
"alias", a,
|
||||||
))
|
))
|
||||||
|
|
||||||
projectCache.DeleteAlias(projectCache.GetAliasByName(a))
|
utils.Cache().DeleteAlias(utils.Cache().GetAliasByName(a))
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println(projectCache.ProjectString(project))
|
fmt.Println(utils.Cache().ProjectString(project))
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
aliasCmd.AddCommand(aliasDeleteCmd)
|
|
||||||
aliasDeleteCmd.PersistentFlags().Int("projectID", 0, "Specify a project by ID")
|
aliasDeleteCmd.PersistentFlags().Int("projectID", 0, "Specify a project by ID")
|
||||||
|
|
||||||
aliasDeleteCmd.RegisterFlagCompletionFunc("projectID", validProjectIdFunc)
|
aliasDeleteCmd.RegisterFlagCompletionFunc("projectID", util.ValidProjectIdFunc)
|
||||||
|
|
||||||
viper.BindPFlag("alias.delete.projectID", aliasDeleteCmd.Flag("projectID"))
|
viper.BindPFlag("alias.delete.projectID", aliasDeleteCmd.Flag("projectID"))
|
||||||
}
|
}
|
@ -1,8 +1,9 @@
|
|||||||
package cmd
|
package alias
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||||
"github.com/pterm/pterm"
|
"github.com/pterm/pterm"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
@ -14,7 +15,7 @@ var aliasListCmd = &cobra.Command{
|
|||||||
Aliases: []string{"dump", "show", "ls", "ll", "l"},
|
Aliases: []string{"dump", "show", "ls", "ll", "l"},
|
||||||
Short: "List Aliases",
|
Short: "List Aliases",
|
||||||
PreRun: mustHaveAliases,
|
PreRun: mustHaveAliases,
|
||||||
Long: aliasListCmdLong,
|
Long: util.AliasListCmdLong,
|
||||||
Run: runListAliasCmd,
|
Run: runListAliasCmd,
|
||||||
}
|
}
|
||||||
|
|
4
cmd/cache.go → cmd/cache/cache.go
vendored
4
cmd/cache.go → cmd/cache/cache.go
vendored
@ -5,12 +5,8 @@ import (
|
|||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
|
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/cache"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var projectCache *cache.Cache
|
|
||||||
|
|
||||||
var cacheCmd = &cobra.Command{
|
var cacheCmd = &cobra.Command{
|
||||||
Use: "cache",
|
Use: "cache",
|
||||||
Aliases: []string{"a", "ln"},
|
Aliases: []string{"a", "ln"},
|
@ -4,6 +4,7 @@ import (
|
|||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"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/cache"
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
||||||
)
|
)
|
||||||
@ -15,16 +16,16 @@ var projectCmd = &cobra.Command{
|
|||||||
Args: cobra.MaximumNArgs(1),
|
Args: cobra.MaximumNArgs(1),
|
||||||
ArgAliases: []string{"alias"},
|
ArgAliases: []string{"alias"},
|
||||||
Long: projCmdLong,
|
Long: projCmdLong,
|
||||||
PersistentPreRun: initProjectCmd,
|
PersistentPreRun: initProjects,
|
||||||
PersistentPostRun: postProjectCmd,
|
PersistentPostRun: postProjectCmd,
|
||||||
// Run: projectGoCmdRun,
|
// Run: projectGoCmdRun,
|
||||||
}
|
}
|
||||||
|
|
||||||
func getProject(args []string) *projects.Project {
|
func getProject(args []string) *projects.Project {
|
||||||
remotes := viper.GetStringSlice(FlagRemote)
|
remotes := viper.GetStringSlice(util.FlagRemote)
|
||||||
fzfOpts := &fzfProjectOpts{
|
fzfOpts := &util.FzfProjectOpts{
|
||||||
Ctx: rootCmd.Context(),
|
Ctx: rootCmd.Context(),
|
||||||
Search: searchStringFromArgs(args),
|
Search: utils.SearchStringFromArgs(args),
|
||||||
Remotes: remotes,
|
Remotes: remotes,
|
||||||
}
|
}
|
||||||
project := fzfFindProject(fzfOpts)
|
project := fzfFindProject(fzfOpts)
|
||||||
@ -47,21 +48,7 @@ func getProject(args []string) *projects.Project {
|
|||||||
return project
|
return project
|
||||||
}
|
}
|
||||||
|
|
||||||
func initProjectCmd(cmd *cobra.Command, args []string) {
|
|
||||||
initProjectCache(cmd, args)
|
|
||||||
mustHaveProjects(cmd, args)
|
|
||||||
}
|
|
||||||
|
|
||||||
func postProjectCmd(cmd *cobra.Command, args []string) {
|
|
||||||
postProjectCache(cmd, args)
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
rootCmd.AddCommand(projectCmd)
|
rootCmd.AddCommand(projectCmd)
|
||||||
}
|
}
|
||||||
|
|
||||||
func mustHaveProjects(cmd *cobra.Command, _ []string) {
|
|
||||||
if len(projectCache.Projects) == 0 {
|
|
||||||
plog.Fatal("No projects to " + cmd.Name() + ", try running cache load")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
136
cmd/root.go
136
cmd/root.go
@ -12,28 +12,27 @@ import (
|
|||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
|
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/cmd/util"
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/config"
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/config"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
|
||||||
conf config.Config
|
|
||||||
plog *pterm.Logger
|
|
||||||
)
|
|
||||||
|
|
||||||
var rootCmd = &cobra.Command{
|
var rootCmd = &cobra.Command{
|
||||||
Use: "git-project-manager",
|
Use: "git-project-manager",
|
||||||
Aliases: []string{"gpm"},
|
Aliases: []string{"gpm"},
|
||||||
Short: "Find and use Git projects locally",
|
Short: "Find and use Git projects locally",
|
||||||
Long: rootCmdLong,
|
Long: util.RootCmdLong,
|
||||||
PersistentPreRun: initRootCmd,
|
PersistentPreRun: initRootCmd,
|
||||||
}
|
}
|
||||||
|
|
||||||
var configExemptCommands = regexp.MustCompile(`^(doc|conf)`)
|
var (
|
||||||
|
utils *util.Utils
|
||||||
|
configExemptCommands = regexp.MustCompile(`^(doc|conf)`)
|
||||||
|
)
|
||||||
|
|
||||||
// Hook traversal is enabled, so this will be run for all
|
// Hook traversal is enabled, so this will be run for all
|
||||||
// sub-commands regardless of their registered pre-hooks
|
// sub-commands regardless of their registered pre-hooks
|
||||||
func initRootCmd(cmd *cobra.Command, args []string) {
|
func initRootCmd(cmd *cobra.Command, args []string) {
|
||||||
initProjectPath(cmd, args)
|
utils.InitProjectPath(cmd, args)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute adds all child commands to the root command and sets flags appropriately.
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
||||||
@ -42,6 +41,9 @@ func Execute() {
|
|||||||
ctx, cncl := signal.NotifyContext(context.Background(), os.Kill, os.Interrupt)
|
ctx, cncl := signal.NotifyContext(context.Background(), os.Kill, os.Interrupt)
|
||||||
defer cncl()
|
defer cncl()
|
||||||
|
|
||||||
|
// Store provided util.Utils in context
|
||||||
|
ctx = util.AddToCtx(ctx, utils)
|
||||||
|
|
||||||
err := rootCmd.ExecuteContext(ctx)
|
err := rootCmd.ExecuteContext(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
pterm.Error.Printfln("%s", pterm.LightYellow("Command failed, "+err.Error()))
|
pterm.Error.Printfln("%s", pterm.LightYellow("Command failed, "+err.Error()))
|
||||||
@ -51,73 +53,79 @@ func Execute() {
|
|||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
cobra.EnableTraverseRunHooks = true
|
cobra.EnableTraverseRunHooks = true
|
||||||
cobra.OnInitialize(initConfig)
|
|
||||||
|
utils = &util.Utils{}
|
||||||
|
cobra.OnInitialize(getInitConfigFunc(utils))
|
||||||
|
|
||||||
// Global flags
|
// Global flags
|
||||||
rootCmd.PersistentFlags().String(FlagConfig, "",
|
rootCmd.PersistentFlags().String(util.FlagConfig, "",
|
||||||
"config file (default is "+defConfigPath+")")
|
"config file (default is "+util.DefConfigPath+")")
|
||||||
rootCmd.PersistentFlags().String(FlagPath, "",
|
rootCmd.PersistentFlags().String(util.FlagPath, "",
|
||||||
"Sets a path for local clones of projects")
|
"Sets a path for local clones of projects")
|
||||||
rootCmd.PersistentFlags().String(FlagLogLevel, defLogLevel,
|
rootCmd.PersistentFlags().String(util.FlagLogLevel, util.DefLogLevel,
|
||||||
"Default log level -- info, warn, error, debug")
|
"Default log level -- info, warn, error, debug")
|
||||||
rootCmd.PersistentFlags().StringSlice(FlagRemote, []string{},
|
rootCmd.PersistentFlags().StringSlice(util.FlagRemote, []string{},
|
||||||
"Specify remotes by host for any sub-command. Provide multiple times or comma delimited.")
|
"Specify remotes by host for any sub-command. Provide multiple times or comma delimited.")
|
||||||
|
|
||||||
// Flag autocompletion
|
// Flag autocompletion
|
||||||
rootCmd.RegisterFlagCompletionFunc(FlagLogLevel, validLogLevelsFunc)
|
rootCmd.RegisterFlagCompletionFunc(util.FlagLogLevel, util.ValidLogLevelsFunc)
|
||||||
rootCmd.RegisterFlagCompletionFunc(FlagRemote, validRemotesFunc)
|
rootCmd.RegisterFlagCompletionFunc(util.FlagRemote, utils.ValidRemotesFunc)
|
||||||
|
|
||||||
viper.BindPFlags(rootCmd.PersistentFlags())
|
viper.BindPFlags(rootCmd.PersistentFlags())
|
||||||
}
|
}
|
||||||
|
|
||||||
// initConfig reads in config file and ENV variables if set.
|
// initConfig reads in config file and ENV variables if set.
|
||||||
func initConfig() {
|
func getInitConfigFunc(utils *util.Utils) func() {
|
||||||
cfgFile := viper.GetString(FlagConfig)
|
return func() {
|
||||||
if cfgFile != "" {
|
cfgFile := viper.GetString(util.FlagConfig)
|
||||||
// Use config file from the flag.
|
if cfgFile != "" {
|
||||||
viper.SetConfigFile(cfgFile)
|
// Use config file from the flag.
|
||||||
} else {
|
viper.SetConfigFile(cfgFile)
|
||||||
// Find home directory.
|
} else {
|
||||||
home, err := os.UserHomeDir()
|
// Find home directory.
|
||||||
cobra.CheckErr(err)
|
home, err := os.UserHomeDir()
|
||||||
|
cobra.CheckErr(err)
|
||||||
|
|
||||||
// Search config in home directory with name ".git-project-manager" (without extension).
|
// Search config in home directory with name ".git-project-manager" (without extension).
|
||||||
configPath := filepath.Join(home, ".config")
|
configPath := filepath.Join(home, ".config")
|
||||||
viper.AddConfigPath(configPath)
|
viper.AddConfigPath(configPath)
|
||||||
viper.SetConfigType("yaml")
|
viper.SetConfigType("yaml")
|
||||||
viper.SetConfigName(getConfigName(configPath))
|
viper.SetConfigName(util.GetConfigName(configPath))
|
||||||
|
}
|
||||||
|
|
||||||
|
viper.AutomaticEnv()
|
||||||
|
viper.ReadInConfig()
|
||||||
|
|
||||||
|
// Configure pretty logger
|
||||||
|
plog := pterm.DefaultLogger.
|
||||||
|
WithLevel(getPtermLogLevel(viper.GetString(util.FlagLogLevel))).
|
||||||
|
WithWriter(os.Stderr)
|
||||||
|
if plog.Level == pterm.LogLevelDebug {
|
||||||
|
pterm.EnableDebugMessages()
|
||||||
|
}
|
||||||
|
utils.SetLogger(plog)
|
||||||
|
|
||||||
|
// Load into struct to not be so darn pythonic, retrieving
|
||||||
|
// settings by untyped string "name"
|
||||||
|
conf := new(config.Config)
|
||||||
|
if err := viper.Unmarshal(&conf); err != nil {
|
||||||
|
plog.Error("Failed loading config", plog.Args("err", err))
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(os.Args) > 0 && configExemptCommands.Match([]byte(os.Args[1])) {
|
||||||
|
plog.Debug("Permitting missing config for config sub-command")
|
||||||
|
return
|
||||||
|
} else if conf.ProjectPath == "" {
|
||||||
|
plog.Fatal("Minimal configuration missing, must have projectPath", plog.Args(
|
||||||
|
"do",
|
||||||
|
"Try running `git-project-manager config default > "+util.DefConfigPath,
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
checkConfigPerms(viper.ConfigFileUsed()) // Abort on world-readable config
|
||||||
|
|
||||||
|
utils.Logger().Debug("Configuration loaded", plog.Args("conf", conf))
|
||||||
}
|
}
|
||||||
|
|
||||||
viper.AutomaticEnv()
|
|
||||||
viper.ReadInConfig()
|
|
||||||
|
|
||||||
// Configure pretty logger
|
|
||||||
plog = pterm.DefaultLogger.
|
|
||||||
WithLevel(getPtermLogLevel(viper.GetString(FlagLogLevel))).
|
|
||||||
WithWriter(os.Stderr)
|
|
||||||
if plog.Level == pterm.LogLevelDebug {
|
|
||||||
pterm.EnableDebugMessages()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Load into struct to not be so darn pythonic, retrieving
|
|
||||||
// settings by untyped string "name"
|
|
||||||
if err := viper.Unmarshal(&conf); err != nil {
|
|
||||||
plog.Error("Failed loading config", plog.Args("err", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(os.Args) > 0 && configExemptCommands.Match([]byte(os.Args[1])) {
|
|
||||||
plog.Debug("Permitting missing config for config sub-command")
|
|
||||||
return
|
|
||||||
} else if conf.ProjectPath == "" {
|
|
||||||
plog.Fatal("Minimal configuration missing, must have projectPath", plog.Args(
|
|
||||||
"do",
|
|
||||||
"Try running `git-project-manager config default > "+defConfigPath,
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
checkConfigPerms(viper.ConfigFileUsed()) // Abort on world-readable config
|
|
||||||
|
|
||||||
plog.Debug("Configuration loaded", plog.Args("conf", conf))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func getPtermLogLevel(level string) pterm.LogLevel {
|
func getPtermLogLevel(level string) pterm.LogLevel {
|
||||||
@ -141,12 +149,12 @@ func getPtermLogLevel(level string) pterm.LogLevel {
|
|||||||
func checkConfigPerms(file string) {
|
func checkConfigPerms(file string) {
|
||||||
stat, err := os.Stat(file)
|
stat, err := os.Stat(file)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
plog.Error("Failure reading configuration", plog.Args("err", err))
|
utils.Logger().Error("Failure reading configuration", utils.Logger().Args("err", err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if stat.Mode().Perm()&0o004 == 0o004 {
|
if stat.Mode().Perm()&0o004 == 0o004 {
|
||||||
plog.Error("Configuration is world-readable. Recomment 0400.",
|
utils.Logger().Error("Configuration is world-readable. Recomment 0400.",
|
||||||
plog.Args("mode", stat.Mode().String()))
|
utils.Logger().Args("mode", stat.Mode().String()))
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
121
cmd/util/util.go
Normal file
121
cmd/util/util.go
Normal file
@ -0,0 +1,121 @@
|
|||||||
|
// Common utilities used by various subcommands
|
||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/cache"
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/config"
|
||||||
|
"github.com/pterm/pterm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Utils struct {
|
||||||
|
ctx context.Context
|
||||||
|
config *config.Config
|
||||||
|
logger *pterm.Logger
|
||||||
|
cache *cache.Cache
|
||||||
|
sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
type UtilOpts struct {
|
||||||
|
Ctx context.Context
|
||||||
|
Config *config.Config
|
||||||
|
Logger *pterm.Logger
|
||||||
|
Cache *cache.Cache
|
||||||
|
}
|
||||||
|
|
||||||
|
type utilsCtxKey uint8
|
||||||
|
|
||||||
|
const UtilsKey utilsCtxKey = iota
|
||||||
|
|
||||||
|
func AddToCtx(ctx context.Context, u *Utils) context.Context {
|
||||||
|
return context.WithValue(ctx, UtilsKey, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
func MustFromCtx(ctx context.Context) *Utils {
|
||||||
|
utils, err := FromCtx(ctx)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return utils
|
||||||
|
}
|
||||||
|
|
||||||
|
func FromCtx(ctx context.Context) (*Utils, error) {
|
||||||
|
utils, avail := ctx.Value(UtilsKey).(*Utils)
|
||||||
|
if !avail {
|
||||||
|
return nil, errors.New("invalid util in context")
|
||||||
|
} else if utils == nil {
|
||||||
|
return nil, errors.New("util never set in context")
|
||||||
|
}
|
||||||
|
|
||||||
|
return utils, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) Init(opts *UtilOpts) {
|
||||||
|
u.ctx = opts.Ctx
|
||||||
|
u.SetConfig(opts.Config)
|
||||||
|
u.SetLogger(opts.Logger)
|
||||||
|
u.SetCache(opts.Cache)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) SetCache(c *cache.Cache) {
|
||||||
|
if c == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Lock()
|
||||||
|
defer u.Unlock()
|
||||||
|
u.cache = c
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) Cache() *cache.Cache {
|
||||||
|
u.RLock()
|
||||||
|
defer u.RUnlock()
|
||||||
|
return u.cache
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) SetLogger(l *pterm.Logger) {
|
||||||
|
if l == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Lock()
|
||||||
|
defer u.Unlock()
|
||||||
|
u.logger = l
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) Logger() *pterm.Logger {
|
||||||
|
u.RLock()
|
||||||
|
defer u.RUnlock()
|
||||||
|
return u.logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) SetConfig(conf *config.Config) {
|
||||||
|
if conf == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Lock()
|
||||||
|
defer u.Unlock()
|
||||||
|
u.config = conf
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) Config() *config.Config {
|
||||||
|
u.RLock()
|
||||||
|
defer u.RUnlock()
|
||||||
|
return u.config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) SetContext(c context.Context) {
|
||||||
|
u.Lock()
|
||||||
|
defer u.Unlock()
|
||||||
|
u.ctx = c
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) Context() context.Context {
|
||||||
|
u.RLock()
|
||||||
|
defer u.RUnlock()
|
||||||
|
return u.ctx
|
||||||
|
}
|
71
cmd/util/util_completion.go
Normal file
71
cmd/util/util_completion.go
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
"golang.org/x/exp/slices"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (u *Utils) ValidProjectsFunc(cmd *cobra.Command, args []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
u.InitProjectCache(cmd, args)
|
||||||
|
return u.Cache().ProjectStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) ValidAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
u.InitProjectCache(cmd, args)
|
||||||
|
return u.Cache().AliasStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) ValidProjectsOrAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
projectStrings, _ := u.ValidAliasesFunc(cmd, args, toComplete)
|
||||||
|
aliasStrings, _ := u.ValidProjectsFunc(cmd, args, toComplete)
|
||||||
|
return append(projectStrings, aliasStrings...), cobra.ShellCompDirectiveDefault
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) ValidRemotesFunc(_ *cobra.Command, _ []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
remotes := make([]string, 0, len(u.Config().Remotes))
|
||||||
|
for _, remote := range u.Config().Remotes {
|
||||||
|
if strings.HasPrefix(remote.Host, toComplete) {
|
||||||
|
remotes = append(remotes, remote.Host)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return remotes, cobra.ShellCompDirectiveNoFileComp
|
||||||
|
}
|
||||||
|
|
||||||
|
func ValidLogLevelsFunc(_ *cobra.Command, _ []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
levels := []string{"info", "warn", "error", "debug"}
|
||||||
|
matchingLevels := make([]string, 0, len(levels))
|
||||||
|
for _, level := range levels {
|
||||||
|
if strings.HasPrefix(level, toComplete) {
|
||||||
|
matchingLevels = append(matchingLevels, level)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return matchingLevels, cobra.ShellCompDirectiveNoFileComp
|
||||||
|
}
|
||||||
|
|
||||||
|
func ValidProjectIdFunc(cmd *cobra.Command, args []string, toComplete string) (
|
||||||
|
[]string, cobra.ShellCompDirective,
|
||||||
|
) {
|
||||||
|
u := MustFromCtx(cmd.Context())
|
||||||
|
u.InitProjectCache(cmd, args)
|
||||||
|
matchingIds := make([]string, 0, len(u.Cache().Projects))
|
||||||
|
for _, p := range u.Cache().Projects {
|
||||||
|
idString := strconv.FormatInt(int64(p.ID), 10)
|
||||||
|
if strings.HasPrefix(idString, toComplete) {
|
||||||
|
matchingIds = append(matchingIds, idString)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return slices.Clip(matchingIds), cobra.ShellCompDirectiveNoFileComp
|
||||||
|
}
|
@ -1,4 +1,4 @@
|
|||||||
package cmd
|
package util
|
||||||
|
|
||||||
const (
|
const (
|
||||||
// Cobra Flags
|
// Cobra Flags
|
||||||
@ -25,62 +25,62 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defGitlabHost = "https://gitlab.com"
|
DefGitlabHost = "https://gitlab.com"
|
||||||
defLogLevel = "info"
|
DefLogLevel = "info"
|
||||||
defConfigPath = "~/.config/git-project-manager.yaml"
|
DefConfigPath = "~/.config/git-project-manager.yaml"
|
||||||
configName = "git-project-manager"
|
ConfigName = "git-project-manager"
|
||||||
legacyConfigName = "gitlab-project-manager"
|
LegacyConfigName = "gitlab-project-manager"
|
||||||
)
|
)
|
||||||
|
|
||||||
const aliasCmdLong = `Manages project aliases, with options for
|
const AliasCmdLong = `Manages project aliases, with options for
|
||||||
listing, adding, and deleting.`
|
listing, adding, and deleting.`
|
||||||
|
|
||||||
const aliasListCmdLong = `Lists all aliases by project`
|
const AliasListCmdLong = `Lists all aliases by project`
|
||||||
|
|
||||||
const aliasAddCmdLong = `Adds a project alias to a project
|
const AliasAddCmdLong = `Adds a project alias to a project
|
||||||
project ID can be provided, or will otherwise use fuzzy find`
|
project ID can be provided, or will otherwise use fuzzy find`
|
||||||
|
|
||||||
const aliasDeleteCmdLong = `Deletes aliases from projects
|
const AliasDeleteCmdLong = `Deletes aliases from projects
|
||||||
project ID can be provided, or will otherwise use fuzzy find`
|
project ID can be provided, or will otherwise use fuzzy find`
|
||||||
|
|
||||||
const cacheCmdLong = `Contains sub-commands for managing project cache.
|
const CacheCmdLong = `Contains sub-commands for managing project cache.
|
||||||
The project cache keeps this speedy, without smashing against the Git
|
The project cache keeps this speedy, without smashing against the Git
|
||||||
API every time a new project is added / searched for`
|
API every time a new project is added / searched for`
|
||||||
|
|
||||||
const rootCmdLong = `Finds Git projects using fuzzy-find, remembering
|
const RootCmdLong = `Finds Git projects using fuzzy-find, remembering
|
||||||
your chosen term for the project as an alias, and offers helpful
|
your chosen term for the project as an alias, and offers helpful
|
||||||
shortcuts for moving around in projects and opening your code`
|
shortcuts for moving around in projects and opening your code`
|
||||||
|
|
||||||
const projCmdLong = `Switches to a Git project by name or alias
|
const ProjCmdLong = `Switches to a Git project by name or alias
|
||||||
If not found, will enter fzf mode. If not cloned, will clone
|
If not found, will enter fzf mode. If not cloned, will clone
|
||||||
the project locally.`
|
the project locally.`
|
||||||
|
|
||||||
const projGoCmdLong = `Go to a project, searching by alias
|
const ProjGoCmdLong = `Go to a project, searching by alias
|
||||||
If project is not already cloned, its path will be built and it
|
If project is not already cloned, its path will be built and it
|
||||||
will be cloned from source control.
|
will be cloned from source control.
|
||||||
|
|
||||||
If conf.projects.alwaysPull, a git pull will be ran automatically`
|
If conf.projects.alwaysPull, a git pull will be ran automatically`
|
||||||
|
|
||||||
const projRunCmdLong = `Runs the current project. Tries to detect
|
const ProjRunCmdLong = `Runs the current project. Tries to detect
|
||||||
the language and runs accordingly (e.g. go run .)`
|
the language and runs accordingly (e.g. go run .)`
|
||||||
|
|
||||||
const projListCmdLong = `List locally cloned projects. Optionally
|
const ProjListCmdLong = `List locally cloned projects. Optionally
|
||||||
lists all projects in project cache`
|
lists all projects in project cache`
|
||||||
|
|
||||||
const projAddCmdLong = `Adds a new project to the local project path
|
const ProjAddCmdLong = `Adds a new project to the local project path
|
||||||
uses fuzzy find to locate the project`
|
uses fuzzy find to locate the project`
|
||||||
|
|
||||||
const projShowCmdLong = `Shows detail for a particular project
|
const ProjShowCmdLong = `Shows detail for a particular project
|
||||||
Will always fuzzy find`
|
Will always fuzzy find`
|
||||||
|
|
||||||
const projOpenCmdLong = `Opens the given project directory in the editor
|
const ProjOpenCmdLong = `Opens the given project directory in the editor
|
||||||
of your choice. Will find certain well-known entrypoints (e.g. main.go).
|
of your choice. Will find certain well-known entrypoints (e.g. main.go).
|
||||||
|
|
||||||
If your editor is set in your config file, it will be used, otherwise
|
If your editor is set in your config file, it will be used, otherwise
|
||||||
one will be found in your path from a list of known defaults.`
|
one will be found in your path from a list of known defaults.`
|
||||||
|
|
||||||
const configCmdLong = `Commands for managing configuration, particulary
|
const ConfigCmdLong = `Commands for managing configuration, particulary
|
||||||
useful for seeding a new config file`
|
useful for seeding a new config file`
|
||||||
|
|
||||||
const configGenCmdLong = `Produces yaml to stdout that can be used
|
const ConfigGenCmdLong = `Produces yaml to stdout that can be used
|
||||||
to seed the configuration file`
|
to seed the configuration file`
|
@ -1,4 +1,4 @@
|
|||||||
package cmd
|
package util
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
@ -10,7 +10,7 @@ import (
|
|||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/projects"
|
||||||
)
|
)
|
||||||
|
|
||||||
type fzfProjectOpts struct {
|
type FzfProjectOpts struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Search string
|
Search string
|
||||||
MustHaveAlias bool
|
MustHaveAlias bool
|
||||||
@ -19,14 +19,14 @@ type fzfProjectOpts struct {
|
|||||||
|
|
||||||
// This will try to find a project by alias if a search term
|
// This will try to find a project by alias if a search term
|
||||||
// is given, otherwise will fuzzy find by project
|
// is given, otherwise will fuzzy find by project
|
||||||
func fzfFindProject(opts *fzfProjectOpts) *projects.Project {
|
func (u *Utils) FzfFindProject(opts *FzfProjectOpts) *projects.Project {
|
||||||
var project *projects.Project
|
var project *projects.Project
|
||||||
|
|
||||||
if opts.Search != "" {
|
if opts.Search != "" {
|
||||||
project = fzfSearchProjectAliases(opts)
|
project = u.FzfSearchProjectAliases(opts)
|
||||||
} else {
|
} else {
|
||||||
var err error
|
var err error
|
||||||
project, err = fzfProject(opts)
|
project, err = u.FzfProject(opts)
|
||||||
if project == nil || err != nil {
|
if project == nil || err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@ -38,35 +38,35 @@ func fzfFindProject(opts *fzfProjectOpts) *projects.Project {
|
|||||||
// If . is given as a project, will open project from the
|
// If . is given as a project, will open project from the
|
||||||
// current working directory. Otherwise, will attempt to fuzzy-find
|
// current working directory. Otherwise, will attempt to fuzzy-find
|
||||||
// a project given a search term if provided
|
// a project given a search term if provided
|
||||||
func fzfCwdOrSearchProjectAliases(opts *fzfProjectOpts) *projects.Project {
|
func (u *Utils) FzfCwdOrSearchProjectAliases(opts *FzfProjectOpts) *projects.Project {
|
||||||
var project *projects.Project
|
var project *projects.Project
|
||||||
if opts.Search == "." {
|
if opts.Search == "." {
|
||||||
project, _ = projectCache.GetProjectFromCwd()
|
project, _ = u.Cache().GetProjectFromCwd()
|
||||||
} else {
|
} else {
|
||||||
project = fzfSearchProjectAliases(opts)
|
project = u.FzfSearchProjectAliases(opts)
|
||||||
}
|
}
|
||||||
return project
|
return project
|
||||||
}
|
}
|
||||||
|
|
||||||
// This will fuzzy search only aliases, preferring an exact
|
// This will fuzzy search only aliases, preferring an exact
|
||||||
// match if one is given
|
// match if one is given
|
||||||
func fzfSearchProjectAliases(opts *fzfProjectOpts) *projects.Project {
|
func (u *Utils) FzfSearchProjectAliases(opts *FzfProjectOpts) *projects.Project {
|
||||||
var project *projects.Project
|
var project *projects.Project
|
||||||
var alias *cache.ProjectAlias
|
var alias *cache.ProjectAlias
|
||||||
if alias = projectCache.GetAliasByName(opts.Search, opts.Remotes...); alias != nil {
|
if alias = u.Cache().GetAliasByName(opts.Search, opts.Remotes...); alias != nil {
|
||||||
project = projectCache.GetProjectByAlias(alias)
|
project = u.Cache().GetProjectByAlias(alias)
|
||||||
plog.Info("Perfect alias match... flawless")
|
u.Logger().Info("Perfect alias match... flawless")
|
||||||
} else {
|
} else {
|
||||||
// Get fuzzy if we don't have an exact match
|
// Get fuzzy if we don't have an exact match
|
||||||
aliases := projectCache.FuzzyFindAlias(opts.Search)
|
aliases := u.Cache().FuzzyFindAlias(opts.Search)
|
||||||
if len(aliases) > 1 {
|
if len(aliases) > 1 {
|
||||||
// If multiple aliases were found, switch over to project
|
// If multiple aliases were found, switch over to project
|
||||||
// by alias mode with merging
|
// by alias mode with merging
|
||||||
// alias = fzfAliasFromAliases(rootCmd.Context(), aliases)
|
// alias = fzfAliasFromAliases(rootCmd.Context(), aliases)
|
||||||
project, _ = fzfProjectFromAliases(opts, aliases)
|
project, _ = u.FzfProjectFromAliases(opts, aliases)
|
||||||
} else if len(aliases) == 1 {
|
} else if len(aliases) == 1 {
|
||||||
alias = aliases[0]
|
alias = aliases[0]
|
||||||
project = projectCache.GetProjectByAlias(alias)
|
project = u.Cache().GetProjectByAlias(alias)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return project
|
return project
|
||||||
@ -76,18 +76,18 @@ func fzfSearchProjectAliases(opts *fzfProjectOpts) *projects.Project {
|
|||||||
// a single one. Replaced by fzfProjectFromAliases in fzfSearchProjectAliases
|
// a single one. Replaced by fzfProjectFromAliases in fzfSearchProjectAliases
|
||||||
// as merging is preferred, but can be used if it's ever desirable to
|
// as merging is preferred, but can be used if it's ever desirable to
|
||||||
// return a single alias from all aliases
|
// return a single alias from all aliases
|
||||||
func fzfAliasFromAliases(opts *fzfProjectOpts, aliases []*cache.ProjectAlias) *cache.ProjectAlias {
|
func (u *Utils) FzfAliasFromAliases(opts *FzfProjectOpts, aliases []*cache.ProjectAlias) *cache.ProjectAlias {
|
||||||
var alias *cache.ProjectAlias
|
var alias *cache.ProjectAlias
|
||||||
i, err := fzf.Find(
|
i, err := fzf.Find(
|
||||||
aliases,
|
aliases,
|
||||||
func(i int) string {
|
func(i int) string {
|
||||||
return aliases[i].Alias + " -> " + projectCache.GetProjectByAlias(aliases[i]).PathWithNamespace
|
return aliases[i].Alias + " -> " + u.Cache().GetProjectByAlias(aliases[i]).PathWithNamespace
|
||||||
},
|
},
|
||||||
fzf.WithContext(opts.Ctx),
|
fzf.WithContext(opts.Ctx),
|
||||||
fzf.WithHeader("Choose an Alias"),
|
fzf.WithHeader("Choose an Alias"),
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
plog.Error("Failed to fzf alias slice", plog.Args("error", err))
|
u.Logger().Error("Failed to fzf alias slice", u.Logger().Args("error", err))
|
||||||
} else {
|
} else {
|
||||||
alias = aliases[i]
|
alias = aliases[i]
|
||||||
}
|
}
|
||||||
@ -96,21 +96,21 @@ func fzfAliasFromAliases(opts *fzfProjectOpts, aliases []*cache.ProjectAlias) *c
|
|||||||
|
|
||||||
// Given a list of aliases, merge them together and use the resulting
|
// Given a list of aliases, merge them together and use the resulting
|
||||||
// list of projects to return a project
|
// list of projects to return a project
|
||||||
func fzfProjectFromAliases(opts *fzfProjectOpts, aliases []*cache.ProjectAlias) (
|
func (u *Utils) FzfProjectFromAliases(opts *FzfProjectOpts, aliases []*cache.ProjectAlias) (
|
||||||
*projects.Project, error,
|
*projects.Project, error,
|
||||||
) {
|
) {
|
||||||
mergedProjects := projectsFromAliases(aliases)
|
mergedProjects := u.projectsFromAliases(aliases)
|
||||||
if len(mergedProjects) == 1 {
|
if len(mergedProjects) == 1 {
|
||||||
return mergedProjects[0], nil
|
return mergedProjects[0], nil
|
||||||
}
|
}
|
||||||
return fzfProjectFromProjects(opts, mergedProjects)
|
return u.FzfProjectFromProjects(opts, mergedProjects)
|
||||||
}
|
}
|
||||||
|
|
||||||
func projectsFromAliases(aliases []*cache.ProjectAlias) []*projects.Project {
|
func (u *Utils) projectsFromAliases(aliases []*cache.ProjectAlias) []*projects.Project {
|
||||||
projects := make([]*projects.Project, 0, len(aliases))
|
projects := make([]*projects.Project, 0, len(aliases))
|
||||||
|
|
||||||
for _, a := range aliases {
|
for _, a := range aliases {
|
||||||
project := projectCache.GetProjectByAlias(a)
|
project := u.Cache().GetProjectByAlias(a)
|
||||||
if project != nil && !slices.Contains(projects, project) {
|
if project != nil && !slices.Contains(projects, project) {
|
||||||
projects = append(projects, project)
|
projects = append(projects, project)
|
||||||
}
|
}
|
||||||
@ -121,30 +121,30 @@ func projectsFromAliases(aliases []*cache.ProjectAlias) []*projects.Project {
|
|||||||
|
|
||||||
// If opts.MustHaveAlias, will only allow selection of projects
|
// If opts.MustHaveAlias, will only allow selection of projects
|
||||||
// that have at least one alias defined
|
// that have at least one alias defined
|
||||||
func fzfProject(opts *fzfProjectOpts) (*projects.Project, error) {
|
func (u *Utils) FzfProject(opts *FzfProjectOpts) (*projects.Project, error) {
|
||||||
var searchableProjects []*projects.Project
|
var searchableProjects []*projects.Project
|
||||||
if opts.MustHaveAlias {
|
if opts.MustHaveAlias {
|
||||||
searchableProjects = projectCache.GetProjectsWithAliases()
|
searchableProjects = u.Cache().GetProjectsWithAliases()
|
||||||
} else {
|
} else {
|
||||||
searchableProjects = projectCache.Projects
|
searchableProjects = u.Cache().Projects
|
||||||
}
|
}
|
||||||
// Filter out unwanted remotes if provided
|
// Filter out unwanted remotes if provided
|
||||||
searchableProjects = filterProjectsWithRemotes(searchableProjects, opts.Remotes...)
|
searchableProjects = u.FilterProjectsWithRemotes(searchableProjects, opts.Remotes...)
|
||||||
return fzfProjectFromProjects(opts, searchableProjects)
|
return u.FzfProjectFromProjects(opts, searchableProjects)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Takes a list of projects and performs a fuzzyfind
|
// Takes a list of projects and performs a fuzzyfind
|
||||||
func fzfProjectFromProjects(opts *fzfProjectOpts, projects []*projects.Project) (
|
func (u *Utils) FzfProjectFromProjects(opts *FzfProjectOpts, projects []*projects.Project) (
|
||||||
*projects.Project, error,
|
*projects.Project, error,
|
||||||
) {
|
) {
|
||||||
i, err := fzf.Find(projects,
|
i, err := fzf.Find(projects,
|
||||||
func(i int) string {
|
func(i int) string {
|
||||||
// Display the project along with its aliases
|
// Display the project along with its aliases
|
||||||
return projectCache.GetProjectStringWithAliases(projects[i])
|
return u.Cache().GetProjectStringWithAliases(projects[i])
|
||||||
},
|
},
|
||||||
fzf.WithPreviewWindow(
|
fzf.WithPreviewWindow(
|
||||||
func(i, width, height int) string {
|
func(i, width, height int) string {
|
||||||
return projectCache.ProjectString(projects[i])
|
return u.Cache().ProjectString(projects[i])
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
fzf.WithContext(opts.Ctx),
|
fzf.WithContext(opts.Ctx),
|
||||||
@ -156,12 +156,12 @@ func fzfProjectFromProjects(opts *fzfProjectOpts, projects []*projects.Project)
|
|||||||
return projects[i], nil
|
return projects[i], nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func fzfPreviewWindow(i, _, _ int) string {
|
func (u *Utils) FzfPreviewWindow(i, _, _ int) string {
|
||||||
p := projectCache.Projects[i]
|
p := u.Cache().Projects[i]
|
||||||
return projectCache.ProjectString(p)
|
return u.Cache().ProjectString(p)
|
||||||
}
|
}
|
||||||
|
|
||||||
func filterProjectsWithRemotes(gitProjects []*projects.Project, remotes ...string) []*projects.Project {
|
func (u *Utils) FilterProjectsWithRemotes(gitProjects []*projects.Project, remotes ...string) []*projects.Project {
|
||||||
filteredProjects := make([]*projects.Project, 0, len(gitProjects))
|
filteredProjects := make([]*projects.Project, 0, len(gitProjects))
|
||||||
if len(remotes) > 0 {
|
if len(remotes) > 0 {
|
||||||
for _, p := range gitProjects {
|
for _, p := range gitProjects {
|
||||||
@ -177,7 +177,7 @@ func filterProjectsWithRemotes(gitProjects []*projects.Project, remotes ...strin
|
|||||||
|
|
||||||
// Nearly useless function that simply returns either an
|
// Nearly useless function that simply returns either an
|
||||||
// empty string, or a string from the first arg if one is provided
|
// empty string, or a string from the first arg if one is provided
|
||||||
func searchStringFromArgs(args []string) string {
|
func (u *Utils) SearchStringFromArgs(args []string) string {
|
||||||
var term string
|
var term string
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
term = args[0]
|
term = args[0]
|
163
cmd/util/util_init.go
Normal file
163
cmd/util/util_init.go
Normal file
@ -0,0 +1,163 @@
|
|||||||
|
// This file contains init methods that may be used by
|
||||||
|
// multiple sub-commands. For instance, the cach and projects
|
||||||
|
// sub-commands both depend on a cache and may both call the initProjectCache
|
||||||
|
// func (u *Util) from their PersistentPreRun commands
|
||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"os/user"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/pterm/pterm"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
"golang.org/x/sys/unix"
|
||||||
|
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/cache"
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/config"
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes"
|
||||||
|
gitearemote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/gitea"
|
||||||
|
githubremote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/github"
|
||||||
|
gitlabremote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/gitlab"
|
||||||
|
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/remote"
|
||||||
|
)
|
||||||
|
|
||||||
|
func InitProjects(cmd *cobra.Command, args []string) {
|
||||||
|
utils, _ := FromCtx(cmd.Context())
|
||||||
|
utils.InitProjectCache(cmd, args)
|
||||||
|
utils.mustHaveProjects(cmd, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) mustHaveProjects(cmd *cobra.Command, _ []string) {
|
||||||
|
if len(u.Cache().Projects) == 0 {
|
||||||
|
u.Logger().Fatal("No projects to " + cmd.Name() + ", try running cache load")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) InitProjectCache(cmd *cobra.Command, _ []string) {
|
||||||
|
var err error
|
||||||
|
u.Logger().Debug("Running pre-run for cacheCmd")
|
||||||
|
u.Config().Cache.File = u.Config().ProjectPath + "/.cache.yaml"
|
||||||
|
|
||||||
|
gitRemotes := remotes.NewRemotes()
|
||||||
|
gitRemotes.AddRemotes(*u.GetRemotes(cmd)...)
|
||||||
|
|
||||||
|
cacheOpts := &cache.CacheOpts{
|
||||||
|
ProjectsPath: u.Config().ProjectPath,
|
||||||
|
Path: u.Config().Cache.File,
|
||||||
|
TTL: u.Config().Cache.Ttl,
|
||||||
|
Logger: u.Logger(),
|
||||||
|
Remotes: gitRemotes,
|
||||||
|
Config: u.Config(),
|
||||||
|
}
|
||||||
|
|
||||||
|
projectCache, err := cache.NewProjectCache(cacheOpts)
|
||||||
|
if err != nil {
|
||||||
|
u.Logger().Error("Failed to prepare project cache", u.Logger().Args("error", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
u.SetCache(projectCache)
|
||||||
|
|
||||||
|
if err := u.Cache().Read(); err != nil {
|
||||||
|
u.Logger().Error("Cache load failed", u.Logger().Args("error", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Logger().Debug("Remotes Loaded", u.Logger().Args("remotes", cacheOpts.Remotes))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generically loads remotes from info.RemoteInfo in config.Remotes
|
||||||
|
func (u *Utils) GetRemotes(cmd *cobra.Command) *remotes.Remotes {
|
||||||
|
gitRemotes := new(remotes.Remotes)
|
||||||
|
*gitRemotes = make([]remote.Remote, 0)
|
||||||
|
for _, r := range u.Config().Remotes {
|
||||||
|
// Create a copy, set context
|
||||||
|
gitRemoteInfo := r
|
||||||
|
gitRemoteInfo.SetContext(cmd.Context())
|
||||||
|
var gitRemote remote.Remote
|
||||||
|
var err error
|
||||||
|
switch r.Type {
|
||||||
|
case "gitlab":
|
||||||
|
gitRemote, err = gitlabremote.NewGitlabRemote(&gitRemoteInfo)
|
||||||
|
case "gitea":
|
||||||
|
gitRemote, err = gitearemote.NewGiteaRemote(&gitRemoteInfo)
|
||||||
|
case "github":
|
||||||
|
gitRemote, err = githubremote.NewGithubRemote(&gitRemoteInfo)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
u.Logger().Error("Failed to prepare remote", u.Logger().Args(
|
||||||
|
"error", err,
|
||||||
|
"type", r.Type))
|
||||||
|
} else {
|
||||||
|
*gitRemotes = append(*gitRemotes, gitRemote)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return gitRemotes
|
||||||
|
}
|
||||||
|
|
||||||
|
func PostProjectCmd(cmd *cobra.Command, args []string) {
|
||||||
|
utils, _ := FromCtx(cmd.Context())
|
||||||
|
utils.PostProjectCache(cmd, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) PostProjectCache(_ *cobra.Command, _ []string) {
|
||||||
|
u.Cache().Write()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *Utils) InitProjectPath(_ *cobra.Command, _ []string) {
|
||||||
|
u.Logger().Debug("Running persistent pre-run for rootCmd")
|
||||||
|
var err error
|
||||||
|
if u.Config().ProjectPath == "" {
|
||||||
|
u.Config().ProjectPath = config.DefaultConfig.ProjectPath
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if u.Config().ProjectPath, err = ResolvePath(u.Config().ProjectPath); err != nil {
|
||||||
|
u.Logger().Error("Failed to determine project path", u.Logger().Args("error", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
_, err = os.Stat(u.Config().ProjectPath)
|
||||||
|
if err != nil {
|
||||||
|
u.Logger().Error("Failed to stat project path, trying to create", u.Logger().Args("error", err))
|
||||||
|
if err = os.MkdirAll(u.Config().ProjectPath, 0o750); err != nil {
|
||||||
|
u.Logger().Error("Failed to create project path", u.Logger().Args("error", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
u.Logger().Info("Project path created", u.Logger().Args("path", u.Config().ProjectPath))
|
||||||
|
} else {
|
||||||
|
if err = unix.Access(u.Config().ProjectPath, unix.W_OK); err != nil {
|
||||||
|
u.Logger().Error("Unable to write to project path", u.Logger().Args(
|
||||||
|
"path", u.Config().ProjectPath,
|
||||||
|
"error", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ResolvePath(path string) (string, error) {
|
||||||
|
if strings.HasPrefix(path, "~/") {
|
||||||
|
usr, _ := user.Current()
|
||||||
|
path = filepath.Join(usr.HomeDir, path[2:])
|
||||||
|
}
|
||||||
|
return filepath.Abs(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetConfigName(configPath string) string {
|
||||||
|
// Check existing config
|
||||||
|
for _, ext := range []string{"yml", "yaml"} {
|
||||||
|
configFile := fmt.Sprintf("%s/%s.%s", configPath, ConfigName, ext)
|
||||||
|
legacyConfigFile := fmt.Sprintf("%s/%s.%s", configPath, LegacyConfigName, ext)
|
||||||
|
|
||||||
|
if _, err := os.Stat(configFile); err == nil {
|
||||||
|
return ConfigName
|
||||||
|
} else if _, err := os.Stat(legacyConfigFile); err == nil {
|
||||||
|
pterm.DefaultLogger.WithWriter(os.Stderr).
|
||||||
|
Warn(fmt.Sprintf("using legacy config path, suggest using %s/%s.yaml", configPath, ConfigName))
|
||||||
|
return LegacyConfigName
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Nothing found, do what we want
|
||||||
|
return ConfigName
|
||||||
|
}
|
@ -1,64 +0,0 @@
|
|||||||
package cmd
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
"golang.org/x/exp/slices"
|
|
||||||
)
|
|
||||||
|
|
||||||
func validProjectsFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
initProjectCache(cmd, args)
|
|
||||||
return projectCache.ProjectStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
|
|
||||||
}
|
|
||||||
|
|
||||||
func validAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
initProjectCache(cmd, args)
|
|
||||||
return projectCache.AliasStrings(toComplete), cobra.ShellCompDirectiveNoFileComp
|
|
||||||
}
|
|
||||||
|
|
||||||
func validProjectsOrAliasesFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
projectStrings, _ := validAliasesFunc(cmd, args, toComplete)
|
|
||||||
aliasStrings, _ := validProjectsFunc(cmd, args, toComplete)
|
|
||||||
return append(projectStrings, aliasStrings...), cobra.ShellCompDirectiveDefault
|
|
||||||
}
|
|
||||||
|
|
||||||
func validRemotesFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
remotes := make([]string, 0, len(conf.Remotes))
|
|
||||||
for _, remote := range conf.Remotes {
|
|
||||||
if strings.HasPrefix(remote.Host, toComplete) {
|
|
||||||
remotes = append(remotes, remote.Host)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return remotes, cobra.ShellCompDirectiveNoFileComp
|
|
||||||
}
|
|
||||||
|
|
||||||
func validLogLevelsFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
levels := []string{"info", "warn", "error", "debug"}
|
|
||||||
matchingLevels := make([]string, 0, len(levels))
|
|
||||||
for _, level := range levels {
|
|
||||||
if strings.HasPrefix(level, toComplete) {
|
|
||||||
matchingLevels = append(matchingLevels, level)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return matchingLevels, cobra.ShellCompDirectiveNoFileComp
|
|
||||||
}
|
|
||||||
|
|
||||||
func validProjectIdFunc(cmd *cobra.Command, args []string, toComplete string) (
|
|
||||||
[]string, cobra.ShellCompDirective) {
|
|
||||||
initProjectCache(cmd, args)
|
|
||||||
matchingIds := make([]string, 0, len(projectCache.Projects))
|
|
||||||
for _, p := range projectCache.Projects {
|
|
||||||
idString := strconv.FormatInt(int64(p.ID), 10)
|
|
||||||
if strings.HasPrefix(idString, toComplete) {
|
|
||||||
matchingIds = append(matchingIds, idString)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return slices.Clip(matchingIds), cobra.ShellCompDirectiveNoFileComp
|
|
||||||
}
|
|
143
cmd/util_init.go
143
cmd/util_init.go
@ -1,143 +0,0 @@
|
|||||||
// This file contains init methods that may be used by
|
|
||||||
// multiple sub-commands. For instance, the cach and projects
|
|
||||||
// sub-commands both depend on a cache and may both call the initProjectCache
|
|
||||||
// func from their PersistentPreRun commands
|
|
||||||
package cmd
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"os/user"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/pterm/pterm"
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
"golang.org/x/sys/unix"
|
|
||||||
|
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/cache"
|
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/config"
|
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes"
|
|
||||||
gitearemote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/gitea"
|
|
||||||
githubremote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/github"
|
|
||||||
gitlabremote "gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/gitlab"
|
|
||||||
"gitea.libretechconsulting.com/rmcguire/git-project-manager/internal/remotes/remote"
|
|
||||||
)
|
|
||||||
|
|
||||||
func initProjectCache(cmd *cobra.Command, _ []string) {
|
|
||||||
var err error
|
|
||||||
plog.Debug("Running pre-run for cacheCmd")
|
|
||||||
conf.Cache.File = conf.ProjectPath + "/.cache.yaml"
|
|
||||||
|
|
||||||
gitRemotes := remotes.NewRemotes()
|
|
||||||
gitRemotes.AddRemotes(*getRemotes(cmd)...)
|
|
||||||
|
|
||||||
cacheOpts := &cache.CacheOpts{
|
|
||||||
ProjectsPath: conf.ProjectPath,
|
|
||||||
Path: conf.Cache.File,
|
|
||||||
TTL: conf.Cache.Ttl,
|
|
||||||
Logger: plog,
|
|
||||||
Remotes: gitRemotes,
|
|
||||||
Config: &conf,
|
|
||||||
}
|
|
||||||
if projectCache, err = cache.NewProjectCache(cacheOpts); err != nil {
|
|
||||||
plog.Error("Failed to prepare project cache", plog.Args("error", err))
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := projectCache.Read(); err != nil {
|
|
||||||
plog.Error("Cache load failed", plog.Args("error", err))
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
plog.Debug("Remotes Loaded", plog.Args("remotes", cacheOpts.Remotes))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generically loads remotes from info.RemoteInfo in config.Remotes
|
|
||||||
func getRemotes(cmd *cobra.Command) *remotes.Remotes {
|
|
||||||
gitRemotes := new(remotes.Remotes)
|
|
||||||
*gitRemotes = make([]remote.Remote, 0)
|
|
||||||
for _, r := range conf.Remotes {
|
|
||||||
// Create a copy, set context
|
|
||||||
gitRemoteInfo := r
|
|
||||||
gitRemoteInfo.SetContext(cmd.Context())
|
|
||||||
var gitRemote remote.Remote
|
|
||||||
var err error
|
|
||||||
switch r.Type {
|
|
||||||
case "gitlab":
|
|
||||||
gitRemote, err = gitlabremote.NewGitlabRemote(&gitRemoteInfo)
|
|
||||||
case "gitea":
|
|
||||||
gitRemote, err = gitearemote.NewGiteaRemote(&gitRemoteInfo)
|
|
||||||
case "github":
|
|
||||||
gitRemote, err = githubremote.NewGithubRemote(&gitRemoteInfo)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
plog.Error("Failed to prepare remote", plog.Args(
|
|
||||||
"error", err,
|
|
||||||
"type", r.Type))
|
|
||||||
} else {
|
|
||||||
*gitRemotes = append(*gitRemotes, gitRemote)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return gitRemotes
|
|
||||||
}
|
|
||||||
|
|
||||||
func postProjectCache(_ *cobra.Command, _ []string) {
|
|
||||||
projectCache.Write()
|
|
||||||
}
|
|
||||||
|
|
||||||
func initProjectPath(_ *cobra.Command, _ []string) {
|
|
||||||
plog.Debug("Running persistent pre-run for rootCmd")
|
|
||||||
var err error
|
|
||||||
if conf.ProjectPath == "" {
|
|
||||||
conf.ProjectPath = config.DefaultConfig.ProjectPath
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if conf.ProjectPath, err = resolvePath(conf.ProjectPath); err != nil {
|
|
||||||
plog.Error("Failed to determine project path", plog.Args("error", err))
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
_, err = os.Stat(conf.ProjectPath)
|
|
||||||
if err != nil {
|
|
||||||
plog.Error("Failed to stat project path, trying to create", plog.Args("error", err))
|
|
||||||
if err = os.MkdirAll(conf.ProjectPath, 0o750); err != nil {
|
|
||||||
plog.Error("Failed to create project path", plog.Args("error", err))
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
plog.Info("Project path created", plog.Args("path", conf.ProjectPath))
|
|
||||||
} else {
|
|
||||||
if err = unix.Access(conf.ProjectPath, unix.W_OK); err != nil {
|
|
||||||
plog.Error("Unable to write to project path", plog.Args(
|
|
||||||
"path", conf.ProjectPath,
|
|
||||||
"error", err))
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func resolvePath(path string) (string, error) {
|
|
||||||
if strings.HasPrefix(path, "~/") {
|
|
||||||
usr, _ := user.Current()
|
|
||||||
path = filepath.Join(usr.HomeDir, path[2:])
|
|
||||||
}
|
|
||||||
return filepath.Abs(path)
|
|
||||||
}
|
|
||||||
|
|
||||||
func getConfigName(configPath string) string {
|
|
||||||
// Check existing config
|
|
||||||
for _, ext := range []string{"yml", "yaml"} {
|
|
||||||
configFile := fmt.Sprintf("%s/%s.%s", configPath, configName, ext)
|
|
||||||
legacyConfigFile := fmt.Sprintf("%s/%s.%s", configPath, legacyConfigName, ext)
|
|
||||||
|
|
||||||
if _, err := os.Stat(configFile); err == nil {
|
|
||||||
return configName
|
|
||||||
} else if _, err := os.Stat(legacyConfigFile); err == nil {
|
|
||||||
pterm.DefaultLogger.WithWriter(os.Stderr).
|
|
||||||
Warn(fmt.Sprintf("using legacy config path, suggest using %s/%s.yaml", configPath, configName))
|
|
||||||
return legacyConfigName
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Nothing found, do what we want
|
|
||||||
return configName
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user