posh/cmd/execute.go

package cmd

import (
	intcmd "github.com/foomo/posh/internal/cmd"
	intconfig "github.com/foomo/posh/internal/config"
	"github.com/foomo/posh/pkg/plugin"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
)

// NewExecute represents the exec command
func NewExecute(root *cobra.Command) {
	cmd := &cobra.Command{
		Use:                "execute",
		Short:              "Execute a single Project Oriented Shell command",
		Aliases:            []string{"x"},
		Args:               cobra.ArbitraryArgs,
		DisableFlagParsing: true,
		SilenceUsage:       true,
		SilenceErrors:      true,
		PreRunE: func(cmd *cobra.Command, args []string) error {
			l := intcmd.NewLogger()
			if err := intconfig.Load(l); err != nil {
				return err
			}

			return nil
		},
		RunE: func(cmd *cobra.Command, args []string) error {
			l := intcmd.NewLogger()

			if len(args) == 0 {
				return errors.New("missing [cmd] argument")
			}

			plg, err := pluginProvider(l)
			if err != nil {
				return err
			}

			return plg.Execute(cmd.Context(), args)
		},
		ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
			l := intcmd.NewLogger()
			if err := intconfig.Load(l); err != nil {
				return nil, cobra.ShellCompDirectiveError
			}

			plg, err := pluginProvider(l)
			if err != nil {
				return nil, cobra.ShellCompDirectiveError
			}

			completer, ok := plg.(plugin.Completer)
			if !ok {
				return nil, cobra.ShellCompDirectiveNoFileComp
			}

			return completer.Complete(cmd.Context(), args, toComplete), cobra.ShellCompDirectiveNoFileComp
		},
	}

	root.AddCommand(cmd)
}