decrypt_files_concurrently_async.rs

use rsmp4decrypt::Mp4Decryptor;

const VIDEO_KID: &str = "eb676abbcb345e96bbcf616630f1a3da";
const VIDEO_KEY: &str = "100b6c20940f779a4589152b57d2dacb";
const AUDIO_KID: &str = "63cb5f7184dd4b689a5c5ff11ee6a328";
const AUDIO_KEY: &str = "3bda3329158a4789880816a70e7e436d";

#[tokio::main(flavor = "multi_thread")]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let jobs = [
        (
            Mp4Decryptor::builder()
                .kid_key(VIDEO_KID, VIDEO_KEY)?
                .build_isolated()?,
            "tests/fixtures/cenc-multi/video_1.m4s",
            "tests/fixtures/cenc-multi/video_init.mp4",
            "target/example-async-video.m4s",
        ),
        (
            Mp4Decryptor::builder()
                .kid_key(AUDIO_KID, AUDIO_KEY)?
                .build_isolated()?,
            "tests/fixtures/cenc-multi/audio_1.m4s",
            "tests/fixtures/cenc-multi/audio_init.mp4",
            "target/example-async-audio.m4s",
        ),
    ];

    let mut tasks = tokio::task::JoinSet::new();
    for (decryptor, input, fragments_info, output) in jobs {
        tasks.spawn(async move {
            decryptor
                .decrypt_file_async(input, output, Some(fragments_info))
                .await
        });
    }

    while let Some(result) = tasks.join_next().await {
        result??;
    }

    Ok(())
}