mirrors/mostlymatter
1
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

MM-17426 Properly specify multiple Accept headers for post metadata (#11766)

Просмотр исходного кода 
* MM-17326 Properly specify multiple Accept headers for post metadata

* Add q= weight to text/html
Этот коммит содержится в:
Harrison Healey
2019-08-01 13:52:38 -04:00
коммит произвёл GitHub
родитель 11472e417c
Коммит fe9e8a4493
2 изменённых файлов: 38 добавлений и 7 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

3
app/post_metadata.go
Просмотреть файл

@@ -379,7 +379,8 @@ func (a *App) getLinkMetadata(requestURL string, timestamp int64, isNewPost bool
// /api/v4/image requires authentication, so bypass the API by hitting the proxy directly // /api/v4/image requires authentication, so bypass the API by hitting the proxy directly
body, contentType, err = a.ImageProxy.GetImageDirect(a.ImageProxy.GetUnproxiedImageURL(request.URL.String())) body, contentType, err = a.ImageProxy.GetImageDirect(a.ImageProxy.GetUnproxiedImageURL(request.URL.String()))
} else { } else {
request.Header.Add("Accept", "image/*, text/html") request.Header.Add("Accept", "image/*")
request.Header.Add("Accept", "text/html;q=0.8")
client := a.HTTPService.MakeClient(false) client := a.HTTPService.MakeClient(false)
client.Timeout = time.Duration(*a.Config().ExperimentalSettings.LinkMetadataTimeoutMilliseconds) * time.Millisecond client.Timeout = time.Duration(*a.Config().ExperimentalSettings.LinkMetadataTimeoutMilliseconds) * time.Millisecond

42
app/post_metadata_test.go
Просмотреть файл

@@ -1372,26 +1372,35 @@ func TestGetLinkMetadata(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
params := r.URL.Query() params := r.URL.Query()
if strings.HasPrefix(r.URL.Path, "/image") { writeImage := func(height, width int) {
height, _ := strconv.ParseInt(params["height"][0], 10, 0)
width, _ := strconv.ParseInt(params["width"][0], 10, 0)
img := image.NewGray(image.Rect(0, 0, int(width), int(height))) img := image.NewGray(image.Rect(0, 0, height, width))
var encoder png.Encoder var encoder png.Encoder
encoder.Encode(w, img) encoder.Encode(w, img)
} else if strings.HasPrefix(r.URL.Path, "/opengraph") { }
writeHTML := func(title string) {
w.Header().Set("Content-Type", "text/html") w.Header().Set("Content-Type", "text/html")
w.Write([]byte(` w.Write([]byte(`
<html prefix="og:http://ogp.me/ns#"> <html prefix="og:http://ogp.me/ns#">
<head> <head>
<meta property="og:title" content="` + params["title"][0] + `" /> <meta property="og:title" content="` + title + `" />
</head> </head>
<body> <body>
</body> </body>
</html>`)) </html>`))
}
if strings.HasPrefix(r.URL.Path, "/image") {
height, _ := strconv.ParseInt(params["height"][0], 10, 0)
width, _ := strconv.ParseInt(params["width"][0], 10, 0)
writeImage(int(height), int(width))
} else if strings.HasPrefix(r.URL.Path, "/opengraph") {
writeHTML(params["title"][0])
} else if strings.HasPrefix(r.URL.Path, "/json") { } else if strings.HasPrefix(r.URL.Path, "/json") {
w.Header().Set("Content-Type", "application/json") w.Header().Set("Content-Type", "application/json")
@@ -1405,6 +1414,14 @@ func TestGetLinkMetadata(t *testing.T) {
case <-r.Context().Done(): case <-r.Context().Done():
} }
w.Write([]byte("</html>")) w.Write([]byte("</html>"))
} else if strings.HasPrefix(r.URL.Path, "/mixed") {
for _, acceptedType := range r.Header["Accept"] {
if strings.HasPrefix(acceptedType, "image/*") || strings.HasPrefix(acceptedType, "image/png") {
writeImage(10, 10)
} else if strings.HasPrefix(acceptedType, "text/html") {
writeHTML("mixed")
}
}
} else { } else {
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
} }
@@ -1868,6 +1885,19 @@ func TestGetLinkMetadata(t *testing.T) {
assert.NotNil(t, err) assert.NotNil(t, err)
assert.IsType(t, imageproxy.Error{}, err) assert.IsType(t, imageproxy.Error{}, err)
}) })
t.Run("should prefer images for mixed content", func(t *testing.T) {
th := setup()
defer th.TearDown()
requestURL := server.URL + "/mixed?name=" + t.Name()
timestamp := int64(1547510400000)
og, img, err := th.App.getLinkMetadata(requestURL, timestamp, true)
assert.Nil(t, og)
assert.NotNil(t, img)
assert.Nil(t, err)
})
} }
func TestResolveMetadataURL(t *testing.T) { func TestResolveMetadataURL(t *testing.T) {