add noowner and federation

This commit is contained in:
Aine
2022-08-22 22:08:32 +03:00
parent 564cd82c6b
commit 0decd4fad6
9 changed files with 129 additions and 13 deletions

View File

@@ -35,6 +35,9 @@ env vars
### optional
* **POSTMOOGLE_NOOWNER** - allow change room settings by any room partisipant
* **POSTMOOGLE_FEDERATION** - allow usage of Postmoogle by users from others homeservers
* **POSTMOOGLE_NOENCRYPTION** - disable encryption support
* **POSTMOOGLE_SENTRY_DSN** - sentry DSN
* **POSTMOOGLE_SENTRY_RATE** - sentry sample rate, from 0 to 100 (default: 20)
* **POSTMOOGLE_LOGLEVEL** - log level

View File

@@ -18,6 +18,8 @@ import (
// Bot represents matrix bot
type Bot struct {
noowner bool
federation bool
prefix string
domain string
rooms map[string]id.RoomID
@@ -27,8 +29,10 @@ type Bot struct {
}
// New creates a new matrix bot
func New(lp *linkpearl.Linkpearl, log *logger.Logger, prefix, domain string) *Bot {
func New(lp *linkpearl.Linkpearl, log *logger.Logger, prefix, domain string, noowner, federation bool) *Bot {
return &Bot{
noowner: noowner,
federation: federation,
roomsmu: &sync.Mutex{},
prefix: prefix,
domain: domain,

View File

@@ -12,6 +12,7 @@ import (
var commands = map[string]string{
"mailbox": "Get or set mailbox of that room",
"owner": "Get or set owner of that room",
"help": "Get help",
}
@@ -20,9 +21,16 @@ func (b *Bot) handleCommand(ctx context.Context, evt *event.Event, command []str
return
}
// ignore requests over federation if disabled
if !b.federation && evt.Sender.Homeserver() != b.lp.GetClient().UserID.Homeserver() {
return
}
switch command[0] {
case "help":
b.sendHelp(ctx, evt.RoomID)
case "owner":
b.handleOwner(ctx, evt, command)
case "mailbox":
b.handleMailbox(ctx, evt, command)
}

View File

@@ -14,6 +14,20 @@ var migrations = []string{}
// settings of a room
type settings struct {
Mailbox string
Owner id.UserID
}
// Allowed checks if change is allowed
func (s *settings) Allowed(noowner bool, userID id.UserID) bool {
if noowner {
return true
}
if s.Owner == "" {
return true
}
return s.Owner == userID
}
func (b *Bot) migrate() error {

View File

@@ -86,6 +86,13 @@ func (b *Bot) setMailbox(ctx context.Context, evt *event.Event, mailbox string)
if cfg == nil {
cfg = &settings{}
}
if !cfg.Allowed(b.noowner, evt.Sender) {
b.Error(span.Context(), evt.RoomID, "you don't have permission to do that")
return
}
cfg.Owner = evt.Sender
cfg.Mailbox = mailbox
err = b.setSettings(span.Context(), evt.RoomID, cfg)
if err != nil {

74
bot/owner.go Normal file
View File

@@ -0,0 +1,74 @@
package bot
import (
"context"
"github.com/getsentry/sentry-go"
"maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/format"
"maunium.net/go/mautrix/id"
)
func (b *Bot) handleOwner(ctx context.Context, evt *event.Event, command []string) {
if len(command) == 1 {
b.getOwner(ctx, evt)
return
}
b.setOwner(ctx, evt, command[1])
}
func (b *Bot) getOwner(ctx context.Context, evt *event.Event) {
span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("getOwner"))
defer span.Finish()
cfg, err := b.getSettings(span.Context(), evt.RoomID)
if err != nil || cfg == nil {
b.Error(span.Context(), evt.RoomID, "owner is not set yet")
return
}
if cfg.Owner == "" {
b.Error(span.Context(), evt.RoomID, "owner is not set yet")
return
}
content := format.RenderMarkdown("Owner of this room is "+cfg.Owner.String(), true, true)
content.MsgType = event.MsgNotice
_, err = b.lp.Send(evt.RoomID, content)
if err != nil {
b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err)
}
}
func (b *Bot) setOwner(ctx context.Context, evt *event.Event, owner string) {
span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("setOwner"))
defer span.Finish()
ownerID := id.UserID(owner)
cfg, err := b.getSettings(span.Context(), evt.RoomID)
if err != nil {
b.log.Warn("cannot get settings: %v", err)
}
if cfg == nil {
cfg = &settings{}
}
if !cfg.Allowed(b.noowner, evt.Sender) {
b.Error(span.Context(), evt.RoomID, "you don't have permission to do that")
return
}
cfg.Owner = ownerID
err = b.setSettings(span.Context(), evt.RoomID, cfg)
if err != nil {
b.Error(span.Context(), evt.RoomID, "cannot update settings: %v", err)
return
}
content := format.RenderMarkdown("Owner of this room set to "+owner, true, true)
content.MsgType = event.MsgNotice
_, err = b.lp.Send(evt.RoomID, content)
if err != nil {
b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err)
}
}

View File

@@ -81,7 +81,7 @@ func initBot(cfg *config.Config) {
// nolint // Fatal = panic, not os.Exit()
log.Fatal("cannot initialize matrix bot: %v", err)
}
mxb = bot.New(lp, mxlog, cfg.Prefix, cfg.Domain)
mxb = bot.New(lp, mxlog, cfg.Prefix, cfg.Domain, cfg.NoOwner, cfg.Federation)
log.Debug("bot has been created")
}

View File

@@ -17,6 +17,8 @@ func New() *Config {
Domain: env.String("domain", defaultConfig.Domain),
Port: env.String("port", defaultConfig.Port),
NoEncryption: env.Bool("noencryption"),
NoOwner: env.Bool("noowner"),
Federation: env.Bool("federation"),
MaxSize: env.Int("maxsize", defaultConfig.MaxSize),
Sentry: Sentry{
DSN: env.String("sentry.dsn", defaultConfig.Sentry.DSN),

View File

@@ -16,6 +16,10 @@ type Config struct {
LogLevel string
// NoEncryption disabled encryption support
NoEncryption bool
// NoOwner allows room settings change by any participant
NoOwner bool
// Federation allows usage of Postmoogle by users from other homeservers
Federation bool
// Prefix for commands
Prefix string
// MaxSize of an email (including attachments)