package sms

// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
// * Use of this source code is governed by a Apache-2.0 license
// * that can be found in the LICENSE file.
// * @Author yangjian102621@163.com
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

import (
	"geekai/core/types"
	"geekai/log"
)

type SmsManager struct {
	aliyun  *AliYunSmsService
	bao     *BaoSmsService
	tencent *TencentSmsService
	active  string
}

var logger = log.GetLogger()

func NewSmsManager(sysConfig *types.SystemConfig, aliyun *AliYunSmsService, bao *BaoSmsService, tencent *TencentSmsService) (*SmsManager, error) {

	return &SmsManager{
		active:  sysConfig.SMS.Active,
		aliyun:  aliyun,
		bao:     bao,
		tencent: tencent,
	}, nil
}

func (m *SmsManager) GetService() Service {
	switch m.active {
	case Ali:
		return m.aliyun
	case Bao:
		return m.bao
	case Tencent:
		return m.tencent
	}
	return nil
}

func (m *SmsManager) SetActive(active string) {
	m.active = active
}

func (m *SmsManager) UpdateConfig(config types.SMSConfig) {
	switch config.Active {
	case Ali:
		m.aliyun.UpdateConfig(config.Ali)
	case Bao:
		m.bao.UpdateConfig(config.Bao)
	case Tencent:
		m.tencent.UpdateConfig(config.Tencent)
	}
	m.active = config.Active
}