From 789951460bf724ac6a2f18097878fc735269ea71 Mon Sep 17 00:00:00 2001
From: Chocobozzz <me@florianbigard.com>
Date: Fri, 29 Jan 2021 13:57:17 +0100
Subject: [PATCH] Try to support other codecs

---
 server/helpers/ffprobe-utils.ts               | 34 +++++++++++++++----
 .../plugin-transcoding-manager.model.ts       |  2 +-
 support/doc/plugins/guide.md                  |  2 +-
 3 files changed, 29 insertions(+), 9 deletions(-)

diff --git a/server/helpers/ffprobe-utils.ts b/server/helpers/ffprobe-utils.ts
index 5b1ad9066..fefed25ae 100644
--- a/server/helpers/ffprobe-utils.ts
+++ b/server/helpers/ffprobe-utils.ts
@@ -91,18 +91,35 @@ async function getVideoStreamCodec (path: string) {
 
   const videoCodec = videoStream.codec_tag_string
 
+  if (videoCodec === 'vp09') return 'vp09.00.50.08'
+
   const baseProfileMatrix = {
-    High: '6400',
-    Main: '4D40',
-    Baseline: '42E0'
+    avc1: {
+      High: '6400',
+      Main: '4D40',
+      Baseline: '42E0'
+    },
+    av01: {
+      High: '1',
+      Main: '0',
+      Professional: '2'
+    }
   }
 
-  let baseProfile = baseProfileMatrix[videoStream.profile]
+  let baseProfile = baseProfileMatrix[videoCodec][videoStream.profile]
   if (!baseProfile) {
     logger.warn('Cannot get video profile codec of %s.', path, { videoStream })
-    baseProfile = baseProfileMatrix['High'] // Fallback
+    baseProfile = baseProfileMatrix[videoCodec]['High'] // Fallback
   }
 
+  if (videoCodec === 'av01') {
+    const level = videoStream.level
+
+    // Guess the tier indicator and bit depth
+    return `${videoCodec}.${baseProfile}.${level}M.08`
+  }
+
+  // Default, h264 codec
   let level = videoStream.level.toString(16)
   if (level.length === 1) level = `0${level}`
 
@@ -114,8 +131,11 @@ async function getAudioStreamCodec (path: string, existingProbe?: ffmpeg.Ffprobe
 
   if (!audioStream) return ''
 
-  const audioCodec = audioStream.codec_name
-  if (audioCodec === 'aac') return 'mp4a.40.2'
+  const audioCodecName = audioStream.codec_name
+
+  if (audioCodecName === 'opus') return 'opus'
+  if (audioCodecName === 'vorbis') return 'vorbis'
+  if (audioCodecName === 'aac') return 'mp4a.40.2'
 
   logger.warn('Cannot get audio codec of %s.', path, { audioStream })
 
diff --git a/shared/models/plugins/plugin-transcoding-manager.model.ts b/shared/models/plugins/plugin-transcoding-manager.model.ts
index 7dfb51ddf..8babccd4e 100644
--- a/shared/models/plugins/plugin-transcoding-manager.model.ts
+++ b/shared/models/plugins/plugin-transcoding-manager.model.ts
@@ -9,5 +9,5 @@ export interface PluginTranscodingManager {
 
   addVODEncoderPriority (streamType: 'audio' | 'video', encoder: string, priority: number): void
 
-  removeAllProfilesAndEncoderPriorities()
+  removeAllProfilesAndEncoderPriorities(): void
 }
diff --git a/support/doc/plugins/guide.md b/support/doc/plugins/guide.md
index 9739d117a..a4a1a136d 100644
--- a/support/doc/plugins/guide.md
+++ b/support/doc/plugins/guide.md
@@ -516,7 +516,7 @@ async function register ({
       }
     }
 
-    // Support libopus and libvpx-vp9 encoders, just for the example (PeerTube player is only compatible with h264/aac)
+    // Support libopus and libvpx-vp9 encoders (these codecs could be incompatible with the player)
     transcodingManager.addVODProfile('libopus', 'test-vod-profile', builder)
 
     // Default priorities are ~100