X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Fvideo.ts;h=f5f645d3eac1997614c177f00c5a64c215f38ef0;hb=75b7117f078461d2507572ba9da6527894e1b734;hp=c90fe06c78e2729174f58f5dee75741d9f194cd5;hpb=97567dd81f508dd6295ac4d73d849aa2ce0a6549;p=github%2FChocobozzz%2FPeerTube.git

diff --git a/server/helpers/video.ts b/server/helpers/video.ts
index c90fe06c7..f5f645d3e 100644
--- a/server/helpers/video.ts
+++ b/server/helpers/video.ts
@@ -1,27 +1,50 @@
-import { VideoModel } from '../models/video/video'
+import { Response } from 'express'
+import { CONFIG } from '@server/initializers/config'
+import { isStreamingPlaylist, MStreamingPlaylistVideo, MVideo } from '@server/types/models'
+import { VideoPrivacy, VideoState } from '@shared/models'
 
-type VideoFetchType = 'all' | 'only-video' | 'only-video-with-rights' | 'id' | 'none'
+function getVideoWithAttributes (res: Response) {
+  return res.locals.videoAPI || res.locals.videoAll || res.locals.onlyVideo
+}
+
+function extractVideo (videoOrPlaylist: MVideo | MStreamingPlaylistVideo) {
+  return isStreamingPlaylist(videoOrPlaylist)
+    ? videoOrPlaylist.Video
+    : videoOrPlaylist
+}
 
-function fetchVideo (id: number | string, fetchType: VideoFetchType, userId?: number) {
-  if (fetchType === 'all') return VideoModel.loadAndPopulateAccountAndServerAndTags(id, undefined, userId)
+function isPrivacyForFederation (privacy: VideoPrivacy) {
+  const castedPrivacy = parseInt(privacy + '', 10)
 
-  if (fetchType === 'only-video-with-rights') return VideoModel.loadWithRights(id)
+  return castedPrivacy === VideoPrivacy.PUBLIC ||
+    (CONFIG.FEDERATION.VIDEOS.FEDERATE_UNLISTED === true && castedPrivacy === VideoPrivacy.UNLISTED)
+}
 
-  if (fetchType === 'only-video') return VideoModel.load(id)
+function isStateForFederation (state: VideoState) {
+  const castedState = parseInt(state + '', 10)
 
-  if (fetchType === 'id' || fetchType === 'none') return VideoModel.loadOnlyId(id)
+  return castedState === VideoState.PUBLISHED || castedState === VideoState.WAITING_FOR_LIVE || castedState === VideoState.LIVE_ENDED
 }
 
-type VideoFetchByUrlType = 'all' | 'only-video'
-function fetchVideoByUrl (url: string, fetchType: VideoFetchByUrlType) {
-  if (fetchType === 'all') return VideoModel.loadByUrlAndPopulateAccount(url)
+function getPrivaciesForFederation () {
+  return (CONFIG.FEDERATION.VIDEOS.FEDERATE_UNLISTED === true)
+    ? [ { privacy: VideoPrivacy.PUBLIC }, { privacy: VideoPrivacy.UNLISTED } ]
+    : [ { privacy: VideoPrivacy.PUBLIC } ]
+}
+
+function getExtFromMimetype (mimeTypes: { [id: string]: string | string[] }, mimeType: string) {
+  const value = mimeTypes[mimeType]
+
+  if (Array.isArray(value)) return value[0]
 
-  if (fetchType === 'only-video') return VideoModel.loadByUrl(url)
+  return value
 }
 
 export {
-  VideoFetchType,
-  VideoFetchByUrlType,
-  fetchVideo,
-  fetchVideoByUrl
+  getVideoWithAttributes,
+  extractVideo,
+  getExtFromMimetype,
+  isStateForFederation,
+  isPrivacyForFederation,
+  getPrivaciesForFederation
 }