fix tunnel

micro · Mar 21, 2022 · 3a63283 · 3a63283
1 parent 9edb03e
commit 3a63283
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tunnel/handler/tunnel.go b/tunnel/handler/tunnel.go
@@ -51,6 +51,13 @@ func New() *Tunnel {
 	}
 	proxy := v.String("")
 
+	v, err = config.Get("tunnel.token")
+	if err != nil {
+		logger.Fatalf("failed to get token: %v", err)
+	}
+
+	token := v.String("")
+
 	v, err = config.Get("tunnel.blocklist")
 	if err != nil {
 		logger.Fatalf("failed to get blocklist: %v", err)
@@ -80,6 +87,7 @@ func New() *Tunnel {
 	return &Tunnel{
 		Blocklist: blocklist,
 		Proxy:     proxy,
+		Token:     token,
 	}
 }