]>
Commit | Line | Data |
---|---|---|
1 | 'use strict' | |
2 | ||
3 | const async = require('async') | |
4 | const config = require('config') | |
5 | const dz = require('dezalgo') | |
6 | const fs = require('fs') | |
7 | const mongoose = require('mongoose') | |
8 | const path = require('path') | |
9 | ||
10 | const logger = require('../helpers/logger') | |
11 | ||
12 | const http = config.get('webserver.https') === true ? 'https' : 'http' | |
13 | const host = config.get('webserver.host') | |
14 | const port = config.get('webserver.port') | |
15 | const uploadDir = path.join(__dirname, '..', '..', config.get('storage.uploads')) | |
16 | ||
17 | // --------------------------------------------------------------------------- | |
18 | ||
19 | const videosSchema = mongoose.Schema({ | |
20 | name: String, | |
21 | namePath: String, | |
22 | description: String, | |
23 | magnetUri: String, | |
24 | podUrl: String, | |
25 | author: String | |
26 | }) | |
27 | const VideosDB = mongoose.model('videos', videosSchema) | |
28 | ||
29 | // --------------------------------------------------------------------------- | |
30 | ||
31 | const Videos = { | |
32 | add: add, | |
33 | addRemotes: addRemotes, | |
34 | get: get, | |
35 | list: list, | |
36 | listOwned: listOwned, | |
37 | removeOwned: removeOwned, | |
38 | removeAllRemotes: removeAllRemotes, | |
39 | removeAllRemotesOf: removeAllRemotesOf, | |
40 | removeRemotesOfByMagnetUris: removeRemotesOfByMagnetUris, | |
41 | search: search | |
42 | } | |
43 | ||
44 | function add (video, callback) { | |
45 | logger.info('Adding %s video to database.', video.name) | |
46 | ||
47 | const params = video | |
48 | params.podUrl = http + '://' + host + ':' + port | |
49 | ||
50 | VideosDB.create(params, function (err, video) { | |
51 | if (err) { | |
52 | logger.error('Cannot insert this video into database.') | |
53 | return callback(err) | |
54 | } | |
55 | ||
56 | callback(null) | |
57 | }) | |
58 | } | |
59 | ||
60 | // TODO: avoid doublons | |
61 | function addRemotes (videos, callback) { | |
62 | if (!callback) callback = function () {} | |
63 | ||
64 | const to_add = [] | |
65 | ||
66 | async.each(videos, function (video, callback_each) { | |
67 | callback_each = dz(callback_each) | |
68 | logger.debug('Add remote video from pod: %s', video.podUrl) | |
69 | ||
70 | const params = { | |
71 | name: video.name, | |
72 | namePath: null, | |
73 | description: video.description, | |
74 | magnetUri: video.magnetUri, | |
75 | podUrl: video.podUrl | |
76 | } | |
77 | ||
78 | to_add.push(params) | |
79 | ||
80 | callback_each() | |
81 | }, function () { | |
82 | VideosDB.create(to_add, function (err, videos) { | |
83 | if (err) { | |
84 | logger.error('Cannot insert this remote video.') | |
85 | return callback(err) | |
86 | } | |
87 | ||
88 | return callback(null, videos) | |
89 | }) | |
90 | }) | |
91 | } | |
92 | ||
93 | function get (id, callback) { | |
94 | VideosDB.findById(id, function (err, video) { | |
95 | if (err) { | |
96 | logger.error('Cannot get this video.') | |
97 | return callback(err) | |
98 | } | |
99 | ||
100 | return callback(null, video) | |
101 | }) | |
102 | } | |
103 | ||
104 | function list (callback) { | |
105 | VideosDB.find(function (err, videos_list) { | |
106 | if (err) { | |
107 | logger.error('Cannot get the list of the videos.') | |
108 | return callback(err) | |
109 | } | |
110 | ||
111 | return callback(null, videos_list) | |
112 | }) | |
113 | } | |
114 | ||
115 | function listOwned (callback) { | |
116 | // If namePath is not null this is *our* video | |
117 | VideosDB.find({ namePath: { $ne: null } }, function (err, videos_list) { | |
118 | if (err) { | |
119 | logger.error('Cannot get the list of owned videos.') | |
120 | return callback(err) | |
121 | } | |
122 | ||
123 | return callback(null, videos_list) | |
124 | }) | |
125 | } | |
126 | ||
127 | function removeOwned (id, callback) { | |
128 | VideosDB.findByIdAndRemove(id, function (err, video) { | |
129 | if (err) { | |
130 | logger.error('Cannot remove the torrent.') | |
131 | return callback(err) | |
132 | } | |
133 | ||
134 | fs.unlink(uploadDir + video.namePath, function (err) { | |
135 | if (err) { | |
136 | logger.error('Cannot remove this video file.') | |
137 | return callback(err) | |
138 | } | |
139 | ||
140 | callback(null) | |
141 | }) | |
142 | }) | |
143 | } | |
144 | ||
145 | function removeAllRemotes (callback) { | |
146 | VideosDB.remove({ namePath: null }, callback) | |
147 | } | |
148 | ||
149 | function removeAllRemotesOf (fromUrl, callback) { | |
150 | VideosDB.remove({ podUrl: fromUrl }, callback) | |
151 | } | |
152 | ||
153 | // Use the magnet Uri because the _id field is not the same on different servers | |
154 | function removeRemotesOfByMagnetUris (fromUrl, magnetUris, callback) { | |
155 | if (callback === undefined) callback = function () {} | |
156 | ||
157 | VideosDB.find({ magnetUri: { $in: magnetUris } }, function (err, videos) { | |
158 | if (err || !videos) { | |
159 | logger.error('Cannot find the torrent URI of these remote videos.') | |
160 | return callback(err) | |
161 | } | |
162 | ||
163 | const to_remove = [] | |
164 | async.each(videos, function (video, callback_async) { | |
165 | callback_async = dz(callback_async) | |
166 | ||
167 | if (video.podUrl !== fromUrl) { | |
168 | logger.error('The pod %s has not the rights on the video of %s.', fromUrl, video.podUrl) | |
169 | } else { | |
170 | to_remove.push(video._id) | |
171 | } | |
172 | ||
173 | callback_async() | |
174 | }, function () { | |
175 | VideosDB.remove({ _id: { $in: to_remove } }, function (err) { | |
176 | if (err) { | |
177 | logger.error('Cannot remove the remote videos.') | |
178 | return callback(err) | |
179 | } | |
180 | ||
181 | logger.info('Removed remote videos from %s.', fromUrl) | |
182 | callback(null) | |
183 | }) | |
184 | }) | |
185 | }) | |
186 | } | |
187 | ||
188 | function search (name, callback) { | |
189 | VideosDB.find({ name: new RegExp(name) }, function (err, videos) { | |
190 | if (err) { | |
191 | logger.error('Cannot search the videos.') | |
192 | return callback(err) | |
193 | } | |
194 | ||
195 | return callback(null, videos) | |
196 | }) | |
197 | } | |
198 | ||
199 | // --------------------------------------------------------------------------- | |
200 | ||
201 | module.exports = Videos |