active_help.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // Copyright 2013-2023 The Cobra Authors
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package cobra
  15. import (
  16. "fmt"
  17. "os"
  18. )
  19. const (
  20. activeHelpMarker = "_activeHelp_ "
  21. // The below values should not be changed: programs will be using them explicitly
  22. // in their user documentation, and users will be using them explicitly.
  23. activeHelpEnvVarSuffix = "ACTIVE_HELP"
  24. activeHelpGlobalEnvVar = configEnvVarGlobalPrefix + "_" + activeHelpEnvVarSuffix
  25. activeHelpGlobalDisable = "0"
  26. )
  27. // AppendActiveHelp adds the specified string to the specified array to be used as ActiveHelp.
  28. // Such strings will be processed by the completion script and will be shown as ActiveHelp
  29. // to the user.
  30. // The array parameter should be the array that will contain the completions.
  31. // This function can be called multiple times before and/or after completions are added to
  32. // the array. Each time this function is called with the same array, the new
  33. // ActiveHelp line will be shown below the previous ones when completion is triggered.
  34. func AppendActiveHelp(compArray []string, activeHelpStr string) []string {
  35. return append(compArray, fmt.Sprintf("%s%s", activeHelpMarker, activeHelpStr))
  36. }
  37. // GetActiveHelpConfig returns the value of the ActiveHelp environment variable
  38. // <PROGRAM>_ACTIVE_HELP where <PROGRAM> is the name of the root command in upper
  39. // case, with all non-ASCII-alphanumeric characters replaced by `_`.
  40. // It will always return "0" if the global environment variable COBRA_ACTIVE_HELP
  41. // is set to "0".
  42. func GetActiveHelpConfig(cmd *Command) string {
  43. activeHelpCfg := os.Getenv(activeHelpGlobalEnvVar)
  44. if activeHelpCfg != activeHelpGlobalDisable {
  45. activeHelpCfg = os.Getenv(activeHelpEnvVar(cmd.Root().Name()))
  46. }
  47. return activeHelpCfg
  48. }
  49. // activeHelpEnvVar returns the name of the program-specific ActiveHelp environment
  50. // variable. It has the format <PROGRAM>_ACTIVE_HELP where <PROGRAM> is the name of the
  51. // root command in upper case, with all non-ASCII-alphanumeric characters replaced by `_`.
  52. func activeHelpEnvVar(name string) string {
  53. return configEnvVar(name, activeHelpEnvVarSuffix)
  54. }