mirror of https://github.com/fatedier/frp.git
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package port
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
NameDelimiter = "_"
|
|
)
|
|
|
|
type NameOption func(*nameBuilder) *nameBuilder
|
|
|
|
type nameBuilder struct {
|
|
name string
|
|
rangePortFrom int
|
|
rangePortTo int
|
|
}
|
|
|
|
func unmarshalFromName(name string) (*nameBuilder, error) {
|
|
var builder nameBuilder
|
|
arrs := strings.Split(name, NameDelimiter)
|
|
switch len(arrs) {
|
|
case 2:
|
|
builder.name = arrs[1]
|
|
case 4:
|
|
builder.name = arrs[1]
|
|
if fromPort, err := strconv.Atoi(arrs[2]); err != nil {
|
|
return nil, fmt.Errorf("error range port from")
|
|
} else {
|
|
builder.rangePortFrom = fromPort
|
|
}
|
|
if toPort, err := strconv.Atoi(arrs[3]); err != nil {
|
|
return nil, fmt.Errorf("error range port to")
|
|
} else {
|
|
builder.rangePortTo = toPort
|
|
}
|
|
default:
|
|
return nil, fmt.Errorf("error port name format")
|
|
}
|
|
return &builder, nil
|
|
}
|
|
|
|
func (builder *nameBuilder) String() string {
|
|
name := fmt.Sprintf("Port%s%s", NameDelimiter, builder.name)
|
|
if builder.rangePortFrom > 0 && builder.rangePortTo > 0 && builder.rangePortTo > builder.rangePortFrom {
|
|
name += fmt.Sprintf("%s%d%s%d", NameDelimiter, builder.rangePortFrom, NameDelimiter, builder.rangePortTo)
|
|
}
|
|
return name
|
|
}
|
|
|
|
func WithRangePorts(from, to int) NameOption {
|
|
return func(builder *nameBuilder) *nameBuilder {
|
|
builder.rangePortFrom = from
|
|
builder.rangePortTo = to
|
|
return builder
|
|
}
|
|
}
|
|
|
|
func GenName(name string, options ...NameOption) string {
|
|
name = strings.ReplaceAll(name, "-", "")
|
|
name = strings.ReplaceAll(name, "_", "")
|
|
builder := &nameBuilder{name: name}
|
|
for _, option := range options {
|
|
builder = option(builder)
|
|
}
|
|
return builder.String()
|
|
}
|