feat(ipinfo): support GeoIP (#26)

Co-authored-by: E99p1ant <[email protected]>

internal/ipinfo/qqwry/download.go

@@ -0,0 +1,99 @@
+package qqwry
+
+import (
+	"bytes"
+	"compress/zlib"
+	"crypto/tls"
+	"encoding/binary"
+	"errors"
+	"io"
+	"net/http"
+	"os"
+	"sync"
+	"time"
+
+	log "unknwon.dev/clog/v2"
+)
+
+const (
+	CopyWriteUrl = "https://qqwry.mirror.noc.one/copywrite.rar"
+	Url          = "https://qqwry.mirror.noc.one/qqwry.rar"
+)
+
+func get(url string) (b []byte, err error) {
+	client := http.Client{
+		Timeout: 90 * time.Second,
+		Transport: &http.Transport{
+			TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, // disable verify
+		}}
+	request, _ := http.NewRequest("GET", url, nil)
+	request.Header.Add("User-Agent", "Nali/2.1.2 (Nali CLI, https://nali.skk.moe)")
+
+	resp, err := client.Do(request)
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+
+	return io.ReadAll(resp.Body)
+}
+
+func getKey(b []byte) (key uint32, err error) {
+	if len(b) != 280 {
+		return 0, errors.New("copywrite.rar is corrupt")
+	}
+	key = binary.LittleEndian.Uint32(b[20:])
+	return key, err
+}
+
+func decrypt(b []byte, key uint32) (_ []byte, err error) {
+	for i := 0; i < 0x200; i++ {
+		key *= uint32(0x805)
+		key++
+		key &= uint32(0xff)
+		b[i] = b[i] ^ byte(key)
+	}
+	rc, err := zlib.NewReader(bytes.NewBuffer(b))
+	if err != nil {
+		return
+	}
+	defer rc.Close()
+	return io.ReadAll(rc)
+}
+
+func download() (err error) {
+	var (
+		copyWriteData, qqwryData []byte
+		wg                       sync.WaitGroup
+		key                      uint32
+	)
+	log.Info("Downloading qqwry.dat...")
+
+	wg.Add(2)
+	go func() {
+		defer wg.Done()
+		if copyWriteData, err = get(CopyWriteUrl); err != nil {
+			return
+		}
+	}()
+
+	go func() {
+		defer wg.Done()
+		if qqwryData, err = get(Url); err != nil {
+			return
+		}
+	}()
+	wg.Wait()
+
+	if err != nil {
+		return err
+	}
+	if key, err = getKey(copyWriteData); err != nil {
+		return err
+	}
+	if b, err := decrypt(qqwryData, key); err != nil {
+		return err
+	} else {
+		return os.WriteFile("qqwry.dat", b, 0644)
+	}
+}

internal/ipinfo/qqwry/wry.go

@@ -0,0 +1,66 @@
+package qqwry
+
+import (
+	"os"
+	"sync"
+	"time"
+
+	"github.com/sinlov/qqwry-golang/qqwry"
+	log "unknwon.dev/clog/v2"
+)
+
+type Database struct {
+	wry *qqwry.QQwry
+}
+
+// Area returns IpArea according to ip
+func (db *Database) Area(ip string) string {
+	defer func() {
+		_ = recover()
+	}()
+	if db.wry == nil {
+		return ""
+	}
+	ipData := db.wry.SearchByIPv4(ip)
+	if ipData.Area == " CZ88.NET" {
+		return ipData.Country
+	}
+	return ipData.Country + " " + ipData.Area
+}
+
+var wry *qqwry.QQwry
+var once sync.Once
+
+func checkUpdate() {
+	info, err := os.Stat("qqwry.dat")
+	if err != nil {
+		if os.IsNotExist(err) {
+			err := download()
+			if err != nil {
+				log.Warn("Download qqwry.dat failed, caused by:%v, recommend to download it by yourself otherwise the `IpArea` will be null", err)
+			}
+		}
+	} else if -time.Until(info.ModTime()) > 7*24*time.Hour {
+		log.Info("Updating qqwry.dat...")
+		err := download()
+		if err != nil {
+			log.Warn("Update qqwry.dat failed, please download qqwry.dat by yourself")
+		}
+	}
+}
+
+func New() *Database {
+	once.Do(func() {
+		checkUpdate()
+		qqwry.DatData.FilePath = "qqwry.dat"
+		init := qqwry.DatData.InitDatFile()
+		if v, ok := init.(error); ok {
+			if v != nil {
+				log.Warn("qqwry init failed")
+				wry = nil
+			}
+		}
+		wry = qqwry.NewQQwry()
+	})
+	return &Database{wry: wry}
+}