2017-05-26 09:57:09 +02:00
// Copyright 2017 Vector Creations Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package routing
import (
"net/http"
2020-06-16 19:31:38 +02:00
"strings"
2017-05-26 09:57:09 +02:00
2020-06-16 15:10:55 +02:00
userapi "github.com/matrix-org/dendrite/userapi/api"
2017-09-28 15:50:40 +02:00
2017-05-26 09:57:09 +02:00
"github.com/gorilla/mux"
2020-05-21 15:40:13 +02:00
"github.com/matrix-org/dendrite/internal/config"
2020-06-12 15:55:57 +02:00
"github.com/matrix-org/dendrite/internal/httputil"
2017-05-26 16:49:54 +02:00
"github.com/matrix-org/dendrite/mediaapi/storage"
2017-05-26 09:57:09 +02:00
"github.com/matrix-org/dendrite/mediaapi/types"
"github.com/matrix-org/gomatrixserverlib"
"github.com/matrix-org/util"
"github.com/prometheus/client_golang/prometheus"
2019-12-17 17:47:45 +01:00
"github.com/prometheus/client_golang/prometheus/promauto"
"github.com/prometheus/client_golang/prometheus/promhttp"
2017-05-26 09:57:09 +02:00
)
2020-05-22 12:43:17 +02:00
const pathPrefixR0 = "/media/r0"
2020-06-16 15:29:11 +02:00
const pathPrefixV1 = "/media/v1" // TODO: remove when synapse is fixed
2017-05-26 09:57:09 +02:00
2017-08-03 16:10:39 +02:00
// Setup registers the media API HTTP handlers
2019-07-03 17:38:50 +02:00
//
// Due to Setup being used to call many other functions, a gocyclo nolint is
// applied:
// nolint: gocyclo
2017-09-21 17:20:10 +02:00
func Setup (
2020-05-22 12:43:17 +02:00
publicAPIMux * mux . Router ,
2020-08-10 15:18:04 +02:00
cfg * config . MediaAPI ,
2020-01-03 15:07:05 +01:00
db storage . Database ,
2020-06-16 15:10:55 +02:00
userAPI userapi . UserInternalAPI ,
2017-09-21 17:20:10 +02:00
client * gomatrixserverlib . Client ,
) {
2020-05-22 12:43:17 +02:00
r0mux := publicAPIMux . PathPrefix ( pathPrefixR0 ) . Subrouter ( )
2020-06-16 15:29:11 +02:00
v1mux := publicAPIMux . PathPrefix ( pathPrefixV1 ) . Subrouter ( )
2017-06-07 01:12:49 +02:00
activeThumbnailGeneration := & types . ActiveThumbnailGeneration {
PathToResult : map [ string ] * types . ThumbnailGenerationResult { } ,
}
2020-07-08 17:39:50 +02:00
uploadHandler := httputil . MakeAuthAPI (
2020-06-16 15:10:55 +02:00
"upload" , userAPI ,
func ( req * http . Request , _ * userapi . Device ) util . JSONResponse {
2017-10-11 19:16:53 +02:00
return Upload ( req , cfg , db , activeThumbnailGeneration )
2017-09-28 15:50:40 +02:00
} ,
2020-07-08 17:39:50 +02:00
)
r0mux . Handle ( "/upload" , uploadHandler ) . Methods ( http . MethodPost , http . MethodOptions )
v1mux . Handle ( "/upload" , uploadHandler ) . Methods ( http . MethodPost , http . MethodOptions )
2017-05-26 09:57:09 +02:00
activeRemoteRequests := & types . ActiveRemoteRequests {
2017-05-31 17:56:11 +02:00
MXCToResult : map [ string ] * types . RemoteRequestResult { } ,
2017-05-26 09:57:09 +02:00
}
2020-06-16 15:29:11 +02:00
downloadHandler := makeDownloadAPI ( "download" , cfg , db , client , activeRemoteRequests , activeThumbnailGeneration )
r0mux . Handle ( "/download/{serverName}/{mediaId}" , downloadHandler ) . Methods ( http . MethodGet , http . MethodOptions )
2020-06-17 12:53:26 +02:00
r0mux . Handle ( "/download/{serverName}/{mediaId}/{downloadName}" , downloadHandler ) . Methods ( http . MethodGet , http . MethodOptions )
v1mux . Handle ( "/download/{serverName}/{mediaId}" , downloadHandler ) . Methods ( http . MethodGet , http . MethodOptions ) // TODO: remove when synapse is fixed
v1mux . Handle ( "/download/{serverName}/{mediaId}/{downloadName}" , downloadHandler ) . Methods ( http . MethodGet , http . MethodOptions ) // TODO: remove when synapse is fixed
2020-06-16 15:29:11 +02:00
2017-06-07 01:12:49 +02:00
r0mux . Handle ( "/thumbnail/{serverName}/{mediaId}" ,
2017-09-21 17:20:10 +02:00
makeDownloadAPI ( "thumbnail" , cfg , db , client , activeRemoteRequests , activeThumbnailGeneration ) ,
2018-03-13 16:55:45 +01:00
) . Methods ( http . MethodGet , http . MethodOptions )
2017-05-26 09:57:09 +02:00
}
2017-06-07 01:12:49 +02:00
2017-09-21 17:20:10 +02:00
func makeDownloadAPI (
name string ,
2020-08-10 15:18:04 +02:00
cfg * config . MediaAPI ,
2020-01-03 15:07:05 +01:00
db storage . Database ,
2017-09-21 17:20:10 +02:00
client * gomatrixserverlib . Client ,
activeRemoteRequests * types . ActiveRemoteRequests ,
activeThumbnailGeneration * types . ActiveThumbnailGeneration ,
) http . HandlerFunc {
2019-12-18 16:10:53 +01:00
counterVec := promauto . NewCounterVec (
prometheus . CounterOpts {
Name : name ,
Help : "Total number of media_api requests for either thumbnails or full downloads" ,
2019-12-17 17:47:45 +01:00
} ,
2019-12-18 16:10:53 +01:00
[ ] string { "code" } ,
)
httpHandler := func ( w http . ResponseWriter , req * http . Request ) {
req = util . RequestWithLogging ( req )
2020-05-21 15:40:13 +02:00
// Set internal headers returned regardless of the outcome of the request
2019-12-18 16:10:53 +01:00
util . SetCORSHeaders ( w )
// Content-Type will be overridden in case of returning file data, else we respond with JSON-formatted errors
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2020-06-16 19:31:38 +02:00
2020-06-12 15:55:57 +02:00
vars , _ := httputil . URLDecodeMapValues ( mux . Vars ( req ) )
2020-06-16 19:31:38 +02:00
serverName := gomatrixserverlib . ServerName ( vars [ "serverName" ] )
// For the purposes of loop avoidance, we will return a 404 if allow_remote is set to
// false in the query string and the target server name isn't our own.
// https://github.com/matrix-org/matrix-doc/pull/1265
if allowRemote := req . URL . Query ( ) . Get ( "allow_remote" ) ; strings . ToLower ( allowRemote ) == "false" {
if serverName != cfg . Matrix . ServerName {
w . WriteHeader ( http . StatusNotFound )
return
}
}
2019-12-18 16:10:53 +01:00
Download (
w ,
req ,
2020-06-16 19:31:38 +02:00
serverName ,
2019-12-18 16:10:53 +01:00
types . MediaID ( vars [ "mediaId" ] ) ,
cfg ,
db ,
client ,
activeRemoteRequests ,
activeThumbnailGeneration ,
name == "thumbnail" ,
2020-06-17 12:53:26 +02:00
vars [ "downloadName" ] ,
2019-12-18 16:10:53 +01:00
)
}
return promhttp . InstrumentHandlerCounter ( counterVec , http . HandlerFunc ( httpHandler ) )
2017-06-07 01:12:49 +02:00
}