quininer / epub2txt Goto Github PK

use std::fs;

fn main() {
    // 创建vb文件夹
    if let Err(e) = fs::create_dir("vb") {
        if e.kind() != std::io::ErrorKind::AlreadyExists {
            panic!("Failed to create directory: {:?}", e);
        }
    }

    // 读取vb.txt文件
    let content = match fs::read_to_string("vb.txt") {
        Ok(s) => s,
        Err(e) => panic!("Failed to read file: {:?}", e),
    };

    // 以四个中横杠隔开文本
    let parts: Vec<&str> = content.split("-----").collect();

    // 保存每个部分文本到文件
    for (i, part) in parts.iter().enumerate() {
        // 生成文件名
        let filename = format!("vb/vb_part{}.txt", i+1);
        // 写入文件
        match fs::write(filename, part) {
            Ok(_) => {},
            Err(e) => panic!("Failed to write file: {:?}", e),
        };
    }
}