// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
//go:build plus

package plus

import (
	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
	plusutils "github.com/TeaOSLab/EdgePlus/pkg/utils"
	"github.com/iwind/TeaGo/lists"
	"net"
	"sync"
)

// 已经开通的组件
var plusLocker = sync.Mutex{}
var plusComponents = []string{}

var PlusEdition = ""

// UpdateComponents 修改开通的组件
func UpdateComponents(edition string, components []string) {
	plusLocker.Lock()
	PlusEdition = edition
	plusComponents = components
	plusLocker.Unlock()
}

// AllowComponent 检查是否允许使用某个组件
func AllowComponent(component ComponentCode) bool {
	if !teaconst.IsPlus {
		return false
	}

	// 允许不指定组件，表示只要是plus用户就可以
	if len(component) == 0 {
		return true
	}

	plusLocker.Lock()
	defer plusLocker.Unlock()

	if len(plusComponents) == 0 {
		return false
	}

	for _, c := range plusComponents {
		if c == "*" || c == component {
			// 如果不是最新版本的授权，则忽略一些权限
			if c == "*" && (len(PlusEdition) == 0 || plusutils.CompareEdition(PlusEdition, plusutils.EditionPro) < 0) && component == ComponentCodeAntiDDoS {
				continue
			}

			return true
		}
	}
	return false
}

// ValidateMac 校验MAC地址
func ValidateMac(mac string) bool {
	if len(mac) == 0 || mac == "*" {
		return true
	}

	for _, addr := range FindAllMacAddresses() {
		if addr == mac {
			return true
		}
	}
	return false
}

// FindAllMacAddresses 获取MAC地址
func FindAllMacAddresses() []string {
	var macAddresses = []string{}
	interfaces, _ := net.Interfaces()
	for _, i := range interfaces {
		var addr = i.HardwareAddr.String()
		if len(addr) == 0 {
			continue
		}
		if lists.ContainsString(macAddresses, addr) {
			continue
		}
		if i.Flags&net.FlagLoopback == 1 {
			continue
		}
		if i.Flags&net.FlagUp == 0 {
			continue
		}
		ipAddrs, _ := i.Addrs()
		if len(ipAddrs) == 0 {
			continue
		}
		macAddresses = append(macAddresses, addr)
	}
	return macAddresses
}