add noowner and federation

bot/bot.go

@@ -18,22 +18,26 @@ import (
 
 // Bot represents matrix bot
 type Bot struct {
-	prefix  string
-	domain  string
-	rooms   map[string]id.RoomID
-	roomsmu *sync.Mutex
-	log     *logger.Logger
-	lp      *linkpearl.Linkpearl
+	noowner    bool
+	federation bool
+	prefix     string
+	domain     string
+	rooms      map[string]id.RoomID
+	roomsmu    *sync.Mutex
+	log        *logger.Logger
+	lp         *linkpearl.Linkpearl
 }
 
 // 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{
-		roomsmu: &sync.Mutex{},
-		prefix:  prefix,
-		domain:  domain,
-		log:     log,
-		lp:      lp,
+		noowner:    noowner,
+		federation: federation,
+		roomsmu:    &sync.Mutex{},
+		prefix:     prefix,
+		domain:     domain,
+		log:        log,
+		lp:         lp,
 	}
 }
 

bot/command.go

@@ -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)
 	}

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 {

bot/mailbox.go

@@ -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 {

bot/owner.go

@@ -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)
+	}
+}