Altair-sh
105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package html
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"serveg/internal/http"
|
|
"serveg/internal/log"
|
|
sh "serveg/internal/string_helper"
|
|
)
|
|
|
|
type listingRow struct {
|
|
name string // html-escaped
|
|
url string // url-encoded
|
|
mtime string // formatted with DateTimeTZ
|
|
size string // formatted with HumanBytes
|
|
isDir bool
|
|
}
|
|
|
|
func (row *listingRow) genHtml(sb *sh.StringBuilder) {
|
|
sb.AddS(`<tr>`)
|
|
sb.AddS(`<td><a href="`).AddS(row.url).AddB('"')
|
|
if !row.isDir {
|
|
// tell browser to download file instead of opening in new tab
|
|
sb.AddS(" download")
|
|
}
|
|
sb.AddB('>').AddS(row.name).AddS(`</a></td>`)
|
|
sb.AddS(`<td class="size">`).AddS(row.size).AddS(`</td>`)
|
|
sb.AddS(`<td class="date">`).AddS(row.mtime).AddS(`</td>`)
|
|
sb.AddS(`</tr>`)
|
|
}
|
|
|
|
func GenDirListingHtml(logger *log.LoggerContext, dirPath, reqUrl, pageTitle string) ([]byte, error) {
|
|
dirEntries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var dirs, files []listingRow
|
|
var estimatedLen = 460 // html for page without entries
|
|
for _, e := range dirEntries {
|
|
einf, err := e.Info()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
row := listingRow{
|
|
name: htmlEscape(e.Name()),
|
|
url: http.PercentEcodeUrl(path.Join(reqUrl, e.Name())),
|
|
mtime: einf.ModTime().Format(DateTimeTZ),
|
|
isDir: e.IsDir(),
|
|
}
|
|
if row.isDir {
|
|
row.size = "DIR"
|
|
dirs = append(dirs, row)
|
|
} else {
|
|
row.size = sh.HumanBytes(einf.Size())
|
|
files = append(files, row)
|
|
}
|
|
|
|
estimatedLen += len(row.name) + len(row.url) + len(row.mtime) + len(row.size)
|
|
estimatedLen += 100 // html for row
|
|
}
|
|
|
|
sb := sh.StringBuilder{Buf: make([]byte, 0, estimatedLen)}
|
|
sb.AddS(`<!doctype html>
|
|
<html>
|
|
<head>
|
|
<meta charset="utf-8">
|
|
<title>`)
|
|
sb.AddS(htmlEscape(pageTitle + " " + reqUrl))
|
|
sb.AddS(`
|
|
</title>
|
|
<style>
|
|
body { font-family: sans-serif; margin: 2em; }
|
|
td { padding: .2em 1em .2em 0; }
|
|
td.size, td.date { color: #666; white-space: nowrap; }
|
|
</style>
|
|
</head>
|
|
<body>`)
|
|
sb.AddS("<h1>").AddS(htmlEscape(reqUrl)).AddS("</h1>")
|
|
sb.AddS(`<table id="listing">`)
|
|
sb.AddS(`<tr><td><a href="`)
|
|
sb.AddS(http.PercentEcodeUrl(path.Dir(reqUrl)))
|
|
sb.AddS(`">../</a></td></tr>`)
|
|
|
|
for i := 0; i < len(dirs); i++ {
|
|
dirs[i].genHtml(&sb)
|
|
}
|
|
for i := 0; i < len(files); i++ {
|
|
files[i].genHtml(&sb)
|
|
}
|
|
|
|
sb.AddS(`
|
|
</table>
|
|
</body>
|
|
</html>`)
|
|
|
|
realLen := sb.Len()
|
|
if estimatedLen < realLen {
|
|
logger.Warn("genDirListingHtml(): size prediction failed. estimatedLen: ",
|
|
estimatedLen, " realLen: ", realLen)
|
|
}
|
|
return sb.Buf, nil
|
|
}
|