EdgeAdmin/internal/web/actions/default/servers/serverutils/utils.go

package serverutils

import (
	"github.com/iwind/TeaGo/lists"
	"github.com/iwind/TeaGo/maps"
)

type ServerType = string

const (
	ServerTypeHTTPProxy ServerType = "httpProxy"
	ServerTypeHTTPWeb   ServerType = "httpWeb"
	ServerTypeTCPProxy  ServerType = "tcpProxy"
	ServerTypeUnixProxy ServerType = "unixProxy"
	ServerTypeUDPProxy  ServerType = "udp"
)

// 获取所有的服务类型
func AllServerTypes() []maps.Map {
	return []maps.Map{
		{
			"name": "HTTP反向代理",
			"code": ServerTypeHTTPProxy,
		},
		{
			"name": "HTTP Web服务",
			"code": ServerTypeHTTPWeb,
		},
		{
			"name": "TCP反向代理",
			"code": ServerTypeTCPProxy,
		},
		/**{
			"name": "UNIX协议反向代理",
			"code": ServerTypeUnixProxy,
		},
		{
			"name": "UDP反向代理",
			"code": ServerTypeUDPProxy,
		},**/
	}
}

// 查找服务类型
func FindServerType(code string) maps.Map {
	for _, m := range AllServerTypes() {
		if m.GetString("code") == code {
			return m
		}
	}
	return nil
}

// 获取所有协议
func AllServerProtocolsForType(serverType ServerType) []maps.Map {
	protocols := []maps.Map{
		{
			"name":        "HTTP",
			"code":        "http",
			"serverTypes": []ServerType{ServerTypeHTTPProxy, ServerTypeHTTPWeb},
		},
		{
			"name":        "HTTPS",
			"code":        "https",
			"serverTypes": []ServerType{ServerTypeHTTPProxy, ServerTypeHTTPWeb},
		},
		{
			"name":        "TCP",
			"code":        "tcp",
			"serverTypes": []ServerType{ServerTypeTCPProxy},
		},
		{
			"name":        "TLS",
			"code":        "tls",
			"serverTypes": []ServerType{ServerTypeTCPProxy},
		},
		{
			"name":        "Unix",
			"code":        "unix",
			"serverTypes": []ServerType{ServerTypeUnixProxy},
		},
		{
			"name":        "UDP",
			"code":        "udp",
			"serverTypes": []ServerType{ServerTypeUDPProxy},
		},
	}

	result := []maps.Map{}
	for _, p := range protocols {
		serverTypes := p.GetSlice("serverTypes")
		if lists.Contains(serverTypes, serverType) {
			result = append(result, p)
		}
	}
	return result
}