jh163888/sing-geosite
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-geosite.git synced 2025-02-11 23:32:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support domain attr

Browse Source
This commit is contained in:
世界 2022-09-04 12:40:46 +08:00
parent ec6544f157
commit 50c4bf766c
No known key found for this signature in database
GPG Key ID: CD109927C34A63C4
1 changed files with 40 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
main.go
View File

@ -88,9 +88,14 @@ func parse(vGeositeData []byte) (map[string][]geosite.Item, error) {
} }
domainMap := make(map[string][]geosite.Item) domainMap := make(map[string][]geosite.Item)
for _, vGeositeEntry := range vGeositeList.Entry { for _, vGeositeEntry := range vGeositeList.Entry {
code := strings.ToLower(vGeositeEntry.CountryCode)
domains := make([]geosite.Item, 0, len(vGeositeEntry.Domain)*2) domains := make([]geosite.Item, 0, len(vGeositeEntry.Domain)*2)
attributes := make(map[string][]*routercommon.Domain)
for _, domain := range vGeositeEntry.Domain { for _, domain := range vGeositeEntry.Domain {
if len(domain.Attribute) > 0 { if len(domain.Attribute) > 0 {
for _, attribute := range domain.Attribute {
attributes[attribute.Key] = append(attributes[attribute.Key], domain)
}
continue continue
} }
switch domain.Type { switch domain.Type {
@ -122,7 +127,41 @@ func parse(vGeositeData []byte) (map[string][]geosite.Item, error) {
}) })
} }
} }
domainMap[strings.ToLower(vGeositeEntry.CountryCode)] = common.Uniq(domains) domainMap[code] = common.Uniq(domains)
for attribute, attributeEntries := range attributes {
attributeDomains := make([]geosite.Item, 0, len(attributeEntries)*2)
for _, domain := range attributeEntries {
switch domain.Type {
case routercommon.Domain_Plain:
attributeDomains = append(attributeDomains, geosite.Item{
Type: geosite.RuleTypeDomainKeyword,
Value: domain.Value,
})
case routercommon.Domain_Regex:
attributeDomains = append(attributeDomains, geosite.Item{
Type: geosite.RuleTypeDomainRegex,
Value: domain.Value,
})
case routercommon.Domain_RootDomain:
if strings.Contains(domain.Value, ".") {
attributeDomains = append(attributeDomains, geosite.Item{
Type: geosite.RuleTypeDomain,
Value: domain.Value,
})
}
attributeDomains = append(attributeDomains, geosite.Item{
Type: geosite.RuleTypeDomainSuffix,
Value: "." + domain.Value,
})
case routercommon.Domain_Full:
attributeDomains = append(attributeDomains, geosite.Item{
Type: geosite.RuleTypeDomain,
Value: domain.Value,
})
}
}
domainMap[code+"@"+attribute] = common.Uniq(attributeDomains)
}
} }
return domainMap, nil return domainMap, nil
} }