Allow plugins to load a video by id

3 files changed, 12 insertions, 2 deletions

diff --git a/server/lib/plugins/plugin-helpers-builder.ts b/server/lib/plugins/plugin-helpers-builder.ts
index 39773f693..dac6b3185 100644
--- a/server/lib/plugins/plugin-helpers-builder.ts
+++ b/server/lib/plugins/plugin-helpers-builder.ts
@@ -63,6 +63,10 @@ function buildVideosHelpers () {
       return VideoModel.loadByUrl(url)
     },
 
+    loadByIdOrUUID: (id: number | string) => {
+      return VideoModel.load(id)
+    },
+
     removeVideo: (id: number) => {
       return sequelizeTypescript.transaction(async t => {
         const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(id, t)
diff --git a/server/tests/fixtures/peertube-plugin-test-four/main.js b/server/tests/fixtures/peertube-plugin-test-four/main.js
index 067c3fe15..8df456c8a 100644
--- a/server/tests/fixtures/peertube-plugin-test-four/main.js
+++ b/server/tests/fixtures/peertube-plugin-test-four/main.js
@@ -24,8 +24,13 @@ async function register ({
     registerHook({
       target: 'action:api.video.viewed',
       handler: async ({ video }) => {
-        const videoFromDB = await peertubeHelpers.videos.loadByUrl(video.url)
-        logger.info('video from DB uuid is %s.', videoFromDB.uuid)
+        const videoFromDB1 = await peertubeHelpers.videos.loadByUrl(video.url)
+        const videoFromDB2 = await peertubeHelpers.videos.loadByIdOrUUID(video.id)
+        const videoFromDB3 = await peertubeHelpers.videos.loadByIdOrUUID(video.uuid)
+
+        if (videoFromDB1.uuid !== videoFromDB2.uuid || videoFromDB2.uuid !== videoFromDB3.uuid) return
+
+        logger.info('video from DB uuid is %s.', videoFromDB1.uuid)
 
         await peertubeHelpers.videos.removeVideo(video.id)
 
diff --git a/server/types/plugins/register-server-option.model.ts b/server/types/plugins/register-server-option.model.ts
index ccd5a060d..1ca17e4ab 100644
--- a/server/types/plugins/register-server-option.model.ts
+++ b/server/types/plugins/register-server-option.model.ts
@@ -30,6 +30,7 @@ export type PeerTubeHelpers = {
 
   videos: {
     loadByUrl: (url: string) => Promise<MVideoThumbnail>
+    loadByIdOrUUID: (id: number | string) => Promise<MVideoThumbnail>
 
     removeVideo: (videoId: number) => Promise<void>
   }