From 27081ae5990b0780a096618f7bdba33b34577e5d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=89=8B=E7=93=9C=E4=B8=80=E5=8D=81=E9=9B=AA?=
 <nanaeonn@outlook.com>
Date: Mon, 14 Oct 2024 22:04:02 +0800
Subject: [PATCH] fix #391

---
 src/onebot/action/file/GetRecord.ts | 41 +++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/src/onebot/action/file/GetRecord.ts b/src/onebot/action/file/GetRecord.ts
index 933ba784..4e9e69b0 100644
--- a/src/onebot/action/file/GetRecord.ts
+++ b/src/onebot/action/file/GetRecord.ts
@@ -1,5 +1,9 @@
 import { GetFileBase, GetFilePayload, GetFileResponse } from './GetFile';
 import { ActionName } from '../types';
+import { spawn } from 'node:child_process';
+import { promises as fs } from 'fs';
+
+const FFMPEG_PATH = process.env.FFMPEG_PATH || 'ffmpeg';
 
 interface Payload extends GetFilePayload {
     out_format: 'mp3' | 'amr' | 'wma' | 'm4a' | 'spx' | 'ogg' | 'wav' | 'flac';
@@ -9,7 +13,40 @@ export default class GetRecord extends GetFileBase {
     actionName = ActionName.GetRecord;
 
     async _handle(payload: Payload): Promise<GetFileResponse> {
-        const res = super._handle(payload);
+        const res = await super._handle(payload);
+        if (payload.out_format && typeof payload.out_format === 'string') {
+            const inputFile = res.file;
+            const outputFile = `${inputFile}.${payload.out_format}`;
+            if (!inputFile) throw new Error('file not found');
+            try {
+                await fs.access(inputFile);
+                await this.convertFile(inputFile, outputFile, payload.out_format);
+                const base64Data = await fs.readFile(outputFile, { encoding: 'base64' });
+                res.file = outputFile;
+                res.url = outputFile;
+                res.base64 = base64Data;
+            } catch (error) {
+                console.error('Error processing file:', error);
+            }
+        }
         return res;
     }
-}
+
+    private convertFile(inputFile: string, outputFile: string, format: string): Promise<void> {
+        return new Promise((resolve, reject) => {
+            const ffmpeg = spawn(FFMPEG_PATH, ['-i', inputFile, outputFile]);
+
+            ffmpeg.on('close', (code) => {
+                if (code === 0) {
+                    resolve();
+                } else {
+                    reject(new Error(`ffmpeg process exited with code ${code}`));
+                }
+            });
+
+            ffmpeg.on('error', (error) => {
+                reject(error);
+            });
+        });
+    }
+}
\ No newline at end of file