forked from argoproj/argo-cd
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: nitishfy <[email protected]>
- Loading branch information
Showing
3 changed files
with
102 additions
and
93 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,94 @@ | ||
| package util | ||
|
|
||
| import ( | ||
| pluginError "errors" | ||
| "fmt" | ||
| "os" | ||
| "os/exec" | ||
| "path/filepath" | ||
| "runtime" | ||
| "syscall" | ||
| ) | ||
|
|
||
| type ArgoCDCLIOptions struct { | ||
| PluginHandler PluginHandler | ||
| Arguments []string | ||
| } | ||
|
|
||
| // PluginHandler parses command line arguments | ||
| // and performs executable filename lookups to search | ||
| // for valid plugin files, and execute found plugins. | ||
| type PluginHandler interface { | ||
| // LookForPlugin will iterate over a list of given prefixes | ||
| // in order to recognize valid plugin filenames. | ||
| // The first filepath to match a prefix is returned. | ||
| LookForPlugin(filename string) (string, bool) | ||
| // ExecutePlugin receives an executable's filepath, a slice | ||
| // of arguments, and a slice of environment variables | ||
| // to relay to the executable. | ||
| ExecutePlugin(executablePath string, cmdArgs, environment []string) error | ||
| } | ||
|
|
||
| // DefaultPluginHandler implements the PluginHandler interface | ||
| type DefaultPluginHandler struct { | ||
| ValidPrefixes []string | ||
| } | ||
|
|
||
| func NewDefaultPluginHandler(validPrefixes []string) *DefaultPluginHandler { | ||
| return &DefaultPluginHandler{ | ||
| ValidPrefixes: validPrefixes, | ||
| } | ||
| } | ||
|
|
||
| // LookForPlugin implements PluginHandler | ||
| func (h *DefaultPluginHandler) LookForPlugin(filename string) (string, bool) { | ||
| for _, prefix := range h.ValidPrefixes { | ||
| path, err := exec.LookPath(fmt.Sprintf("%s-%s", prefix, filename)) | ||
| if shouldSkipOnLookPathErr(err) || len(path) == 0 { | ||
| continue | ||
| } | ||
| return path, true | ||
| } | ||
| return "", false | ||
| } | ||
|
|
||
| // ExecutePlugin implements PluginHandler | ||
| func (h *DefaultPluginHandler) ExecutePlugin(executablePath string, cmdArgs, environment []string) error { | ||
| // Windows does not support exec syscall. | ||
| if runtime.GOOS == "windows" { | ||
| cmd := Command(executablePath, cmdArgs...) | ||
| cmd.Stdout = os.Stdout | ||
| cmd.Stderr = os.Stderr | ||
| cmd.Stdin = os.Stdin | ||
| cmd.Env = environment | ||
| err := cmd.Run() | ||
| if err == nil { | ||
| os.Exit(0) | ||
| } | ||
| return err | ||
| } | ||
|
|
||
| return syscall.Exec(executablePath, append([]string{executablePath}, cmdArgs...), environment) | ||
| } | ||
|
|
||
| func Command(name string, arg ...string) *exec.Cmd { | ||
| cmd := &exec.Cmd{ | ||
| Path: name, | ||
| Args: append([]string{name}, arg...), | ||
| } | ||
| if filepath.Base(name) == name { | ||
| lp, err := exec.LookPath(name) | ||
| if lp != "" && !shouldSkipOnLookPathErr(err) { | ||
| // Update cmd.Path even if err is non-nil. | ||
| // If err is ErrDot (especially on Windows), lp may include a resolved | ||
| // extension (like .exe or .bat) that should be preserved. | ||
| cmd.Path = lp | ||
| } | ||
| } | ||
| return cmd | ||
| } | ||
|
|
||
| // shouldSkipOnLookPathErr checks if the error is nil and it is of type ErrDot | ||
| func shouldSkipOnLookPathErr(err error) bool { | ||
| return err != nil && !pluginError.Is(err, exec.ErrDot) | ||
| } |