81 lines
2 KiB
Go
81 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
plugin "github.com/hashicorp/go-plugin"
|
|
|
|
"github.com/slack-go/slack"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type PluginConfig struct {
|
|
Name string `yaml:"name"`
|
|
Webhook string `yaml:"webhook"`
|
|
LogLevel *string `yaml:"log_level"`
|
|
}
|
|
type Notify struct {
|
|
ConfigByName map[string]PluginConfig
|
|
}
|
|
|
|
var logger hclog.Logger = hclog.New(&hclog.LoggerOptions{
|
|
Name: "slack-plugin",
|
|
Level: hclog.LevelFromString("DEBUG"),
|
|
Output: os.Stderr,
|
|
JSONFormat: true,
|
|
})
|
|
|
|
func (n *Notify) Notify(ctx context.Context, notification *Notification) (*Empty, error) {
|
|
if _, ok := n.ConfigByName[notification.Name]; !ok {
|
|
return nil, fmt.Errorf("invalid plugin config name %s", notification.Name)
|
|
}
|
|
cfg := n.ConfigByName[notification.Name]
|
|
if cfg.LogLevel != nil && *cfg.LogLevel != "" {
|
|
logger.SetLevel(hclog.LevelFromString(*cfg.LogLevel))
|
|
} else {
|
|
logger.SetLevel(hclog.Info)
|
|
}
|
|
|
|
logger.Info(fmt.Sprintf("found notify signal for %s config", notification.Name))
|
|
logger.Debug(fmt.Sprintf("posting to %s webhook, message %s", cfg.Webhook, notification.Text))
|
|
err := slack.PostWebhook(n.ConfigByName[notification.Name].Webhook, &slack.WebhookMessage{
|
|
Text: notification.Text,
|
|
})
|
|
if err != nil {
|
|
logger.Error(err.Error())
|
|
}
|
|
|
|
return &Empty{}, err
|
|
}
|
|
|
|
func (n *Notify) Configure(ctx context.Context, config *Config) (*Empty, error) {
|
|
d := PluginConfig{}
|
|
if err := yaml.Unmarshal(config.Config, &d); err != nil {
|
|
return nil, err
|
|
}
|
|
n.ConfigByName[d.Name] = d
|
|
return &Empty{}, nil
|
|
}
|
|
|
|
func main() {
|
|
var handshake = plugin.HandshakeConfig{
|
|
ProtocolVersion: 1,
|
|
MagicCookieKey: "CROWDSEC_PLUGIN_KEY",
|
|
MagicCookieValue: os.Getenv("CROWDSEC_PLUGIN_KEY"),
|
|
}
|
|
|
|
plugin.Serve(&plugin.ServeConfig{
|
|
HandshakeConfig: handshake,
|
|
Plugins: map[string]plugin.Plugin{
|
|
"slack": &NotifierPlugin{
|
|
Impl: &Notify{ConfigByName: make(map[string]PluginConfig)},
|
|
},
|
|
},
|
|
GRPCServer: plugin.DefaultGRPCServer,
|
|
Logger: logger,
|
|
})
|
|
}
|