diff options
Diffstat (limited to 'server')
-rw-r--r-- | server/controllers/api/remote.js | 95 | ||||
-rw-r--r-- | server/controllers/api/videos.js | 99 | ||||
-rw-r--r-- | server/controllers/client.js | 2 | ||||
-rw-r--r-- | server/initializers/database.js | 13 | ||||
-rw-r--r-- | server/lib/friends.js | 10 | ||||
-rw-r--r-- | server/models/pod.js | 3 | ||||
-rw-r--r-- | server/models/request.js | 6 | ||||
-rw-r--r-- | server/models/tag.js | 30 | ||||
-rw-r--r-- | server/models/video.js | 97 | ||||
-rw-r--r-- | server/models/videoTag.js | 9 | ||||
-rw-r--r-- | server/tests/api/check-params.js | 4 | ||||
-rw-r--r-- | server/tests/api/requests.js | 15 | ||||
-rw-r--r-- | server/tests/api/single-pod.js | 90 |
13 files changed, 334 insertions, 139 deletions
diff --git a/server/controllers/api/remote.js b/server/controllers/api/remote.js index a6753a2b0..c7a5658e8 100644 --- a/server/controllers/api/remote.js +++ b/server/controllers/api/remote.js | |||
@@ -50,28 +50,35 @@ function remoteVideos (req, res, next) { | |||
50 | return res.type('json').status(204).end() | 50 | return res.type('json').status(204).end() |
51 | } | 51 | } |
52 | 52 | ||
53 | function addRemoteVideo (videoToCreateData, fromHost, callback) { | 53 | function addRemoteVideo (videoToCreateData, fromHost, finalCallback) { |
54 | logger.debug('Adding remote video "%s".', videoToCreateData.name) | 54 | logger.debug('Adding remote video "%s".', videoToCreateData.name) |
55 | 55 | ||
56 | waterfall([ | 56 | waterfall([ |
57 | 57 | ||
58 | function findOrCreatePod (callback) { | 58 | function startTransaction (callback) { |
59 | db.sequelize.transaction().asCallback(function (err, t) { | ||
60 | return callback(err, t) | ||
61 | }) | ||
62 | }, | ||
63 | |||
64 | function findOrCreatePod (t, callback) { | ||
59 | const query = { | 65 | const query = { |
60 | where: { | 66 | where: { |
61 | host: fromHost | 67 | host: fromHost |
62 | }, | 68 | }, |
63 | defaults: { | 69 | defaults: { |
64 | host: fromHost | 70 | host: fromHost |
65 | } | 71 | }, |
72 | transaction: t | ||
66 | } | 73 | } |
67 | 74 | ||
68 | db.Pod.findOrCreate(query).asCallback(function (err, result) { | 75 | db.Pod.findOrCreate(query).asCallback(function (err, result) { |
69 | // [ instance, wasCreated ] | 76 | // [ instance, wasCreated ] |
70 | return callback(err, result[0]) | 77 | return callback(err, t, result[0]) |
71 | }) | 78 | }) |
72 | }, | 79 | }, |
73 | 80 | ||
74 | function findOrCreateAuthor (pod, callback) { | 81 | function findOrCreateAuthor (t, pod, callback) { |
75 | const username = videoToCreateData.author | 82 | const username = videoToCreateData.author |
76 | 83 | ||
77 | const query = { | 84 | const query = { |
@@ -82,16 +89,45 @@ function addRemoteVideo (videoToCreateData, fromHost, callback) { | |||
82 | defaults: { | 89 | defaults: { |
83 | name: username, | 90 | name: username, |
84 | podId: pod.id | 91 | podId: pod.id |
85 | } | 92 | }, |
93 | transaction: t | ||
86 | } | 94 | } |
87 | 95 | ||
88 | db.Author.findOrCreate(query).asCallback(function (err, result) { | 96 | db.Author.findOrCreate(query).asCallback(function (err, result) { |
89 | // [ instance, wasCreated ] | 97 | // [ instance, wasCreated ] |
90 | return callback(err, result[0]) | 98 | return callback(err, t, result[0]) |
91 | }) | 99 | }) |
92 | }, | 100 | }, |
93 | 101 | ||
94 | function createVideoObject (author, callback) { | 102 | function findOrCreateTags (t, author, callback) { |
103 | const tags = videoToCreateData.tags | ||
104 | const tagInstances = [] | ||
105 | |||
106 | each(tags, function (tag, callbackEach) { | ||
107 | const query = { | ||
108 | where: { | ||
109 | name: tag | ||
110 | }, | ||
111 | defaults: { | ||
112 | name: tag | ||
113 | }, | ||
114 | transaction: t | ||
115 | } | ||
116 | |||
117 | db.Tag.findOrCreate(query).asCallback(function (err, res) { | ||
118 | if (err) return callbackEach(err) | ||
119 | |||
120 | // res = [ tag, isCreated ] | ||
121 | const tag = res[0] | ||
122 | tagInstances.push(tag) | ||
123 | return callbackEach() | ||
124 | }) | ||
125 | }, function (err) { | ||
126 | return callback(err, t, author, tagInstances) | ||
127 | }) | ||
128 | }, | ||
129 | |||
130 | function createVideoObject (t, author, tagInstances, callback) { | ||
95 | const videoData = { | 131 | const videoData = { |
96 | name: videoToCreateData.name, | 132 | name: videoToCreateData.name, |
97 | remoteId: videoToCreateData.remoteId, | 133 | remoteId: videoToCreateData.remoteId, |
@@ -99,31 +135,58 @@ function addRemoteVideo (videoToCreateData, fromHost, callback) { | |||
99 | infoHash: videoToCreateData.infoHash, | 135 | infoHash: videoToCreateData.infoHash, |
100 | description: videoToCreateData.description, | 136 | description: videoToCreateData.description, |
101 | authorId: author.id, | 137 | authorId: author.id, |
102 | duration: videoToCreateData.duration, | 138 | duration: videoToCreateData.duration |
103 | tags: videoToCreateData.tags | ||
104 | } | 139 | } |
105 | 140 | ||
106 | const video = db.Video.build(videoData) | 141 | const video = db.Video.build(videoData) |
107 | 142 | ||
108 | return callback(null, video) | 143 | return callback(null, t, tagInstances, video) |
109 | }, | 144 | }, |
110 | 145 | ||
111 | function generateThumbnail (video, callback) { | 146 | function generateThumbnail (t, tagInstances, video, callback) { |
112 | db.Video.generateThumbnailFromBase64(video, videoToCreateData.thumbnailBase64, function (err) { | 147 | db.Video.generateThumbnailFromBase64(video, videoToCreateData.thumbnailBase64, function (err) { |
113 | if (err) { | 148 | if (err) { |
114 | logger.error('Cannot generate thumbnail from base 64 data.', { error: err }) | 149 | logger.error('Cannot generate thumbnail from base 64 data.', { error: err }) |
115 | return callback(err) | 150 | return callback(err) |
116 | } | 151 | } |
117 | 152 | ||
118 | video.save().asCallback(callback) | 153 | return callback(err, t, tagInstances, video) |
119 | }) | 154 | }) |
120 | }, | 155 | }, |
121 | 156 | ||
122 | function insertIntoDB (video, callback) { | 157 | function insertVideoIntoDB (t, tagInstances, video, callback) { |
123 | video.save().asCallback(callback) | 158 | const options = { |
159 | transaction: t | ||
160 | } | ||
161 | |||
162 | video.save(options).asCallback(function (err, videoCreated) { | ||
163 | return callback(err, t, tagInstances, videoCreated) | ||
164 | }) | ||
165 | }, | ||
166 | |||
167 | function associateTagsToVideo (t, tagInstances, video, callback) { | ||
168 | const options = { transaction: t } | ||
169 | |||
170 | video.setTags(tagInstances, options).asCallback(function (err) { | ||
171 | return callback(err, t) | ||
172 | }) | ||
124 | } | 173 | } |
125 | 174 | ||
126 | ], callback) | 175 | ], function (err, t) { |
176 | if (err) { | ||
177 | logger.error('Cannot insert the remote video.') | ||
178 | |||
179 | // Abort transaction? | ||
180 | if (t) t.rollback() | ||
181 | |||
182 | return finalCallback(err) | ||
183 | } | ||
184 | |||
185 | // Commit transaction | ||
186 | t.commit() | ||
187 | |||
188 | return finalCallback() | ||
189 | }) | ||
127 | } | 190 | } |
128 | 191 | ||
129 | function removeRemoteVideo (videoToRemoveData, fromHost, callback) { | 192 | function removeRemoteVideo (videoToRemoveData, fromHost, callback) { |
diff --git a/server/controllers/api/videos.js b/server/controllers/api/videos.js index a61f2b2c9..992f03db0 100644 --- a/server/controllers/api/videos.js +++ b/server/controllers/api/videos.js | |||
@@ -1,5 +1,6 @@ | |||
1 | 'use strict' | 1 | 'use strict' |
2 | 2 | ||
3 | const each = require('async/each') | ||
3 | const express = require('express') | 4 | const express = require('express') |
4 | const fs = require('fs') | 5 | const fs = require('fs') |
5 | const multer = require('multer') | 6 | const multer = require('multer') |
@@ -87,7 +88,13 @@ function addVideo (req, res, next) { | |||
87 | 88 | ||
88 | waterfall([ | 89 | waterfall([ |
89 | 90 | ||
90 | function findOrCreateAuthor (callback) { | 91 | function startTransaction (callback) { |
92 | db.sequelize.transaction().asCallback(function (err, t) { | ||
93 | return callback(err, t) | ||
94 | }) | ||
95 | }, | ||
96 | |||
97 | function findOrCreateAuthor (t, callback) { | ||
91 | const username = res.locals.oauth.token.user.username | 98 | const username = res.locals.oauth.token.user.username |
92 | 99 | ||
93 | const query = { | 100 | const query = { |
@@ -98,75 +105,125 @@ function addVideo (req, res, next) { | |||
98 | defaults: { | 105 | defaults: { |
99 | name: username, | 106 | name: username, |
100 | podId: null // null because it is OUR pod | 107 | podId: null // null because it is OUR pod |
101 | } | 108 | }, |
109 | transaction: t | ||
102 | } | 110 | } |
103 | 111 | ||
104 | db.Author.findOrCreate(query).asCallback(function (err, result) { | 112 | db.Author.findOrCreate(query).asCallback(function (err, result) { |
105 | // [ instance, wasCreated ] | 113 | // [ instance, wasCreated ] |
106 | return callback(err, result[0]) | 114 | return callback(err, t, result[0]) |
115 | }) | ||
116 | }, | ||
117 | |||
118 | function findOrCreateTags (t, author, callback) { | ||
119 | const tags = videoInfos.tags | ||
120 | const tagInstances = [] | ||
121 | |||
122 | each(tags, function (tag, callbackEach) { | ||
123 | const query = { | ||
124 | where: { | ||
125 | name: tag | ||
126 | }, | ||
127 | defaults: { | ||
128 | name: tag | ||
129 | }, | ||
130 | transaction: t | ||
131 | } | ||
132 | |||
133 | db.Tag.findOrCreate(query).asCallback(function (err, res) { | ||
134 | if (err) return callbackEach(err) | ||
135 | |||
136 | // res = [ tag, isCreated ] | ||
137 | const tag = res[0] | ||
138 | tagInstances.push(tag) | ||
139 | return callbackEach() | ||
140 | }) | ||
141 | }, function (err) { | ||
142 | return callback(err, t, author, tagInstances) | ||
107 | }) | 143 | }) |
108 | }, | 144 | }, |
109 | 145 | ||
110 | function createVideoObject (author, callback) { | 146 | function createVideoObject (t, author, tagInstances, callback) { |
111 | const videoData = { | 147 | const videoData = { |
112 | name: videoInfos.name, | 148 | name: videoInfos.name, |
113 | remoteId: null, | 149 | remoteId: null, |
114 | extname: path.extname(videoFile.filename), | 150 | extname: path.extname(videoFile.filename), |
115 | description: videoInfos.description, | 151 | description: videoInfos.description, |
116 | duration: videoFile.duration, | 152 | duration: videoFile.duration, |
117 | tags: videoInfos.tags, | ||
118 | authorId: author.id | 153 | authorId: author.id |
119 | } | 154 | } |
120 | 155 | ||
121 | const video = db.Video.build(videoData) | 156 | const video = db.Video.build(videoData) |
122 | 157 | ||
123 | return callback(null, author, video) | 158 | return callback(null, t, author, tagInstances, video) |
124 | }, | 159 | }, |
125 | 160 | ||
126 | // Set the videoname the same as the id | 161 | // Set the videoname the same as the id |
127 | function renameVideoFile (author, video, callback) { | 162 | function renameVideoFile (t, author, tagInstances, video, callback) { |
128 | const videoDir = constants.CONFIG.STORAGE.VIDEOS_DIR | 163 | const videoDir = constants.CONFIG.STORAGE.VIDEOS_DIR |
129 | const source = path.join(videoDir, videoFile.filename) | 164 | const source = path.join(videoDir, videoFile.filename) |
130 | const destination = path.join(videoDir, video.getVideoFilename()) | 165 | const destination = path.join(videoDir, video.getVideoFilename()) |
131 | 166 | ||
132 | fs.rename(source, destination, function (err) { | 167 | fs.rename(source, destination, function (err) { |
133 | return callback(err, author, video) | 168 | return callback(err, t, author, tagInstances, video) |
134 | }) | 169 | }) |
135 | }, | 170 | }, |
136 | 171 | ||
137 | function insertIntoDB (author, video, callback) { | 172 | function insertVideoIntoDB (t, author, tagInstances, video, callback) { |
138 | video.save().asCallback(function (err, videoCreated) { | 173 | const options = { transaction: t } |
174 | |||
175 | // Add tags association | ||
176 | video.save(options).asCallback(function (err, videoCreated) { | ||
177 | if (err) return callback(err) | ||
178 | |||
139 | // Do not forget to add Author informations to the created video | 179 | // Do not forget to add Author informations to the created video |
140 | videoCreated.Author = author | 180 | videoCreated.Author = author |
141 | 181 | ||
142 | return callback(err, videoCreated) | 182 | return callback(err, t, tagInstances, videoCreated) |
143 | }) | 183 | }) |
144 | }, | 184 | }, |
145 | 185 | ||
146 | function sendToFriends (video, callback) { | 186 | function associateTagsToVideo (t, tagInstances, video, callback) { |
187 | const options = { transaction: t } | ||
188 | |||
189 | video.setTags(tagInstances, options).asCallback(function (err) { | ||
190 | video.Tags = tagInstances | ||
191 | |||
192 | return callback(err, t, video) | ||
193 | }) | ||
194 | }, | ||
195 | |||
196 | function sendToFriends (t, video, callback) { | ||
147 | video.toRemoteJSON(function (err, remoteVideo) { | 197 | video.toRemoteJSON(function (err, remoteVideo) { |
148 | if (err) return callback(err) | 198 | if (err) return callback(err) |
149 | 199 | ||
150 | // Now we'll add the video's meta data to our friends | 200 | // Now we'll add the video's meta data to our friends |
151 | friends.addVideoToFriends(remoteVideo) | 201 | friends.addVideoToFriends(remoteVideo) |
152 | 202 | ||
153 | return callback(null) | 203 | return callback(null, t) |
154 | }) | 204 | }) |
155 | } | 205 | } |
156 | 206 | ||
157 | ], function andFinally (err) { | 207 | ], function andFinally (err, t) { |
158 | if (err) { | 208 | if (err) { |
159 | logger.error('Cannot insert the video.') | 209 | logger.error('Cannot insert the video.') |
210 | |||
211 | // Abort transaction? | ||
212 | if (t) t.rollback() | ||
213 | |||
160 | return next(err) | 214 | return next(err) |
161 | } | 215 | } |
162 | 216 | ||
217 | // Commit transaction | ||
218 | t.commit() | ||
219 | |||
163 | // TODO : include Location of the new video -> 201 | 220 | // TODO : include Location of the new video -> 201 |
164 | return res.type('json').status(204).end() | 221 | return res.type('json').status(204).end() |
165 | }) | 222 | }) |
166 | } | 223 | } |
167 | 224 | ||
168 | function getVideo (req, res, next) { | 225 | function getVideo (req, res, next) { |
169 | db.Video.loadAndPopulateAuthorAndPod(req.params.id, function (err, video) { | 226 | db.Video.loadAndPopulateAuthorAndPodAndTags(req.params.id, function (err, video) { |
170 | if (err) return next(err) | 227 | if (err) return next(err) |
171 | 228 | ||
172 | if (!video) { | 229 | if (!video) { |
@@ -222,12 +279,14 @@ function removeVideo (req, res, next) { | |||
222 | } | 279 | } |
223 | 280 | ||
224 | function searchVideos (req, res, next) { | 281 | function searchVideos (req, res, next) { |
225 | db.Video.searchAndPopulateAuthorAndPod(req.params.value, req.query.field, req.query.start, req.query.count, req.query.sort, | 282 | db.Video.searchAndPopulateAuthorAndPodAndTags( |
226 | function (err, videosList, videosTotal) { | 283 | req.params.value, req.query.field, req.query.start, req.query.count, req.query.sort, |
227 | if (err) return next(err) | 284 | function (err, videosList, videosTotal) { |
285 | if (err) return next(err) | ||
228 | 286 | ||
229 | res.json(getFormatedVideos(videosList, videosTotal)) | 287 | res.json(getFormatedVideos(videosList, videosTotal)) |
230 | }) | 288 | } |
289 | ) | ||
231 | } | 290 | } |
232 | 291 | ||
233 | // --------------------------------------------------------------------------- | 292 | // --------------------------------------------------------------------------- |
diff --git a/server/controllers/client.js b/server/controllers/client.js index a5fac5626..8c242af07 100644 --- a/server/controllers/client.js +++ b/server/controllers/client.js | |||
@@ -93,7 +93,7 @@ function generateWatchHtmlPage (req, res, next) { | |||
93 | }, | 93 | }, |
94 | 94 | ||
95 | video: function (callback) { | 95 | video: function (callback) { |
96 | db.Video.loadAndPopulateAuthorAndPod(videoId, callback) | 96 | db.Video.loadAndPopulateAuthorAndPodAndTags(videoId, callback) |
97 | } | 97 | } |
98 | }, function (err, results) { | 98 | }, function (err, results) { |
99 | if (err) return next(err) | 99 | if (err) return next(err) |
diff --git a/server/initializers/database.js b/server/initializers/database.js index cc6f59b63..9642231b9 100644 --- a/server/initializers/database.js +++ b/server/initializers/database.js | |||
@@ -6,13 +6,24 @@ const Sequelize = require('sequelize') | |||
6 | 6 | ||
7 | const constants = require('../initializers/constants') | 7 | const constants = require('../initializers/constants') |
8 | const logger = require('../helpers/logger') | 8 | const logger = require('../helpers/logger') |
9 | const utils = require('../helpers/utils') | ||
9 | 10 | ||
10 | const database = {} | 11 | const database = {} |
11 | 12 | ||
12 | const sequelize = new Sequelize(constants.CONFIG.DATABASE.DBNAME, 'peertube', 'peertube', { | 13 | const sequelize = new Sequelize(constants.CONFIG.DATABASE.DBNAME, 'peertube', 'peertube', { |
13 | dialect: 'postgres', | 14 | dialect: 'postgres', |
14 | host: constants.CONFIG.DATABASE.HOSTNAME, | 15 | host: constants.CONFIG.DATABASE.HOSTNAME, |
15 | port: constants.CONFIG.DATABASE.PORT | 16 | port: constants.CONFIG.DATABASE.PORT, |
17 | benchmark: utils.isTestInstance(), | ||
18 | |||
19 | logging: function (message, benchmark) { | ||
20 | let newMessage = message | ||
21 | if (benchmark !== undefined) { | ||
22 | newMessage += ' | ' + benchmark + 'ms' | ||
23 | } | ||
24 | |||
25 | logger.debug(newMessage) | ||
26 | } | ||
16 | }) | 27 | }) |
17 | 28 | ||
18 | const modelDirectory = path.join(__dirname, '..', 'models') | 29 | const modelDirectory = path.join(__dirname, '..', 'models') |
diff --git a/server/lib/friends.js b/server/lib/friends.js index 3ed29f651..ad9e4fdae 100644 --- a/server/lib/friends.js +++ b/server/lib/friends.js | |||
@@ -66,10 +66,12 @@ function makeFriends (hosts, callback) { | |||
66 | function quitFriends (callback) { | 66 | function quitFriends (callback) { |
67 | // Stop pool requests | 67 | // Stop pool requests |
68 | db.Request.deactivate() | 68 | db.Request.deactivate() |
69 | // Flush pool requests | ||
70 | db.Request.flush() | ||
71 | 69 | ||
72 | waterfall([ | 70 | waterfall([ |
71 | function flushRequests (callbackAsync) { | ||
72 | db.Request.flush(callbackAsync) | ||
73 | }, | ||
74 | |||
73 | function getPodsList (callbackAsync) { | 75 | function getPodsList (callbackAsync) { |
74 | return db.Pod.list(callbackAsync) | 76 | return db.Pod.list(callbackAsync) |
75 | }, | 77 | }, |
@@ -118,7 +120,7 @@ function removeVideoToFriends (videoParams) { | |||
118 | } | 120 | } |
119 | 121 | ||
120 | function sendOwnedVideosToPod (podId) { | 122 | function sendOwnedVideosToPod (podId) { |
121 | db.Video.listOwnedAndPopulateAuthor(function (err, videosList) { | 123 | db.Video.listOwnedAndPopulateAuthorAndTags(function (err, videosList) { |
122 | if (err) { | 124 | if (err) { |
123 | logger.error('Cannot get the list of videos we own.') | 125 | logger.error('Cannot get the list of videos we own.') |
124 | return | 126 | return |
@@ -226,7 +228,7 @@ function makeRequestsToWinningPods (cert, podsList, callback) { | |||
226 | } | 228 | } |
227 | 229 | ||
228 | // Add our videos to the request scheduler | 230 | // Add our videos to the request scheduler |
229 | sendOwnedVideosToPod(podCreated._id) | 231 | sendOwnedVideosToPod(podCreated.id) |
230 | 232 | ||
231 | return callbackEach() | 233 | return callbackEach() |
232 | }) | 234 | }) |
diff --git a/server/models/pod.js b/server/models/pod.js index 2c1f56203..fff6970a7 100644 --- a/server/models/pod.js +++ b/server/models/pod.js | |||
@@ -19,7 +19,6 @@ module.exports = function (sequelize, DataTypes) { | |||
19 | type: DataTypes.INTEGER, | 19 | type: DataTypes.INTEGER, |
20 | defaultValue: constants.FRIEND_SCORE.BASE | 20 | defaultValue: constants.FRIEND_SCORE.BASE |
21 | } | 21 | } |
22 | // Check createdAt | ||
23 | }, | 22 | }, |
24 | { | 23 | { |
25 | classMethods: { | 24 | classMethods: { |
@@ -68,7 +67,7 @@ function associate (models) { | |||
68 | this.belongsToMany(models.Request, { | 67 | this.belongsToMany(models.Request, { |
69 | foreignKey: 'podId', | 68 | foreignKey: 'podId', |
70 | through: models.RequestToPod, | 69 | through: models.RequestToPod, |
71 | onDelete: 'CASCADE' | 70 | onDelete: 'cascade' |
72 | }) | 71 | }) |
73 | } | 72 | } |
74 | 73 | ||
diff --git a/server/models/request.js b/server/models/request.js index 882f747b7..70aa32610 100644 --- a/server/models/request.js +++ b/server/models/request.js | |||
@@ -79,9 +79,11 @@ function deactivate () { | |||
79 | timer = null | 79 | timer = null |
80 | } | 80 | } |
81 | 81 | ||
82 | function flush () { | 82 | function flush (callback) { |
83 | removeAll.call(this, function (err) { | 83 | removeAll.call(this, function (err) { |
84 | if (err) logger.error('Cannot flush the requests.', { error: err }) | 84 | if (err) logger.error('Cannot flush the requests.', { error: err }) |
85 | |||
86 | return callback(err) | ||
85 | }) | 87 | }) |
86 | } | 88 | } |
87 | 89 | ||
@@ -298,7 +300,7 @@ function listWithLimitAndRandom (limit, callback) { | |||
298 | 300 | ||
299 | function removeAll (callback) { | 301 | function removeAll (callback) { |
300 | // Delete all requests | 302 | // Delete all requests |
301 | this.destroy({ truncate: true }).asCallback(callback) | 303 | this.truncate({ cascade: true }).asCallback(callback) |
302 | } | 304 | } |
303 | 305 | ||
304 | function removeWithEmptyTo (callback) { | 306 | function removeWithEmptyTo (callback) { |
diff --git a/server/models/tag.js b/server/models/tag.js new file mode 100644 index 000000000..874e88842 --- /dev/null +++ b/server/models/tag.js | |||
@@ -0,0 +1,30 @@ | |||
1 | 'use strict' | ||
2 | |||
3 | // --------------------------------------------------------------------------- | ||
4 | |||
5 | module.exports = function (sequelize, DataTypes) { | ||
6 | const Tag = sequelize.define('Tag', | ||
7 | { | ||
8 | name: { | ||
9 | type: DataTypes.STRING | ||
10 | } | ||
11 | }, | ||
12 | { | ||
13 | classMethods: { | ||
14 | associate | ||
15 | } | ||
16 | } | ||
17 | ) | ||
18 | |||
19 | return Tag | ||
20 | } | ||
21 | |||
22 | // --------------------------------------------------------------------------- | ||
23 | |||
24 | function associate (models) { | ||
25 | this.belongsToMany(models.Video, { | ||
26 | foreignKey: 'tagId', | ||
27 | through: models.VideoTag, | ||
28 | onDelete: 'cascade' | ||
29 | }) | ||
30 | } | ||
diff --git a/server/models/video.js b/server/models/video.js index 8ef07c9e6..0023a24e1 100644 --- a/server/models/video.js +++ b/server/models/video.js | |||
@@ -4,6 +4,7 @@ const createTorrent = require('create-torrent') | |||
4 | const ffmpeg = require('fluent-ffmpeg') | 4 | const ffmpeg = require('fluent-ffmpeg') |
5 | const fs = require('fs') | 5 | const fs = require('fs') |
6 | const magnetUtil = require('magnet-uri') | 6 | const magnetUtil = require('magnet-uri') |
7 | const map = require('lodash/map') | ||
7 | const parallel = require('async/parallel') | 8 | const parallel = require('async/parallel') |
8 | const parseTorrent = require('parse-torrent') | 9 | const parseTorrent = require('parse-torrent') |
9 | const pathUtils = require('path') | 10 | const pathUtils = require('path') |
@@ -41,9 +42,6 @@ module.exports = function (sequelize, DataTypes) { | |||
41 | }, | 42 | }, |
42 | duration: { | 43 | duration: { |
43 | type: DataTypes.INTEGER | 44 | type: DataTypes.INTEGER |
44 | }, | ||
45 | tags: { | ||
46 | type: DataTypes.ARRAY(DataTypes.STRING) | ||
47 | } | 45 | } |
48 | }, | 46 | }, |
49 | { | 47 | { |
@@ -54,12 +52,12 @@ module.exports = function (sequelize, DataTypes) { | |||
54 | getDurationFromFile, | 52 | getDurationFromFile, |
55 | listForApi, | 53 | listForApi, |
56 | listByHostAndRemoteId, | 54 | listByHostAndRemoteId, |
57 | listOwnedAndPopulateAuthor, | 55 | listOwnedAndPopulateAuthorAndTags, |
58 | listOwnedByAuthor, | 56 | listOwnedByAuthor, |
59 | load, | 57 | load, |
60 | loadAndPopulateAuthor, | 58 | loadAndPopulateAuthor, |
61 | loadAndPopulateAuthorAndPod, | 59 | loadAndPopulateAuthorAndPodAndTags, |
62 | searchAndPopulateAuthorAndPod | 60 | searchAndPopulateAuthorAndPodAndTags |
63 | }, | 61 | }, |
64 | instanceMethods: { | 62 | instanceMethods: { |
65 | generateMagnetUri, | 63 | generateMagnetUri, |
@@ -170,6 +168,12 @@ function associate (models) { | |||
170 | }, | 168 | }, |
171 | onDelete: 'cascade' | 169 | onDelete: 'cascade' |
172 | }) | 170 | }) |
171 | |||
172 | this.belongsToMany(models.Tag, { | ||
173 | foreignKey: 'videoId', | ||
174 | through: models.VideoTag, | ||
175 | onDelete: 'cascade' | ||
176 | }) | ||
173 | } | 177 | } |
174 | 178 | ||
175 | function generateMagnetUri () { | 179 | function generateMagnetUri () { |
@@ -248,7 +252,7 @@ function toFormatedJSON () { | |||
248 | magnetUri: this.generateMagnetUri(), | 252 | magnetUri: this.generateMagnetUri(), |
249 | author: this.Author.name, | 253 | author: this.Author.name, |
250 | duration: this.duration, | 254 | duration: this.duration, |
251 | tags: this.tags, | 255 | tags: map(this.Tags, 'name'), |
252 | thumbnailPath: constants.STATIC_PATHS.THUMBNAILS + '/' + this.getThumbnailName(), | 256 | thumbnailPath: constants.STATIC_PATHS.THUMBNAILS + '/' + this.getThumbnailName(), |
253 | createdAt: this.createdAt | 257 | createdAt: this.createdAt |
254 | } | 258 | } |
@@ -275,7 +279,7 @@ function toRemoteJSON (callback) { | |||
275 | author: self.Author.name, | 279 | author: self.Author.name, |
276 | duration: self.duration, | 280 | duration: self.duration, |
277 | thumbnailBase64: new Buffer(thumbnailData).toString('base64'), | 281 | thumbnailBase64: new Buffer(thumbnailData).toString('base64'), |
278 | tags: self.tags, | 282 | tags: map(self.Tags, 'name'), |
279 | createdAt: self.createdAt, | 283 | createdAt: self.createdAt, |
280 | extname: self.extname | 284 | extname: self.extname |
281 | } | 285 | } |
@@ -310,12 +314,15 @@ function listForApi (start, count, sort, callback) { | |||
310 | const query = { | 314 | const query = { |
311 | offset: start, | 315 | offset: start, |
312 | limit: count, | 316 | limit: count, |
317 | distinct: true, // For the count, a video can have many tags | ||
313 | order: [ modelUtils.getSort(sort) ], | 318 | order: [ modelUtils.getSort(sort) ], |
314 | include: [ | 319 | include: [ |
315 | { | 320 | { |
316 | model: this.sequelize.models.Author, | 321 | model: this.sequelize.models.Author, |
317 | include: [ this.sequelize.models.Pod ] | 322 | include: [ { model: this.sequelize.models.Pod, required: false } ] |
318 | } | 323 | }, |
324 | |||
325 | this.sequelize.models.Tag | ||
319 | ] | 326 | ] |
320 | } | 327 | } |
321 | 328 | ||
@@ -337,6 +344,7 @@ function listByHostAndRemoteId (fromHost, remoteId, callback) { | |||
337 | include: [ | 344 | include: [ |
338 | { | 345 | { |
339 | model: this.sequelize.models.Pod, | 346 | model: this.sequelize.models.Pod, |
347 | required: true, | ||
340 | where: { | 348 | where: { |
341 | host: fromHost | 349 | host: fromHost |
342 | } | 350 | } |
@@ -349,13 +357,13 @@ function listByHostAndRemoteId (fromHost, remoteId, callback) { | |||
349 | return this.findAll(query).asCallback(callback) | 357 | return this.findAll(query).asCallback(callback) |
350 | } | 358 | } |
351 | 359 | ||
352 | function listOwnedAndPopulateAuthor (callback) { | 360 | function listOwnedAndPopulateAuthorAndTags (callback) { |
353 | // If remoteId is null this is *our* video | 361 | // If remoteId is null this is *our* video |
354 | const query = { | 362 | const query = { |
355 | where: { | 363 | where: { |
356 | remoteId: null | 364 | remoteId: null |
357 | }, | 365 | }, |
358 | include: [ this.sequelize.models.Author ] | 366 | include: [ this.sequelize.models.Author, this.sequelize.models.Tag ] |
359 | } | 367 | } |
360 | 368 | ||
361 | return this.findAll(query).asCallback(callback) | 369 | return this.findAll(query).asCallback(callback) |
@@ -391,23 +399,26 @@ function loadAndPopulateAuthor (id, callback) { | |||
391 | return this.findById(id, options).asCallback(callback) | 399 | return this.findById(id, options).asCallback(callback) |
392 | } | 400 | } |
393 | 401 | ||
394 | function loadAndPopulateAuthorAndPod (id, callback) { | 402 | function loadAndPopulateAuthorAndPodAndTags (id, callback) { |
395 | const options = { | 403 | const options = { |
396 | include: [ | 404 | include: [ |
397 | { | 405 | { |
398 | model: this.sequelize.models.Author, | 406 | model: this.sequelize.models.Author, |
399 | include: [ this.sequelize.models.Pod ] | 407 | include: [ { model: this.sequelize.models.Pod, required: false } ] |
400 | } | 408 | }, |
409 | this.sequelize.models.Tag | ||
401 | ] | 410 | ] |
402 | } | 411 | } |
403 | 412 | ||
404 | return this.findById(id, options).asCallback(callback) | 413 | return this.findById(id, options).asCallback(callback) |
405 | } | 414 | } |
406 | 415 | ||
407 | function searchAndPopulateAuthorAndPod (value, field, start, count, sort, callback) { | 416 | function searchAndPopulateAuthorAndPodAndTags (value, field, start, count, sort, callback) { |
408 | const podInclude = { | 417 | const podInclude = { |
409 | model: this.sequelize.models.Pod | 418 | model: this.sequelize.models.Pod, |
419 | required: false | ||
410 | } | 420 | } |
421 | |||
411 | const authorInclude = { | 422 | const authorInclude = { |
412 | model: this.sequelize.models.Author, | 423 | model: this.sequelize.models.Author, |
413 | include: [ | 424 | include: [ |
@@ -415,55 +426,61 @@ function searchAndPopulateAuthorAndPod (value, field, start, count, sort, callba | |||
415 | ] | 426 | ] |
416 | } | 427 | } |
417 | 428 | ||
429 | const tagInclude = { | ||
430 | model: this.sequelize.models.Tag | ||
431 | } | ||
432 | |||
418 | const query = { | 433 | const query = { |
419 | where: {}, | 434 | where: {}, |
420 | include: [ | ||
421 | authorInclude | ||
422 | ], | ||
423 | offset: start, | 435 | offset: start, |
424 | limit: count, | 436 | limit: count, |
437 | distinct: true, // For the count, a video can have many tags | ||
425 | order: [ modelUtils.getSort(sort) ] | 438 | order: [ modelUtils.getSort(sort) ] |
426 | } | 439 | } |
427 | 440 | ||
428 | // TODO: include our pod for podHost searches (we are not stored in the database) | ||
429 | // Make an exact search with the magnet | 441 | // Make an exact search with the magnet |
430 | if (field === 'magnetUri') { | 442 | if (field === 'magnetUri') { |
431 | const infoHash = magnetUtil.decode(value).infoHash | 443 | const infoHash = magnetUtil.decode(value).infoHash |
432 | query.where.infoHash = infoHash | 444 | query.where.infoHash = infoHash |
433 | } else if (field === 'tags') { | 445 | } else if (field === 'tags') { |
434 | query.where[field] = value | 446 | const escapedValue = this.sequelize.escape('%' + value + '%') |
435 | } else if (field === 'host') { | 447 | query.where = { |
436 | const whereQuery = { | 448 | id: { |
437 | '$Author.Pod.host$': { | 449 | $in: this.sequelize.literal( |
438 | $like: '%' + value + '%' | 450 | '(SELECT "VideoTags"."videoId" FROM "Tags" INNER JOIN "VideoTags" ON "Tags"."id" = "VideoTags"."tagId" WHERE name LIKE ' + escapedValue + ')' |
451 | ) | ||
439 | } | 452 | } |
440 | } | 453 | } |
441 | 454 | } else if (field === 'host') { | |
442 | // Include our pod? (not stored in the database) | 455 | // FIXME: Include our pod? (not stored in the database) |
443 | if (constants.CONFIG.WEBSERVER.HOST.indexOf(value) !== -1) { | 456 | podInclude.where = { |
444 | query.where = { | 457 | host: { |
445 | $or: [ | 458 | $like: '%' + value + '%' |
446 | whereQuery, | ||
447 | { | ||
448 | remoteId: null | ||
449 | } | ||
450 | ] | ||
451 | } | 459 | } |
452 | } else { | ||
453 | query.where = whereQuery | ||
454 | } | 460 | } |
461 | podInclude.required = true | ||
455 | } else if (field === 'author') { | 462 | } else if (field === 'author') { |
456 | query.where = { | 463 | authorInclude.where = { |
457 | '$Author.name$': { | 464 | name: { |
458 | $like: '%' + value + '%' | 465 | $like: '%' + value + '%' |
459 | } | 466 | } |
460 | } | 467 | } |
468 | |||
469 | // authorInclude.or = true | ||
461 | } else { | 470 | } else { |
462 | query.where[field] = { | 471 | query.where[field] = { |
463 | $like: '%' + value + '%' | 472 | $like: '%' + value + '%' |
464 | } | 473 | } |
465 | } | 474 | } |
466 | 475 | ||
476 | query.include = [ | ||
477 | authorInclude, tagInclude | ||
478 | ] | ||
479 | |||
480 | if (tagInclude.where) { | ||
481 | // query.include.push([ this.sequelize.models.Tag ]) | ||
482 | } | ||
483 | |||
467 | return this.findAndCountAll(query).asCallback(function (err, result) { | 484 | return this.findAndCountAll(query).asCallback(function (err, result) { |
468 | if (err) return callback(err) | 485 | if (err) return callback(err) |
469 | 486 | ||
diff --git a/server/models/videoTag.js b/server/models/videoTag.js new file mode 100644 index 000000000..0f2b20838 --- /dev/null +++ b/server/models/videoTag.js | |||
@@ -0,0 +1,9 @@ | |||
1 | 'use strict' | ||
2 | |||
3 | // --------------------------------------------------------------------------- | ||
4 | |||
5 | module.exports = function (sequelize, DataTypes) { | ||
6 | const VideoTag = sequelize.define('VideoTag', {}, {}) | ||
7 | |||
8 | return VideoTag | ||
9 | } | ||
diff --git a/server/tests/api/check-params.js b/server/tests/api/check-params.js index d9e51770c..9aecc3720 100644 --- a/server/tests/api/check-params.js +++ b/server/tests/api/check-params.js | |||
@@ -456,7 +456,7 @@ describe('Test parameters validator', function () { | |||
456 | }) | 456 | }) |
457 | }) | 457 | }) |
458 | 458 | ||
459 | it('Should fail without a mongodb id', function (done) { | 459 | it('Should fail without a correct uuid', function (done) { |
460 | request(server.url) | 460 | request(server.url) |
461 | .get(path + 'coucou') | 461 | .get(path + 'coucou') |
462 | .set('Accept', 'application/json') | 462 | .set('Accept', 'application/json') |
@@ -481,7 +481,7 @@ describe('Test parameters validator', function () { | |||
481 | .expect(400, done) | 481 | .expect(400, done) |
482 | }) | 482 | }) |
483 | 483 | ||
484 | it('Should fail without a mongodb id', function (done) { | 484 | it('Should fail without a correct uuid', function (done) { |
485 | request(server.url) | 485 | request(server.url) |
486 | .delete(path + 'hello') | 486 | .delete(path + 'hello') |
487 | .set('Authorization', 'Bearer ' + server.accessToken) | 487 | .set('Authorization', 'Bearer ' + server.accessToken) |
diff --git a/server/tests/api/requests.js b/server/tests/api/requests.js index 7e790b54b..933ed29b4 100644 --- a/server/tests/api/requests.js +++ b/server/tests/api/requests.js | |||
@@ -79,15 +79,16 @@ describe('Test requests stats', function () { | |||
79 | uploadVideo(server, function (err) { | 79 | uploadVideo(server, function (err) { |
80 | if (err) throw err | 80 | if (err) throw err |
81 | 81 | ||
82 | getRequestsStats(server, function (err, res) { | 82 | setTimeout(function () { |
83 | if (err) throw err | 83 | getRequestsStats(server, function (err, res) { |
84 | if (err) throw err | ||
84 | 85 | ||
85 | const body = res.body | 86 | const body = res.body |
86 | expect(body.totalRequests).to.equal(1) | 87 | expect(body.totalRequests).to.equal(1) |
87 | 88 | ||
88 | // Wait one cycle | 89 | done() |
89 | setTimeout(done, 10000) | 90 | }) |
90 | }) | 91 | }, 1000) |
91 | }) | 92 | }) |
92 | }) | 93 | }) |
93 | 94 | ||
diff --git a/server/tests/api/single-pod.js b/server/tests/api/single-pod.js index aedecacf3..66b762f82 100644 --- a/server/tests/api/single-pod.js +++ b/server/tests/api/single-pod.js | |||
@@ -153,31 +153,32 @@ describe('Test a single pod', function () { | |||
153 | }) | 153 | }) |
154 | }) | 154 | }) |
155 | 155 | ||
156 | it('Should search the video by podHost', function (done) { | 156 | // Not implemented yet |
157 | videosUtils.searchVideo(server.url, '9001', 'host', function (err, res) { | 157 | // it('Should search the video by podHost', function (done) { |
158 | if (err) throw err | 158 | // videosUtils.searchVideo(server.url, '9001', 'host', function (err, res) { |
159 | 159 | // if (err) throw err | |
160 | expect(res.body.total).to.equal(1) | 160 | |
161 | expect(res.body.data).to.be.an('array') | 161 | // expect(res.body.total).to.equal(1) |
162 | expect(res.body.data.length).to.equal(1) | 162 | // expect(res.body.data).to.be.an('array') |
163 | 163 | // expect(res.body.data.length).to.equal(1) | |
164 | const video = res.body.data[0] | 164 | |
165 | expect(video.name).to.equal('my super name') | 165 | // const video = res.body.data[0] |
166 | expect(video.description).to.equal('my super description') | 166 | // expect(video.name).to.equal('my super name') |
167 | expect(video.podHost).to.equal('localhost:9001') | 167 | // expect(video.description).to.equal('my super description') |
168 | expect(video.author).to.equal('root') | 168 | // expect(video.podHost).to.equal('localhost:9001') |
169 | expect(video.isLocal).to.be.true | 169 | // expect(video.author).to.equal('root') |
170 | expect(video.tags).to.deep.equal([ 'tag1', 'tag2', 'tag3' ]) | 170 | // expect(video.isLocal).to.be.true |
171 | expect(miscsUtils.dateIsValid(video.createdAt)).to.be.true | 171 | // expect(video.tags).to.deep.equal([ 'tag1', 'tag2', 'tag3' ]) |
172 | 172 | // expect(miscsUtils.dateIsValid(video.createdAt)).to.be.true | |
173 | videosUtils.testVideoImage(server.url, 'video_short.webm', video.thumbnailPath, function (err, test) { | 173 | |
174 | if (err) throw err | 174 | // videosUtils.testVideoImage(server.url, 'video_short.webm', video.thumbnailPath, function (err, test) { |
175 | expect(test).to.equal(true) | 175 | // if (err) throw err |
176 | 176 | // expect(test).to.equal(true) | |
177 | done() | 177 | |
178 | }) | 178 | // done() |
179 | }) | 179 | // }) |
180 | }) | 180 | // }) |
181 | // }) | ||
181 | 182 | ||
182 | it('Should search the video by tag', function (done) { | 183 | it('Should search the video by tag', function (done) { |
183 | videosUtils.searchVideo(server.url, 'tag1', 'tags', function (err, res) { | 184 | videosUtils.searchVideo(server.url, 'tag1', 'tags', function (err, res) { |
@@ -230,7 +231,7 @@ describe('Test a single pod', function () { | |||
230 | }) | 231 | }) |
231 | 232 | ||
232 | it('Should not find a search by tag', function (done) { | 233 | it('Should not find a search by tag', function (done) { |
233 | videosUtils.searchVideo(server.url, 'tag', 'tags', function (err, res) { | 234 | videosUtils.searchVideo(server.url, 'hello', 'tags', function (err, res) { |
234 | if (err) throw err | 235 | if (err) throw err |
235 | 236 | ||
236 | expect(res.body.total).to.equal(0) | 237 | expect(res.body.total).to.equal(0) |
@@ -424,29 +425,30 @@ describe('Test a single pod', function () { | |||
424 | }) | 425 | }) |
425 | }) | 426 | }) |
426 | 427 | ||
427 | it('Should search all the 9001 port videos', function (done) { | 428 | // Not implemented yet |
428 | videosUtils.searchVideoWithPagination(server.url, '9001', 'host', 0, 15, function (err, res) { | 429 | // it('Should search all the 9001 port videos', function (done) { |
429 | if (err) throw err | 430 | // videosUtils.searchVideoWithPagination(server.url, '9001', 'host', 0, 15, function (err, res) { |
431 | // if (err) throw err | ||
430 | 432 | ||
431 | const videos = res.body.data | 433 | // const videos = res.body.data |
432 | expect(res.body.total).to.equal(6) | 434 | // expect(res.body.total).to.equal(6) |
433 | expect(videos.length).to.equal(6) | 435 | // expect(videos.length).to.equal(6) |
434 | 436 | ||
435 | done() | 437 | // done() |
436 | }) | 438 | // }) |
437 | }) | 439 | // }) |
438 | 440 | ||
439 | it('Should search all the localhost videos', function (done) { | 441 | // it('Should search all the localhost videos', function (done) { |
440 | videosUtils.searchVideoWithPagination(server.url, 'localhost', 'host', 0, 15, function (err, res) { | 442 | // videosUtils.searchVideoWithPagination(server.url, 'localhost', 'host', 0, 15, function (err, res) { |
441 | if (err) throw err | 443 | // if (err) throw err |
442 | 444 | ||
443 | const videos = res.body.data | 445 | // const videos = res.body.data |
444 | expect(res.body.total).to.equal(6) | 446 | // expect(res.body.total).to.equal(6) |
445 | expect(videos.length).to.equal(6) | 447 | // expect(videos.length).to.equal(6) |
446 | 448 | ||
447 | done() | 449 | // done() |
448 | }) | 450 | // }) |
449 | }) | 451 | // }) |
450 | 452 | ||
451 | it('Should search the good magnetUri video', function (done) { | 453 | it('Should search the good magnetUri video', function (done) { |
452 | const video = videosListBase[0] | 454 | const video = videosListBase[0] |