package oauth_controller

import (
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
	"schisandra-cloud-album/common/constant"
	"schisandra-cloud-album/common/redis"
	"schisandra-cloud-album/common/result"
	"schisandra-cloud-album/global"
	"schisandra-cloud-album/service/impl"
	"schisandra-cloud-album/utils"
	"sync"
	"time"
)

type OAuthController struct{}

var userSocialService = impl.UserSocialServiceImpl{}
var userService = impl.UserServiceImpl{}

type Token struct {
	AccessToken string `json:"access_token"`
}

var script = `
        <script>
        window.opener.postMessage('%s', '%s');
        window.close();
        </script>
        `

func HandleLoginResponse(c *gin.Context, uid string) {
	// 查询用户信息
	user := userService.QueryUserByUuidService(&uid)

	var accessToken, refreshToken string
	var expiresAt int64
	var err error
	var wg sync.WaitGroup
	var accessTokenErr error

	wg.Add(2) // 增加计数器，等待两个协程完成

	// 使用goroutine生成accessToken
	go func() {
		defer wg.Done() // 完成时减少计数器
		accessToken, accessTokenErr = utils.GenerateAccessToken(utils.AccessJWTPayload{UserID: &uid})
	}()

	// 使用goroutine生成refreshToken
	go func() {
		defer wg.Done() // 完成时减少计数器
		refreshToken, expiresAt = utils.GenerateRefreshToken(utils.RefreshJWTPayload{UserID: &uid}, time.Hour*24*7)
	}()

	// 等待两个协程完成
	wg.Wait()

	// 检查生成accessToken时是否有错误
	if accessTokenErr != nil {
		global.LOG.Error(accessTokenErr)
		return
	}

	data := ResponseData{
		AccessToken:  accessToken,
		RefreshToken: refreshToken,
		ExpiresAt:    expiresAt,
		UID:          &uid,
		UserInfo: UserInfo{
			Username: user.Username,
			Nickname: user.Nickname,
			Avatar:   user.Avatar,
			Email:    user.Email,
			Phone:    user.Phone,
			Gender:   user.Gender,
			Status:   user.Status,
			CreateAt: *user.CreatedTime,
		},
	}

	if err = utils.SetSession(c, constant.SessionKey, data); err != nil {
		return
	}

	// 将数据存入redis
	if err = redis.Set(constant.UserLoginTokenRedisKey+uid, data, time.Hour*24*7).Err(); err != nil {
		global.LOG.Error(err)
		return
	}

	responseData := result.Response{
		Data:    data,
		Message: "success",
		Code:    200,
		Success: true,
	}

	tokenData, err := json.Marshal(responseData)
	if err != nil {
		global.LOG.Error(err)
		return
	}

	formattedScript := fmt.Sprintf(script, tokenData, global.CONFIG.System.WebURL())
	c.Data(http.StatusOK, "text/html; charset=utf-8", []byte(formattedScript))
	return
}