mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2025-12-20 00:50:06 +08:00
ae7967f662
Some checks are pending
Test / test (1.20, macos-13) (push) Waiting to run
Test / test (1.20, macos-latest) (push) Waiting to run
Test / test (1.20, ubuntu-24.04-arm) (push) Waiting to run
Test / test (1.20, ubuntu-latest) (push) Waiting to run
Test / test (1.20, windows-latest) (push) Waiting to run
Test / test (1.21, macos-13) (push) Waiting to run
Test / test (1.21, macos-latest) (push) Waiting to run
Test / test (1.21, ubuntu-24.04-arm) (push) Waiting to run
Test / test (1.21, ubuntu-latest) (push) Waiting to run
Test / test (1.21, windows-latest) (push) Waiting to run
Test / test (1.22, macos-13) (push) Waiting to run
Test / test (1.22, macos-latest) (push) Waiting to run
Test / test (1.22, ubuntu-24.04-arm) (push) Waiting to run
Test / test (1.22, ubuntu-latest) (push) Waiting to run
Test / test (1.22, windows-latest) (push) Waiting to run
Test / test (1.23, macos-13) (push) Waiting to run
Test / test (1.23, macos-latest) (push) Waiting to run
Test / test (1.23, ubuntu-24.04-arm) (push) Waiting to run
Test / test (1.23, ubuntu-latest) (push) Waiting to run
Test / test (1.23, windows-latest) (push) Waiting to run
Test / test (1.24, macos-13) (push) Waiting to run
Test / test (1.24, macos-latest) (push) Waiting to run
Test / test (1.24, ubuntu-24.04-arm) (push) Waiting to run
Test / test (1.24, ubuntu-latest) (push) Waiting to run
Test / test (1.24, windows-latest) (push) Waiting to run
Trigger CMFA Update / trigger-CMFA-update (push) Waiting to run
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/metacubex/mihomo/component/geodata"
|
|
_ "github.com/metacubex/mihomo/component/geodata/memconservative"
|
|
"github.com/metacubex/mihomo/component/geodata/router"
|
|
_ "github.com/metacubex/mihomo/component/geodata/standard"
|
|
C "github.com/metacubex/mihomo/constant"
|
|
"github.com/metacubex/mihomo/log"
|
|
)
|
|
|
|
type GEOSITE struct {
|
|
*Base
|
|
country string
|
|
adapter string
|
|
recodeSize int
|
|
}
|
|
|
|
func (gs *GEOSITE) RuleType() C.RuleType {
|
|
return C.GEOSITE
|
|
}
|
|
|
|
func (gs *GEOSITE) Match(metadata *C.Metadata, helper C.RuleMatchHelper) (bool, string) {
|
|
return gs.MatchDomain(metadata.RuleHost()), gs.adapter
|
|
}
|
|
|
|
// MatchDomain implements C.DomainMatcher
|
|
func (gs *GEOSITE) MatchDomain(domain string) bool {
|
|
if len(domain) == 0 {
|
|
return false
|
|
}
|
|
matcher, err := gs.GetDomainMatcher()
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return matcher.ApplyDomain(domain)
|
|
}
|
|
|
|
func (gs *GEOSITE) Adapter() string {
|
|
return gs.adapter
|
|
}
|
|
|
|
func (gs *GEOSITE) Payload() string {
|
|
return gs.country
|
|
}
|
|
|
|
func (gs *GEOSITE) GetDomainMatcher() (router.DomainMatcher, error) {
|
|
matcher, err := geodata.LoadGeoSiteMatcher(gs.country)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("load GeoSite data error, %w", err)
|
|
}
|
|
return matcher, nil
|
|
}
|
|
|
|
func (gs *GEOSITE) GetRecodeSize() int {
|
|
if matcher, err := gs.GetDomainMatcher(); err == nil {
|
|
return matcher.Count()
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func NewGEOSITE(country string, adapter string) (*GEOSITE, error) {
|
|
if err := geodata.InitGeoSite(); err != nil {
|
|
log.Errorln("can't initial GeoSite: %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
geoSite := &GEOSITE{
|
|
Base: &Base{},
|
|
country: country,
|
|
adapter: adapter,
|
|
}
|
|
|
|
matcher, err := geoSite.GetDomainMatcher() // test load
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Infoln("Finished initial GeoSite rule %s => %s, records: %d", country, adapter, matcher.Count())
|
|
|
|
return geoSite, nil
|
|
}
|
|
|
|
var _ C.Rule = (*GEOSITE)(nil)
|