preparing for v0.5

src/server/instance.go

@@ -0,0 +1,25 @@
+package server
+
+var proxyServer, apiServer *server
+
+func InitProxyServer(opt Options) *server {
+	if proxyServer == nil {
+		proxyServer = NewServer(opt)
+	}
+	return proxyServer
+}
+
+func InitAPIServer(opt Options) *server {
+	if apiServer == nil {
+		apiServer = NewServer(opt)
+	}
+	return apiServer
+}
+
+func GetProxyServer() *server {
+	return proxyServer
+}
+
+func GetAPIServer() *server {
+	return apiServer
+}

src/server/server.go

@@ -0,0 +1,157 @@
+package server
+
+import (
+	"crypto/tls"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/sirupsen/logrus"
+	"github.com/yusing/go-proxy/autocert"
+	"golang.org/x/net/context"
+)
+
+type server struct {
+	Name         string
+	CertProvider *autocert.Provider
+	http         *http.Server
+	https        *http.Server
+	httpStarted  bool
+	httpsStarted bool
+	startTime    time.Time
+}
+
+type Options struct {
+	Name string
+	// port (with leading colon)
+	HTTPPort string
+	// port (with leading colon)
+	HTTPSPort       string
+	CertProvider    *autocert.Provider
+	RedirectToHTTPS bool
+	Handler         http.Handler
+}
+
+type LogrusWrapper struct {
+	*logrus.Entry
+}
+
+func (l LogrusWrapper) Write(b []byte) (int, error) {
+	return l.Logger.WriterLevel(logrus.ErrorLevel).Write(b)
+}
+
+func NewServer(opt Options) (s *server) {
+	var httpSer, httpsSer *http.Server
+	var httpHandler http.Handler
+
+	logger := log.Default()
+	logger.SetOutput(LogrusWrapper{
+		logrus.WithFields(logrus.Fields{"?": "server", "name": opt.Name}),
+	})
+
+	_, err := opt.CertProvider.GetCert(nil)
+	certAvailable := err == nil
+	if certAvailable && opt.RedirectToHTTPS && opt.HTTPSPort != "" {
+		httpHandler = redirectToTLSHandler(opt.HTTPSPort)
+	} else {
+		httpHandler = opt.Handler
+	}
+
+	if opt.HTTPPort != "" {
+		httpSer = &http.Server{
+			Addr:     opt.HTTPPort,
+			Handler:  httpHandler,
+			ErrorLog: logger,
+		}
+	}
+	if certAvailable && opt.HTTPSPort != "" {
+		httpsSer = &http.Server{
+			Addr:     opt.HTTPSPort,
+			Handler:  opt.Handler,
+			ErrorLog: logger,
+			TLSConfig: &tls.Config{
+				GetCertificate: opt.CertProvider.GetCert,
+			},
+		}
+	}
+	return &server{
+		Name:         opt.Name,
+		CertProvider: opt.CertProvider,
+		http:         httpSer,
+		https:        httpsSer,
+	}
+}
+
+func (s *server) Start() {
+	if s.http == nil && s.https == nil {
+		return
+	}
+
+	s.startTime = time.Now()
+	if s.http != nil {
+		s.httpStarted = true
+		logrus.Printf("starting http %s server on %s", s.Name, s.http.Addr)
+		go func() {
+			s.handleErr("http", s.http.ListenAndServe())
+		}()
+	}
+
+	if s.https != nil {
+		s.httpsStarted = true
+		logrus.Printf("starting https %s server on %s", s.Name, s.https.Addr)
+		go func() {
+			s.handleErr("https", s.https.ListenAndServeTLS(s.CertProvider.GetCertPath(), s.CertProvider.GetKeyPath()))
+		}()
+	}
+}
+
+func (s *server) Stop() {
+	if s.http == nil && s.https == nil {
+		return
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	if s.http != nil && s.httpStarted {
+		s.handleErr("http", s.http.Shutdown(ctx))
+		s.httpStarted = false
+		logger.Debugf("HTTP server %q stopped", s.Name)
+	}
+
+	if s.https != nil && s.httpsStarted {
+		s.handleErr("https", s.https.Shutdown(ctx))
+		s.httpsStarted = false
+		logger.Debugf("HTTPS server %q stopped", s.Name)
+	}
+}
+
+func (s *server) Uptime() time.Duration {
+	return time.Since(s.startTime)
+}
+
+func (s *server) handleErr(scheme string, err error) {
+	switch err {
+	case nil, http.ErrServerClosed:
+		return
+	default:
+		logrus.Fatalf("failed to start %s %s server: %s", scheme, s.Name, err)
+	}
+}
+
+func redirectToTLSHandler(port string) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		r.URL.Scheme = "https"
+		r.URL.Host = r.URL.Hostname() + port
+
+		var redirectCode int
+		if r.Method == http.MethodGet {
+			redirectCode = http.StatusMovedPermanently
+		} else {
+			redirectCode = http.StatusPermanentRedirect
+		}
+		http.Redirect(w, r, r.URL.String(), redirectCode)
+	}
+}
+
+var logger = logrus.WithField("?", "server")