fix: Geoip lang select

Signed-off-by: zu1k <[email protected]>
lmafwo · May 9, 2022 · 9e049a4 · 9e049a4
1 parent 45878d4
commit 9e049a4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/pkg/geoip/geoip.go b/pkg/geoip/geoip.go
@@ -7,6 +7,8 @@ import (
 	"net"
 	"os"
 
+	"github.com/spf13/viper"
+
 	"github.com/oschwald/geoip2-golang"
 )
 
@@ -41,11 +43,9 @@ func (g GeoIP) Find(query string, params ...string) (result fmt.Stringer, err er
 		return
 	}
 
-	lang := "zh-CN"
-	if len(params) > 0 {
-		if _, ok := record.Country.Names[params[0]]; ok {
-			lang = params[0]
-		}
+	lang := viper.GetString("selected.lang")
+	if lang == "" {
+		lang = "zh-CN"
 	}
 
 	result = Result{