southfox/miniflux
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Minor simplification of internal/proxy/proxy.go

Browse source 
- re-use ProxifiedUrl to implement AbsoluteProxifyURL, reducing the copy-pasta
- reduce the internal indentation of ProxifiedUrl by inverting some conditions
This commit is contained in:
jvoisin 2024-03-13 23:16:58 +01:00 committed by Frédéric Guillot
parent e0ee28c013
commit 66b8483791
1 changed files with 16 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
internal/proxy/proxy.go
View file

@ -19,51 +19,34 @@ import (
// ProxifyURL generates a relative URL for a proxified resource. // ProxifyURL generates a relative URL for a proxified resource.
func ProxifyURL(router *mux.Router, link string) string { func ProxifyURL(router *mux.Router, link string) string {
if link != "" { if link == "" {
proxyImageUrl := config.Opts.ProxyUrl() return ""
}
if proxyImageUrl == "" {
mac := hmac.New(sha256.New, config.Opts.ProxyPrivateKey())
mac.Write([]byte(link))
digest := mac.Sum(nil)
return route.Path(router, "proxy", "encodedDigest", base64.URLEncoding.EncodeToString(digest), "encodedURL", base64.URLEncoding.EncodeToString([]byte(link)))
}
if proxyImageUrl := config.Opts.ProxyUrl(); proxyImageUrl != "" {
proxyUrl, err := url.Parse(proxyImageUrl) proxyUrl, err := url.Parse(proxyImageUrl)
if err != nil { if err != nil {
return "" return ""
} }
proxyUrl.Path = path.Join(proxyUrl.Path, base64.URLEncoding.EncodeToString([]byte(link))) proxyUrl.Path = path.Join(proxyUrl.Path, base64.URLEncoding.EncodeToString([]byte(link)))
return proxyUrl.String() return proxyUrl.String()
} }
return ""
mac := hmac.New(sha256.New, config.Opts.ProxyPrivateKey())
mac.Write([]byte(link))
digest := mac.Sum(nil)
return route.Path(router, "proxy", "encodedDigest", base64.URLEncoding.EncodeToString(digest), "encodedURL", base64.URLEncoding.EncodeToString([]byte(link)))
} }
// AbsoluteProxifyURL generates an absolute URL for a proxified resource. // AbsoluteProxifyURL generates an absolute URL for a proxified resource.
func AbsoluteProxifyURL(router *mux.Router, host, link string) string { func AbsoluteProxifyURL(router *mux.Router, host, link string) string {
if link != "" { proxifiedUrl := ProxifyURL(router, link)
proxyImageUrl := config.Opts.ProxyUrl()
if proxyImageUrl == "" { if config.Opts.ProxyUrl() == "" {
mac := hmac.New(sha256.New, config.Opts.ProxyPrivateKey()) return proxifiedUrl
mac.Write([]byte(link))
digest := mac.Sum(nil)
path := route.Path(router, "proxy", "encodedDigest", base64.URLEncoding.EncodeToString(digest), "encodedURL", base64.URLEncoding.EncodeToString([]byte(link)))
if config.Opts.HTTPS {
return "https://" + host + path
} else {
return "http://" + host + path
}
}
proxyUrl, err := url.Parse(proxyImageUrl)
if err != nil {
return ""
}
proxyUrl.Path = path.Join(proxyUrl.Path, base64.URLEncoding.EncodeToString([]byte(link)))
return proxyUrl.String()
} }
return "" if config.Opts.HTTPS {
return "https://" + host + proxifiedUrl
}
return "http://" + host + proxifiedUrl
} }