feat: add customizable file tree

jabuxas · Sep 19, 2024 · 39c6a1f · 39c6a1f
1 parent 914e303
commit 39c6a1f
Show file tree

Hide file tree

Showing 3 changed files with 157 additions and 13 deletions.
diff --git a/abyss.go b/abyss.go
@@ -33,10 +33,6 @@ func main() {
 		log.Fatal("basic auth password must be provided")
 	}
 
-	if app.url == "" {
-		slog.Warn("no root url detected, defaulting to localhost.")
-	}
-
 	if app.key == "" {
 		slog.Warn("no upload key detected")
 	}
@@ -54,11 +50,19 @@ func main() {
 		app.port = ":" + app.port
 	}
 
+	if app.url == "" {
+		slog.Warn("no root url detected, defaulting to localhost.")
+		app.url = "localhost" + app.port
+	}
+
 	mux := http.NewServeMux()
 	mux.HandleFunc("/", app.indexHandler)
-	mux.HandleFunc(
+	mux.Handle(
 		"/tree/",
-		app.basicAuth(app.treeHandler),
+		http.StripPrefix(
+			"/tree",
+			app.basicAuth(app.fileListingHandler),
+		),
 	)
 	mux.HandleFunc("/last", app.lastHandler)
 

diff --git a/handlers.go b/handlers.go
@@ -7,9 +7,11 @@ import (
 	"encoding/hex"
 	"fmt"
 	"io"
+	"log/slog"
 	"net/http"
 	"os"
 	"path/filepath"
+	"text/template"
 )
 
 type Application struct {
@@ -24,8 +26,63 @@ type Application struct {
 	lastUploadedFile string
 }
 
-func (app *Application) treeHandler(w http.ResponseWriter, r *http.Request) {
-	http.StripPrefix("/tree/", http.FileServer(http.Dir(app.filesDir))).ServeHTTP(w, r)
+type FileInfo struct {
+	Name          string
+	Path          string
+	Size          int64
+	FormattedSize string
+}
+
+type TemplateData struct {
+	Files []FileInfo
+	URL   string
+}
+
+func (app *Application) fileListingHandler(w http.ResponseWriter, r *http.Request) {
+	dir := app.filesDir + r.URL.Path
+
+	files, err := os.ReadDir(dir)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	var fileInfos []FileInfo
+	for _, file := range files {
+		filePath := filepath.Join(dir, file.Name())
+		info, err := os.Stat(filePath)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		fileInfos = append(fileInfos, FileInfo{
+			Name:          file.Name(),
+			Path:          filepath.Join(r.URL.Path, file.Name()),
+			Size:          info.Size(),
+			FormattedSize: formatFileSize(info.Size()),
+		})
+	}
+
+	tmpl := template.Must(template.ParseFiles("templates/dirlist.html"))
+	templateData := TemplateData{
+		Files: fileInfos,
+		URL:   app.url,
+	}
+	if err := tmpl.Execute(w, templateData); err != nil {
+		slog.Warn(error.Error(err))
+	}
+}
+
+func formatFileSize(size int64) string {
+	if size < 1024 {
+		return fmt.Sprintf("%d B", size)
+	} else if size < 1024*1024 {
+		return fmt.Sprintf("%.2f KB", float64(size)/1024)
+	} else if size < 1024*1024*1024 {
+		return fmt.Sprintf("%.2f MB", float64(size)/(1024*1024))
+	}
+	return fmt.Sprintf("%.2f GB", float64(size)/(1024*1024*1024))
 }
 
 func (app *Application) indexHandler(w http.ResponseWriter, r *http.Request) {
@@ -137,11 +194,7 @@ func (app *Application) uploadHandler(w http.ResponseWriter, r *http.Request) {
 
 	app.lastUploadedFile = filepath
 
-	if app.url == "" {
-		fmt.Fprintf(w, "http://localhost%s/%s\n", app.port, filename)
-	} else {
-		fmt.Fprintf(w, "http://%s/%s\n", app.url, filename)
-	}
+	fmt.Fprintf(w, "http://%s/%s\n", app.url, filename)
 }
 
 func (app *Application) checkAuth(r *http.Request) bool {

diff --git a/templates/dirlist.html b/templates/dirlist.html
@@ -0,0 +1,87 @@
+<!doctype html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>dir listing</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            margin: 20px;
+        }
+
+        h1 {
+            font-size: 24px;
+            margin-bottom: 10px;
+        }
+
+        header {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+        }
+
+        .file-listing {
+            width: 100%;
+            border-collapse: collapse;
+            margin-top: 20px;
+        }
+
+        .file-listing th,
+        .file-listing td {
+            border: 1px solid #ddd;
+            padding: 12px;
+            text-align: left;
+        }
+
+        .file-listing th {
+            background-color: #333;
+            color: white;
+        }
+
+        .file-listing tr:nth-child(even) {
+            background-color: #f9f9f9;
+        }
+
+        .file-listing tr:hover {
+            background-color: #f1f1f1;
+        }
+
+        .file-listing td a {
+            color: #0288d1;
+            text-decoration: none;
+        }
+
+        .file-listing td a:hover {
+            text-decoration: underline;
+        }
+    </style>
+</head>
+
+<body>
+    <header>
+        <h1>{{ .URL }}</h1>
+    </header>
+
+    <table class="file-listing">
+        <thead>
+            <tr>
+                <th>Name</th>
+                <th>Size</th>
+            </tr>
+        </thead>
+        <tbody>
+            {{range .Files}}
+            <tr>
+                <td>
+                    <a href="{{.Path}}">{{.Name}}</a>
+                </td>
+                <td>{{.FormattedSize}}</td>
+            </tr>
+            {{end}}
+        </tbody>
+    </table>
+</body>
+
+</html>