package bot

import (
	"context"
	"fmt"
	"net"
	"sort"
	"strconv"
	"strings"

	"gitlab.com/etke.cc/go/secgen"
	"maunium.net/go/mautrix/id"

	"gitlab.com/etke.cc/postmoogle/utils"
)

func (b *Bot) sendMailboxes(ctx context.Context) {
	evt := eventFromContext(ctx)
	mailboxes := map[string]roomSettings{}
	slice := []string{}
	b.rooms.Range(func(key any, value any) bool {
		if key == nil {
			return true
		}
		if value == nil {
			return true
		}

		mailbox, ok := key.(string)
		if !ok {
			return true
		}
		roomID, ok := value.(id.RoomID)
		if !ok {
			return true
		}
		config, err := b.getRoomSettings(roomID)
		if err != nil {
			b.log.Error("cannot retrieve settings: %v", err)
		}

		mailboxes[mailbox] = config
		slice = append(slice, mailbox)
		return true
	})
	sort.Strings(slice)

	if len(slice) == 0 {
		b.SendNotice(ctx, evt.RoomID, "No mailboxes are managed by the bot so far, kupo!")
		return
	}

	var msg strings.Builder
	msg.WriteString("The following mailboxes are managed by the bot:\n")
	for _, mailbox := range slice {
		cfg := mailboxes[mailbox]
		msg.WriteString("* `")
		msg.WriteString(utils.EmailsList(mailbox, cfg.Domain()))
		msg.WriteString("` by ")
		msg.WriteString(cfg.Owner())
		msg.WriteString("\n")
	}

	b.SendNotice(ctx, evt.RoomID, msg.String())
}

func (b *Bot) runDelete(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	if len(commandSlice) < 2 {
		b.SendNotice(ctx, evt.RoomID, fmt.Sprintf("Usage: `%s delete MAILBOX`", b.prefix))
		return
	}
	mailbox := utils.Mailbox(commandSlice[1])

	v, ok := b.rooms.Load(mailbox)
	if v == nil || !ok {
		b.SendError(ctx, evt.RoomID, "mailbox does not exists, kupo")
		return
	}
	roomID := v.(id.RoomID)

	b.rooms.Delete(mailbox)
	err := b.setRoomSettings(roomID, roomSettings{})
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot update settings: %v", err)
		return
	}

	b.SendNotice(ctx, evt.RoomID, "mailbox has been deleted")
}

func (b *Bot) runUsers(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	cfg := b.getBotSettings()
	if len(commandSlice) < 2 {
		var msg strings.Builder
		users := cfg.Users()
		if len(users) > 0 {
			msg.WriteString("Currently: `")
			msg.WriteString(strings.Join(users, " "))
			msg.WriteString("`\n\n")
		}
		msg.WriteString("Usage: `")
		msg.WriteString(b.prefix)
		msg.WriteString(" users PATTERN1 PATTERN2 PATTERN3...`")
		msg.WriteString("where each pattern is like `@someone:example.com`, ")
		msg.WriteString("`@bot.*:example.com`, `@*:another.com`, or `@*:*`\n")

		b.SendNotice(ctx, evt.RoomID, msg.String())
		return
	}

	_, homeserver, err := b.lp.GetClient().UserID.Parse()
	if err != nil {
		b.SendError(ctx, evt.RoomID, fmt.Sprintf("invalid userID: %v", err))
	}

	patterns := commandSlice[1:]
	allowedUsers, err := parseMXIDpatterns(patterns, "@*:"+homeserver)
	if err != nil {
		b.SendError(ctx, evt.RoomID, fmt.Sprintf("invalid patterns: %v", err))
		return
	}

	cfg.Set(botOptionUsers, strings.Join(patterns, " "))

	err = b.setBotSettings(cfg)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set bot config: %v", err)
	}
	b.allowedUsers = allowedUsers
	b.SendNotice(ctx, evt.RoomID, "allowed users updated")
}

func (b *Bot) runDKIM(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	cfg := b.getBotSettings()
	if len(commandSlice) > 1 && commandSlice[1] == "reset" {
		cfg.Set(botOptionDKIMPrivateKey, "")
		cfg.Set(botOptionDKIMSignature, "")
	}

	signature := cfg.DKIMSignature()
	if signature == "" {
		var private string
		var derr error
		signature, private, derr = secgen.DKIM()
		if derr != nil {
			b.Error(ctx, evt.RoomID, "cannot generate DKIM signature: %v", derr)
			return
		}
		cfg.Set(botOptionDKIMSignature, signature)
		cfg.Set(botOptionDKIMPrivateKey, private)
		err := b.setBotSettings(cfg)
		if err != nil {
			b.Error(ctx, evt.RoomID, "cannot save bot options: %v", err)
			return
		}
	}

	b.SendNotice(ctx, evt.RoomID, fmt.Sprintf(
		"DKIM signature is: `%s`.\n"+
			"You need to add it to DNS records of all domains added to postmoogle (if not already):\n"+
			"Add new DNS record with type = `TXT`, key (subdomain/from): `postmoogle._domainkey` and value (to):\n ```\n%s\n```\n"+
			"Without that record other email servers may reject your emails as spam, kupo.\n"+
			"To reset the signature, send `%s dkim reset`",
		signature, signature, b.prefix))
}

func (b *Bot) runCatchAll(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	cfg := b.getBotSettings()
	if len(commandSlice) < 2 {
		var msg strings.Builder
		msg.WriteString("Currently: `")
		if cfg.CatchAll() != "" {
			msg.WriteString(cfg.CatchAll())
			msg.WriteString(" (")
			msg.WriteString(utils.EmailsList(cfg.CatchAll(), ""))
			msg.WriteString(")")
		} else {
			msg.WriteString("not set")
		}
		msg.WriteString("`\n\n")
		msg.WriteString("Usage: `")
		msg.WriteString(b.prefix)
		msg.WriteString(" catch-all MAILBOX`")
		msg.WriteString("where mailbox is valid and existing mailbox name\n")

		b.SendNotice(ctx, evt.RoomID, msg.String())
		return
	}

	mailbox := utils.Mailbox(commandSlice[1])
	_, ok := b.GetMapping(mailbox)
	if !ok {
		b.SendError(ctx, evt.RoomID, "mailbox does not exist, kupo.")
		return
	}

	cfg.Set(botOptionCatchAll, mailbox)
	err := b.setBotSettings(cfg)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot save bot options: %v", err)
		return
	}

	b.SendNotice(ctx, evt.RoomID, fmt.Sprintf("Catch-all is set to: `%s` (%s).", mailbox, utils.EmailsList(mailbox, "")))
}

func (b *Bot) printGreylist(ctx context.Context, roomID id.RoomID) {
	cfg := b.getBotSettings()
	greylist := b.getGreylist()
	var msg strings.Builder
	size := len(greylist)
	duration := cfg.Greylist()
	msg.WriteString("Currently: `")
	if duration == 0 {
		msg.WriteString("disabled")
	} else {
		msg.WriteString(cfg.Get(botOptionGreylist))
		msg.WriteString("min")
	}
	msg.WriteString("`")
	if size > 0 {
		msg.WriteString(", total known: ")
		msg.WriteString(strconv.Itoa(size))
		msg.WriteString(" hosts (`")
		msg.WriteString(strings.Join(greylist.Slice(), "`, `"))
		msg.WriteString("`)\n\n")
	}
	if duration == 0 {
		msg.WriteString("\n\nTo enable greylist: `")
		msg.WriteString(b.prefix)
		msg.WriteString(" greylist MIN`")
		msg.WriteString("where `MIN` is duration in minutes for automatic greylisting\n")
	}

	b.SendNotice(ctx, roomID, msg.String())
}

func (b *Bot) runGreylist(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	if len(commandSlice) < 2 {
		b.printGreylist(ctx, evt.RoomID)
		return
	}
	cfg := b.getBotSettings()
	value := utils.SanitizeIntString(commandSlice[1])
	cfg.Set(botOptionGreylist, value)
	err := b.setBotSettings(cfg)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set bot config: %v", err)
	}
	b.SendNotice(ctx, evt.RoomID, "greylist duration has been updated")
}

func (b *Bot) runBanlist(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	cfg := b.getBotSettings()
	if len(commandSlice) < 2 {
		banlist := b.getBanlist()
		var msg strings.Builder
		size := len(banlist)
		if size > 0 {
			msg.WriteString("Currently: `")
			msg.WriteString(cfg.Get(botOptionBanlistEnabled))
			msg.WriteString("`, total: ")
			msg.WriteString(strconv.Itoa(size))
			msg.WriteString(" hosts (`")
			msg.WriteString(strings.Join(banlist.Slice(), "`, `"))
			msg.WriteString("`)\n\n")
		}
		if !cfg.BanlistEnabled() {
			msg.WriteString("To enable banlist, send `")
			msg.WriteString(b.prefix)
			msg.WriteString(" banlist true`\n\n")
		}
		msg.WriteString("To ban somebody: `")
		msg.WriteString(b.prefix)
		msg.WriteString(" banlist:add IP1 IP2 IP3...`")
		msg.WriteString("where each ip is IPv4 or IPv6\n")

		b.SendNotice(ctx, evt.RoomID, msg.String())
		return
	}
	value := utils.SanitizeBoolString(commandSlice[1])
	cfg.Set(botOptionBanlistEnabled, value)
	err := b.setBotSettings(cfg)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set bot config: %v", err)
	}
	b.syncBanlist()
	b.SendNotice(ctx, evt.RoomID, "banlist has been updated")
}

func (b *Bot) runBanlistAdd(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	if len(commandSlice) < 2 {
		b.runBanlist(ctx, commandSlice)
		return
	}
	banlist := b.getBanlist()

	ips := commandSlice[1:]
	for _, ip := range ips {
		addr, err := net.ResolveIPAddr("ip", ip)
		if err != nil {
			b.Error(ctx, evt.RoomID, "cannot add %s to banlist: %v", ip, err)
			return
		}
		banlist.Add(addr)
	}

	err := b.setBanlist(banlist)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set banlist: %v", err)
		return
	}

	b.SendNotice(ctx, evt.RoomID, "banlist has been updated, kupo")
}

func (b *Bot) runBanlistRemove(ctx context.Context, commandSlice []string) {
	evt := eventFromContext(ctx)
	if len(commandSlice) < 2 {
		b.runBanlist(ctx, commandSlice)
		return
	}
	banlist := b.getBanlist()

	ips := commandSlice[1:]
	for _, ip := range ips {
		addr, err := net.ResolveIPAddr("ip", ip)
		if err != nil {
			b.Error(ctx, evt.RoomID, "cannot remove %s from banlist: %v", ip, err)
			return
		}
		banlist.Remove(addr)
	}

	err := b.setBanlist(banlist)
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set banlist: %v", err)
		return
	}

	b.SendNotice(ctx, evt.RoomID, "banlist has been updated, kupo")
}

func (b *Bot) runBanlistReset(ctx context.Context) {
	evt := eventFromContext(ctx)

	err := b.setBanlist(bglist{})
	if err != nil {
		b.Error(ctx, evt.RoomID, "cannot set banlist: %v", err)
		return
	}

	b.SendNotice(ctx, evt.RoomID, "banlist has been reset, kupo")
}