be-vpn/internal/server/server.go

package server

import (
	"be-vpn/internal/dto"
	"be-vpn/internal/model"
	"be-vpn/internal/storage"
	"fmt"
	"github.com/gin-gonic/gin"
	"log"
	"math/rand"
	"net/http"
	"sort"
	"strconv"
	"sync"
	"time"
)

var nodes = make([]*model.Node, 0)
var locker = sync.RWMutex{}
var totalFreeDuration = uint64((time.Hour * 1).Milliseconds() / 1000)

func Config(c *gin.Context) {
	deviceId := c.Query("deviceId")
	usedDuration, err := storage.GetUsedDuration(deviceId)
	if err != nil {
		dto.Error(c, err)
		return
	}
	var node *model.Node
	if len(healthNodes()) > 0 {
		node = healthNodes()[0]
	}
	freeTrialDuration := uint64(0)
	if usedDuration <= totalFreeDuration {
		freeTrialDuration = totalFreeDuration - usedDuration
	}

	c.JSON(http.StatusOK, dto.ConfigResponse{
		Response: dto.NewOkResponse(),
		Data: dto.ConfigResult{
			FreeTrialDuration: freeTrialDuration,
			Timestamp:         time.Now().Unix(),
			Node:              convert2DtoNode(node, 0),
		},
	})
}

func AddUsedDuration(c *gin.Context) {
	deviceId := c.Query("deviceId")
	usedDurationStr := c.Query("usedDuration")
	log.Printf("deviceId: %s, usedDuration: %s", deviceId, usedDurationStr)

	usedDuration, err := strconv.ParseUint(usedDurationStr, 10, 64)
	if err != nil {
		dto.Error(c, err)
		return
	}
	if existed, err := storage.AddUsedDuration(deviceId, usedDuration); err != nil {
		dto.Error(c, err)
		return
	} else {
		freeTrialDuration := totalFreeDuration - existed
		if freeTrialDuration > totalFreeDuration || freeTrialDuration < 0 {
			freeTrialDuration = 0
		}
		c.JSON(http.StatusOK, dto.ConfigResponse{
			Response: dto.NewOkResponse(),
			Data: dto.ConfigResult{
				FreeTrialDuration: freeTrialDuration,
				Timestamp:         time.Now().Unix(),
				Node:              convert2DtoNode(healthNodes()[0], 0),
			},
		})
	}
}

func Register(c *gin.Context) {
	locker.Lock()
	defer locker.Unlock()

	var request dto.RegisterRequest
	if err := c.ShouldBindJSON(&request); err != nil {
		dto.BadRequest(c, err)
		return
	}

	for _, node := range nodes {
		if node.Ip == request.Ip {
			node.Ip = request.Ip
			node.Secret = request.Secret
			node.CountryCode = request.CountryCode
			node.CountryName = request.CountryName
			node.City = request.City
			node.LastUpdateTime = time.Now()
			c.JSON(http.StatusOK, dto.RegisterResponse{
				Response: dto.NewOkResponse(),
				Data: dto.RegisterResult{
					Success: true,
				},
			})
			return
		}
	}
	node := &model.Node{
		Ip:             request.Ip,
		Secret:         request.Secret,
		LastUpdateTime: time.Now(),
	}
	nodes = append(nodes, node)
	log.Printf("update nodes: %+v", nodes)
}

func List(c *gin.Context) {
	locker.RLock()
	defer locker.RUnlock()

	nodes := healthNodes()
	sort.SliceStable(nodes, func(i, j int) bool {
		return nodes[i].CountryCode > nodes[j].CountryCode
	})

	countryLabelSeqs := make(map[string]int)
	dtoNodes := make([]*dto.Node, 0)
	for _, node := range nodes {
		seq, ok := countryLabelSeqs[node.CountryCode]
		if !ok {
			countryLabelSeqs[node.CountryCode] = 0
		} else {
			countryLabelSeqs[node.CountryCode] = seq + 1
		}
		dtoNodes = append(dtoNodes, convert2DtoNode(node, countryLabelSeqs[node.CountryCode]))
	}

	c.JSON(http.StatusOK, dto.ListResponse{
		Response: dto.NewOkResponse(),
		Data:     dtoNodes,
	})
}

func Group(c *gin.Context) {
	locker.RLock()
	defer locker.RUnlock()

	nodes := healthNodes()
	sort.SliceStable(nodes, func(i, j int) bool {
		return nodes[i].CountryCode > nodes[j].CountryCode
	})

	countryLabelSeqs := make(map[string]int)
	dtoNodes := make([]*dto.Node, 0)
	for _, node := range nodes {
		seq, ok := countryLabelSeqs[node.CountryCode]
		if !ok {
			countryLabelSeqs[node.CountryCode] = 0
		} else {
			countryLabelSeqs[node.CountryCode] = seq + 1
		}
		dtoNodes = append(dtoNodes, convert2DtoNode(node, countryLabelSeqs[node.CountryCode]))
	}

	continents := make(map[string]bool)
	for _, node := range nodes {
		continents[node.CountryCode] = true
	}

	// 再遍历出每个 node 对应的 groups
	groupDtos := make([]*dto.Group, 0)
	for continent := range continents {
		groupDto := &dto.Group{
			Continent: continent,
			Nodes:     make([]*dto.Node, 0),
		}
		for _, dtoNode := range dtoNodes {
			if dtoNode.Continent == continent {
				groupDto.Nodes = append(groupDto.Nodes, dtoNode)
			}
		}
		groupDtos = append(groupDtos, groupDto)
	}

	c.JSON(http.StatusOK, dto.GroupResponse{
		Response: dto.NewOkResponse(),
		Data:     groupDtos,
	})
}

func SecretRandom(c *gin.Context) {
	locker.RLock()
	defer locker.RUnlock()

	random := rand.Intn(len(nodes))

	for i, node := range nodes {
		if i == random {
			c.Header("Content-Disposition", "attachment; filename=client.ovpn")
			c.Data(http.StatusOK, "plain/text", []byte(node.Secret))
			return
		}
	}
	c.JSON(http.StatusNotFound, gin.H{
		"message": "not found",
	})
}

func Secret(c *gin.Context) {
	locker.RLock()
	defer locker.RUnlock()

	var request dto.DetailRequest
	if err := c.ShouldBindQuery(&request); err != nil {
		dto.BadRequest(c, err)
		return
	}

	for _, node := range nodes {
		if node.Ip == request.Ip {
			//secret, err := util.AesEncrypt([]byte(node.Secret))
			//if err != nil {
			//	dto.Error(c, err)
			//	return
			//}
			c.Header("Content-Disposition", "attachment; filename=client.ovpn")
			c.Data(http.StatusOK, "plain/text", []byte(node.Secret))
			return
		}
	}
	c.JSON(http.StatusNotFound, gin.H{
		"message": "not found ip",
	})
}

func Health(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{"status": "up"})
}

func healthNodes() []*model.Node {
	healthNodes := make([]*model.Node, 0)
	for _, node := range nodes {
		if node.LastUpdateTime.Add(10 * time.Second).After(time.Now()) {
			healthNodes = append(healthNodes, node)
		}
	}
	return healthNodes
}

func convert2DtoNode(node *model.Node, seq int) *dto.Node {
	if node == nil {
		return nil
	}
	icons := map[string]string{
		"BR": "http://v.starttransfernow.com/static/BR.jpg",
		"DE": "http://v.starttransfernow.com/static/DE.jpg",
		"HK": "http://v.starttransfernow.com/static/HK.jpg",
		"JP": "http://v.starttransfernow.com/static/JP.jpg",
		"US": "http://v.starttransfernow.com/static/US.jpg",
		"UK": "http://v.starttransfernow.com/static/UK.jpg",
		"GB": "http://v.starttransfernow.com/static/UK.jpg",
		"AU": "http://v.starttransfernow.com/static/AU.png",
		"CA": "http://v.starttransfernow.com/static/CA.png",
		"KR": "http://v.starttransfernow.com/static/KR.png",
		"SA": "http://v.starttransfernow.com/static/SA.png",
		"SG": "http://v.starttransfernow.com/static/SG.png",
		"VN": "http://v.starttransfernow.com/static/VN.png",
		"AE": "http://v.starttransfernow.com/static/AE.png",
		"BH": "http://v.starttransfernow.com/static/BH.png",
		"FR": "http://v.starttransfernow.com/static/FR.png",
		"IN": "http://v.starttransfernow.com/static/IN.png",
		"NL": "http://v.starttransfernow.com/static/NL.png",
	}
	countryLabels := map[string]string{
		"BR": "Brazil",               // aws
		"DE": "Germany",              // aws
		"HK": "Hong Kong",            // aws
		"JP": "Japan",                // aws
		"US": "United States",        // digitalocean
		"UK": "United Kingdom",       // aws
		"GB": "United Kingdom",       // aws
		"AU": "Australia",            // aws
		"CA": "Canada",               // digitalocean
		"KR": "South Korea",          // aws
		"SA": "Saudi Arabia",         // 无
		"SG": "Singapore",            //  digitalocean
		"VN": "Vietnam",              // 无
		"AE": "United Arab Emirates", // aws
		"BH": "Bahrain",              // aws
		"FR": "France",               // aws
		"IN": "India",                //  digitalocean
		"NL": "Netherlands",          //  digitalocean
	}
	countryContinents := map[string]string{
		"BR": "southam",
		"DE": "eu",
		"HK": "asia",
		"JP": "asia",
		"US": "northam",
		"UK": "eu",
		"GB": "eu",
		"AU": "oce",
		"CA": "northam",
		"KR": "asia",
		"SA": "asia",
		"SG": "asia",
		"VN": "asia",
		"AE": "asia",
		"BH": "asia",
		"FR": "eu",
		"IN": "asia",
		"NL": "eu",
	}

	countryLabel := fmt.Sprintf("%s", countryLabels[node.CountryCode])

	return &dto.Node{
		Ip:           node.Ip,
		CountryCode:  node.CountryCode,
		CountryName:  node.CountryName,
		City:         node.City,
		CountryLabel: countryLabel,
		Icon:         icons[node.CountryCode],
		SecretUrl:    fmt.Sprintf("http://v.starttransfernow.com/secret?ip=%s", node.Ip),
		Continent:    countryContinents[node.CountryCode],
	}
}