]>
Commit | Line | Data |
---|---|---|
90a8bd30 C |
1 | import * as cors from 'cors' |
2 | import * as express from 'express' | |
3 | import { VideosTorrentCache } from '@server/lib/files-cache/videos-torrent-cache' | |
4 | import { getVideoFilePath } from '@server/lib/video-paths' | |
5 | import { MVideoFile, MVideoFullLight } from '@server/types/models' | |
6 | import { HttpStatusCode } from '@shared/core-utils/miscs/http-error-codes' | |
7 | import { VideoStreamingPlaylistType } from '@shared/models' | |
8 | import { STATIC_DOWNLOAD_PATHS } from '../initializers/constants' | |
9 | import { asyncMiddleware, videosDownloadValidator } from '../middlewares' | |
10 | ||
11 | const downloadRouter = express.Router() | |
12 | ||
13 | downloadRouter.use(cors()) | |
14 | ||
15 | downloadRouter.use( | |
16 | STATIC_DOWNLOAD_PATHS.TORRENTS + ':filename', | |
17 | downloadTorrent | |
18 | ) | |
19 | ||
20 | downloadRouter.use( | |
21 | STATIC_DOWNLOAD_PATHS.VIDEOS + ':id-:resolution([0-9]+).:extension', | |
22 | asyncMiddleware(videosDownloadValidator), | |
23 | downloadVideoFile | |
24 | ) | |
25 | ||
26 | downloadRouter.use( | |
27 | STATIC_DOWNLOAD_PATHS.HLS_VIDEOS + ':id-:resolution([0-9]+)-fragmented.:extension', | |
28 | asyncMiddleware(videosDownloadValidator), | |
29 | downloadHLSVideoFile | |
30 | ) | |
31 | ||
32 | // --------------------------------------------------------------------------- | |
33 | ||
34 | export { | |
35 | downloadRouter | |
36 | } | |
37 | ||
38 | // --------------------------------------------------------------------------- | |
39 | ||
40 | async function downloadTorrent (req: express.Request, res: express.Response) { | |
41 | const result = await VideosTorrentCache.Instance.getFilePath(req.params.filename) | |
42 | if (!result) return res.sendStatus(HttpStatusCode.NOT_FOUND_404) | |
43 | ||
44 | return res.download(result.path, result.downloadName) | |
45 | } | |
46 | ||
47 | function downloadVideoFile (req: express.Request, res: express.Response) { | |
48 | const video = res.locals.videoAll | |
49 | ||
50 | const videoFile = getVideoFile(req, video.VideoFiles) | |
51 | if (!videoFile) return res.status(HttpStatusCode.NOT_FOUND_404).end() | |
52 | ||
53 | return res.download(getVideoFilePath(video, videoFile), `${video.name}-${videoFile.resolution}p${videoFile.extname}`) | |
54 | } | |
55 | ||
56 | function downloadHLSVideoFile (req: express.Request, res: express.Response) { | |
57 | const video = res.locals.videoAll | |
58 | const playlist = getHLSPlaylist(video) | |
59 | if (!playlist) return res.status(HttpStatusCode.NOT_FOUND_404).end | |
60 | ||
61 | const videoFile = getVideoFile(req, playlist.VideoFiles) | |
62 | if (!videoFile) return res.status(HttpStatusCode.NOT_FOUND_404).end() | |
63 | ||
64 | const filename = `${video.name}-${videoFile.resolution}p-${playlist.getStringType()}${videoFile.extname}` | |
65 | return res.download(getVideoFilePath(playlist, videoFile), filename) | |
66 | } | |
67 | ||
68 | function getVideoFile (req: express.Request, files: MVideoFile[]) { | |
69 | const resolution = parseInt(req.params.resolution, 10) | |
70 | return files.find(f => f.resolution === resolution) | |
71 | } | |
72 | ||
73 | function getHLSPlaylist (video: MVideoFullLight) { | |
74 | const playlist = video.VideoStreamingPlaylists.find(p => p.type === VideoStreamingPlaylistType.HLS) | |
75 | if (!playlist) return undefined | |
76 | ||
77 | return Object.assign(playlist, { Video: video }) | |
78 | } |