frp/test/e2e/pkg/port/util.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 {
	builder := &nameBuilder{name: name}
	for _, option := range options {
		option(builder)
	}
	return builder.String()
}