mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
dce6cb601f
* MM-14686 Implement /api/v4/image when proxy is disabled * MM-14686 Send all image proxy requests through /api/v4/image * Update unit tests
67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package imageproxy
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type AtmosCamoBackend struct {
|
|
proxy *ImageProxy
|
|
}
|
|
|
|
func makeAtmosCamoBackend(proxy *ImageProxy) *AtmosCamoBackend {
|
|
return &AtmosCamoBackend{
|
|
proxy: proxy,
|
|
}
|
|
}
|
|
|
|
func (backend *AtmosCamoBackend) GetImage(w http.ResponseWriter, r *http.Request, imageURL string) {
|
|
http.Redirect(w, r, backend.getAtmosCamoImageURL(imageURL), http.StatusFound)
|
|
}
|
|
|
|
func (backend *AtmosCamoBackend) GetImageDirect(imageURL string) (io.ReadCloser, string, error) {
|
|
req, err := http.NewRequest("GET", backend.getAtmosCamoImageURL(imageURL), nil)
|
|
if err != nil {
|
|
return nil, "", Error{err}
|
|
}
|
|
|
|
client := backend.proxy.HTTPService.MakeClient(false)
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, "", Error{err}
|
|
}
|
|
|
|
// Note that we don't do any additional validation of the received data since we expect the image proxy to do that
|
|
return resp.Body, resp.Header.Get("Content-Type"), nil
|
|
}
|
|
|
|
func (backend *AtmosCamoBackend) getAtmosCamoImageURL(imageURL string) string {
|
|
cfg := *backend.proxy.ConfigService.Config()
|
|
siteURL := *cfg.ServiceSettings.SiteURL
|
|
proxyURL := *cfg.ImageProxySettings.RemoteImageProxyURL
|
|
options := *cfg.ImageProxySettings.RemoteImageProxyOptions
|
|
|
|
return getAtmosCamoImageURL(imageURL, siteURL, proxyURL, options)
|
|
}
|
|
|
|
func getAtmosCamoImageURL(imageURL, siteURL, proxyURL, options string) string {
|
|
// Don't proxy blank images, relative URLs, absolute URLs on this server, or URLs that are already going through the proxy
|
|
if imageURL == "" || imageURL[0] == '/' || (siteURL != "" && strings.HasPrefix(imageURL, siteURL)) || strings.HasPrefix(imageURL, proxyURL) {
|
|
return imageURL
|
|
}
|
|
|
|
mac := hmac.New(sha1.New, []byte(options))
|
|
mac.Write([]byte(imageURL))
|
|
digest := hex.EncodeToString(mac.Sum(nil))
|
|
|
|
return proxyURL + "/" + digest + "/" + hex.EncodeToString([]byte(imageURL))
|
|
}
|