add noowner and federation
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
|
||||
14
bot/data.go
14
bot/data.go
@@ -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 {
|
||||
|
||||
@@ -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
74
bot/owner.go
Normal 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)
|
||||
}
|
||||
}
|
||||
@@ -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")
|
||||
}
|
||||
|
||||
|
||||
@@ -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),
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user