From 5559e0134382a141f5edabdacf1dc81f12b55c27 Mon Sep 17 00:00:00 2001 From: Brian Picciano Date: Sat, 15 Apr 2023 21:35:06 +0200 Subject: Implement asset.Loader This moved a bunch of logic out of http and into the asset package, making it available for gmit too. --- src/http/assets.go | 222 ++++++----------------------------------------------- 1 file changed, 23 insertions(+), 199 deletions(-) (limited to 'src/http/assets.go') diff --git a/src/http/assets.go b/src/http/assets.go index 1f5f0d6..1a8f520 100644 --- a/src/http/assets.go +++ b/src/http/assets.go @@ -2,14 +2,8 @@ package http import ( "bytes" - "compress/gzip" "errors" "fmt" - "image" - "image/jpeg" - "image/png" - "io" - "io/fs" "net/http" "path/filepath" "strings" @@ -17,52 +11,8 @@ import ( "github.com/mediocregopher/blog.mediocregopher.com/srv/http/apiutil" "github.com/mediocregopher/blog.mediocregopher.com/srv/post/asset" - "github.com/omeid/go-tarfs" - "golang.org/x/image/draw" ) -func isImgResizable(path string) bool { - switch strings.ToLower(filepath.Ext(path)) { - case ".jpg", ".jpeg", ".png": - return true - default: - return false - } -} - -func resizeImage(out io.Writer, in io.Reader, maxWidth float64) error { - - img, format, err := image.Decode(in) - if err != nil { - return fmt.Errorf("decoding image: %w", err) - } - - imgRect := img.Bounds() - imgW, imgH := float64(imgRect.Dx()), float64(imgRect.Dy()) - - if imgW > maxWidth { - - newH := imgH * maxWidth / imgW - newImg := image.NewRGBA(image.Rect(0, 0, int(maxWidth), int(newH))) - - // Resize - draw.BiLinear.Scale( - newImg, newImg.Bounds(), img, img.Bounds(), draw.Over, nil, - ) - - img = newImg - } - - switch format { - case "jpeg": - return jpeg.Encode(out, img, nil) - case "png": - return png.Encode(out, img) - default: - return fmt.Errorf("unknown image format %q", format) - } -} - func (a *api) managePostAssetsHandler() http.Handler { tpl := a.mustParseBasedTpl("post-assets-manage.html") @@ -88,173 +38,47 @@ func (a *api) managePostAssetsHandler() http.Handler { }) } -type postAssetArchiveInfo struct { - path string - id string - subPath string - isGzipped bool -} - -func extractPostAssetArchiveInfo(path string) (postAssetArchiveInfo, bool) { - - var info postAssetArchiveInfo - - info.path = strings.TrimPrefix(path, "/") - - info.id, info.subPath, _ = strings.Cut(info.path, "/") - - switch { - - case strings.HasSuffix(info.id, ".tar.gz"), - strings.HasSuffix(info.id, ".tgz"): - info.isGzipped = true - - case strings.HasSuffix(info.id, ".tar"): - // ok - - default: - // unsupported - return postAssetArchiveInfo{}, false - } - - return info, true -} - -func (a *api) writePostAsset( - rw http.ResponseWriter, - r *http.Request, - path string, - from io.ReadSeeker, -) { - - maxWidth, err := apiutil.StrToInt(r.FormValue("w"), 0) - if err != nil { - apiutil.BadRequest(rw, r, fmt.Errorf("invalid w parameter: %w", err)) - return - } - - if maxWidth == 0 { - http.ServeContent(rw, r, path, time.Time{}, from) - return - } - - if !isImgResizable(path) { - apiutil.BadRequest(rw, r, fmt.Errorf("cannot resize asset %q", path)) - return - } - - resizedBuf := new(bytes.Buffer) - - if err := resizeImage(resizedBuf, from, float64(maxWidth)); err != nil { - apiutil.InternalServerError( - rw, r, - fmt.Errorf( - "resizing image %q to size %d: %w", - path, maxWidth, err, - ), - ) - } - - http.ServeContent( - rw, r, path, time.Time{}, bytes.NewReader(resizedBuf.Bytes()), - ) -} - -func (a *api) handleGetPostAssetArchive( - rw http.ResponseWriter, - r *http.Request, - info postAssetArchiveInfo, -) { - - buf := new(bytes.Buffer) - - err := a.params.PostAssetStore.Get(info.id, buf) - - if errors.Is(err, asset.ErrNotFound) { - http.Error(rw, "asset not found", 404) - return - } else if err != nil { - apiutil.InternalServerError( - rw, r, - fmt.Errorf("fetching archive asset with id %q: %w", info.id, err), - ) - return - } - - var from io.Reader = buf - - if info.isGzipped { - - if from, err = gzip.NewReader(from); err != nil { - apiutil.InternalServerError( - rw, r, - fmt.Errorf("decompressing archive asset with id %q: %w", info.id, err), - ) - return - } - } - - tarFS, err := tarfs.New(from) - - if err != nil { - apiutil.InternalServerError( - rw, r, - fmt.Errorf("reading archive asset with id %q as fs: %w", info.id, err), - ) - return - } - - f, err := tarFS.Open(info.subPath) - - if errors.Is(err, fs.ErrExist) { - http.Error(rw, "Asset not found", 404) - return - - } else if err != nil { - - apiutil.InternalServerError( - rw, r, - fmt.Errorf( - "opening path %q from archive asset with id %q as fs: %w", - info.subPath, info.id, err, - ), - ) - return - } - - defer f.Close() - - a.writePostAsset(rw, r, info.path, f) -} - func (a *api) getPostAssetHandler() http.Handler { return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { - archiveInfo, ok := extractPostAssetArchiveInfo(r.URL.Path) - - if ok { - a.handleGetPostAssetArchive(rw, r, archiveInfo) + maxWidth, err := apiutil.StrToInt(r.FormValue("w"), 0) + if err != nil { + apiutil.BadRequest(rw, r, fmt.Errorf("invalid w parameter: %w", err)) return } - id := filepath.Base(r.URL.Path) - - buf := new(bytes.Buffer) + var ( + path = strings.TrimPrefix(r.URL.Path, "/") + buf = new(bytes.Buffer) + ) - err := a.params.PostAssetStore.Get(id, buf) + err = a.params.PostAssetLoader.Load( + path, + buf, + asset.LoadOpts{ + ImageWidth: maxWidth, + }, + ) if errors.Is(err, asset.ErrNotFound) { http.Error(rw, "Asset not found", 404) return + + } else if errors.Is(err, asset.ErrCannotResize) { + http.Error(rw, "Image resizing not supported", 400) + return + } else if err != nil { apiutil.InternalServerError( - rw, r, fmt.Errorf("fetching asset with id %q: %w", id, err), + rw, r, fmt.Errorf("fetching asset at path %q: %w", path, err), ) return } - a.writePostAsset(rw, r, id, bytes.NewReader(buf.Bytes())) + http.ServeContent( + rw, r, path, time.Time{}, bytes.NewReader(buf.Bytes()), + ) }) } -- cgit v1.2.3