aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/ui/controller/proxy.go
blob: 6c85bc22d98887be4e8ab7fb82367bc80d0e190c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// 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/miniflux/helper"
	"github.com/miniflux/miniflux/http"
	"github.com/miniflux/miniflux/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)
}