feat: introduce audio merging and trimming tools with support for multiple formats

This commit is contained in:
AshAnand34
2025-07-07 15:50:06 -07:00
parent a1b929e45c
commit 7962bba04f
13 changed files with 851 additions and 1 deletions

View File

@@ -0,0 +1,112 @@
import { Box, FormControlLabel, Radio, RadioGroup } from '@mui/material';
import React, { useState } from 'react';
import ToolContent from '@components/ToolContent';
import { ToolComponentProps } from '@tools/defineTool';
import { GetGroupsType } from '@components/options/ToolOptions';
import { InitialValuesType } from './types';
import ToolMultipleAudioInput, {
MultiAudioInput
} from '@components/input/ToolMultipleAudioInput';
import ToolFileResult from '@components/result/ToolFileResult';
import { mergeAudioFiles } from './service';
const initialValues: InitialValuesType = {
outputFormat: 'mp3'
};
const formatOptions = [
{ label: 'MP3', value: 'mp3' },
{ label: 'AAC', value: 'aac' },
{ label: 'WAV', value: 'wav' }
];
export default function MergeAudio({
title,
longDescription
}: ToolComponentProps) {
const [input, setInput] = useState<MultiAudioInput[]>([]);
const [result, setResult] = useState<File | null>(null);
const [loading, setLoading] = useState(false);
const compute = async (
optionsValues: InitialValuesType,
input: MultiAudioInput[]
) => {
if (input.length === 0) return;
setLoading(true);
try {
const files = input.map((item) => item.file);
const mergedFile = await mergeAudioFiles(files, optionsValues);
setResult(mergedFile);
} catch (err) {
console.error(`Failed to merge audio: ${err}`);
setResult(null);
} finally {
setLoading(false);
}
};
const getGroups: GetGroupsType<InitialValuesType> | null = ({
values,
updateField
}) => [
{
title: 'Output Format',
component: (
<Box mt={2}>
<RadioGroup
row
value={values.outputFormat}
onChange={(e) =>
updateField(
'outputFormat',
e.target.value as 'mp3' | 'aac' | 'wav'
)
}
>
{formatOptions.map((opt) => (
<FormControlLabel
key={opt.value}
value={opt.value}
control={<Radio />}
label={opt.label}
/>
))}
</RadioGroup>
</Box>
)
}
];
return (
<ToolContent
title={title}
input={input}
inputComponent={
<ToolMultipleAudioInput
value={input}
onChange={setInput}
accept={['audio/*', '.mp3', '.wav', '.aac']}
title={'Input Audio Files'}
type="audio"
/>
}
resultComponent={
loading ? (
<ToolFileResult title="Merging Audio" value={null} loading={true} />
) : (
<ToolFileResult
title="Merged Audio"
value={result}
extension={result ? result.name.split('.').pop() : undefined}
/>
)
}
initialValues={initialValues}
getGroups={getGroups}
setInput={setInput}
compute={compute}
toolInfo={{ title: `What is a ${title}?`, description: longDescription }}
/>
);
}

View File

@@ -0,0 +1,73 @@
import { expect, describe, it, vi } from 'vitest';
// Mock FFmpeg since it doesn't support Node.js
vi.mock('@ffmpeg/ffmpeg', () => ({
FFmpeg: vi.fn().mockImplementation(() => ({
loaded: false,
load: vi.fn().mockResolvedValue(undefined),
writeFile: vi.fn().mockResolvedValue(undefined),
exec: vi.fn().mockResolvedValue(undefined),
readFile: vi.fn().mockResolvedValue(new Uint8Array([1, 2, 3, 4, 5])),
deleteFile: vi.fn().mockResolvedValue(undefined)
}))
}));
vi.mock('@ffmpeg/util', () => ({
fetchFile: vi.fn().mockResolvedValue(new Uint8Array([1, 2, 3, 4, 5]))
}));
import { mergeAudioFiles } from './service';
describe('mergeAudioFiles', () => {
it('should merge multiple audio files', async () => {
// Create mock audio files
const mockAudioData1 = new Uint8Array([0, 1, 2, 3, 4, 5]);
const mockAudioData2 = new Uint8Array([6, 7, 8, 9, 10, 11]);
const mockFile1 = new File([mockAudioData1], 'test1.mp3', {
type: 'audio/mp3'
});
const mockFile2 = new File([mockAudioData2], 'test2.mp3', {
type: 'audio/mp3'
});
const options = {
outputFormat: 'mp3' as const
};
const result = await mergeAudioFiles([mockFile1, mockFile2], options);
expect(result).toBeInstanceOf(File);
expect(result.name).toBe('merged_audio.mp3');
expect(result.type).toBe('audio/mp3');
});
it('should handle different output formats', async () => {
const mockAudioData = new Uint8Array([0, 1, 2, 3, 4, 5]);
const mockFile = new File([mockAudioData], 'test.wav', {
type: 'audio/wav'
});
const options = {
outputFormat: 'aac' as const
};
const result = await mergeAudioFiles([mockFile], options);
expect(result).toBeInstanceOf(File);
expect(result.name).toBe('merged_audio.aac');
expect(result.type).toBe('audio/aac');
});
it('should throw error when no input files provided', async () => {
const options = {
outputFormat: 'mp3' as const
};
try {
await mergeAudioFiles([], options);
expect.fail('Should have thrown an error');
} catch (error) {
expect(error).toBeInstanceOf(Error);
expect((error as Error).message).toBe('No input files provided');
}
});
});

View File

@@ -0,0 +1,26 @@
import { defineTool } from '@tools/defineTool';
import { lazy } from 'react';
export const tool = defineTool('audio', {
name: 'Merge Audio',
path: 'merge-audio',
icon: 'mdi:music-note-multiple',
description:
'Combine multiple audio files into a single audio file by concatenating them in sequence.',
shortDescription: 'Merge multiple audio files into one (MP3, AAC, WAV).',
keywords: [
'merge',
'audio',
'combine',
'concatenate',
'join',
'mp3',
'aac',
'wav',
'audio editing',
'multiple files'
],
longDescription:
'This tool allows you to merge multiple audio files into a single file by concatenating them in the order you upload them. Perfect for combining podcast segments, music tracks, or any audio files that need to be joined together. Supports various audio formats including MP3, AAC, and WAV.',
component: lazy(() => import('./index'))
});

View File

@@ -0,0 +1,94 @@
import { FFmpeg } from '@ffmpeg/ffmpeg';
import { fetchFile } from '@ffmpeg/util';
import { InitialValuesType } from './types';
const ffmpeg = new FFmpeg();
export async function mergeAudioFiles(
inputs: File[],
options: InitialValuesType
): Promise<File> {
if (!ffmpeg.loaded) {
await ffmpeg.load({
wasmURL:
'https://cdn.jsdelivr.net/npm/@ffmpeg/core@0.12.9/dist/esm/ffmpeg-core.wasm'
});
}
if (inputs.length === 0) {
throw new Error('No input files provided');
}
const { outputFormat } = options;
const outputName = `output.${outputFormat}`;
// Write all input files to FFmpeg
const inputNames: string[] = [];
for (let i = 0; i < inputs.length; i++) {
const inputName = `input${i}.mp3`;
await ffmpeg.writeFile(inputName, await fetchFile(inputs[i]));
inputNames.push(inputName);
}
// Create a file list for concatenation
const fileListName = 'filelist.txt';
const fileListContent = inputNames.map((name) => `file '${name}'`).join('\n');
await ffmpeg.writeFile(fileListName, fileListContent);
// Build FFmpeg arguments for merging
let args: string[] = ['-f', 'concat', '-safe', '0', '-i', fileListName];
// Add format-specific arguments
if (outputFormat === 'mp3') {
args.push(
'-ar',
'44100',
'-ac',
'2',
'-b:a',
'192k',
'-f',
'mp3',
outputName
);
} else if (outputFormat === 'aac') {
args.push('-c:a', 'aac', '-b:a', '192k', '-f', 'adts', outputName);
} else if (outputFormat === 'wav') {
args.push(
'-acodec',
'pcm_s16le',
'-ar',
'44100',
'-ac',
'2',
'-f',
'wav',
outputName
);
}
await ffmpeg.exec(args);
const mergedAudio = await ffmpeg.readFile(outputName);
let mimeType = 'audio/mp3';
if (outputFormat === 'aac') mimeType = 'audio/aac';
if (outputFormat === 'wav') mimeType = 'audio/wav';
// Clean up files
for (const inputName of inputNames) {
await ffmpeg.deleteFile(inputName);
}
await ffmpeg.deleteFile(fileListName);
await ffmpeg.deleteFile(outputName);
return new File(
[
new Blob([mergedAudio], {
type: mimeType
})
],
`merged_audio.${outputFormat}`,
{ type: mimeType }
);
}

View File

@@ -0,0 +1,3 @@
export type InitialValuesType = {
outputFormat: 'mp3' | 'aac' | 'wav';
};