feat: add customizable file tree
This commit is contained in:
parent
914e303667
commit
39c6a1f570
16
abyss.go
16
abyss.go
@ -33,10 +33,6 @@ func main() {
|
|||||||
log.Fatal("basic auth password must be provided")
|
log.Fatal("basic auth password must be provided")
|
||||||
}
|
}
|
||||||
|
|
||||||
if app.url == "" {
|
|
||||||
slog.Warn("no root url detected, defaulting to localhost.")
|
|
||||||
}
|
|
||||||
|
|
||||||
if app.key == "" {
|
if app.key == "" {
|
||||||
slog.Warn("no upload key detected")
|
slog.Warn("no upload key detected")
|
||||||
}
|
}
|
||||||
@ -54,11 +50,19 @@ func main() {
|
|||||||
app.port = ":" + app.port
|
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 := http.NewServeMux()
|
||||||
mux.HandleFunc("/", app.indexHandler)
|
mux.HandleFunc("/", app.indexHandler)
|
||||||
mux.HandleFunc(
|
mux.Handle(
|
||||||
"/tree/",
|
"/tree/",
|
||||||
app.basicAuth(app.treeHandler),
|
http.StripPrefix(
|
||||||
|
"/tree",
|
||||||
|
app.basicAuth(app.fileListingHandler),
|
||||||
|
),
|
||||||
)
|
)
|
||||||
mux.HandleFunc("/last", app.lastHandler)
|
mux.HandleFunc("/last", app.lastHandler)
|
||||||
|
|
||||||
|
|||||||
67
handlers.go
67
handlers.go
@ -7,9 +7,11 @@ import (
|
|||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"text/template"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Application struct {
|
type Application struct {
|
||||||
@ -24,8 +26,63 @@ type Application struct {
|
|||||||
lastUploadedFile string
|
lastUploadedFile string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (app *Application) treeHandler(w http.ResponseWriter, r *http.Request) {
|
type FileInfo struct {
|
||||||
http.StripPrefix("/tree/", http.FileServer(http.Dir(app.filesDir))).ServeHTTP(w, r)
|
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) {
|
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
|
app.lastUploadedFile = filepath
|
||||||
|
|
||||||
if app.url == "" {
|
fmt.Fprintf(w, "http://%s/%s\n", app.url, filename)
|
||||||
fmt.Fprintf(w, "http://localhost%s/%s\n", app.port, filename)
|
|
||||||
} else {
|
|
||||||
fmt.Fprintf(w, "http://%s/%s\n", app.url, filename)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (app *Application) checkAuth(r *http.Request) bool {
|
func (app *Application) checkAuth(r *http.Request) bool {
|
||||||
|
|||||||
87
templates/dirlist.html
Normal file
87
templates/dirlist.html
Normal file
@ -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>
|
||||||
Loading…
Reference in New Issue
Block a user