package pack

import (
	"errors"
	"net/http"
	"strings"

	"github.com/tamnd/kage/zim"
)

// Handler serves a ZIM archive over HTTP. "/" maps to the archive's main page;
// "/a/b.png" maps to the C/a/b.png content entry. Because the saved HTML's links
// are mirror-relative paths, and those are exactly the C urls, a click in a
// served page hits the right entry with no rewriting. A miss is a plain 404.
func Handler(r *zim.Reader) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
		p := strings.TrimPrefix(req.URL.Path, "/")
		if p == "" {
			blob, err := r.MainPage()
			if err != nil {
				http.NotFound(w, req)
				return
			}
			serveBlob(w, blob)
			return
		}
		blob, err := r.Get(zim.NamespaceContent, p)
		if errors.Is(err, zim.ErrNotFound) {
			http.NotFound(w, req)
			return
		}
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		serveBlob(w, blob)
	})
}

func serveBlob(w http.ResponseWriter, b zim.Blob) {
	if b.MimeType != "" {
		w.Header().Set("Content-Type", b.MimeType)
	}
	_, _ = w.Write(b.Data)
}