package service

import (
	"github.com/sirupsen/logrus"
	"hy2xs-admin/dao"
	"hy2xs-admin/model/bo"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/proxy"
	"hy2xs-admin/util"
	"strconv"
	"sync"
	"time"
)

var trafficMutex sync.Mutex
var kickMutex sync.Mutex

func CronHandleAccount() {
	go func() {
		hysteriaEnable, err := dao.GetConfig("key = ?", constant.Hysteria2Enable)
		if err != nil {
			return
		}
		if hysteriaEnable.Value != nil && *hysteriaEnable.Value == "1" {
			apiPort, err := GetHysteria2ApiPort()
			if err != nil {
				return
			}

			trafficSecretConfig, err := dao.GetConfig("key = ?", constant.Hysteria2TrafficStatsSecret)
			if err != nil {
				return
			}

			// Сохранение данных трафика
			go saveAccountTraffic(apiPort, *trafficSecretConfig.Value)

			// Принудительное отключение
			go kickAccount(apiPort, *trafficSecretConfig.Value)
		}
	}()
}

func CronResetTraffic() {
	accounts, err := dao.ListAccount(nil, nil)
	if err != nil {
		return
	}
	var ids []int64
	for _, item := range accounts {
		ids = append(ids, *item.Id)
	}
	idsList := util.SplitArr(ids, 100)
	for _, item := range idsList {
		if err := dao.UpdateAccount(item, map[string]interface{}{"download": 0, "upload": 0}); err != nil {
			continue
		}
	}
}

func saveAccountTraffic(apiPort int64, trafficStatsSecret string) {
	if !trafficMutex.TryLock() {
		return
	}
	defer trafficMutex.Unlock()

	hysteria2TrafficTime, err := dao.GetConfig("key = ?", constant.Hysteria2TrafficTime)
	if err != nil {
		return
	}
	hysteria2TrafficTimeFloat, err := strconv.ParseFloat(*hysteria2TrafficTime.Value, 64)
	if err != nil {
		logrus.Errorf("hysteria2TrafficTime string conv int64 err: %v", err)
		return
	}

	users, err := proxy.NewHysteria2Api(apiPort).ListUsers(true, trafficStatsSecret)
	if err != nil {
		return
	}
	if len(users) > 0 {
		userLists := util.SplitMap(users, 10)
		var wg sync.WaitGroup
		for _, userList := range userLists {
			wg.Add(1)
			go func(userList map[string]bo.Hysteria2UserTraffic) {
				defer wg.Done()
				for username, traffic := range userList {
					if err = dao.UpdateAccountTraffic(username, int64(float64(traffic.Rx)*hysteria2TrafficTimeFloat), int64(float64(traffic.Tx)*hysteria2TrafficTimeFloat)); err != nil {
						continue
					}
				}
			}(userList)
		}
		wg.Wait()
	}
}

func kickAccount(apiPort int64, trafficStatsSecret string) {
	if !kickMutex.TryLock() {
		return
	}
	defer kickMutex.Unlock()

	users, err := proxy.NewHysteria2Api(apiPort).OnlineUsers(trafficStatsSecret)
	if err != nil {
		return
	}
	if len(users) > 0 {
		i := 0
		usernames := make([]string, len(users))
		for k := range users {
			usernames[i] = k
			i++
		}
		usernameLists := util.SplitArr(usernames, 10)
		var wg sync.WaitGroup
		for _, usernameList := range usernameLists {
			wg.Add(1)
			go func(usernameList []string) {
				defer wg.Done()
				now := time.Now().UnixMilli()
				accounts, err := dao.ListAccount("username in ? and (deleted = 1 or (quota > 0 and quota < download + upload)) or ? > expire_time or ? < kick_util_time", usernameList, now, now)
				if err != nil {
					return
				}
				kickUsernames := make([]string, len(accounts))
				j := 0
				for _, item := range accounts {
					kickUsernames[j] = *item.Username
					j++
				}
				if err = proxy.NewHysteria2Api(apiPort).KickUsers(kickUsernames, trafficStatsSecret); err != nil {
					return
				}
			}(usernameList)
		}
		wg.Wait()
	}
}