drone-matrix/plugin/impl.go

112 lines
2.9 KiB
Go
Raw Normal View History

2020-11-25 19:44:45 +00:00
// Copyright (c) 2020, the Drone Plugins project authors.
// Copyright (c) 2021, Robert Kaussow <mail@thegeeklab.de>
2020-11-25 19:44:45 +00:00
// Use of this source code is governed by an Apache 2.0 license that can be
// found in the LICENSE file.
package plugin
import (
"errors"
2020-11-25 19:44:45 +00:00
"fmt"
"strings"
"github.com/matrix-org/gomatrix"
"github.com/microcosm-cc/bluemonday"
"github.com/russross/blackfriday/v2"
"github.com/sirupsen/logrus"
2023-02-08 15:30:46 +00:00
"github.com/thegeeklab/drone-plugin-lib/v2/template"
"maunium.net/go/mautrix"
"maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/id"
2020-11-25 19:44:45 +00:00
)
// Settings for the plugin.
type Settings struct {
Username string
Password string
UserID string
AccessToken string
Homeserver string
RoomID string
Template string
}
var ErrAuthSourceNotSet = errors.New("either username and password or userid and accesstoken are required")
2020-11-25 19:44:45 +00:00
// Validate handles the settings validation of the plugin.
func (p *Plugin) Validate() error {
if (p.settings.Username == "" || p.settings.Password == "") &&
(p.settings.UserID == "" || p.settings.AccessToken == "") {
return ErrAuthSourceNotSet
}
2020-11-25 19:44:45 +00:00
return nil
}
// Execute provides the implementation of the plugin.
func (p *Plugin) Execute() error {
muid := id.NewUserID(prepend("@", p.settings.UserID), p.settings.Homeserver)
client, err := mautrix.NewClient(p.settings.Homeserver, muid, p.settings.AccessToken)
2020-11-25 19:44:45 +00:00
if err != nil {
return fmt.Errorf("failed to initialize client: %w", err)
}
if p.settings.UserID == "" || p.settings.AccessToken == "" {
_, err := client.Login(&mautrix.ReqLogin{
2020-11-25 19:44:45 +00:00
Type: "m.login.password",
Identifier: mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: p.settings.Username},
2020-11-25 19:44:45 +00:00
Password: p.settings.Password,
InitialDeviceDisplayName: "Drone",
StoreCredentials: true,
2020-11-25 19:44:45 +00:00
})
if err != nil {
return fmt.Errorf("failed to authenticate user: %w", err)
}
}
logrus.Info("logged in successfully")
2020-11-25 19:44:45 +00:00
joined, err := client.JoinRoom(prepend("!", p.settings.RoomID), "", nil)
2020-11-25 19:44:45 +00:00
if err != nil {
return fmt.Errorf("failed to join room: %w", err)
}
2023-02-08 15:30:46 +00:00
message, err := template.RenderTrim(p.network.Context, *p.network.Client, p.settings.Template, p.pipeline)
2020-11-25 19:44:45 +00:00
if err != nil {
return fmt.Errorf("failed to render template: %w", err)
}
formatted := bluemonday.UGCPolicy().SanitizeBytes(
blackfriday.Run([]byte(message)),
)
content := gomatrix.HTMLMessage{
Body: message,
MsgType: "m.notice",
Format: "org.matrix.custom.html",
FormattedBody: string(formatted),
}
if _, err := client.SendMessageEvent(joined.RoomID, event.EventMessage, content); err != nil {
2020-11-25 19:44:45 +00:00
return fmt.Errorf("failed to submit message: %w", err)
}
logrus.Info("message sent successfully")
2020-11-25 19:44:45 +00:00
return nil
}
func prepend(prefix, input string) string {
if strings.TrimSpace(input) == "" {
return input
2020-11-25 19:44:45 +00:00
}
if strings.HasPrefix(input, prefix) {
return input
2020-11-25 19:44:45 +00:00
}
return prefix + input
2020-11-25 19:44:45 +00:00
}