package main import ( "fmt" "time" log "github.com/sirupsen/logrus" "github.com/spf13/cobra" "gopkg.in/tomb.v2" "github.com/crowdsecurity/go-cs-lib/ptr" "github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/require" "github.com/crowdsecurity/crowdsec/pkg/apiserver" "github.com/crowdsecurity/crowdsec/pkg/database" ) type cliPapi struct { cfg configGetter } func NewCLIPapi(cfg configGetter) *cliPapi { return &cliPapi{ cfg: cfg, } } func (cli *cliPapi) NewCommand() *cobra.Command { cmd := &cobra.Command{ Use: "papi [action]", Short: "Manage interaction with Polling API (PAPI)", Args: cobra.MinimumNArgs(1), DisableAutoGenTag: true, PersistentPreRunE: func(_ *cobra.Command, _ []string) error { cfg := cli.cfg() if err := require.LAPI(cfg); err != nil { return err } if err := require.CAPI(cfg); err != nil { return err } if err := require.PAPI(cfg); err != nil { return err } return nil }, } cmd.AddCommand(cli.NewStatusCmd()) cmd.AddCommand(cli.NewSyncCmd()) return cmd } func (cli *cliPapi) NewStatusCmd() *cobra.Command { cmd := &cobra.Command{ Use: "status", Short: "Get status of the Polling API", Args: cobra.MinimumNArgs(0), DisableAutoGenTag: true, RunE: func(_ *cobra.Command, _ []string) error { var err error cfg := cli.cfg() dbClient, err = database.NewClient(cfg.DbConfig) if err != nil { return fmt.Errorf("unable to initialize database client: %s", err) } apic, err := apiserver.NewAPIC(cfg.API.Server.OnlineClient, dbClient, cfg.API.Server.ConsoleConfig, cfg.API.Server.CapiWhitelists) if err != nil { return fmt.Errorf("unable to initialize API client: %s", err) } papi, err := apiserver.NewPAPI(apic, dbClient, cfg.API.Server.ConsoleConfig, log.GetLevel()) if err != nil { return fmt.Errorf("unable to initialize PAPI client: %s", err) } perms, err := papi.GetPermissions() if err != nil { return fmt.Errorf("unable to get PAPI permissions: %s", err) } var lastTimestampStr *string lastTimestampStr, err = dbClient.GetConfigItem(apiserver.PapiPullKey) if err != nil { lastTimestampStr = ptr.Of("never") } log.Infof("You can successfully interact with Polling API (PAPI)") log.Infof("Console plan: %s", perms.Plan) log.Infof("Last order received: %s", *lastTimestampStr) log.Infof("PAPI subscriptions:") for _, sub := range perms.Categories { log.Infof(" - %s", sub) } return nil }, } return cmd } func (cli *cliPapi) NewSyncCmd() *cobra.Command { cmd := &cobra.Command{ Use: "sync", Short: "Sync with the Polling API, pulling all non-expired orders for the instance", Args: cobra.MinimumNArgs(0), DisableAutoGenTag: true, RunE: func(_ *cobra.Command, _ []string) error { var err error cfg := cli.cfg() t := tomb.Tomb{} dbClient, err = database.NewClient(cfg.DbConfig) if err != nil { return fmt.Errorf("unable to initialize database client: %s", err) } apic, err := apiserver.NewAPIC(cfg.API.Server.OnlineClient, dbClient, cfg.API.Server.ConsoleConfig, cfg.API.Server.CapiWhitelists) if err != nil { return fmt.Errorf("unable to initialize API client: %s", err) } t.Go(apic.Push) papi, err := apiserver.NewPAPI(apic, dbClient, cfg.API.Server.ConsoleConfig, log.GetLevel()) if err != nil { return fmt.Errorf("unable to initialize PAPI client: %s", err) } t.Go(papi.SyncDecisions) err = papi.PullOnce(time.Time{}, true) if err != nil { return fmt.Errorf("unable to sync decisions: %s", err) } log.Infof("Sending acknowledgements to CAPI") apic.Shutdown() papi.Shutdown() t.Wait() time.Sleep(5 * time.Second) //FIXME: the push done by apic.Push is run inside a sub goroutine, sleep to make sure it's done return nil }, } return cmd }