package tasks

import (
	"time"

	"git.rosy.net.cn/jx-callback/globals/globals2"

	"git.rosy.net.cn/baseapi/utils"
	"git.rosy.net.cn/jx-callback/globals"

	"git.rosy.net.cn/jx-callback/business/models"
	"github.com/astaxie/beego/orm"
)

const (
	weixinTokenExpires = 7200 * time.Second
	elmTokenExpires    = 20 * 24 * 3600 * time.Second
)

func RefreshConfig(configKey string, expiresTime time.Duration, configGetter func() string) {
	go func() {
		sleepGap := expiresTime / 10
		needRefreshGap := expiresTime * 8 / 10

		for {
			curConfig := &models.Config{
				Thirdparty: configKey,
			}

			db := orm.NewOrm()
			// 0: don't refresh, 1 update, 2 insert
			handleType := 0
			if err := db.Read(curConfig, "Thirdparty"); err != nil {
				if err != orm.ErrNoRows {
					globals.SugarLogger.Error("db error:%v", err)
				} else {
					curConfig.Token = configGetter()
					handleType = 2
				}
			} else {
				latestTimeStr := utils.Time2Str(time.Now().Add(-needRefreshGap))
				if curConfig.Date <= latestTimeStr {
					curConfig.Token = configGetter()
					handleType = 1
				}
			}

			if handleType != 0 {
				globals.SugarLogger.Debug(configKey)
				curConfig.Date = utils.GetCurTimeStr()
				var num int64
				var err error

				if handleType == 1 {
					num, err = db.Update(curConfig, "Token", "Date")
				} else {
					num, err = db.Insert(curConfig)
				}
				if err != nil || num == 0 {
					globals.SugarLogger.Errorf("db error:%v, num:%d", err, num)
				} else {
					if configKey == "wechat" {
						globals2.FreshFoodAPI.AccessFreshFood("weixin/accesstoken/update", nil)
					} else if configKey == "eleme" {
						globals2.FreshFoodAPI.AccessFreshFood("eleme/token/update", nil)
					}
				}
			}
			time.Sleep(sleepGap)
		}
	}()
}

func RefreshWeixinToken() {
	RefreshConfig("wechat", weixinTokenExpires, func() string {
		if tokenInfo, err := globals2.WeixinAPI.RefreshToken(); err == nil {
			return tokenInfo.AccessToken
		}
		return ""
	})
}

func RefreshElmToken() {
	RefreshConfig("eleme", elmTokenExpires, func() string {
		if tokenInfo, err := globals2.ElmAPI.RefreshToken(); err == nil {
			return string(utils.MustMarshal(tokenInfo))
		}
		return ""
	})
}