1 files changed, 14 insertions, 3 deletions

diff --git a/gopherproxy.go b/gopherproxy.go
index 8c85448..fcce1ee 100644
--- a/gopherproxy.go
+++ b/gopherproxy.go
@@ -23,9 +23,16 @@ type tplRow struct {
 type Handler func(w http.ResponseWriter, req *http.Request)
 
 func renderDirectory(w http.ResponseWriter, tpl *template.Template, hostport string, d gopher.Directory) error {
-        out := make([]tplRow, len(d))
+        var title string
+
+        out := make([]tplRow, len(d.Items))
+
+        for i, x := range d.Items {
+                if x.Type == gopher.INFO && x.Selector == "TITLE" {
+                        title = x.Description
+                        continue
+                }
 
-        for i, x := range d {
                 tr := tplRow{
                         Text: x.Description,
                         Type: x.Type.String(),
@@ -60,10 +67,14 @@ func renderDirectory(w http.ResponseWriter, tpl *template.Template, hostport str
                 out[i] = tr
         }
 
+        if title == "" {
+                title = hostport
+        }
+
         return tpl.Execute(w, struct {
                 Title string
                 Lines []tplRow
-        }{hostport, out})
+        }{title, out})
 }
 
 // MakeGopherProxyHandler returns a Handler that proxies requests