miniflux-v2/server/ui/controller/proxy.go

57 lines
1.4 KiB
Go

// Copyright 2017 Frédéric Guillot. All rights reserved.
// Use of this source code is governed by the Apache 2.0
// license that can be found in the LICENSE file.
package controller
import (
"encoding/base64"
"errors"
"io/ioutil"
"log"
"time"
"github.com/miniflux/miniflux2/helper"
"github.com/miniflux/miniflux2/http"
"github.com/miniflux/miniflux2/server/core"
)
// ImageProxy fetch an image from a remote server and sent it back to the browser.
func (c *Controller) ImageProxy(ctx *core.Context, request *core.Request, response *core.Response) {
// If we receive a "If-None-Match" header we assume the image in stored in browser cache
if request.Request().Header.Get("If-None-Match") != "" {
response.NotModified()
return
}
encodedURL := request.StringParam("encodedURL", "")
if encodedURL == "" {
response.HTML().BadRequest(errors.New("No URL provided"))
return
}
decodedURL, err := base64.StdEncoding.DecodeString(encodedURL)
if err != nil {
response.HTML().BadRequest(errors.New("Unable to decode this URL"))
return
}
client := http.NewClient(string(decodedURL))
resp, err := client.Get()
if err != nil {
log.Println("[ImageProxy]", err)
response.HTML().NotFound()
return
}
if resp.HasServerFailure() {
response.HTML().NotFound()
return
}
body, _ := ioutil.ReadAll(resp.Body)
etag := helper.HashFromBytes(body)
response.Cache(resp.ContentType, etag, body, 72*time.Hour)
}