github.com/nxtrace/NTrace-core@v1.3.1-0.20240513132635-39169291e8c9/pow/pow.go (about)

     1  package pow
     2  
     3  import (
     4  	"fmt"
     5  	"github.com/nxtrace/NTrace-core/util"
     6  	"github.com/tsosunchia/powclient"
     7  	"net/url"
     8  	"os"
     9  )
    10  
    11  const (
    12  	baseURL = "/v3/challenge"
    13  )
    14  
    15  func GetToken(fastIp string, host string, port string) (string, error) {
    16  	getTokenParams := powclient.NewGetTokenParams()
    17  	u := url.URL{Scheme: "https", Host: fastIp + ":" + port, Path: baseURL}
    18  	getTokenParams.BaseUrl = u.String()
    19  	getTokenParams.SNI = host
    20  	getTokenParams.Host = host
    21  	getTokenParams.UserAgent = util.UserAgent
    22  	proxyUrl := util.GetProxy()
    23  	if proxyUrl != nil {
    24  		getTokenParams.Proxy = proxyUrl
    25  	}
    26  	var (
    27  		token string
    28  		err   error
    29  	)
    30  	// 尝试三次RetToken，如果都失败了，异常退出
    31  	for i := 0; i < 3; i++ {
    32  		token, err = powclient.RetToken(getTokenParams)
    33  		if err != nil {
    34  			continue
    35  		}
    36  		//fmt.Println("GetToken success", token, getTokenParams.UserAgent)
    37  		return token, nil
    38  	}
    39  	if err != nil {
    40  		fmt.Println(err)
    41  	}
    42  	fmt.Println("RetToken failed 3 times, please try again after a while, exit")
    43  	os.Exit(1)
    44  	return "", err
    45  }