Converter from twitchchatdownloader to srt.

csv2srt.rs

#!/usr/bin/env -S cargo +nightly -Zscript

//! Convert Twitch subtitles exported as CSV from https://www.twitchchatdownloader.com to srt format
//!

//! ```cargo
//! [package]
//! edition="2021"
//! [dependencies]
//! csv = "1"
//! anyhow="1"
//! srtlib = "0.1.0"
//! unicode-segmentation="1"
//! ```

pub fn main() -> anyhow::Result<()> {
    use unicode_segmentation::UnicodeSegmentation;

    let subs = Vec::with_capacity(1000);
    let mut num = 1;
    let mut subs = srtlib::Subtitles::new_from_vec(subs);
    let mut r = csv::Reader::from_reader(std::io::stdin());
    let header = r.headers()?;
    
    anyhow::ensure!(header == &csv::StringRecord::from(vec!["time", "user_name", "user_color", "message"]));
    let enef = ||anyhow::anyhow!("No enough fields in a csv line");
    for l in r.records() {
        let l = l?;
        let t : i32 = l.get(0).ok_or_else(enef)?.parse()?;
        let name    : &str = &l.get(1).ok_or_else(enef)?;
        let comment : &str = &l.get(3).ok_or_else(enef)?;

        let mut start_time = srtlib::Timestamp::new(0,0,0,0);
        start_time.add_seconds(t);
        let mut end_time = start_time.clone();
        end_time.add_seconds(1);
        end_time.add_seconds((comment.graphemes(true).count() / 20) as i32 );

        let s = srtlib::Subtitle::new(
            num, 
            start_time,
            end_time, 
            format!("[{}] {}", name, comment)
        );
        subs.push(s);
        num+=1;
    }
    
    println!("{}", subs.to_string());
    Ok(())
}