ffmpeg-progress 1.3.0 → 1.4.0

package/README.md

@@ -42,6 +42,15 @@ await convertFile({
 timer.end()
 ```
 
+**Get video resolution**:
+
+```typescript
+import { getVideoResolution } from 'ffmpeg-progress'
+
+console.log(await getVideoResolution('test/rotate.mp4'))
+// { width: 3024, height: 4032 }
+```
+
 ## Typescript Types
 
 ```typescript
@@ -88,6 +97,16 @@ export function attachChildProcess(
     childProcess: ChildProcessWithoutNullStreams
   } & ProgressArgs,
 ): Promise<void>
+
+/**
+ * take a image frame from the video,
+ * this is more accurate than parsing the resolution string in ffmpeg,
+ * because the video has rotation metadata.
+ */
+export function getVideoResolution(video_file: string): Promise<{
+  width: number
+  height: number
+}>
 ```
 
 ## License

package/core.d.ts

@@ -15,7 +15,11 @@ export type ScanVideoResult = {
     duration: string;
     /** @description e.g. 180.03 or 0 */
     seconds: number;
-    /** @description e.g. "4032x3024" */
+    /**
+     * @description e.g. "4032x3024"
+     * This is the stored resolution of the video, not the resolution of the video when it is played.
+     * The display image may be rotated if there are rotation metadata in the video.
+     */
     resolution: string | null;
     /** @description e.g. 44100 for 44.1kHz */
     audioSampleRate: number | null;
@@ -49,3 +53,12 @@ export declare function estimateOutSize(args: {
     currentSeconds: number;
     totalSeconds: number;
 }): number;
+/**
+ * take a image frame from the video,
+ * this is more accurate than parsing the resolution string in ffmpeg,
+ * because the video has rotation metadata.
+ */
+export declare function getVideoResolution(video_file: string): Promise<{
+    width: number;
+    height: number;
+}>;

package/core.js

@@ -7,7 +7,10 @@ exports.scanVideo = scanVideo;
 exports.convertFile = convertFile;
 exports.attachChildProcess = attachChildProcess;
 exports.estimateOutSize = estimateOutSize;
+exports.getVideoResolution = getVideoResolution;
 const child_process_1 = require("child_process");
+const promises_1 = require("fs/promises");
+const path_1 = require("path");
 /** @description
  * from "00:01:00.03" to 60.03;
  * from "N/A" to NaN;
@@ -194,3 +197,68 @@ function estimateOutSize(args) {
     let estimatedOutSize = args.currentOutSize + estimatedRate * remindSeconds;
     return estimatedOutSize;
 }
+/**
+ * take a image frame from the video,
+ * this is more accurate than parsing the resolution string in ffmpeg,
+ * because the video has rotation metadata.
+ */
+async function getVideoResolution(video_file) {
+    let image_file = (0, path_1.join)('/tmp/', (0, path_1.basename)(video_file) + '.jpg');
+    // ffmpeg -ss 0 -i video.mp4 -frames:v 1 -f image2 -update 1 -y image.jpg
+    let childProcess = (0, child_process_1.spawn)('ffmpeg', [
+        '-ss',
+        '0',
+        '-i',
+        video_file,
+        '-frames:v',
+        '1',
+        '-f',
+        'image2',
+        '-update',
+        '1',
+        '-y',
+        image_file,
+    ]);
+    var { stdout, stderr, code, signal } = await waitChildProcess(childProcess);
+    if (code != 0) {
+        throw new Error(`ffmpeg exit abnormally, exit code: ${code}, signal: ${signal}, stdout: ${stdout}, stderr: ${stderr}`);
+    }
+    childProcess = (0, child_process_1.spawn)('file', [image_file]);
+    var { stdout, stderr, code, signal } = await waitChildProcess(childProcess);
+    if (code != 0) {
+        throw new Error(`file exit abnormally, exit code: ${code}, signal: ${signal}, stdout: ${stdout}, stderr: ${stderr}`);
+    }
+    let resolution = parseImageResolution(stdout);
+    await (0, promises_1.unlink)(image_file);
+    return resolution;
+}
+function waitChildProcess(childProcess) {
+    return new Promise((resolve, reject) => {
+        let stdout = '';
+        let stderr = '';
+        childProcess.stdout.on('data', (data) => {
+            stdout += data.toString();
+        });
+        childProcess.stderr.on('data', (data) => {
+            stderr += data.toString();
+        });
+        childProcess.on('exit', (code, signal) => {
+            resolve({ stdout, stderr, code, signal });
+        });
+    });
+}
+// e.g. sample.jpg: JPEG image data, baseline, precision 8, 3024x4032, components 3
+function parseImageResolution(text) {
+    let parts = text.split(', ').reverse();
+    for (let part of parts) {
+        let parts = part.split('x');
+        if (parts.length == 2) {
+            let width = +parts[0];
+            let height = +parts[1];
+            if (width && height) {
+                return { width, height };
+            }
+        }
+    }
+    throw new Error('failed to parse image resolution: ' + JSON.stringify(text));
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ffmpeg-progress",
-  "version": "1.3.0",
+  "version": "1.4.0",
   "description": "Extract progress from ffmpeg child_process",
   "keywords": [
     "ffmpeg",