mihomo/config/initial.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())
		}
	}
}