api

type AddPortRequestMessage struct {
	Type          string `json:"type"`
	ID            string `json:"id"`
	RequestedPort int    `json:"port"`
}

type AddPortResponseMessage struct {
	Type         string `json:"tcp"`
	ID           string `json:"id"`
	AssignedPort int    `json:"port"`
	Address      string `json:"address"`
}

type Client interface {
	Connect(ctx context.Context) error
	AddTCPPort(id string, requestedPort int) (net.Listener, error)
}

type ControlStream interface {
	Stream
	AddMessageHandler(messageType MessageType, handler MessageHandler)
}

type Dialer interface {
	DialContext(ctx context.Context, network, address string) (net.Conn, error)
}

type Message struct {
	ID   string      `json:"id"`
	Type MessageType `json:"type"`
	Data []byte      `json:"data"`
}

type MessageHandler func(msg []byte) error

type MessageType string

type OpenTCPStreamMessage struct {
	ID string `json:"id"`
}

type PortProvider interface {
	GetFreePort() int
	GetPort(port int) bool
	ReleasePort(int)
}

type RequestConnectionMessage struct {
	PortID        string `json:"portID"`
	Identifier    string `json:"identifier"`
	RemoteAddress string `json:"remoteAddress"`
	LocalAddress  string `json:"localAddress"`
}

type Server interface {
	Start(ctx context.Context) error
}

type Session interface {
	Close() error
	Handle() error
	GetControlStream() io.ReadWriter
	OpenTCPStream(id string) (net.Conn, error)
}

type Stream interface {
	io.ReadWriteCloser
	Handle() error
}