]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/commitdiff
Fix import captions test
authorChocobozzz <me@florianbigard.com>
Wed, 15 Apr 2020 12:15:44 +0000 (14:15 +0200)
committerChocobozzz <me@florianbigard.com>
Wed, 15 Apr 2020 12:15:44 +0000 (14:15 +0200)
server/controllers/api/videos/import.ts
server/helpers/youtube-dl.ts
server/lib/activitypub/actor.ts
server/models/video/video-import.ts
server/tests/api/videos/video-imports.ts

index e9b9d68d73fd07e5b48ed13c0b213205e0aa2728..fb9d73140601dfeceaacfa96c1833e4a40f06def 100644 (file)
@@ -172,11 +172,12 @@ async function addYoutubeDLImport (req: express.Request, res: express.Response)
     user
   })
 
-
   // Get video subtitles
   try {
     const subtitles = await getYoutubeDLSubs(targetUrl)
 
+    logger.info('Will create %s subtitles from youtube import %s.', subtitles.length, targetUrl)
+
     for (const subtitle of subtitles) {
       const videoCaption = new VideoCaptionModel({
         videoId: video.id,
index 277422645cdaa923f88202f1ce952ac2d33f0127..6d2e6f6d188ea686e64aaf813b7b8f0f9d187a48 100644 (file)
@@ -21,8 +21,8 @@ export type YoutubeDLInfo = {
 }
 
 export type YoutubeDLSubs = {
-  language: string,
-  filename: string,
+  language: string
+  filename: string
   path: string
 }[]
 
@@ -61,15 +61,20 @@ function getYoutubeDLSubs (url: string, opts?: object): Promise<YoutubeDLSubs> {
         youtubeDL.getSubs(url, options, (err, files) => {
           if (err) return rej(err)
 
+          logger.debug('Get subtitles from youtube dl.', { url, files })
+
           const subtitles = files.reduce((acc, filename) => {
             const matched = filename.match(/\.([a-z]{2})\.(vtt|ttml)/i)
 
             if (matched[1]) {
-              return [...acc, {
-                language: matched[1],
-                path: join(cwd, filename),
-                filename
-              }]
+              return [
+                ...acc,
+                {
+                  language: matched[1],
+                  path: join(cwd, filename),
+                  filename
+                }
+              ]
             }
           }, [])
 
index c3598b75b4fa3b2cd256bd613714f59ff8f2dde6..8132ac135ca8e374506c3086b590bbe18e4bb0d2 100644 (file)
@@ -117,7 +117,7 @@ async function getOrCreateActorAndServerAndModel (
   if (actor.VideoChannel) (actor as MActorAccountChannelIdActor).VideoChannel.Actor = actor
 
   const { actor: actorRefreshed, refreshed } = await retryTransactionWrapper(refreshActorIfNeeded, actor, fetchType)
-  if (!actorRefreshed) throw new Error('Actor ' + actorRefreshed.url + ' does not exist anymore.')
+  if (!actorRefreshed) throw new Error('Actor ' + actor.url + ' does not exist anymore.')
 
   if ((created === true || refreshed === true) && updateCollections === true) {
     const payload = { uri: actor.outboxUrl, type: 'activity' as 'activity' }
index af5314ce9be806fb7be7bbfbbf50f2224189b898..fbe0ee0a740844ae5094b538951f4645ed32be69 100644 (file)
@@ -129,6 +129,7 @@ export class VideoImportModel extends Model<VideoImportModel> {
       distinct: true,
       include: [
         {
+          attributes: [ 'id' ],
           model: UserModel.unscoped(), // FIXME: Without this, sequelize try to COUNT(DISTINCT(*)) which is an invalid SQL query
           required: true
         }
index 1e97cc6ca818bff7832ebe75a5d03594375b6d6b..8e179b8254bc50129b9c280b4b28f58b6e46dfa9 100644 (file)
@@ -62,11 +62,14 @@ describe('Test video imports', function () {
 
     expect(videoTorrent.name).to.contain('你好 世界 720p.mp4')
     expect(videoMagnet.name).to.contain('super peertube2 video')
+
+    const resCaptions = await listVideoCaptions(url, idHttp)
+    expect(resCaptions.body.total).to.equal(2)
   }
 
   async function checkVideoServer2 (url: string, id: number | string) {
     const res = await getVideo(url, id)
-    const video = res.body
+    const video: VideoDetails = res.body
 
     expect(video.name).to.equal('my super name')
     expect(video.category.label).to.equal('Entertainment')
@@ -77,6 +80,9 @@ describe('Test video imports', function () {
     expect(video.tags).to.deep.equal([ 'supertag1', 'supertag2' ])
 
     expect(video.files).to.have.lengthOf(1)
+
+    const resCaptions = await listVideoCaptions(url, id)
+    expect(resCaptions.body.total).to.equal(2)
   }
 
   before(async function () {
@@ -114,44 +120,42 @@ describe('Test video imports', function () {
       expect(res.body.video.name).to.equal('small video - youtube')
 
       const resCaptions = await listVideoCaptions(servers[0].url, res.body.video.id)
-      const videoCaptions: VideoCaption[] = resCaptions.body
+      const videoCaptions: VideoCaption[] = resCaptions.body.data
       expect(videoCaptions).to.have.lengthOf(2)
 
-      const enCaption = videoCaptions.filter(caption => caption.language.label === 'en')[0]
-      expect(enCaption).to.not(undefined)
-      expect(enCaption.language.label).to.equal('en')
+      const enCaption = videoCaptions.find(caption => caption.language.id === 'en')
+      expect(enCaption).to.exist
+      expect(enCaption.language.label).to.equal('English')
       expect(enCaption.captionPath).to.equal(`/static/video-captions/${res.body.video.uuid}-en.vtt`)
       await testCaptionFile(servers[0].url, enCaption.captionPath, `WEBVTT
+Kind: captions
+Language: en
 
-      1
-      00:00:01.600 --> 00:00:04.200
-      English (US)
+00:00:01.600 --> 00:00:04.200
+English (US)
 
-      2
-      00:00:05.900 --> 00:00:07.999
-      This is a subtitle in American English
+00:00:05.900 --> 00:00:07.999
+This is a subtitle in American English
 
-      3
-      00:00:10.000 --> 00:00:14.000
-      Adding subtitles is very easy to do`)
+00:00:10.000 --> 00:00:14.000
+Adding subtitles is very easy to do`)
 
-      const frCaption = videoCaptions.filter(caption => caption.language.label === 'fr')[0]
-      expect(frCaption).to.not(undefined)
-      expect(frCaption.language.label).to.equal('fr')
-      expect(frCaption.captionPath).to.equal(`/static/video-captions/${res.body.video.uuid}-en.vtt`)
+      const frCaption = videoCaptions.find(caption => caption.language.id === 'fr')
+      expect(frCaption).to.exist
+      expect(frCaption.language.label).to.equal('French')
+      expect(frCaption.captionPath).to.equal(`/static/video-captions/${res.body.video.uuid}-fr.vtt`)
       await testCaptionFile(servers[0].url, frCaption.captionPath, `WEBVTT
+Kind: captions
+Language: fr
 
-      1
-      00:00:01,600 --> 00:00:04.200
-      Français (FR)
+00:00:01.600 --> 00:00:04.200
+Français (FR)
 
-      2
-      00:00:05,900 --> 00:00:07.999
-      C'est un sous-titre français
+00:00:05.900 --> 00:00:07.999
+C'est un sous-titre français
 
-      3
-      00:00:10,000 --> 00:00:14.000
-      Ajouter un sous-titre est vraiment facile`)
+00:00:10.000 --> 00:00:14.000
+Ajouter un sous-titre est vraiment facile`)
     }
 
     {