spec.go

package relay

import (
	"net/http"

	"github.com/btwiuse/muxr"
	"github.com/webteleport/webteleport/edge"
)

// dispatch to other http.Handler implementations
type Dispatcher interface {
	Dispatch(r *http.Request) http.Handler
}

// wrapping Dispatcher into http.Handler
type DispatcherFunc func(*http.Request) http.Handler

func (d DispatcherFunc) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	d(r).ServeHTTP(w, r)
}

// edge.Edge multiplexer with builtin HTTPUpgrader
type Relayer interface {
	// dispatch to HTTPUpgrader and Storage
	Dispatcher

	// builtin HTTPUpgrader
	edge.HTTPUpgrader

	// edge.Edge multiplexer
	Storage
}

// edge.Edge multiplexer
type Storage interface {
	// Dispatch to edge.Edge
	Dispatcher

	// get Session wrapped by http.Transport
	GetRoundTripper(k string) (http.RoundTripper, bool)

	// record Info
	RecordsHandler(w http.ResponseWriter, r *http.Request)

	// subscribe to incoming stream of edge.Edge
	edge.Subscriber

	// apply middleware to dispatcher
	Use(middlewares ...muxr.Middleware)

	// shortcut to dispatcher
	http.Handler
}