// Copyright 2021 GoEdge goedge.cdn@gmail.com. All rights reserved.

package teaconst

import (
	"os"
	"strings"

	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
)

var (
	// 流量统计

	InTrafficBytes  = uint64(0)
	OutTrafficBytes = uint64(0)

	NodeId       int64 = 0
	NodeIdString       = ""
	IsMain             = checkMain()

	// Is301 是否 301 节点
	Is301 = false

	// BypassMobile 是否过移动
	BypassMobile int32 = 0

	GlobalProductName = nodeconfigs.DefaultProductName

	IsQuiting    = false // 是否正在退出
	EnableDBStat = false // 是否开启本地数据库统计
)

// 检查是否为主程序
func checkMain() bool {
	if len(os.Args) == 1 ||
		(len(os.Args) >= 2 && os.Args[1] == "pprof") {
		return true
	}
	exe, _ := os.Executable()
	return strings.HasSuffix(exe, ".test") ||
		strings.HasSuffix(exe, ".test.exe") ||
		strings.Contains(exe, "___")
}