From ca0affb42b61836c738c565ca4a14c29379256ba Mon Sep 17 00:00:00 2001 From: Liang Ding Date: Fri, 24 Feb 2023 10:44:14 +0800 Subject: [PATCH] =?UTF-8?q?:art:=20=E6=94=B9=E8=BF=9B=E6=B5=8F=E8=A7=88?= =?UTF-8?q?=E5=99=A8=E5=89=AA=E8=97=8F=E6=89=A9=E5=B1=95=E8=BD=AC=E6=8D=A2?= =?UTF-8?q?=E6=9C=AC=E5=9C=B0=E5=9B=BE=E7=89=87=20https://github.com/siyua?= =?UTF-8?q?n-note/siyuan/issues/7464?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- kernel/api/format.go | 6 +++++- kernel/model/assets.go | 3 ++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/kernel/api/format.go b/kernel/api/format.go index 6b8076941..a7eb4c0ac 100644 --- a/kernel/api/format.go +++ b/kernel/api/format.go @@ -35,7 +35,11 @@ func netImg2LocalAssets(c *gin.Context) { } id := arg["id"].(string) - err := model.NetImg2LocalAssets(id) + var url string + if urlArg := arg["url"]; nil != urlArg { + url = urlArg.(string) + } + err := model.NetImg2LocalAssets(id, url) if nil != err { ret.Code = -1 ret.Msg = err.Error() diff --git a/kernel/model/assets.go b/kernel/model/assets.go index cc415cc57..d8feb64a7 100644 --- a/kernel/model/assets.go +++ b/kernel/model/assets.go @@ -70,7 +70,7 @@ func DocImageAssets(rootID string) (ret []string, err error) { return } -func NetImg2LocalAssets(rootID string) (err error) { +func NetImg2LocalAssets(rootID, originalURL string) (err error) { tree, err := loadTreeByBlockID(rootID) if nil != err { return @@ -138,6 +138,7 @@ func NetImg2LocalAssets(rootID string) (err error) { } util.PushUpdateMsg(msgId, fmt.Sprintf(Conf.Language(119), u), 15000) request := httpclient.NewBrowserRequest() + request.SetHeader("Referer", originalURL) resp, reqErr := request.Get(u) if nil != reqErr { logging.LogErrorf("download net img [%s] failed: %s", u, reqErr)