]>
Commit | Line | Data |
---|---|---|
06aad801 | 1 | import { ffprobe, FfprobeData } from 'fluent-ffmpeg' |
4638cd71 | 2 | import { forceNumber } from '@shared/core-utils' |
44e702de | 3 | import { VideoFileMetadata, VideoResolution } from '@shared/models/videos' |
06aad801 | 4 | |
5 | /** | |
6 | * | |
7 | * Helpers to run ffprobe and extract data from the JSON output | |
8 | * | |
9 | */ | |
10 | ||
11 | function ffprobePromise (path: string) { | |
12 | return new Promise<FfprobeData>((res, rej) => { | |
13 | ffprobe(path, (err, data) => { | |
14 | if (err) return rej(err) | |
15 | ||
16 | return res(data) | |
17 | }) | |
18 | }) | |
19 | } | |
20 | ||
c729caf6 C |
21 | // --------------------------------------------------------------------------- |
22 | // Audio | |
23 | // --------------------------------------------------------------------------- | |
24 | ||
f686f5ed C |
25 | const imageCodecs = new Set([ |
26 | 'ansi', 'apng', 'bintext', 'bmp', 'brender_pix', 'dpx', 'exr', 'fits', 'gem', 'gif', 'jpeg2000', 'jpgls', 'mjpeg', 'mjpegb', 'msp2', | |
27 | 'pam', 'pbm', 'pcx', 'pfm', 'pgm', 'pgmyuv', 'pgx', 'photocd', 'pictor', 'png', 'ppm', 'psd', 'sgi', 'sunrast', 'svg', 'targa', 'tiff', | |
28 | 'txd', 'webp', 'xbin', 'xbm', 'xface', 'xpm', 'xwd' | |
29 | ]) | |
30 | ||
482b2623 | 31 | async function isAudioFile (path: string, existingProbe?: FfprobeData) { |
c729caf6 | 32 | const videoStream = await getVideoStream(path, existingProbe) |
f686f5ed C |
33 | if (!videoStream) return true |
34 | ||
35 | if (imageCodecs.has(videoStream.codec_name)) return true | |
482b2623 | 36 | |
f686f5ed | 37 | return false |
482b2623 C |
38 | } |
39 | ||
c729caf6 C |
40 | async function hasAudioStream (path: string, existingProbe?: FfprobeData) { |
41 | const { audioStream } = await getAudioStream(path, existingProbe) | |
42 | ||
43 | return !!audioStream | |
44 | } | |
45 | ||
06aad801 | 46 | async function getAudioStream (videoPath: string, existingProbe?: FfprobeData) { |
47 | // without position, ffprobe considers the last input only | |
48 | // we make it consider the first input only | |
49 | // if you pass a file path to pos, then ffprobe acts on that file directly | |
50 | const data = existingProbe || await ffprobePromise(videoPath) | |
51 | ||
52 | if (Array.isArray(data.streams)) { | |
53 | const audioStream = data.streams.find(stream => stream['codec_type'] === 'audio') | |
54 | ||
55 | if (audioStream) { | |
56 | return { | |
57 | absolutePath: data.format.filename, | |
58 | audioStream, | |
4638cd71 | 59 | bitrate: forceNumber(audioStream['bit_rate']) |
06aad801 | 60 | } |
61 | } | |
62 | } | |
63 | ||
64 | return { absolutePath: data.format.filename } | |
65 | } | |
66 | ||
67 | function getMaxAudioBitrate (type: 'aac' | 'mp3' | string, bitrate: number) { | |
68 | const maxKBitrate = 384 | |
69 | const kToBits = (kbits: number) => kbits * 1000 | |
70 | ||
71 | // If we did not manage to get the bitrate, use an average value | |
72 | if (!bitrate) return 256 | |
73 | ||
74 | if (type === 'aac') { | |
75 | switch (true) { | |
76 | case bitrate > kToBits(maxKBitrate): | |
77 | return maxKBitrate | |
78 | ||
79 | default: | |
80 | return -1 // we interpret it as a signal to copy the audio stream as is | |
81 | } | |
82 | } | |
83 | ||
84 | /* | |
85 | a 192kbit/sec mp3 doesn't hold as much information as a 192kbit/sec aac. | |
86 | That's why, when using aac, we can go to lower kbit/sec. The equivalences | |
87 | made here are not made to be accurate, especially with good mp3 encoders. | |
88 | */ | |
89 | switch (true) { | |
90 | case bitrate <= kToBits(192): | |
91 | return 128 | |
92 | ||
93 | case bitrate <= kToBits(384): | |
94 | return 256 | |
95 | ||
96 | default: | |
97 | return maxKBitrate | |
98 | } | |
99 | } | |
100 | ||
c729caf6 C |
101 | // --------------------------------------------------------------------------- |
102 | // Video | |
103 | // --------------------------------------------------------------------------- | |
06aad801 | 104 | |
c729caf6 C |
105 | async function getVideoStreamDimensionsInfo (path: string, existingProbe?: FfprobeData) { |
106 | const videoStream = await getVideoStream(path, existingProbe) | |
44e702de C |
107 | if (!videoStream) { |
108 | return { | |
109 | width: 0, | |
110 | height: 0, | |
111 | ratio: 0, | |
112 | resolution: VideoResolution.H_NOVIDEO, | |
113 | isPortraitMode: false | |
114 | } | |
115 | } | |
06aad801 | 116 | |
117 | return { | |
c729caf6 C |
118 | width: videoStream.width, |
119 | height: videoStream.height, | |
120 | ratio: Math.max(videoStream.height, videoStream.width) / Math.min(videoStream.height, videoStream.width), | |
121 | resolution: Math.min(videoStream.height, videoStream.width), | |
122 | isPortraitMode: videoStream.height > videoStream.width | |
06aad801 | 123 | } |
124 | } | |
125 | ||
c729caf6 C |
126 | async function getVideoStreamFPS (path: string, existingProbe?: FfprobeData) { |
127 | const videoStream = await getVideoStream(path, existingProbe) | |
128 | if (!videoStream) return 0 | |
06aad801 | 129 | |
130 | for (const key of [ 'avg_frame_rate', 'r_frame_rate' ]) { | |
131 | const valuesText: string = videoStream[key] | |
132 | if (!valuesText) continue | |
133 | ||
134 | const [ frames, seconds ] = valuesText.split('/') | |
135 | if (!frames || !seconds) continue | |
136 | ||
137 | const result = parseInt(frames, 10) / parseInt(seconds, 10) | |
138 | if (result > 0) return Math.round(result) | |
139 | } | |
140 | ||
141 | return 0 | |
142 | } | |
143 | ||
c729caf6 | 144 | async function buildFileMetadata (path: string, existingProbe?: FfprobeData) { |
06aad801 | 145 | const metadata = existingProbe || await ffprobePromise(path) |
146 | ||
147 | return new VideoFileMetadata(metadata) | |
148 | } | |
149 | ||
c729caf6 C |
150 | async function getVideoStreamBitrate (path: string, existingProbe?: FfprobeData): Promise<number> { |
151 | const metadata = await buildFileMetadata(path, existingProbe) | |
06aad801 | 152 | |
153 | let bitrate = metadata.format.bit_rate as number | |
154 | if (bitrate && !isNaN(bitrate)) return bitrate | |
155 | ||
c729caf6 | 156 | const videoStream = await getVideoStream(path, existingProbe) |
06aad801 | 157 | if (!videoStream) return undefined |
158 | ||
159 | bitrate = videoStream?.bit_rate | |
160 | if (bitrate && !isNaN(bitrate)) return bitrate | |
161 | ||
162 | return undefined | |
163 | } | |
164 | ||
c729caf6 C |
165 | async function getVideoStreamDuration (path: string, existingProbe?: FfprobeData) { |
166 | const metadata = await buildFileMetadata(path, existingProbe) | |
06aad801 | 167 | |
168 | return Math.round(metadata.format.duration) | |
169 | } | |
170 | ||
c729caf6 C |
171 | async function getVideoStream (path: string, existingProbe?: FfprobeData) { |
172 | const metadata = await buildFileMetadata(path, existingProbe) | |
06aad801 | 173 | |
c729caf6 | 174 | return metadata.streams.find(s => s.codec_type === 'video') |
06aad801 | 175 | } |
176 | ||
177 | // --------------------------------------------------------------------------- | |
178 | ||
179 | export { | |
c729caf6 C |
180 | getVideoStreamDimensionsInfo, |
181 | buildFileMetadata, | |
06aad801 | 182 | getMaxAudioBitrate, |
c729caf6 C |
183 | getVideoStream, |
184 | getVideoStreamDuration, | |
06aad801 | 185 | getAudioStream, |
c729caf6 | 186 | getVideoStreamFPS, |
482b2623 | 187 | isAudioFile, |
06aad801 | 188 | ffprobePromise, |
c729caf6 C |
189 | getVideoStreamBitrate, |
190 | hasAudioStream | |
06aad801 | 191 | } |