Move code to pkg

pkg/server/router/tcp/router.go

@@ -0,0 +1,139 @@
+package tcp
+
+import (
+	"context"
+	"crypto/tls"
+	"net/http"
+
+	"github.com/containous/traefik/pkg/config"
+	"github.com/containous/traefik/pkg/log"
+	"github.com/containous/traefik/pkg/rules"
+	"github.com/containous/traefik/pkg/server/internal"
+	tcpservice "github.com/containous/traefik/pkg/server/service/tcp"
+	"github.com/containous/traefik/pkg/tcp"
+)
+
+// NewManager Creates a new Manager
+func NewManager(routers map[string]*config.TCPRouter,
+	serviceManager *tcpservice.Manager,
+	httpHandlers map[string]http.Handler,
+	httpsHandlers map[string]http.Handler,
+	tlsConfig *tls.Config,
+) *Manager {
+	return &Manager{
+		configs:        routers,
+		serviceManager: serviceManager,
+		httpHandlers:   httpHandlers,
+		httpsHandlers:  httpsHandlers,
+		tlsConfig:      tlsConfig,
+	}
+}
+
+// Manager is a route/router manager
+type Manager struct {
+	configs        map[string]*config.TCPRouter
+	serviceManager *tcpservice.Manager
+	httpHandlers   map[string]http.Handler
+	httpsHandlers  map[string]http.Handler
+	tlsConfig      *tls.Config
+}
+
+// BuildHandlers builds the handlers for the given entrypoints
+func (m *Manager) BuildHandlers(rootCtx context.Context, entryPoints []string) map[string]*tcp.Router {
+	entryPointsRouters := m.filteredRouters(rootCtx, entryPoints)
+
+	entryPointHandlers := make(map[string]*tcp.Router)
+	for _, entryPointName := range entryPoints {
+		entryPointName := entryPointName
+
+		routers := entryPointsRouters[entryPointName]
+
+		ctx := log.With(rootCtx, log.Str(log.EntryPointName, entryPointName))
+
+		handler, err := m.buildEntryPointHandler(ctx, routers, m.httpHandlers[entryPointName], m.httpsHandlers[entryPointName])
+		if err != nil {
+			log.FromContext(ctx).Error(err)
+			continue
+		}
+		entryPointHandlers[entryPointName] = handler
+	}
+	return entryPointHandlers
+}
+
+func (m *Manager) buildEntryPointHandler(ctx context.Context, configs map[string]*config.TCPRouter, handlerHTTP http.Handler, handlerHTTPS http.Handler) (*tcp.Router, error) {
+	router := &tcp.Router{}
+
+	router.HTTPHandler(handlerHTTP)
+	router.HTTPSHandler(handlerHTTPS, m.tlsConfig)
+	for routerName, routerConfig := range configs {
+		ctxRouter := log.With(ctx, log.Str(log.RouterName, routerName))
+		logger := log.FromContext(ctxRouter)
+
+		ctxRouter = internal.AddProviderInContext(ctxRouter, routerName)
+
+		handler, err := m.serviceManager.BuildTCP(ctxRouter, routerConfig.Service)
+		if err != nil {
+			logger.Error(err)
+			continue
+		}
+
+		domains, err := rules.ParseHostSNI(routerConfig.Rule)
+		if err != nil {
+			log.WithoutContext().Debugf("Unknown rule %s", routerConfig.Rule)
+			continue
+		}
+		for _, domain := range domains {
+			log.WithoutContext().Debugf("Add route %s on TCP", domain)
+			switch {
+			case routerConfig.TLS != nil:
+				if routerConfig.TLS.Passthrough {
+					router.AddRoute(domain, handler)
+				} else {
+					router.AddRouteTLS(domain, handler, m.tlsConfig)
+
+				}
+			case domain == "*":
+				router.AddCatchAllNoTLS(handler)
+			default:
+				logger.Warn("TCP Router ignored, cannot specify a Host rule without TLS")
+			}
+		}
+	}
+
+	return router, nil
+}
+
+func contains(entryPoints []string, entryPointName string) bool {
+	for _, name := range entryPoints {
+		if name == entryPointName {
+			return true
+		}
+	}
+	return false
+}
+
+func (m *Manager) filteredRouters(ctx context.Context, entryPoints []string) map[string]map[string]*config.TCPRouter {
+	entryPointsRouters := make(map[string]map[string]*config.TCPRouter)
+
+	for rtName, rt := range m.configs {
+		eps := rt.EntryPoints
+		if len(eps) == 0 {
+			eps = entryPoints
+		}
+		for _, entryPointName := range eps {
+			if !contains(entryPoints, entryPointName) {
+				log.FromContext(log.With(ctx, log.Str(log.EntryPointName, entryPointName))).
+					Errorf("entryPoint %q doesn't exist", entryPointName)
+				continue
+			}
+
+			if _, ok := entryPointsRouters[entryPointName]; !ok {
+				entryPointsRouters[entryPointName] = make(map[string]*config.TCPRouter)
+			}
+
+			entryPointsRouters[entryPointName][rtName] = rt
+		}
+	}
+
+	return entryPointsRouters
+}