package db

import (
	"github.com/go-pg/pg"
)

type MarketConfigStatus string

const MarketConfigEnabled = "enabled"
const MarketConfigDisabled = "disabled"
const MarketConfigInvalidCredentials = "invalid_credentials"

type MarketConfig struct {
	Id         int64
	MarketName string
	UserId     int64
	Status     MarketConfigStatus
	Config     map[string]string
}

func InsertMarketConfig(config *MarketConfig) error {
	return DB.Insert(config)
}

func GetUserMarketConfig(userId int64, market string) (*MarketConfig, error) {
	var config MarketConfig

	err := DB.Model(&config).Where("user_id = ?", userId).Where("market_name = ?", market).First()

	if err != nil && err != pg.ErrNoRows {
		return nil, err
	}

	if err == pg.ErrNoRows {
		return nil, nil
	} else {
		return &config, nil
	}
}

func SetUserMarketConfig(userId int64, market string, newConfig map[string]string) (*MarketConfig, error) {
	config := MarketConfig{
		UserId:     userId,
		MarketName: market,
		Config:     newConfig,
	}

	_, err := DB.Model(&config).
		OnConflict("(user_id, market_name) DO UPDATE").
		Set("config = ?", newConfig).
		Insert()

	return &config, err
}

func SetMarketConfigStatus(marketConfig MarketConfig, status MarketConfigStatus) (*MarketConfig, error) {
	marketConfig.Status = status
	_, err := DB.Model(&marketConfig).
		OnConflict("(user_id, market_name) DO UPDATE").
		Set("status = ?", status).
		Insert()

	return &marketConfig, err
}