73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/tar"
|
||
|
|
"compress/gzip"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
C "github.com/Dreamacro/clash/constant"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func downloadMMDB(path string) (err error) {
|
||
|
|
resp, err := http.Get("http://geolite.maxmind.com/download/geoip/database/GeoLite2-Country.tar.gz")
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
gr, err := gzip.NewReader(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer gr.Close()
|
||
|
|
|
||
|
|
tr := tar.NewReader(gr)
|
||
|
|
for {
|
||
|
|
h, err := tr.Next()
|
||
|
|
if err == io.EOF {
|
||
|
|
break
|
||
|
|
} else if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if !strings.HasSuffix(h.Name, "GeoLite2-Country.mmdb") {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0644)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
_, err = io.Copy(f, tr)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Init prepare necessary files
|
||
|
|
func Init() {
|
||
|
|
// initial config.ini
|
||
|
|
if _, err := os.Stat(C.ConfigPath); os.IsNotExist(err) {
|
||
|
|
log.Info("Can't find config, create a empty file")
|
||
|
|
os.OpenFile(C.ConfigPath, os.O_CREATE|os.O_WRONLY, 0644)
|
||
|
|
}
|
||
|
|
|
||
|
|
// initial mmdb
|
||
|
|
if _, err := os.Stat(C.MMDBPath); os.IsNotExist(err) {
|
||
|
|
log.Info("Can't find MMDB, start download")
|
||
|
|
err := downloadMMDB(C.MMDBPath)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Can't download MMDB: %s", err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|