Function waitForMedia

packages/scripts/src/utils/study.ts:4–45

(options?: {
	/**
	 * 视频选择器
	 */
	videoSelector?: string;
	/**
	 * 音频选择器
	 */
	audioSelector?: string;
	/**
	 * 根元素
	 */
	root?: HTMLElement | Document;
	timeout?: number;
	filter?: (video: HTMLVideoElement | HTMLAudioElement) => boolean;
})

 * 等待视频加载并获取视频
 */
export async function waitForMedia(options?: {
	/**
	 * 视频选择器
	 */
	videoSelector?: string;
	/**
	 * 音频选择器
	 */
	audioSelector?: string;
	/**
	 * 根元素
	 */
	root?: HTMLElement | Document;
	timeout?: number;
	filter?: (video: HTMLVideoElement | HTMLAudioElement) => boolean;
}) {
	const timeoutMs = options?.timeout ?? 3 * 60 * 1000;

	const res = await new Promise<HTMLVideoElement | HTMLAudioElement>((resolve, reject) => {
		// eslint-disable-next-line prefer-const
		let timeoutId: ReturnType<typeof setTimeout> | undefined = undefined;
		const interval = setInterval(() => {
			const video = (options?.root || document).querySelector<HTMLVideoElement | HTMLAudioElement>(
				`${options?.videoSelector || 'video'},${options?.audioSelector || 'audio'}`
			);
			if (video && (!options?.filter || options.filter(video))) {
				clearInterval(interval);
				if (timeoutId !== undefined) {
					clearTimeout(timeoutId);
				}
				resolve(video);
			}
		}, 200);

		timeoutId = setTimeout(() => {
			clearInterval(interval);
			reject(new Error('视频/音频未找到，或者加载超时。'));
		}, timeoutMs);
	});

	return res;
}

export function waitForElement(
	selector: string | { (): HTMLElement | undefined },