nali/internal/app/parse.go

package app

import (
	"fmt"
	"path/filepath"

	"github.com/zu1k/nali/constant"

	"github.com/zu1k/nali/internal/ipdb"

	geoip2 "github.com/zu1k/nali/pkg/geoip"
	"github.com/zu1k/nali/pkg/qqwry"
)

var (
	db    ipdb.IPDB
	qqip  qqwry.QQwry
	geoip geoip2.GeoIP
)

// init ip db content
func InitIPDB() {
	qqip = qqwry.NewQQwry(filepath.Join(constant.HomePath, "qqwry.dat"))
	//geoip = geoip2.NewGeoIP(filepath.Join(constant.HomePath, "GeoLite2-City.mmdb"))

	db = qqip
}

// set db to use
func SetDB(dbName ipdb.IPDBType) {
	switch dbName {
	case ipdb.GEOIP2:
		db = geoip
	case ipdb.QQIP:
		db = qqip
	}
}

// parse several ips
func ParseIPs(ips []string) {
	for _, ip := range ips {
		ParseIP(ip)
	}
}

// parse one ip
func ParseIP(ip string) {
	result := db.Find(ip)
	fmt.Println(formatResult(ip, result))
}

func formatResult(ip string, result string) string {
	if result == "" {
		result = "未找到"
	}
	return fmt.Sprintf("%s [%s]", ip, result)
}