package service

import (
	"context"
	"errors"
	"hepa-calc-api/api/dao"
	"hepa-calc-api/api/dto"
	"hepa-calc-api/extend/aliyun"
	"hepa-calc-api/global"
	"math/rand"
	"net/http"
	"strconv"
	"time"
)

type PublicService struct {
}

// GetPhoneCode 获取手机验证码
func (r *PublicService) GetPhoneCode(scene int, phone string) (bool, error) {
	sendCodeCount := 1      // // 获取验证码最大次数
	var code string         // 验证码
	var templateCode string // 短信模版

	// 登录获取验证码
	if scene == 1 {
		// 验证发送次数
		res, _ := global.Redis.Get(context.Background(), "login_code_count_"+phone).Result()
		if res != "" {
			sendCodeCount, err := strconv.Atoi(res)
			if err != nil {
				return false, err
			}

			if sendCodeCount > 3 {
				// 超出规定时间内最大获取次数
				return false, errors.New("手机号超出规定时间内最大获取次数，请您稍后再试")
			}
		}

		// 生成随机数
		rand.New(rand.NewSource(time.Now().UnixNano()))
		code = strconv.Itoa(rand.Intn(9000) + 1000)

		// 模版
		templateCode = "SMS_215344868"

		sendCodeCount = sendCodeCount + 1
	}

	if code == "" || templateCode == "" {
		return false, errors.New("验证码发送失败，请您稍后再试")
	}

	// 发送验证码
	templateParam := make(map[string]interface{})
	templateParam["code"] = code
	err := aliyun.SendSms(phone, templateCode, "获取验证码", templateParam)
	if err != nil {
		return false, err
	}

	// 记录发送次数
	if sendCodeCount != 0 {
		_, err = global.Redis.Set(context.Background(), "login_code_count_"+phone, sendCodeCount, 60*5*time.Second).Result()
		if err != nil {
			return false, errors.New("验证码发送失败，请您稍后再试")
		}
	}

	// 设置验证码有效期
	_, err = global.Redis.Set(context.Background(), "login_code_"+phone, code, 60*5*time.Second).Result()
	if err != nil {
		return false, errors.New("验证码发送失败，请您稍后再试")
	}

	return true, nil
}

// GetIndex 获取首页数据
func (r *PublicService) GetIndex(userId int64) (g *dto.IndexDto, err error) {
	// 获取疾病分类列表
	baseClassDao := dao.BaseClassDao{}

	maps := make(map[string]interface{})
	maps["class_status"] = 1
	baseClasss, err := baseClassDao.GetBaseClassExistsOrderList(maps)
	if err != nil {
		return nil, err
	}

	questionService := &QuestionService{}

	// 获取算一算热榜-人气数最高的9个
	hotQuestions, err := questionService.GetHotList()
	if err != nil {
		return nil, err
	}

	// 获取为你推荐-后台指定的推广
	recommendQuestions, err := questionService.GetRecommendList()
	if err != nil {
		return nil, err
	}

	// 获取猜你喜欢
	guessUserLikes, err := questionService.GetGuessUserLIkeList(userId)
	if err != nil {
		return nil, err
	}

	// 处理返回值
	g = &dto.IndexDto{}

	g.LoadBaseClass(baseClasss)

	g.LoadHotQuestionList(hotQuestions)

	g.LoadRecommendQuestionList(recommendQuestions)

	g.LoadGuessUserLikeList(guessUserLikes)

	return g, nil
}

// GetUserIP 获取用户ip
func (r *PublicService) GetUserIP(h *http.Request) string {
	forwarded := h.Header.Get("X-FORWARDED-FOR")
	if forwarded != "" {
		return forwarded
	}
	return h.RemoteAddr
}

// GetOssSign 获取oss签名
func (a *PublicService) GetOssSign(scene int) (*aliyun.GetOssSignResponse, error) {
	var dir string
	if scene == 1 {
		dir = dir + "user/avatar/"
	}

	if dir == "" {
		return nil, errors.New("获取签名失败")
	}

	// 生成签名
	ossSign, _ := aliyun.GetOssSign(dir)
	return ossSign, nil
}