feat(app): try match unknown hostname to first no hostname vhost

Author: marjune163

README.md

@@ -136,6 +136,12 @@ server [options]
 ,,
     To specify multiple virtual hosts with options, split these hosts' options by this sign.
     Options above can be specified for each virtual host.
+
+    If multiple virtual hosts share same IP and ports,
+    use --hostname to identify them according to the request.
+    If request host name does not match any virtual host,
+    server will try to use first virtual host that has no hostname,
+    otherwise use the first virtual host.
 ```
 
 ## Examples

src/app/listen.go

@@ -8,15 +8,16 @@ import (
 )
 
 type ListenItem struct {
-	proto     string
-	addr      string
-	useTLS    bool
-	hostnames []string
-	certs     []tls.Certificate
-	handler   http.Handler
-	listener  net.Listener
-	server    *http.Server
-	vhosts    []*vhost.VHost
+	proto        string
+	addr         string
+	useTLS       bool
+	hostnames    []string
+	certs        []tls.Certificate
+	handler      http.Handler
+	listener     net.Listener
+	server       *http.Server
+	vhosts       []*vhost.VHost
+	defaultVHost *vhost.VHost
 }
 
 type Listens []*ListenItem

src/app/main.go

@@ -186,16 +186,28 @@ func NewApp(params []*param.Param) *App {
 		}
 	}
 
-	// handler for each listen item
+	// init default vhost for each listen item
 	for _, l := range app.listens {
-		item := l
+		for _, vh := range l.vhosts {
+			if len(vh.Hostnames) == 0 {
+				l.defaultVHost = vh
+				break
+			}
+		}
+
+		if l.defaultVHost == nil {
+			l.defaultVHost = l.vhosts[0]
+		}
+	}
 
-		if len(item.vhosts) == 1 {
-			item.handler = item.vhosts[0].Mux
+	// handler for each listen item
+	for _, l := range app.listens {
+		if len(l.vhosts) == 1 {
+			l.handler = l.defaultVHost.Mux
 			continue
 		}
 
-		item.handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		l.handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			var serveVHost *vhost.VHost
 
 			hostname := r.Host
@@ -204,14 +216,15 @@ func NewApp(params []*param.Param) *App {
 				hostname = hostname[:colonIndex]
 			}
 
-			for _, vh := range item.vhosts {
+			for _, vh := range l.vhosts {
 				if vh.MatchHostname(hostname) {
 					serveVHost = vh
 					break
 				}
 			}
+
 			if serveVHost == nil {
-				serveVHost = item.vhosts[0]
+				serveVHost = l.defaultVHost
 			}
 
 			serveVHost.Mux.ServeHTTP(w, r)

src/vhost/main.go

@@ -60,12 +60,9 @@ func NewVHost(p *param.Param) *VHost {
 	mux := serveMux.NewServeMux(p, logger, errHandler)
 
 	// hostnames
-	hostnames := p.Hostnames
-	for i, length := 0, len(hostnames); i < length; i++ {
-		hostnames[i] = strings.ToLower(hostnames[i])
-	}
-	if len(hostnames) == 0 {
-		hostnames = append(hostnames, "")
+	hostnames := make([]string, 0, len(p.Hostnames))
+	for _, hostname := range p.Hostnames {
+		hostnames = append(hostnames, strings.ToLower(hostname))
 	}
 
 	// determine can use TLS