Fixes #8 Add custom user-agent config file support

cmd/serve.go

@@ -12,6 +12,7 @@ import (
 	"github.com/DataHenHQ/tillup/interceptors"
 	"github.com/DataHenHQ/tillup/logger"
 	"github.com/DataHenHQ/tillup/sessions"
+	"github.com/DataHenHQ/useragent"
 	homedir "github.com/mitchellh/go-homedir"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -39,6 +40,16 @@ var serveCmd = &cobra.Command{
 		if proxy.ForceUA {
 			fmt.Printf("Till is currently configured to override all User-Agent headers with random %v browsers\n", proxy.UAType)
 		}
+		uaConfigFile := viper.GetString("ua-config-file")
+		if uaConfigFile != "" {
+			err := useragent.LoadUAConfig(uaConfigFile)
+			if err != nil {
+				fmt.Println("Problem loading the ua-config-file:", err)
+				fmt.Println("aborting server")
+				return
+			}
+			fmt.Printf("Using ua-config-file to generate user-agent: %v\n", uaConfigFile)
+		}
 
 		// set the Token
 		token := viper.GetString("token")
@@ -183,6 +194,11 @@ func init() {
 		log.Fatal("Unable to bind flag:", err)
 	}
 
+	serveCmd.Flags().String("ua-config-file", "", "Specify the path to a JSON file that contains a custom user-agent configuration.")
+	if err := viper.BindPFlag("ua-config-file", serveCmd.Flags().Lookup("ua-config-file")); err != nil {
+		log.Fatal("Unable to bind flag:", err)
+	}
+
 	serveCmd.Flags().String("proxy-file", "", "Specify the path to a txt file that contains a list of proxies")
 	if err := viper.BindPFlag("proxy-file", serveCmd.Flags().Lookup("proxy-file")); err != nil {
 		log.Fatal("Unable to bind flag:", err)