optimize: pull subscription using custom UA (#233)

cmd/cmd.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"github.com/daeuniverse/dae/config"
 	"github.com/spf13/cobra"
 )
 
@@ -17,6 +18,10 @@ var (
 	}
 )
 
+func init() {
+	config.Version = Version
+}
+
 // Execute executes the root command.
 func Execute() error {
 	return rootCmd.Execute()

common/subscription/subscription.go

@@ -20,6 +20,7 @@ import (
 	"strings"
 
 	"github.com/daeuniverse/dae/common"
+	"github.com/daeuniverse/dae/config"
 	"github.com/sirupsen/logrus"
 )
 
@@ -149,6 +150,7 @@ func ResolveSubscription(log *logrus.Logger, client *http.Client, configDir stri
 	log.Debugf("ResolveSubscription: %v", subscription)
 	var (
 		b    []byte
+		req  *http.Request
 		resp *http.Response
 	)
 	switch u.Scheme {
@@ -160,7 +162,11 @@ func ResolveSubscription(log *logrus.Logger, client *http.Client, configDir stri
 		goto resolve
 	default:
 	}
-	resp, err = client.Get(subscription)
+	req, err = http.NewRequest("GET", subscription, nil)
+	if err != nil {
+		return "", nil, err
+	}
+	req.Header.Set("User-Agent", fmt.Sprintf("dae/%v (like v2rayA/1.0 WebRequestHelper) (like v2rayN/1.0 WebRequestHelper)", config.Version))
 	if err != nil {
 		return "", nil, err
 	}

config/config.go

@@ -13,6 +13,10 @@ import (
 	"github.com/daeuniverse/dae/pkg/config_parser"
 )
 
+var (
+	Version string
+)
+
 type Global struct {
 	TproxyPort        uint16 `mapstructure:"tproxy_port" default:"12345"`
 	TproxyPortProtect bool   `mapstructure:"tproxy_port_protect" default:"true"`