Support '/w/' and '/w/p/' for watch page

[github/Chocobozzz/PeerTube.git] / server / middlewares / validators / oembed.ts

diff --git a/server/middlewares/validators/oembed.ts b/server/middlewares/validators/oembed.ts
index 0bb908d0b6719363a5f0c4e57279f8f940181b2c..165eda6d56addf50a7b9363b78236428ab1ae3d7 100644 (file)
--- a/server/middlewares/validators/oembed.ts
+++ b/server/middlewares/validators/oembed.ts
@@ -1,15 +1,34 @@
 import * as express from 'express'
-import { query } from 'express-validator/check'
+import { query } from 'express-validator'
 import { join } from 'path'
+import { fetchVideo } from '@server/helpers/video'
+import { VideoPlaylistModel } from '@server/models/video/video-playlist'
+import { VideoPlaylistPrivacy, VideoPrivacy } from '@shared/models'
+import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes'
 import { isTestInstance } from '../../helpers/core-utils'
 import { isIdOrUUIDValid } from '../../helpers/custom-validators/misc'
-import { doesVideoExist } from '../../helpers/custom-validators/videos'
 import { logger } from '../../helpers/logger'
-import { areValidationErrors } from './utils'
 import { WEBSERVER } from '../../initializers/constants'
+import { areValidationErrors } from './utils'
+
+const playlistPaths = [
+  join('videos', 'watch', 'playlist'),
+  join('w', 'p')
+]
+
+const videoPaths = [
+  join('videos', 'watch'),
+  'w'
+]
 
-const urlShouldStartWith = WEBSERVER.SCHEME + '://' + join(WEBSERVER.HOST, 'videos', 'watch') + '/'
-const videoWatchRegex = new RegExp('([^/]+)$')
+function buildUrls (paths: string[]) {
+  return paths.map(p => WEBSERVER.SCHEME + '://' + join(WEBSERVER.HOST, p) + '/')
+}
+
+const startPlaylistURLs = buildUrls(playlistPaths)
+const startVideoURLs = buildUrls(videoPaths)
+
+const watchRegex = /([^/]+)$/
 const isURLOptions = {
   require_host: true,
   require_tld: true
@@ -32,30 +51,64 @@ const oembedValidator = [
     if (areValidationErrors(req, res)) return
 
     if (req.query.format !== undefined && req.query.format !== 'json') {
-      return res.status(501)
-                .json({ error: 'Requested format is not implemented on server.' })
-                .end()
+      return res.status(HttpStatusCode.NOT_IMPLEMENTED_501)
+        .json({ error: 'Requested format is not implemented on server.' })
     }
 
-    const startIsOk = req.query.url.startsWith(urlShouldStartWith)
-    const matches = videoWatchRegex.exec(req.query.url)
+    const url = req.query.url as string
+
+    const isPlaylist = startPlaylistURLs.some(u => url.startsWith(u))
+    const isVideo = isPlaylist ? false : startVideoURLs.some(u => url.startsWith(u))
+
+    const startIsOk = isVideo || isPlaylist
+
+    const matches = watchRegex.exec(url)
+
     if (startIsOk === false || matches === null) {
-      return res.status(400)
-                .json({ error: 'Invalid url.' })
-                .end()
+      return res.status(HttpStatusCode.BAD_REQUEST_400)
+        .json({ error: 'Invalid url.' })
+    }
+
+    const elementId = matches[1]
+    if (isIdOrUUIDValid(elementId) === false) {
+      return res.status(HttpStatusCode.BAD_REQUEST_400)
+        .json({ error: 'Invalid video or playlist id.' })
     }
 
-    const videoId = matches[1]
-    if (isIdOrUUIDValid(videoId) === false) {
-      return res.status(400)
-                .json({ error: 'Invalid video id.' })
-                .end()
+    if (isVideo) {
+      const video = await fetchVideo(elementId, 'all')
+
+      if (!video) {
+        return res.status(HttpStatusCode.NOT_FOUND_404)
+          .json({ error: 'Video not found' })
+      }
+
+      if (video.privacy !== VideoPrivacy.PUBLIC) {
+        return res.status(HttpStatusCode.FORBIDDEN_403)
+          .json({ error: 'Video is not public' })
+      }
+
+      res.locals.videoAll = video
+      return next()
     }
 
-    if (!await doesVideoExist(videoId, res)) return
+    // Is playlist
 
+    const videoPlaylist = await VideoPlaylistModel.loadWithAccountAndChannelSummary(elementId, undefined)
+    if (!videoPlaylist) {
+      return res.status(HttpStatusCode.NOT_FOUND_404)
+        .json({ error: 'Video playlist not found' })
+    }
+
+    if (videoPlaylist.privacy !== VideoPlaylistPrivacy.PUBLIC) {
+      return res.status(HttpStatusCode.FORBIDDEN_403)
+        .json({ error: 'Playlist is not public' })
+    }
+
+    res.locals.videoPlaylistSummary = videoPlaylist
     return next()
   }
+
 ]
 
 // ---------------------------------------------------------------------------