98 lines
2.0 KiB
Go
98 lines
2.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var batchTmpl *template.Template
|
|
|
|
func (h *Handler) Batch(w http.ResponseWriter, r *http.Request) {
|
|
batchID := r.PathValue("id")
|
|
if batchID == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
files, err := h.store.GetBatch(batchID)
|
|
if err != nil {
|
|
log.Printf("db get batch error: %v", err)
|
|
http.Error(w, "internal error", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
if len(files) == 0 {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
type fileData struct {
|
|
ID string
|
|
Filename string
|
|
Size int64
|
|
ContentType string
|
|
UploadedAt string
|
|
ThumbType string
|
|
PreviewURL string
|
|
ViewURL string
|
|
DownloadURL string
|
|
}
|
|
|
|
var items []fileData
|
|
var totalSize int64
|
|
for _, f := range files {
|
|
// Skip expired files
|
|
if f.ExpiresAt != nil && f.ExpiresAt.Before(time.Now().UTC()) {
|
|
continue
|
|
}
|
|
|
|
thumbType := ""
|
|
previewURL := ""
|
|
if strings.HasPrefix(f.ContentType, "image/") {
|
|
thumbType = "image"
|
|
} else if strings.HasPrefix(f.ContentType, "video/") {
|
|
thumbType = "video"
|
|
} else if strings.HasPrefix(f.ContentType, "audio/") {
|
|
thumbType = "audio"
|
|
} else if f.ContentType == "application/pdf" {
|
|
thumbType = "pdf"
|
|
}
|
|
if thumbType != "" {
|
|
if url, err := h.r2.PresignGet(r.Context(), f.R2Key, f.Filename, true); err == nil {
|
|
previewURL = url
|
|
}
|
|
}
|
|
|
|
totalSize += f.SizeBytes
|
|
items = append(items, fileData{
|
|
ID: f.ID,
|
|
Filename: f.Filename,
|
|
Size: f.SizeBytes,
|
|
ContentType: f.ContentType,
|
|
UploadedAt: f.UploadedAt.Format("Jan 2, 2006 at 3:04 PM"),
|
|
ThumbType: thumbType,
|
|
PreviewURL: previewURL,
|
|
ViewURL: "/f/" + f.ID + "/view",
|
|
DownloadURL: "/f/" + f.ID + "/dl",
|
|
})
|
|
}
|
|
|
|
if len(items) == 0 {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
data := map[string]any{
|
|
"BatchID": batchID,
|
|
"Files": items,
|
|
"Count": len(items),
|
|
"TotalSize": totalSize,
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|
w.Header().Set("Cache-Control", "no-cache, must-revalidate")
|
|
batchTmpl.Execute(w, data)
|
|
}
|