Add: vip

cmd/slave.go

@@ -4,6 +4,7 @@ import (
 	"be-vpn/internal/dto"
 	"bytes"
 	"encoding/json"
+	"flag"
 	"github.com/tidwall/gjson"
 	"io"
 	"log"
@@ -13,6 +14,9 @@ import (
 )
 
 func main() {
+	vip := flag.Bool("vip", false, "vip if or not")
+	flag.Parse()
+
 	file, err := os.Open("/home/ubuntu/client.ovpn")
 	if err != nil {
 		log.Fatalf("can not open file, err: %+v", err)
@@ -35,7 +39,7 @@ func main() {
 	city := gjson.Get(string(bs), "city").String()
 	log.Printf("ipinfo: %s", string(bs))
 
-	request := dto.RegisterRequest{Ip: ip, CountryCode: countryCode, CountryName: city, Secret: buf.String(), City: city}
+	request := dto.RegisterRequest{Ip: ip, CountryCode: countryCode, CountryName: city, Secret: buf.String(), City: city, Vip: *vip}
 	body, err := json.Marshal(request)
 	if err != nil {
 		log.Printf("err: %+v", err)

internal/dto/request.go

@@ -6,6 +6,7 @@ type RegisterRequest struct {
 	CountryName string `json:"countryName"`
 	City        string `json:"city"`
 	Secret      string `json:"secret"`
+	Vip         bool   `json:"vip"`
 }
 
 type DetailRequest struct {

internal/dto/response.go

@@ -62,6 +62,7 @@ type Node struct {
 	Icon         string `json:"icon"`
 	SecretUrl    string `json:"secretUrl"`
 	Continent    string `json:"continent"`
+	Vip          bool   `json:"vip"`
 }
 
 type GroupResponse struct {

internal/model/node.go

@@ -8,6 +8,7 @@ type Node struct {
 	CountryName string
 	City        string
 	Secret      string
+	Vip         bool
 
 	LastUpdateTime time.Time
 }

internal/server/server.go

@@ -91,6 +91,7 @@ func Register(c *gin.Context) {
 			node.CountryCode = request.CountryCode
 			node.CountryName = request.CountryName
 			node.City = request.City
+			node.Vip = request.Vip
 			node.LastUpdateTime = time.Now()
 			c.JSON(http.StatusOK, dto.RegisterResponse{
 				Response: dto.NewOkResponse(),
@@ -334,5 +335,6 @@ func convert2DtoNode(node *model.Node, seq int) *dto.Node {
 		Icon:         icons[node.CountryCode],
 		SecretUrl:    fmt.Sprintf("http://v.starttransfernow.com/secret?ip=%s", node.Ip),
 		Continent:    countryContinents[node.CountryCode],
+		Vip:          node.Vip,
 	}
 }