mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-08-23 10:42:33 +00:00
81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package geodata
|
||
|
||
import (
|
||
"github.com/Dreamacro/clash/component/geodata/router"
|
||
C "github.com/Dreamacro/clash/constant"
|
||
"strings"
|
||
)
|
||
|
||
var geoLoaderName = "memconservative"
|
||
|
||
// geoLoaderName = "standard"
|
||
|
||
func LoaderName() string {
|
||
return geoLoaderName
|
||
}
|
||
|
||
func SetLoader(newLoader string) {
|
||
geoLoaderName = newLoader
|
||
}
|
||
|
||
func Verify(name string) bool {
|
||
switch name {
|
||
case C.GeositeName:
|
||
_, _, err := LoadGeoSiteMatcher("CN")
|
||
return err == nil
|
||
case C.GeoipName:
|
||
_, _, err := LoadGeoIPMatcher("CN")
|
||
return err == nil
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func LoadGeoSiteMatcher(countryCode string) (*router.DomainMatcher, int, error) {
|
||
geoLoader, err := GetGeoDataLoader(geoLoaderName)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
domains, err := geoLoader.LoadGeoSite(countryCode)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
/**
|
||
linear: linear algorithm
|
||
matcher, err := router.NewDomainMatcher(domains)
|
||
mph:minimal perfect hash algorithm
|
||
*/
|
||
matcher, err := router.NewMphMatcherGroup(domains)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
return matcher, len(domains), nil
|
||
}
|
||
|
||
func LoadGeoIPMatcher(country string) (*router.GeoIPMatcher, int, error) {
|
||
geoLoader, err := GetGeoDataLoader(geoLoaderName)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
records, err := geoLoader.LoadGeoIP(strings.ReplaceAll(country, "!", ""))
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
geoIP := &router.GeoIP{
|
||
CountryCode: country,
|
||
Cidr: records,
|
||
ReverseMatch: strings.Contains(country, "!"),
|
||
}
|
||
|
||
matcher, err := router.NewGeoIPMatcher(geoIP)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
return matcher, len(records), nil
|
||
}
|