lint; rearrange code

bot/command.go

@@ -8,14 +8,29 @@ import (
 	"maunium.net/go/mautrix/event"
 	"maunium.net/go/mautrix/format"
 	"maunium.net/go/mautrix/id"
+
+	"gitlab.com/etke.cc/postmoogle/utils"
 )
 
-var commands = map[string]string{
-	"mailbox":  "Get or set mailbox of that room",
-	"owner":    "Get or set owner of that room",
-	"nosender": "Get or set `nosender` of that room (`true` - hide email sender; `false` - show email sender)",
-	"help":     "Get help",
-}
+type sanitizerFunc func(string) string
+
+var (
+	commands = map[string]string{
+		// special commands
+		"help": "Get help",
+
+		// options commands
+		"mailbox":  "Get or set mailbox of that room",
+		"owner":    "Get or set owner of that room",
+		"nosender": "Get or set `nosender` of that room (`true` - hide email sender; `false` - show email sender)",
+	}
+
+	// sanitizers is map of option name => sanitizer function
+	sanitizers = map[string]sanitizerFunc{
+		"mailbox":  utils.Mailbox,
+		"nosender": utils.SanitizeBoolString,
+	}
+)
 
 func (b *Bot) handleCommand(ctx context.Context, evt *event.Event, command []string) {
 	if _, ok := commands[command[0]]; !ok {
@@ -30,11 +45,7 @@ func (b *Bot) handleCommand(ctx context.Context, evt *event.Event, command []str
 	switch command[0] {
 	case "help":
 		b.sendHelp(ctx, evt.RoomID)
-	case "owner":
-		b.handleOption(ctx, evt, command)
-	case "mailbox":
-		b.handleOption(ctx, evt, command)
-	case "nosender":
+	default:
 		b.handleOption(ctx, evt, command)
 	}
 }
@@ -74,3 +85,82 @@ func (b *Bot) sendHelp(ctx context.Context, roomID id.RoomID) {
 		b.Error(span.Context(), roomID, "cannot send message: %v", err)
 	}
 }
+
+func (b *Bot) handleOption(ctx context.Context, evt *event.Event, command []string) {
+	if len(command) == 1 {
+		b.getOption(ctx, evt, command[0])
+		return
+	}
+	b.setOption(ctx, evt, command[0], command[1])
+}
+
+func (b *Bot) getOption(ctx context.Context, evt *event.Event, name string) {
+	msg := "`%s` of this room is %s"
+	span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("getOption"))
+	defer span.Finish()
+
+	cfg, err := b.getSettings(span.Context(), evt.RoomID)
+	if err != nil {
+		b.Error(span.Context(), evt.RoomID, "failed to retrieve settings: %v", err)
+		return
+	}
+
+	value := cfg.Get(name)
+	if value == "" {
+		b.Notice(span.Context(), evt.RoomID, "`%s` is not set", name)
+		return
+	}
+
+	if name == "mailbox" {
+		msg = msg + "@" + b.domain
+	}
+
+	b.Notice(span.Context(), evt.RoomID, msg, name, value)
+}
+
+func (b *Bot) setOption(ctx context.Context, evt *event.Event, name, value string) {
+	span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("setOption"))
+	defer span.Finish()
+	msg := "`%s` of this room set to %s"
+
+	sanitizer, ok := sanitizers[name]
+	if ok {
+		value = sanitizer(value)
+	}
+
+	if name == "mailbox" {
+		existingID, ok := b.GetMapping(ctx, value)
+		if ok && existingID != "" && existingID != evt.RoomID {
+			b.Notice(span.Context(), evt.RoomID, "Mailbox %s@%s already taken", value, b.domain)
+			return
+		}
+	}
+
+	cfg, err := b.getSettings(span.Context(), evt.RoomID)
+	if err != nil {
+		b.Error(span.Context(), evt.RoomID, "failed to retrieve settings: %v", err)
+		return
+	}
+
+	if !cfg.Allowed(b.noowner, evt.Sender) {
+		b.Notice(span.Context(), evt.RoomID, "you don't have permission to do that")
+		return
+	}
+
+	cfg.Set(name, value)
+	if name == "mailbox" {
+		msg = msg + "@" + b.domain
+		cfg.Set("owner", evt.Sender.String())
+		b.roomsmu.Lock()
+		b.rooms[value] = evt.RoomID
+		b.roomsmu.Unlock()
+	}
+
+	err = b.setSettings(span.Context(), evt.RoomID, cfg)
+	if err != nil {
+		b.Error(span.Context(), evt.RoomID, "cannot update settings: %v", err)
+		return
+	}
+
+	b.Notice(span.Context(), evt.RoomID, msg, name, value)
+}