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)
}