option.go

package teleportation

import (
	"net"

	"github.com/sirupsen/logrus"

	"github.com/PeerXu/meepo/pkg/ofn"
	"github.com/PeerXu/meepo/pkg/transport"
)

type NewTeleportationSourceOption = ofn.OFN
type NewTeleportationSinkOption = ofn.OFN

func WithLogger(logger logrus.FieldLogger) ofn.OFN {
	return func(o ofn.Option) {
		o["logger"] = logger
	}
}

func WithName(name string) ofn.OFN {
	return func(o ofn.Option) {
		o["name"] = name
	}
}

func WithSource(addr net.Addr) ofn.OFN {
	return func(o ofn.Option) {
		o["source"] = addr
	}
}

func WithSink(addr net.Addr) ofn.OFN {
	return func(o ofn.Option) {
		o["sink"] = addr
	}
}

func WithTransport(t transport.Transport) ofn.OFN {
	return func(o ofn.Option) {
		o["transport"] = t
	}
}

type NewDial func(network, address string) (net.Conn, error)

func WithNewDial(f NewDial) ofn.OFN {
	return func(o ofn.Option) {
		o["newDial"] = f
	}
}

type DoTeleportFunc func(label string) error

func WithDoTeleportFunc(f DoTeleportFunc) ofn.OFN {
	return func(o ofn.Option) {
		o["doTeleportFunc"] = f
	}
}

type OnDoTeleportFunc func() error

func WithOnDoTeleportFunc(f OnDoTeleportFunc) ofn.OFN {
	return func(o ofn.Option) {
		o["onDoTeleportFunc"] = f
	}
}

type OnCloseHandler func()

func WithOnCloseHandler(h OnCloseHandler) ofn.OFN {
	return func(o ofn.Option) {
		o["onCloseHandler"] = h
	}
}

type OnErrorHandler func(error)

func WithOnErrorHandler(h OnErrorHandler) ofn.OFN {
	return func(o ofn.Option) {
		o["onErrorHandler"] = h
	}
}

type DialRequest struct {
	Conn net.Conn
	Quit chan struct{}
}

func NewDialRequest(conn net.Conn) *DialRequest {
	return &DialRequest{Conn: conn}
}

func NewDialRequestWithQuit(conn net.Conn, quit chan struct{}) *DialRequest {
	return &DialRequest{Conn: conn, Quit: quit}
}

func SetDialRequestChannel(c chan *DialRequest) ofn.OFN {
	return func(o ofn.Option) {
		o["dialRequestChannel"] = c
	}
}