mirror of
https://github.com/mattermost/mattermost.git
synced 2026-02-19 02:28:08 -05:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package api4
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/dyatlov/go-opengraph/opengraph"
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/utils"
|
|
)
|
|
|
|
const OPEN_GRAPH_METADATA_CACHE_SIZE = 10000
|
|
|
|
var openGraphDataCache = utils.NewLru(OPEN_GRAPH_METADATA_CACHE_SIZE)
|
|
|
|
func (api *API) InitOpenGraph() {
|
|
api.BaseRoutes.OpenGraph.Handle("", api.ApiSessionRequired(getOpenGraphMetadata)).Methods("POST")
|
|
|
|
// Dump the image cache if the proxy settings have changed. (need switch URLs to the correct proxy)
|
|
api.App.AddConfigListener(func(before, after *model.Config) {
|
|
if (before.ServiceSettings.ImageProxyType != after.ServiceSettings.ImageProxyType) ||
|
|
(before.ServiceSettings.ImageProxyURL != after.ServiceSettings.ImageProxyType) {
|
|
openGraphDataCache.Purge()
|
|
}
|
|
})
|
|
}
|
|
|
|
func OpenGraphDataWithProxyAddedToImageURLs(ogdata *opengraph.OpenGraph, toProxyURL func(string) string) *opengraph.OpenGraph {
|
|
for _, image := range ogdata.Images {
|
|
var url string
|
|
if image.SecureURL != "" {
|
|
url = image.SecureURL
|
|
} else {
|
|
url = image.URL
|
|
}
|
|
|
|
image.URL = ""
|
|
image.SecureURL = toProxyURL(url)
|
|
}
|
|
|
|
return ogdata
|
|
}
|
|
|
|
func getOpenGraphMetadata(c *Context, w http.ResponseWriter, r *http.Request) {
|
|
if !*c.App.Config().ServiceSettings.EnableLinkPreviews {
|
|
c.Err = model.NewAppError("getOpenGraphMetadata", "api.post.link_preview_disabled.app_error", nil, "", http.StatusNotImplemented)
|
|
return
|
|
}
|
|
|
|
props := model.StringInterfaceFromJson(r.Body)
|
|
|
|
url := ""
|
|
ok := false
|
|
if url, ok = props["url"].(string); len(url) == 0 || !ok {
|
|
c.SetInvalidParam("url")
|
|
return
|
|
}
|
|
|
|
ogJSONGeneric, ok := openGraphDataCache.Get(url)
|
|
if ok {
|
|
w.Write(ogJSONGeneric.([]byte))
|
|
return
|
|
}
|
|
|
|
og := c.App.GetOpenGraphMetadata(url)
|
|
|
|
// If image proxy enabled modify open graph data to feed though proxy
|
|
if toProxyURL := c.App.ImageProxyAdder(); toProxyURL != nil {
|
|
og = OpenGraphDataWithProxyAddedToImageURLs(og, toProxyURL)
|
|
}
|
|
|
|
ogJSON, err := og.ToJSON()
|
|
openGraphDataCache.AddWithExpiresInSecs(props["url"], ogJSON, 3600) // Cache would expire after 1 hour
|
|
if err != nil {
|
|
w.Write([]byte(`{"url": ""}`))
|
|
return
|
|
}
|
|
|
|
w.Write(ogJSON)
|
|
}
|