package config

import (
	"context"

	"github.com/rs/zerolog"
	"gitlab.com/etke.cc/linkpearl"
	"maunium.net/go/mautrix/id"

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

// Manager of configs
type Manager struct {
	mu  utils.Mutex
	log *zerolog.Logger
	lp  *linkpearl.Linkpearl
}

// New config manager
func New(lp *linkpearl.Linkpearl, log *zerolog.Logger) *Manager {
	m := &Manager{
		mu:  utils.NewMutex(),
		lp:  lp,
		log: log,
	}

	return m
}

// GetBot config
func (m *Manager) GetBot(ctx context.Context) Bot {
	var err error
	var config Bot
	config, err = m.lp.GetAccountData(ctx, acBotKey)
	if err != nil {
		m.log.Error().Err(err).Msg("cannot get bot settings")
	}
	if config == nil {
		config = make(Bot, 0)
		return config
	}

	return config
}

// SetBot config
func (m *Manager) SetBot(ctx context.Context, cfg Bot) error {
	return m.lp.SetAccountData(ctx, acBotKey, cfg)
}

// GetRoom config
func (m *Manager) GetRoom(ctx context.Context, roomID id.RoomID) (Room, error) {
	config, err := m.lp.GetRoomAccountData(ctx, roomID, acRoomKey)
	if err != nil {
		m.log.Warn().Err(err).Str("room_id", roomID.String()).Msg("cannot get room settings")
	}
	if config == nil {
		config = make(Room, 0)
	}

	return config, err
}

// SetRoom config
func (m *Manager) SetRoom(ctx context.Context, roomID id.RoomID, cfg Room) error {
	return m.lp.SetRoomAccountData(ctx, roomID, acRoomKey, cfg)
}

// GetBanlist config
func (m *Manager) GetBanlist(ctx context.Context) List {
	if !m.GetBot(ctx).BanlistEnabled() {
		return make(List, 0)
	}

	m.mu.Lock("banlist")
	defer m.mu.Unlock("banlist")
	config, err := m.lp.GetAccountData(ctx, acBanlistKey)
	if err != nil {
		m.log.Error().Err(err).Msg("cannot get banlist")
	}
	if config == nil {
		config = make(List, 0)
		return config
	}
	return config
}

// SetBanlist config
func (m *Manager) SetBanlist(ctx context.Context, cfg List) error {
	if !m.GetBot(ctx).BanlistEnabled() {
		return nil
	}

	m.mu.Lock("banlist")
	defer m.mu.Unlock("banlist")
	if cfg == nil {
		cfg = make(List, 0)
	}

	return m.lp.SetAccountData(ctx, acBanlistKey, cfg)
}

// GetGreylist config
func (m *Manager) GetGreylist(ctx context.Context) List {
	config, err := m.lp.GetAccountData(ctx, acGreylistKey)
	if err != nil {
		m.log.Error().Err(err).Msg("cannot get banlist")
	}
	if config == nil {
		config = make(List, 0)
		return config
	}

	return config
}

// SetGreylist config
func (m *Manager) SetGreylist(ctx context.Context, cfg List) error {
	return m.lp.SetAccountData(ctx, acGreylistKey, cfg)
}