move plugins/notifications/* to cmd/notification-* (#2429)

This ensures keeping all dependencies in sync, and simplifies
packaging under freebsd/gentoo/etc because there is a single
vendor directory.

cmd/notification-email/main.go

@@ -0,0 +1,149 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/crowdsecurity/crowdsec/pkg/protobufs"
+	"github.com/hashicorp/go-hclog"
+	plugin "github.com/hashicorp/go-plugin"
+	mail "github.com/xhit/go-simple-mail/v2"
+	"gopkg.in/yaml.v2"
+)
+
+var baseLogger hclog.Logger = hclog.New(&hclog.LoggerOptions{
+	Name:       "email-plugin",
+	Level:      hclog.LevelFromString("INFO"),
+	Output:     os.Stderr,
+	JSONFormat: true,
+})
+
+var AuthStringToType map[string]mail.AuthType = map[string]mail.AuthType{
+	"none":    mail.AuthNone,
+	"crammd5": mail.AuthCRAMMD5,
+	"login":   mail.AuthLogin,
+	"plain":   mail.AuthPlain,
+}
+
+var EncryptionStringToType map[string]mail.Encryption = map[string]mail.Encryption{
+	"ssltls":   mail.EncryptionSSLTLS,
+	"starttls": mail.EncryptionSTARTTLS,
+	"none":     mail.EncryptionNone,
+}
+
+type PluginConfig struct {
+	Name     string  `yaml:"name"`
+	LogLevel *string `yaml:"log_level"`
+
+	SMTPHost       string   `yaml:"smtp_host"`
+	SMTPPort       int      `yaml:"smtp_port"`
+	SMTPUsername   string   `yaml:"smtp_username"`
+	SMTPPassword   string   `yaml:"smtp_password"`
+	SenderEmail    string   `yaml:"sender_email"`
+	SenderName     string   `yaml:"sender_name"`
+	ReceiverEmails []string `yaml:"receiver_emails"`
+	EmailSubject   string   `yaml:"email_subject"`
+	EncryptionType string   `yaml:"encryption_type"`
+	AuthType       string   `yaml:"auth_type"`
+	HeloHost       string   `yaml:"helo_host"`
+}
+
+type EmailPlugin struct {
+	ConfigByName map[string]PluginConfig
+}
+
+func (n *EmailPlugin) Configure(ctx context.Context, config *protobufs.Config) (*protobufs.Empty, error) {
+	d := PluginConfig{
+		SMTPPort:       25,
+		SenderName:     "Crowdsec",
+		EmailSubject:   "Crowdsec notification",
+		EncryptionType: "ssltls",
+		AuthType:       "login",
+		SenderEmail:    "crowdsec@crowdsec.local",
+		HeloHost:	"localhost",
+	}
+
+	if err := yaml.Unmarshal(config.Config, &d); err != nil {
+		return nil, err
+	}
+
+	if d.Name == "" {
+		return nil, fmt.Errorf("name is required")
+	}
+
+	if d.SMTPHost == "" {
+		return nil, fmt.Errorf("SMTP host is not set")
+	}
+
+	if d.ReceiverEmails == nil || len(d.ReceiverEmails) == 0 {
+		return nil, fmt.Errorf("Receiver emails are not set")
+	}
+
+	n.ConfigByName[d.Name] = d
+	baseLogger.Debug(fmt.Sprintf("Email plugin '%s' use SMTP host '%s:%d'", d.Name, d.SMTPHost, d.SMTPPort))
+	return &protobufs.Empty{}, nil
+}
+
+func (n *EmailPlugin) Notify(ctx context.Context, notification *protobufs.Notification) (*protobufs.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]
+
+	logger := baseLogger.Named(cfg.Name)
+
+	if cfg.LogLevel != nil && *cfg.LogLevel != "" {
+		logger.SetLevel(hclog.LevelFromString(*cfg.LogLevel))
+	}
+
+	logger.Debug("got notification")
+
+	server := mail.NewSMTPClient()
+	server.Host = cfg.SMTPHost
+	server.Port = cfg.SMTPPort
+	server.Username = cfg.SMTPUsername
+	server.Password = cfg.SMTPPassword
+	server.Encryption = EncryptionStringToType[cfg.EncryptionType]
+	server.Authentication = AuthStringToType[cfg.AuthType]
+	server.Helo = cfg.HeloHost
+
+	logger.Debug("making smtp connection")
+	smtpClient, err := server.Connect()
+	if err != nil {
+		return &protobufs.Empty{}, err
+	}
+	logger.Debug("smtp connection done")
+
+	email := mail.NewMSG()
+	email.SetFrom(fmt.Sprintf("%s <%s>", cfg.SenderName, cfg.SenderEmail)).
+		AddTo(cfg.ReceiverEmails...).
+		SetSubject(cfg.EmailSubject)
+	email.SetBody(mail.TextHTML, notification.Text)
+
+	err = email.Send(smtpClient)
+	if err != nil {
+		return &protobufs.Empty{}, err
+	}
+	logger.Info(fmt.Sprintf("sent email to %v", cfg.ReceiverEmails))
+	return &protobufs.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{
+			"email": &protobufs.NotifierPlugin{
+				Impl: &EmailPlugin{ConfigByName: make(map[string]PluginConfig)},
+			},
+		},
+		GRPCServer: plugin.DefaultGRPCServer,
+		Logger:     baseLogger,
+	})
+}