-
Notifications
You must be signed in to change notification settings - Fork 19
/
server_name_config.go
84 lines (74 loc) · 2.24 KB
/
server_name_config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package serverconfigs
import (
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
"github.com/iwind/TeaGo/lists"
"strings"
)
type ServerNameType = string
const (
ServerNameTypeFull ServerNameType = "full" // 完整的域名,包含通配符等
ServerNameTypePrefix ServerNameType = "prefix" // 前缀
ServerNameTypeSuffix ServerNameType = "suffix" // 后缀
ServerNameTypeMatch ServerNameType = "match" // 正则匹配
)
// ServerNameConfig 主机名(域名)配置
type ServerNameConfig struct {
Name string `yaml:"name" json:"name"` // 名称
Type string `yaml:"type" json:"type"` // 类型
SubNames []string `yaml:"subNames" json:"subNames"` // 子名称,用来支持大量的域名批量管理
}
// Normalize 格式化域名
func (this *ServerNameConfig) Normalize() {
this.Name = strings.ToLower(this.Name)
for index, subName := range this.SubNames {
this.SubNames[index] = strings.ToLower(subName)
}
}
// Match 判断主机名是否匹配
func (this *ServerNameConfig) Match(name string) bool {
if len(this.SubNames) > 0 {
return configutils.MatchDomains(this.SubNames, name)
}
return configutils.MatchDomains([]string{this.Name}, name)
}
// FirstName 获取第一个名称
func (this *ServerNameConfig) FirstName() string {
if len(this.Name) > 0 {
return this.Name
}
if len(this.SubNames) > 0 {
return this.SubNames[0]
}
return ""
}
// Count 计算域名数量
func (this *ServerNameConfig) Count() int {
if len(this.SubNames) > 0 {
return len(this.SubNames)
}
return 1
}
// NormalizeServerNames 格式化一组域名
func NormalizeServerNames(serverNames []*ServerNameConfig) {
for _, serverName := range serverNames {
serverName.Normalize()
}
}
// PlainServerNames 获取所有域名
func PlainServerNames(serverNames []*ServerNameConfig) (result []string) {
NormalizeServerNames(serverNames)
for _, serverName := range serverNames {
if len(serverName.SubNames) == 0 {
if len(serverName.Name) > 0 && !lists.ContainsString(result, serverName.Name) {
result = append(result, serverName.Name)
}
} else {
for _, subName := range serverName.SubNames {
if len(subName) > 0 && !lists.ContainsString(result, subName) {
result = append(result, subName)
}
}
}
}
return result
}