package bot

import (
	"strings"

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

// account data key
const acBotSettingsKey = "cc.etke.postmoogle.config"

// bot options keys
const (
	botOptionUsers = "users"
)

type botSettings map[string]string

// Get option
func (s botSettings) Get(key string) string {
	return s[strings.ToLower(strings.TrimSpace(key))]
}

// Set option
func (s botSettings) Set(key, value string) {
	s[strings.ToLower(strings.TrimSpace(key))] = value
}

// Users option
func (s botSettings) Users() []string {
	value := s.Get(botOptionUsers)
	if strings.Contains(value, " ") {
		return strings.Split(value, " ")
	}
	return []string{value}
}

func (b *Bot) initBotUsers(users []string) error {
	_, homeserver, err := b.lp.GetClient().UserID.Parse()
	if err != nil {
		return err
	}
	config := b.getBotSettings()
	oldUsers := config.Get(botOptionUsers)
	// TODO: remove after migration
	if len(users) > 0 && oldUsers == "" {
		_, err := parseMXIDpatterns(users, "@*:"+homeserver)
		if err != nil {
			return err
		}
		config.Set(botOptionUsers, strings.Join(users, " "))
	}

	allowedUsers, uerr := parseMXIDpatterns(config.Users(), "@*:"+homeserver)
	if uerr != nil {
		return uerr
	}
	b.allowedUsers = allowedUsers

	if oldUsers != config.Get(botOptionUsers) {
		return b.setBotSettings(config)
	}

	return nil
}

func (b *Bot) getBotSettings() botSettings {
	cfg := b.botcfg.Get(acBotSettingsKey)
	if cfg != nil {
		return cfg
	}

	config := botSettings{}
	err := b.lp.GetClient().GetAccountData(acBotSettingsKey, &config)
	if err != nil {
		if strings.Contains(err.Error(), "M_NOT_FOUND") {
			err = nil
		} else {
			b.log.Error("cannot get bot settings: %v", utils.UnwrapError(err))
		}
	}

	if err == nil {
		b.botcfg.Set(acBotSettingsKey, config)
	}

	return config
}

func (b *Bot) setBotSettings(cfg botSettings) error {
	b.botcfg.Set(acBotSettingsKey, cfg)
	return utils.UnwrapError(b.lp.GetClient().SetAccountData(acBotSettingsKey, cfg))
}