Integrate transcription in PeerTube

packages/server-commands/src/runners/runner-jobs-command.ts

@@ -5,9 +5,6 @@ import {
   AcceptRunnerJobResult,
   ErrorRunnerJobBody,
   HttpStatusCode,
-  isHLSTranscodingPayloadSuccess,
-  isLiveRTMPHLSTranscodingUpdatePayload,
-  isWebVideoOrAudioMergeTranscodingPayloadSuccess,
   ListRunnerJobsQuery,
   RequestRunnerJobBody,
   RequestRunnerJobResult,
@@ -22,8 +19,13 @@ import {
   RunnerJobType,
   RunnerJobUpdateBody,
   RunnerJobVODPayload,
+  TranscriptionSuccess,
   VODHLSTranscodingSuccess,
-  VODWebVideoTranscodingSuccess
+  VODWebVideoTranscodingSuccess,
+  isHLSTranscodingPayloadSuccess,
+  isLiveRTMPHLSTranscodingUpdatePayload,
+  isTranscriptionPayloadSuccess,
+  isWebVideoOrAudioMergeTranscodingPayloadSuccess
 } from '@peertube/peertube-models'
 import { unwrapBody } from '../requests/index.js'
 import { waitJobs } from '../server/jobs.js'
@@ -196,6 +198,12 @@ export class RunnerJobsCommand extends AbstractCommand {
       payloadWithoutFiles = omit(payloadWithoutFiles as VODHLSTranscodingSuccess, [ 'resolutionPlaylistFile' ])
     }
 
+    if (isTranscriptionPayloadSuccess(payload) && payload.vttFile) {
+      attaches[`payload[vttFile]`] = payload.vttFile
+
+      payloadWithoutFiles = omit(payloadWithoutFiles as TranscriptionSuccess, [ 'vttFile' ])
+    }
+
     return this.postUploadRequest({
       ...options,
 

packages/server-commands/src/server/config-command.ts

@@ -355,6 +355,29 @@ export class ConfigCommand extends AbstractCommand {
 
   // ---------------------------------------------------------------------------
 
+  enableTranscription ({ remote = false }: { remote?: boolean } = {}) {
+    return this.setTranscriptionEnabled(true, remote)
+  }
+
+  disableTranscription () {
+    return this.setTranscriptionEnabled(false, false)
+  }
+
+  private setTranscriptionEnabled (enabled: boolean, remoteEnabled: boolean) {
+    return this.updateExistingConfig({
+      newConfig: {
+        videoTranscription: {
+          enabled,
+          remoteRunners: {
+            enabled: remoteEnabled
+          }
+        }
+      }
+    })
+  }
+
+  // ---------------------------------------------------------------------------
+
   getConfig (options: OverrideCommandOptions = {}) {
     const path = '/api/v1/config'
 

packages/server-commands/src/shared/abstract-command.ts

@@ -1,7 +1,10 @@
 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/no-floating-promises */
 
-import { isAbsolute } from 'path'
+import { HttpStatusCode, HttpStatusCodeType } from '@peertube/peertube-models'
 import { buildAbsoluteFixturePath, getFileSize } from '@peertube/peertube-node-utils'
+import { expect } from 'chai'
+import got, { Response as GotResponse } from 'got'
+import { isAbsolute } from 'path'
 import {
   makeDeleteRequest,
   makeGetRequest,
@@ -11,12 +14,9 @@ import {
   unwrapBody,
   unwrapText
 } from '../requests/requests.js'
-import { expect } from 'chai'
-import got, { Response as GotResponse } from 'got'
-import { HttpStatusCode, HttpStatusCodeType } from '@peertube/peertube-models'
 
-import type { PeerTubeServer } from '../server/server.js'
 import { createReadStream } from 'fs'
+import type { PeerTubeServer } from '../server/server.js'
 
 export interface OverrideCommandOptions {
   token?: string

packages/server-commands/src/videos/captions-command.ts

@@ -1,3 +1,4 @@
+import { pick } from '@peertube/peertube-core-utils'
 import { HttpStatusCode, ResultList, VideoCaption } from '@peertube/peertube-models'
 import { buildAbsoluteFixturePath } from '@peertube/peertube-node-utils'
 import { AbstractCommand, OverrideCommandOptions } from '../shared/index.js'
@@ -32,6 +33,23 @@ export class CaptionsCommand extends AbstractCommand {
     })
   }
 
+  runGenerate (options: OverrideCommandOptions & {
+    videoId: string | number
+    forceTranscription?: boolean
+  }) {
+    const { videoId } = options
+    const path = '/api/v1/videos/' + videoId + '/captions/generate'
+
+    return this.postBodyRequest({
+      ...options,
+
+      path,
+      fields: pick(options, [ 'forceTranscription' ]),
+      implicitToken: true,
+      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
+    })
+  }
+
   list (options: OverrideCommandOptions & {
     videoId: string | number
     videoPassword?: string

packages/server-commands/src/videos/video-imports-command.ts

@@ -4,12 +4,19 @@ import { AbstractCommand, OverrideCommandOptions } from '../shared/index.js'
 
 export class VideoImportsCommand extends AbstractCommand {
 
-  importVideo (options: OverrideCommandOptions & {
-    attributes: (VideoImportCreate | { torrentfile?: string, previewfile?: string, thumbnailfile?: string })
+  async importVideo (options: OverrideCommandOptions & {
+    attributes: (Partial<VideoImportCreate> | { torrentfile?: string, previewfile?: string, thumbnailfile?: string })
   }) {
     const { attributes } = options
     const path = '/api/v1/videos/imports'
 
+    let defaultChannelId = 1
+
+    try {
+      const { videoChannels } = await this.server.users.getMyInfo({ token: options.token })
+      defaultChannelId = videoChannels[0].id
+    } catch (e) { /* empty */ }
+
     let attaches: any = {}
     if (attributes.torrentfile) attaches = { torrentfile: attributes.torrentfile }
     if (attributes.thumbnailfile) attaches = { thumbnailfile: attributes.thumbnailfile }
@@ -20,7 +27,11 @@ export class VideoImportsCommand extends AbstractCommand {
 
       path,
       attaches,
-      fields: options.attributes,
+      fields: {
+        channelId: defaultChannelId,
+
+        ...options.attributes
+      },
       implicitToken: true,
       defaultExpectedStatus: HttpStatusCode.OK_200
     }))