Spaces:
Runtime error
Runtime error
File size: 2,883 Bytes
92189dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import {getFileName} from '@/common/components/options/ShareUtils';
import {
EncodingCompletedEvent,
EncodingStateUpdateEvent,
} from '@/common/components/video/VideoWorkerBridge';
import useVideo from '@/common/components/video/editor/useVideo';
import {MP4ArrayBuffer} from 'mp4box';
import {useState} from 'react';
type DownloadingState = 'default' | 'started' | 'encoding' | 'completed';
type State = {
state: DownloadingState;
progress: number;
download: (shouldSave?: boolean) => Promise<MP4ArrayBuffer>;
};
export default function useDownloadVideo(): State {
const [downloadingState, setDownloadingState] =
useState<DownloadingState>('default');
const [progress, setProgress] = useState<number>(0);
const video = useVideo();
async function download(shouldSave = true): Promise<MP4ArrayBuffer> {
return new Promise(resolve => {
function onEncodingStateUpdate(event: EncodingStateUpdateEvent) {
setDownloadingState('encoding');
setProgress(event.progress);
}
function onEncodingComplete(event: EncodingCompletedEvent) {
const file = event.file;
if (shouldSave) {
saveVideo(file, getFileName());
}
video?.removeEventListener('encodingCompleted', onEncodingComplete);
video?.removeEventListener(
'encodingStateUpdate',
onEncodingStateUpdate,
);
setDownloadingState('completed');
resolve(file);
}
video?.addEventListener('encodingStateUpdate', onEncodingStateUpdate);
video?.addEventListener('encodingCompleted', onEncodingComplete);
if (downloadingState === 'default' || downloadingState === 'completed') {
setDownloadingState('started');
video?.pause();
video?.encode();
}
});
}
function saveVideo(file: MP4ArrayBuffer, fileName: string) {
const blob = new Blob([file]);
const url = window.URL.createObjectURL(blob);
const a = document.createElement('a');
document.body.appendChild(a);
a.setAttribute('href', url);
a.setAttribute('download', fileName);
a.setAttribute('target', '_self');
a.click();
window.URL.revokeObjectURL(url);
}
return {download, progress, state: downloadingState};
}
|