]>
Commit | Line | Data |
---|---|---|
1 | import * as magnetUtil from 'magnet-uri' | |
2 | import { VideoTorrentObject } from '../../../../shared' | |
3 | import { VideoChannelObject } from '../../../../shared/models/activitypub/objects/video-channel-object' | |
4 | import { VideoPrivacy } from '../../../../shared/models/videos/video-privacy.enum' | |
5 | import { isVideoFileInfoHashValid } from '../../../helpers/custom-validators/videos' | |
6 | import { doRequest } from '../../../helpers/requests' | |
7 | import { database as db } from '../../../initializers' | |
8 | import { ACTIVITY_PUB, VIDEO_MIMETYPE_EXT } from '../../../initializers/constants' | |
9 | import { AccountInstance } from '../../../models/account/account-interface' | |
10 | import { VideoChannelInstance } from '../../../models/video/video-channel-interface' | |
11 | import { VideoFileAttributes } from '../../../models/video/video-file-interface' | |
12 | import { VideoAttributes, VideoInstance } from '../../../models/video/video-interface' | |
13 | import { getOrCreateAccountAndServer } from '../account' | |
14 | ||
15 | function videoChannelActivityObjectToDBAttributes (videoChannelObject: VideoChannelObject, account: AccountInstance) { | |
16 | return { | |
17 | name: videoChannelObject.name, | |
18 | description: videoChannelObject.content, | |
19 | uuid: videoChannelObject.uuid, | |
20 | url: videoChannelObject.id, | |
21 | createdAt: new Date(videoChannelObject.published), | |
22 | updatedAt: new Date(videoChannelObject.updated), | |
23 | remote: true, | |
24 | accountId: account.id | |
25 | } | |
26 | } | |
27 | ||
28 | async function videoActivityObjectToDBAttributes ( | |
29 | videoChannel: VideoChannelInstance, | |
30 | videoObject: VideoTorrentObject, | |
31 | to: string[] = [], | |
32 | cc: string[] = [] | |
33 | ) { | |
34 | let privacy = VideoPrivacy.PRIVATE | |
35 | if (to.indexOf(ACTIVITY_PUB.PUBLIC) !== -1) privacy = VideoPrivacy.PUBLIC | |
36 | else if (cc.indexOf(ACTIVITY_PUB.PUBLIC) !== -1) privacy = VideoPrivacy.UNLISTED | |
37 | ||
38 | const duration = videoObject.duration.replace(/[^\d]+/, '') | |
39 | let language = null | |
40 | if (videoObject.language) { | |
41 | language = parseInt(videoObject.language.identifier, 10) | |
42 | } | |
43 | ||
44 | const videoData: VideoAttributes = { | |
45 | name: videoObject.name, | |
46 | uuid: videoObject.uuid, | |
47 | url: videoObject.id, | |
48 | category: parseInt(videoObject.category.identifier, 10), | |
49 | licence: parseInt(videoObject.licence.identifier, 10), | |
50 | language, | |
51 | nsfw: videoObject.nsfw, | |
52 | description: videoObject.content, | |
53 | channelId: videoChannel.id, | |
54 | duration: parseInt(duration, 10), | |
55 | createdAt: new Date(videoObject.published), | |
56 | // FIXME: updatedAt does not seems to be considered by Sequelize | |
57 | updatedAt: new Date(videoObject.updated), | |
58 | views: videoObject.views, | |
59 | likes: 0, | |
60 | dislikes: 0, | |
61 | remote: true, | |
62 | privacy | |
63 | } | |
64 | ||
65 | return videoData | |
66 | } | |
67 | ||
68 | function videoFileActivityUrlToDBAttributes (videoCreated: VideoInstance, videoObject: VideoTorrentObject) { | |
69 | const mimeTypes = Object.keys(VIDEO_MIMETYPE_EXT) | |
70 | const fileUrls = videoObject.url.filter(u => { | |
71 | return mimeTypes.indexOf(u.mimeType) !== -1 && u.mimeType.startsWith('video/') | |
72 | }) | |
73 | ||
74 | if (fileUrls.length === 0) { | |
75 | throw new Error('Cannot find video files for ' + videoCreated.url) | |
76 | } | |
77 | ||
78 | const attributes: VideoFileAttributes[] = [] | |
79 | for (const fileUrl of fileUrls) { | |
80 | // Fetch associated magnet uri | |
81 | const magnet = videoObject.url.find(u => { | |
82 | return u.mimeType === 'application/x-bittorrent;x-scheme-handler/magnet' && u.width === fileUrl.width | |
83 | }) | |
84 | ||
85 | if (!magnet) throw new Error('Cannot find associated magnet uri for file ' + fileUrl.url) | |
86 | ||
87 | const parsed = magnetUtil.decode(magnet.url) | |
88 | if (!parsed || isVideoFileInfoHashValid(parsed.infoHash) === false) throw new Error('Cannot parse magnet URI ' + magnet.url) | |
89 | ||
90 | const attribute = { | |
91 | extname: VIDEO_MIMETYPE_EXT[fileUrl.mimeType], | |
92 | infoHash: parsed.infoHash, | |
93 | resolution: fileUrl.width, | |
94 | size: fileUrl.size, | |
95 | videoId: videoCreated.id | |
96 | } | |
97 | attributes.push(attribute) | |
98 | } | |
99 | ||
100 | return attributes | |
101 | } | |
102 | ||
103 | async function addVideoShares (instance: VideoInstance, shares: string[]) { | |
104 | for (const share of shares) { | |
105 | // Fetch url | |
106 | const json = await doRequest({ | |
107 | uri: share, | |
108 | json: true | |
109 | }) | |
110 | const actor = json['actor'] | |
111 | if (!actor) continue | |
112 | ||
113 | const account = await getOrCreateAccountAndServer(actor) | |
114 | ||
115 | const entry = { | |
116 | accountId: account.id, | |
117 | videoId: instance.id | |
118 | } | |
119 | ||
120 | await db.VideoShare.findOrCreate({ | |
121 | where: entry, | |
122 | defaults: entry | |
123 | }) | |
124 | } | |
125 | } | |
126 | ||
127 | async function addVideoChannelShares (instance: VideoChannelInstance, shares: string[]) { | |
128 | for (const share of shares) { | |
129 | // Fetch url | |
130 | const json = await doRequest({ | |
131 | uri: share, | |
132 | json: true | |
133 | }) | |
134 | const actor = json['actor'] | |
135 | if (!actor) continue | |
136 | ||
137 | const account = await getOrCreateAccountAndServer(actor) | |
138 | ||
139 | const entry = { | |
140 | accountId: account.id, | |
141 | videoChannelId: instance.id | |
142 | } | |
143 | ||
144 | await db.VideoChannelShare.findOrCreate({ | |
145 | where: entry, | |
146 | defaults: entry | |
147 | }) | |
148 | } | |
149 | } | |
150 | ||
151 | // --------------------------------------------------------------------------- | |
152 | ||
153 | export { | |
154 | videoFileActivityUrlToDBAttributes, | |
155 | videoActivityObjectToDBAttributes, | |
156 | videoChannelActivityObjectToDBAttributes, | |
157 | addVideoChannelShares, | |
158 | addVideoShares | |
159 | } |