add Link column to home.html

Signed-off-by: shaftoes <[email protected]>
tailscale · May 10, 2024 · 1c704ff · 1c704ff
1 parent 41e9dc9
commit 1c704ff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/golink.go b/golink.go
@@ -264,6 +264,7 @@ var (
 
 type visitData struct {
 	Short     string
+	Long      string
 	NumClicks int
 }
 
@@ -404,10 +405,21 @@ func serveHandler() http.Handler {
 func serveHome(w http.ResponseWriter, r *http.Request, short string) {
 	var clicks []visitData
 
+	linkMap := map[string]string{}
+	links, err := db.LoadAll()
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	for _, link := range links {
+		linkMap[link.Short] = link.Long
+	}
+
 	stats.mu.Lock()
 	for short, numClicks := range stats.clicks {
 		clicks = append(clicks, visitData{
 			Short:     short,
+			Long:      linkMap[short],
 			NumClicks: numClicks,
 		})
 	}

diff --git a/tmpl/home.html b/tmpl/home.html
@@ -23,6 +23,7 @@ <h2 class="text-xl font-bold pt-6 pb-2">Popular Links</h2>
         <tr>
           <th class="p-2">Link</th>
           <th class="p-2">Clicks</th>
+          <th class="p-2">URL</th>
         </tr>
       </thead>
       <tbody>
@@ -35,6 +36,7 @@ <h2 class="text-xl font-bold pt-6 pb-2">Popular Links</h2>
             </a>
           </td>
           <td class="p-2">{{.NumClicks}}</td>
+          <td class="p-2">{{.Long}}</td>
         </tr>
       {{end}}
       </tbody>