44 lines
1 KiB
Go
44 lines
1 KiB
Go
package geodata
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/metacubex/mihomo/component/geodata/router"
|
|
C "github.com/metacubex/mihomo/constant"
|
|
)
|
|
|
|
type loader struct {
|
|
LoaderImplementation
|
|
}
|
|
|
|
func (l *loader) LoadGeoSite(list string) ([]*router.Domain, error) {
|
|
return l.LoadSiteByPath(C.GeositeName, list)
|
|
}
|
|
|
|
func (l *loader) LoadGeoIP(country string) ([]*router.CIDR, error) {
|
|
return l.LoadIPByPath(C.GeoipName, country)
|
|
}
|
|
|
|
var loaders map[string]func() LoaderImplementation
|
|
|
|
func RegisterGeoDataLoaderImplementationCreator(name string, loader func() LoaderImplementation) {
|
|
if loaders == nil {
|
|
loaders = map[string]func() LoaderImplementation{}
|
|
}
|
|
loaders[name] = loader
|
|
}
|
|
|
|
func getGeoDataLoaderImplementation(name string) (LoaderImplementation, error) {
|
|
if geoLoader, ok := loaders[name]; ok {
|
|
return geoLoader(), nil
|
|
}
|
|
return nil, fmt.Errorf("unable to locate GeoData loader %s", name)
|
|
}
|
|
|
|
func GetGeoDataLoader(name string) (Loader, error) {
|
|
loadImpl, err := getGeoDataLoaderImplementation(name)
|
|
if err == nil {
|
|
return &loader{loadImpl}, nil
|
|
}
|
|
return nil, err
|
|
}
|