[test]

config/initial.go

@@ -12,7 +12,7 @@ import (
 )
 
 func downloadMMDB(path string) (err error) {
-	resp, err := http.Get("https://cdn.jsdelivr.net/gh/Dreamacro/maxmind-geoip@release/Country.mmdb")
+	resp, err := http.Get("https://cdn.jsdelivr.net/gh/Loyalsoldier/geoip@release/Country.mmdb")
 	if err != nil {
 		return
 	}
@@ -50,6 +50,65 @@ func initMMDB() error {
 	return nil
 }
 
+//func downloadGeoIP(path string) (err error) {
+//	resp, err := http.Get("https://cdn.jsdelivr.net/gh/Loyalsoldier/v2ray-rules-dat@release/geoip.dat")
+//	if err != nil {
+//		return
+//	}
+//	defer resp.Body.Close()
+//
+//	f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0644)
+//	if err != nil {
+//		return err
+//	}
+//	defer f.Close()
+//	_, err = io.Copy(f, resp.Body)
+//
+//	return err
+//}
+
+func downloadGeoSite(path string) (err error) {
+	resp, err := http.Get("https://cdn.jsdelivr.net/gh/Loyalsoldier/v2ray-rules-dat@release/geosite.dat")
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+
+	f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0o644)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	_, err = io.Copy(f, resp.Body)
+
+	return err
+}
+
+//
+//func initGeoIP() error {
+//	if _, err := os.Stat(C.Path.GeoIP()); os.IsNotExist(err) {
+//		log.Infoln("Can't find GeoIP.dat, start download")
+//		if err := downloadGeoIP(C.Path.GeoIP()); err != nil {
+//			return fmt.Errorf("can't download GeoIP.dat: %s", err.Error())
+//		}
+//		log.Infoln("Download GeoIP.dat finish")
+//	}
+//
+//	return nil
+//}
+
+func initGeoSite() error {
+	if _, err := os.Stat(C.Path.GeoSite()); os.IsNotExist(err) {
+		log.Infoln("Can't find GeoSite.dat, start download")
+		if err := downloadGeoSite(C.Path.GeoSite()); err != nil {
+			return fmt.Errorf("can't download GeoSite.dat: %s", err.Error())
+		}
+		log.Infoln("Download GeoSite.dat finish")
+	}
+
+	return nil
+}
+
 // Init prepare necessary files
 func Init(dir string) error {
 	// initial homedir
@@ -70,9 +129,19 @@ func Init(dir string) error {
 		f.Close()
 	}
 
+	//// initial GeoIP
+	//if err := initGeoIP(); err != nil {
+	//	return fmt.Errorf("can't initial GeoIP: %w", err)
+	//}
+
 	// initial mmdb
 	if err := initMMDB(); err != nil {
 		return fmt.Errorf("can't initial MMDB: %w", err)
 	}
+
+	// initial GeoSite
+	if err := initGeoSite(); err != nil {
+		return fmt.Errorf("can't initial GeoSite: %w", err)
+	}
 	return nil
 }