main.go 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. package main
  2. import (
  3. "fmt"
  4. "os"
  5. "path"
  6. "path/filepath"
  7. "strings"
  8. "github.com/confluentinc/bincover"
  9. "github.com/fatih/color"
  10. cc "github.com/ivanpirog/coloredcobra"
  11. log "github.com/sirupsen/logrus"
  12. "github.com/spf13/cobra"
  13. "github.com/spf13/cobra/doc"
  14. "github.com/crowdsecurity/crowdsec/pkg/csconfig"
  15. "github.com/crowdsecurity/crowdsec/pkg/cwhub"
  16. "github.com/crowdsecurity/crowdsec/pkg/cwversion"
  17. "github.com/crowdsecurity/crowdsec/pkg/database"
  18. )
  19. var bincoverTesting = ""
  20. var trace_lvl, dbg_lvl, nfo_lvl, wrn_lvl, err_lvl bool
  21. var ConfigFilePath string
  22. var csConfig *csconfig.Config
  23. var dbClient *database.Client
  24. var OutputFormat string
  25. var OutputColor string
  26. var downloadOnly bool
  27. var forceAction bool
  28. var purge bool
  29. var all bool
  30. var restoreOldBackup bool
  31. var prometheusURL string
  32. func initConfig() {
  33. var err error
  34. if trace_lvl {
  35. log.SetLevel(log.TraceLevel)
  36. } else if dbg_lvl {
  37. log.SetLevel(log.DebugLevel)
  38. } else if nfo_lvl {
  39. log.SetLevel(log.InfoLevel)
  40. } else if wrn_lvl {
  41. log.SetLevel(log.WarnLevel)
  42. } else if err_lvl {
  43. log.SetLevel(log.ErrorLevel)
  44. }
  45. logFormatter := &log.TextFormatter{TimestampFormat: "02-01-2006 15:04:05", FullTimestamp: true}
  46. log.SetFormatter(logFormatter)
  47. if !inSlice(os.Args[1], NoNeedConfig) {
  48. csConfig, err = csconfig.NewConfig(ConfigFilePath, false, false)
  49. if err != nil {
  50. log.Fatal(err)
  51. }
  52. log.Debugf("Using %s as configuration file", ConfigFilePath)
  53. if err := csConfig.LoadCSCLI(); err != nil {
  54. log.Fatal(err)
  55. }
  56. } else {
  57. csConfig = csconfig.NewDefaultConfig()
  58. }
  59. if csConfig.Cscli == nil {
  60. log.Fatalf("missing 'cscli' configuration in '%s', exiting", ConfigFilePath)
  61. }
  62. if cwhub.HubBranch == "" && csConfig.Cscli.HubBranch != "" {
  63. cwhub.HubBranch = csConfig.Cscli.HubBranch
  64. }
  65. if OutputFormat != "" {
  66. csConfig.Cscli.Output = OutputFormat
  67. if OutputFormat != "json" && OutputFormat != "raw" && OutputFormat != "human" {
  68. log.Fatalf("output format %s unknown", OutputFormat)
  69. }
  70. }
  71. if csConfig.Cscli.Output == "" {
  72. csConfig.Cscli.Output = "human"
  73. }
  74. if csConfig.Cscli.Output == "json" {
  75. log.SetFormatter(&log.JSONFormatter{})
  76. log.SetLevel(log.ErrorLevel)
  77. } else if csConfig.Cscli.Output == "raw" {
  78. log.SetLevel(log.ErrorLevel)
  79. }
  80. if OutputColor != "" {
  81. csConfig.Cscli.Color = OutputColor
  82. if OutputColor != "yes" && OutputColor != "no" && OutputColor != "auto" {
  83. log.Fatalf("output color %s unknown", OutputColor)
  84. }
  85. }
  86. }
  87. var validArgs = []string{
  88. "scenarios", "parsers", "collections", "capi", "lapi", "postoverflows", "machines",
  89. "metrics", "bouncers", "alerts", "decisions", "simulation", "hub", "dashboard",
  90. "config", "completion", "version", "console", "notifications", "support",
  91. }
  92. func prepender(filename string) string {
  93. const header = `---
  94. id: %s
  95. title: %s
  96. ---
  97. `
  98. name := filepath.Base(filename)
  99. base := strings.TrimSuffix(name, path.Ext(name))
  100. return fmt.Sprintf(header, base, strings.ReplaceAll(base, "_", " "))
  101. }
  102. func linkHandler(name string) string {
  103. return fmt.Sprintf("/cscli/%s", name)
  104. }
  105. var (
  106. NoNeedConfig = []string{
  107. "help",
  108. "completion",
  109. "version",
  110. "hubtest",
  111. }
  112. )
  113. func main() {
  114. var rootCmd = &cobra.Command{
  115. Use: "cscli",
  116. Short: "cscli allows you to manage crowdsec",
  117. Long: `cscli is the main command to interact with your crowdsec service, scenarios & db.
  118. It is meant to allow you to manage bans, parsers/scenarios/etc, api and generally manage you crowdsec setup.`,
  119. ValidArgs: validArgs,
  120. DisableAutoGenTag: true,
  121. SilenceErrors: true,
  122. SilenceUsage: true,
  123. /*TBD examples*/
  124. }
  125. cc.Init(&cc.Config{
  126. RootCmd: rootCmd,
  127. Headings: cc.Yellow,
  128. Commands: cc.Green + cc.Bold,
  129. CmdShortDescr: cc.Cyan,
  130. Example: cc.Italic,
  131. ExecName: cc.Bold,
  132. Aliases: cc.Bold + cc.Italic,
  133. FlagsDataType: cc.White,
  134. Flags: cc.Green,
  135. FlagsDescr: cc.Cyan,
  136. })
  137. rootCmd.SetOut(color.Output)
  138. var cmdDocGen = &cobra.Command{
  139. Use: "doc",
  140. Short: "Generate the documentation in `./doc/`. Directory must exist.",
  141. Args: cobra.ExactArgs(0),
  142. Hidden: true,
  143. DisableAutoGenTag: true,
  144. Run: func(cmd *cobra.Command, args []string) {
  145. if err := doc.GenMarkdownTreeCustom(rootCmd, "./doc/", prepender, linkHandler); err != nil {
  146. log.Fatalf("Failed to generate cobra doc: %s", err)
  147. }
  148. },
  149. }
  150. rootCmd.AddCommand(cmdDocGen)
  151. /*usage*/
  152. var cmdVersion = &cobra.Command{
  153. Use: "version",
  154. Short: "Display version and exit.",
  155. Args: cobra.ExactArgs(0),
  156. DisableAutoGenTag: true,
  157. Run: func(cmd *cobra.Command, args []string) {
  158. cwversion.Show()
  159. },
  160. }
  161. rootCmd.AddCommand(cmdVersion)
  162. rootCmd.PersistentFlags().StringVarP(&ConfigFilePath, "config", "c", csconfig.DefaultConfigPath("config.yaml"), "path to crowdsec config file")
  163. rootCmd.PersistentFlags().StringVarP(&OutputFormat, "output", "o", "", "Output format: human, json, raw.")
  164. rootCmd.PersistentFlags().StringVarP(&OutputColor, "color", "", "auto", "Output color: yes, no, auto.")
  165. rootCmd.PersistentFlags().BoolVar(&dbg_lvl, "debug", false, "Set logging to debug.")
  166. rootCmd.PersistentFlags().BoolVar(&nfo_lvl, "info", false, "Set logging to info.")
  167. rootCmd.PersistentFlags().BoolVar(&wrn_lvl, "warning", false, "Set logging to warning.")
  168. rootCmd.PersistentFlags().BoolVar(&err_lvl, "error", false, "Set logging to error.")
  169. rootCmd.PersistentFlags().BoolVar(&trace_lvl, "trace", false, "Set logging to trace.")
  170. rootCmd.PersistentFlags().StringVar(&cwhub.HubBranch, "branch", "", "Override hub branch on github")
  171. if err := rootCmd.PersistentFlags().MarkHidden("branch"); err != nil {
  172. log.Fatalf("failed to hide flag: %s", err)
  173. }
  174. if len(os.Args) > 1 {
  175. cobra.OnInitialize(initConfig)
  176. }
  177. /*don't sort flags so we can enforce order*/
  178. rootCmd.Flags().SortFlags = false
  179. rootCmd.PersistentFlags().SortFlags = false
  180. rootCmd.AddCommand(NewConfigCmd())
  181. rootCmd.AddCommand(NewHubCmd())
  182. rootCmd.AddCommand(NewMetricsCmd())
  183. rootCmd.AddCommand(NewDashboardCmd())
  184. rootCmd.AddCommand(NewDecisionsCmd())
  185. rootCmd.AddCommand(NewAlertsCmd())
  186. // rootCmd.AddCommand(NewInspectCmd())
  187. rootCmd.AddCommand(NewSimulationCmds())
  188. rootCmd.AddCommand(NewBouncersCmd())
  189. rootCmd.AddCommand(NewMachinesCmd())
  190. rootCmd.AddCommand(NewParsersCmd())
  191. rootCmd.AddCommand(NewScenariosCmd())
  192. rootCmd.AddCommand(NewCollectionsCmd())
  193. rootCmd.AddCommand(NewPostOverflowsCmd())
  194. rootCmd.AddCommand(NewCapiCmd())
  195. rootCmd.AddCommand(NewLapiCmd())
  196. rootCmd.AddCommand(NewCompletionCmd())
  197. rootCmd.AddCommand(NewConsoleCmd())
  198. rootCmd.AddCommand(NewExplainCmd())
  199. rootCmd.AddCommand(NewHubTestCmd())
  200. rootCmd.AddCommand(NewNotificationsCmd())
  201. rootCmd.AddCommand(NewSupportCmd())
  202. if err := rootCmd.Execute(); err != nil {
  203. if bincoverTesting != "" {
  204. log.Debug("coverage report is enabled")
  205. }
  206. exitCode := 1
  207. log.NewEntry(log.StandardLogger()).Log(log.FatalLevel, err)
  208. if bincoverTesting == "" {
  209. os.Exit(exitCode)
  210. }
  211. bincover.ExitCode = exitCode
  212. }
  213. }