Fixed tls host lookup

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

armor.go

@@ -42,6 +42,7 @@ type (
 
 	TLS struct {
 		Address      string `json:"address"`
+		Port         string `json:"-"`
 		CertFile     string `json:"cert_file"`
 		KeyFile      string `json:"key_file"`
 		Auto         bool   `json:"auto"`
@@ -134,9 +135,11 @@ func (a *Armor) FindHost(name string, add bool) (h *Host) {
 
 	// Initialize host
 	if !h.initialized {
+		h.Name = name
 		h.Paths = make(Paths)
 		h.Group = a.Echo.Host(net.JoinHostPort(name, a.Port))
-		h.Name = name
+		routers := a.Echo.Routers()
+		routers[net.JoinHostPort(name, a.TLS.Port)] = routers[name]
 		h.initialized = true
 	}
 

http.go

@@ -43,6 +43,7 @@ func (a *Armor) NewHTTP() (h *HTTP) {
 		WriteTimeout: a.WriteTimeout * time.Second,
 	}
 	if a.TLS != nil {
+		_, a.TLS.Port, _ = net.SplitHostPort(a.TLS.Address)
 		e.TLSServer = &http.Server{
 			Addr:         a.TLS.Address,
 			TLSConfig:    a.setupTLSConfig(),
@@ -89,8 +90,7 @@ func (h *HTTP) Start() error {
 		a.Colorer.Printf("⇨ serving from %s (Local)\n", a.Colorer.Green("http://localhost"+a.Address))
 		ip := util.PrivateIP()
 		if ip != "" {
-			_, port, _ := net.SplitHostPort(a.Address)
-			a.Colorer.Printf("⇨ serving from %s (Intranet)\n", a.Colorer.Green(fmt.Sprintf("http://%s:%s", ip, port)))
+			a.Colorer.Printf("⇨ serving from %s (Intranet)\n", a.Colorer.Green(fmt.Sprintf("http://%s:%s", ip, a.Port)))
 		}
 	} else {
 		a.Colorer.Printf("⇨ http server started on %s\n", a.Colorer.Green(a.Address))