move CORS allowance to the REST API router

[github/Chocobozzz/PeerTube.git] / server / controllers / static.ts

diff --git a/server/controllers/static.ts b/server/controllers/static.ts
index ccae6051727177c18d92290a53ef3d5b0a5d637a..139ba67cc290b09db1c9867911f14f43669ee04b 100644 (file)
--- a/server/controllers/static.ts
+++ b/server/controllers/static.ts
@@ -1,8 +1,9 @@
 import * as cors from 'cors'
 import * as express from 'express'
-import { CONFIG, STATIC_MAX_AGE, STATIC_PATHS } from '../initializers'
+import { CONFIG, STATIC_DOWNLOAD_PATHS, STATIC_MAX_AGE, STATIC_PATHS } from '../initializers'
 import { VideosPreviewCache } from '../lib/cache'
-import { asyncMiddleware } from '../middlewares'
+import { asyncMiddleware, videosGetValidator } from '../middlewares'
+import { VideoModel } from '../models/video/video'
 
 const staticRouter = express.Router()
 
@@ -16,6 +17,11 @@ staticRouter.use(
   cors(),
   express.static(torrentsPhysicalPath, { maxAge: 0 }) // Don't cache because we could regenerate the torrent file
 )
+staticRouter.use(
+  STATIC_DOWNLOAD_PATHS.TORRENTS + ':id-:resolution([0-9]+).torrent',
+  asyncMiddleware(videosGetValidator),
+  asyncMiddleware(downloadTorrent)
+)
 
 // Videos path for webseeding
 const videosPhysicalPath = CONFIG.STORAGE.VIDEOS_DIR
@@ -24,6 +30,11 @@ staticRouter.use(
   cors(),
   express.static(videosPhysicalPath, { maxAge: STATIC_MAX_AGE })
 )
+staticRouter.use(
+  STATIC_DOWNLOAD_PATHS.VIDEOS + ':id-:resolution([0-9]+).:extension',
+  asyncMiddleware(videosGetValidator),
+  asyncMiddleware(downloadVideoFile)
+)
 
 // Thumbnails path for express
 const thumbnailsPhysicalPath = CONFIG.STORAGE.THUMBNAILS_DIR
@@ -32,12 +43,24 @@ staticRouter.use(
   express.static(thumbnailsPhysicalPath, { maxAge: STATIC_MAX_AGE })
 )
 
+const avatarsPhysicalPath = CONFIG.STORAGE.AVATARS_DIR
+staticRouter.use(
+  STATIC_PATHS.AVATARS,
+  express.static(avatarsPhysicalPath, { maxAge: STATIC_MAX_AGE })
+)
+
 // Video previews path for express
 staticRouter.use(
   STATIC_PATHS.PREVIEWS + ':uuid.jpg',
   asyncMiddleware(getPreview)
 )
 
+// robots.txt service
+staticRouter.get('/robots.txt', (req: express.Request, res: express.Response) => {
+  res.type('text/plain')
+  return res.send(CONFIG.INSTANCE.ROBOTS)
+})
+
 // ---------------------------------------------------------------------------
 
 export {
@@ -52,3 +75,26 @@ async function getPreview (req: express.Request, res: express.Response, next: ex
 
   return res.sendFile(path, { maxAge: STATIC_MAX_AGE })
 }
+
+async function downloadTorrent (req: express.Request, res: express.Response, next: express.NextFunction) {
+  const { video, videoFile } = getVideoAndFile(req, res)
+  if (!videoFile) return res.status(404).end()
+
+  return res.download(video.getTorrentFilePath(videoFile), `${video.name}-${videoFile.resolution}p.torrent`)
+}
+
+async function downloadVideoFile (req: express.Request, res: express.Response, next: express.NextFunction) {
+  const { video, videoFile } = getVideoAndFile(req, res)
+  if (!videoFile) return res.status(404).end()
+
+  return res.download(video.getVideoFilePath(videoFile), `${video.name}-${videoFile.resolution}p${videoFile.extname}`)
+}
+
+function getVideoAndFile (req: express.Request, res: express.Response) {
+  const resolution = parseInt(req.params.resolution, 10)
+  const video: VideoModel = res.locals.video
+
+  const videoFile = video.VideoFiles.find(f => f.resolution === resolution)
+
+  return { video, videoFile }
+}