132 lines
4.7 KiB
Go
132 lines
4.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/cwhub"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func NewPostOverflowsCmd() *cobra.Command {
|
|
var cmdPostOverflows = &cobra.Command{
|
|
Use: "postoverflows [action] [config]",
|
|
Short: "Install/Remove/Upgrade/Inspect postoverflow(s) from hub",
|
|
Example: `cscli postoverflows install crowdsecurity/cdn-whitelist
|
|
cscli postoverflows inspect crowdsecurity/cdn-whitelist
|
|
cscli postoverflows upgrade crowdsecurity/cdn-whitelist
|
|
cscli postoverflows list
|
|
cscli postoverflows remove crowdsecurity/cdn-whitelist`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := csConfig.LoadHub(); err != nil {
|
|
log.Fatalf(err.Error())
|
|
}
|
|
if csConfig.Hub == nil {
|
|
return fmt.Errorf("you must configure cli before interacting with hub")
|
|
}
|
|
|
|
if err := setHubBranch(); err != nil {
|
|
return fmt.Errorf("error while setting hub branch: %s", err)
|
|
}
|
|
|
|
if err := cwhub.GetHubIdx(csConfig.Hub); err != nil {
|
|
log.Fatalf("Failed to get Hub index : %v", err)
|
|
log.Infoln("Run 'sudo cscli hub update' to get the hub index")
|
|
}
|
|
return nil
|
|
},
|
|
PersistentPostRun: func(cmd *cobra.Command, args []string) {
|
|
if cmd.Name() == "inspect" || cmd.Name() == "list" {
|
|
return
|
|
}
|
|
log.Infof("Run 'sudo systemctl reload crowdsec' for the new configuration to be effective.")
|
|
},
|
|
}
|
|
|
|
var cmdPostOverflowsInstall = &cobra.Command{
|
|
Use: "install [config]",
|
|
Short: "Install given postoverflow(s)",
|
|
Long: `Fetch and install given postoverflow(s) from hub`,
|
|
Example: `cscli postoverflows install crowdsec/xxx crowdsec/xyz`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
for _, name := range args {
|
|
InstallItem(name, cwhub.PARSERS_OVFLW, forceAction)
|
|
}
|
|
},
|
|
}
|
|
cmdPostOverflowsInstall.PersistentFlags().BoolVarP(&downloadOnly, "download-only", "d", false, "Only download packages, don't enable")
|
|
cmdPostOverflowsInstall.PersistentFlags().BoolVar(&forceAction, "force", false, "Force install : Overwrite tainted and outdated files")
|
|
cmdPostOverflows.AddCommand(cmdPostOverflowsInstall)
|
|
|
|
var cmdPostOverflowsRemove = &cobra.Command{
|
|
Use: "remove [config]",
|
|
Short: "Remove given postoverflow(s)",
|
|
Long: `remove given postoverflow(s)`,
|
|
Example: `cscli postoverflows remove crowdsec/xxx crowdsec/xyz`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if all {
|
|
RemoveMany(cwhub.PARSERS_OVFLW, "")
|
|
} else {
|
|
for _, name := range args {
|
|
RemoveMany(cwhub.PARSERS_OVFLW, name)
|
|
}
|
|
}
|
|
},
|
|
}
|
|
cmdPostOverflowsRemove.PersistentFlags().BoolVar(&purge, "purge", false, "Delete source file too")
|
|
cmdPostOverflowsRemove.PersistentFlags().BoolVar(&forceAction, "force", false, "Force remove : Remove tainted and outdated files")
|
|
cmdPostOverflowsRemove.PersistentFlags().BoolVar(&all, "all", false, "Delete all the postoverflows")
|
|
cmdPostOverflows.AddCommand(cmdPostOverflowsRemove)
|
|
|
|
var cmdPostOverflowsUpgrade = &cobra.Command{
|
|
Use: "upgrade [config]",
|
|
Short: "Upgrade given postoverflow(s)",
|
|
Long: `Fetch and Upgrade given postoverflow(s) from hub`,
|
|
Example: `cscli postoverflows upgrade crowdsec/xxx crowdsec/xyz`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if all {
|
|
UpgradeConfig(cwhub.PARSERS_OVFLW, "", forceAction)
|
|
} else {
|
|
for _, name := range args {
|
|
UpgradeConfig(cwhub.PARSERS_OVFLW, name, forceAction)
|
|
}
|
|
}
|
|
},
|
|
}
|
|
cmdPostOverflowsUpgrade.PersistentFlags().BoolVarP(&all, "all", "a", false, "Upgrade all the postoverflows")
|
|
cmdPostOverflowsUpgrade.PersistentFlags().BoolVar(&forceAction, "force", false, "Force upgrade : Overwrite tainted and outdated files")
|
|
cmdPostOverflows.AddCommand(cmdPostOverflowsUpgrade)
|
|
|
|
var cmdPostOverflowsInspect = &cobra.Command{
|
|
Use: "inspect [config]",
|
|
Short: "Inspect given postoverflow",
|
|
Long: `Inspect given postoverflow`,
|
|
Example: `cscli postoverflows inspect crowdsec/xxx crowdsec/xyz`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
InspectItem(args[0], cwhub.PARSERS_OVFLW)
|
|
},
|
|
}
|
|
cmdPostOverflows.AddCommand(cmdPostOverflowsInspect)
|
|
|
|
var cmdPostOverflowsList = &cobra.Command{
|
|
Use: "list [config]",
|
|
Short: "List all postoverflows or given one",
|
|
Long: `List all postoverflows or given one`,
|
|
Example: `cscli postoverflows list
|
|
cscli postoverflows list crowdsecurity/xxx`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
ListItem(cwhub.PARSERS_OVFLW, args)
|
|
},
|
|
}
|
|
cmdPostOverflowsList.PersistentFlags().BoolVarP(&all, "all", "a", false, "List as well disabled items")
|
|
cmdPostOverflows.AddCommand(cmdPostOverflowsList)
|
|
|
|
return cmdPostOverflows
|
|
}
|