chore: upgrade goVirtualHost

Author: marjune163

src/goVirtualHost/listenable.go

@@ -0,0 +1,43 @@
+package goVirtualHost
+
+import (
+	"net"
+	"os"
+)
+
+func newListenable(proto, ip, port string) *listenable {
+	return &listenable{
+		proto: proto,
+		ip:    ip,
+		port:  port,
+	}
+}
+
+func (l *listenable) open() error {
+	addr := l.ip + l.port
+	if l.proto == "unix" {
+		sockInfo, _ := os.Lstat(addr)
+		if sockInfo != nil && (sockInfo.Mode()&os.ModeSocket != 0) {
+			os.Remove(addr)
+		}
+	}
+
+	listener, err := net.Listen(l.proto, addr)
+	l.listener = listener
+
+	if l.proto == "unix" && err == nil {
+		os.Chmod(addr, 0660)
+	}
+
+	return err
+}
+
+func (l *listenable) close() error {
+	if l.listener == nil {
+		return nil
+	}
+
+	err := l.listener.Close()
+	l.listener = nil
+	return err
+}

src/goVirtualHost/listeners.go renamed to src/goVirtualHost/listenables.go

@@ -1,7 +1,7 @@
 package goVirtualHost
 
-func (listeners listeners) find(proto, ip, port string) *listener {
-	for _, l := range listeners {
+func (ls listenables) find(proto, ip, port string) *listenable {
+	for _, l := range ls {
 		if l.proto == proto && l.ip == ip && l.port == port {
 			return l
 		}

src/goVirtualHost/listener.go

@@ -0,0 +1,88 @@
+package goVirtualHost
+
+import (
+	"context"
+	"crypto/tls"
+	"net/http"
+)
+
+func newServeable(useTLS bool) *serveable {
+	return &serveable{
+		useTLS:       useTLS,
+		vhosts:       vhosts{},
+		defaultVhost: nil,
+
+		server: &http.Server{},
+	}
+}
+
+func (serveable *serveable) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	hostname := extractHostName(r.Host)
+	for i := range serveable.vhosts {
+		if serveable.vhosts[i].matchHostName(hostname) {
+			serveable.vhosts[i].handler.ServeHTTP(w, r)
+			return
+		}
+	}
+
+	serveable.defaultVhost.handler.ServeHTTP(w, r)
+}
+
+func (serveable *serveable) getDefaultVhost() *vhost {
+	for _, vh := range serveable.vhosts {
+		if len(vh.hostNames) == 0 {
+			return vh
+		}
+	}
+
+	if len(serveable.vhosts) > 0 {
+		return serveable.vhosts[0]
+	}
+
+	return nil
+}
+
+func (serveable *serveable) updateDefaultVhost() {
+	serveable.defaultVhost = serveable.getDefaultVhost()
+}
+
+func (serveable *serveable) updateHttpServerTLSConfig() {
+	if !serveable.useTLS {
+		return
+	}
+
+	certs := make([]tls.Certificate, 0, len(serveable.vhosts))
+
+	for _, vhost := range serveable.vhosts {
+		certs = append(certs, vhost.certs...)
+	}
+
+	serveable.server.TLSConfig = &tls.Config{
+		Certificates: certs,
+	}
+}
+
+func (serveable *serveable) updateHttpServerHandler() {
+	if len(serveable.vhosts) == 1 {
+		serveable.server.Handler = serveable.defaultVhost.handler
+		return
+	}
+
+	serveable.server.Handler = serveable
+}
+
+func (serveable *serveable) open(l *listenable) error {
+	if serveable.useTLS {
+		return serveable.server.ServeTLS(l.listener, "", "")
+	} else {
+		return serveable.server.Serve(l.listener)
+	}
+}
+
+func (serveable *serveable) shutdown(ctx context.Context) error {
+	return serveable.server.Shutdown(ctx)
+}
+
+func (serveable *serveable) close() error {
+	return serveable.server.Close()
+}