addr.go

package glx

import (
	"fmt"
	"net"
	"strconv"
	"strings"

	"github.com/spf13/pflag"
)

type Port uint16

type Addr struct {
	IP   net.IP
	Host string
	Port Port
}

var _ pflag.Value = (*Addr)(nil)

func (a *Addr) String() string {
	var h string
	if a.IP != nil {
		h = a.IP.String()
	} else if a.Host != "" {
		h = a.Host
	}
	return fmt.Sprintf("%s:%d", h, a.Port)
}

func (a *Addr) Set(in string) error {
	chunks := strings.Split(in, ":")

	if c := strings.Join(chunks[:len(chunks)-1], ":"); c != "" {
		if ip := net.ParseIP(c); ip != nil {
			a.IP = ip
		} else {
			a.Host = c
		}
	}

	if c := chunks[len(chunks)-1]; c != "" {
		port, err := strconv.ParseUint(c, 10, 16)
		if err != nil {
			return err
		}
		a.Port = Port(port)
	}

	if a.Port == 0 {
		return fmt.Errorf("invalid address format: %s", in)
	}

	return nil
}

func (Addr) Type() string { return "Addr" }