吃得饱系列-用 Rust 写个 CLI 解决 UWP Bilibili 下载的视频无法播放问题
发现问题
众所周知，使用 wINDOWS 微软商店下载的 Bilibili 客户端是有视频下载功能的，不过它有个问题，下载后的视频无法直接使用一般的打开。
稍微研究（使用搜索引擎）一下发觉，原来是该客户端下载后的文件开头补了三个 0xFF。
解决方案
最简单的方法就是直接使用一些支持 hex 编辑的 editor 手动把它们删掉再保存，不过如果有多个文件（很明显这是常态），那一个一个手动处理太机械了，于是直接写个命令行工具解决。实现原理也很简单，就是直接跳过读取开头三个 0xFF 再写入到新文件。
开始编码
创建项目
创建 Rust 项目先，直接执行 cargo new xxx 就把项目生成好啦。然后添加几个第三方库，首先自然是 clap，然后是错误处理，这种小工具就不手写自定义错误处理啦，使用 anyhow，还可以加个进度条指示器indictif，因为要批处理文件，文件夹内可能有非视频文件，所以加个 regex 包用正则过滤一下文件。
其中 clap 用得是它的派生宏版本。
cargo add clap -F derive
cargo add anyhow
cargo add indicatif
cargo add regex
可以 cat 查看一下 Cargo.toml 内容
$ cat ./Cargo.toml
---------------------------
[package]
name = "xxx"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
anyhow = "1.0.70"
clap = { version = "4.2.1", features = ["derive"] }
indicatif = "0.17.3"
regex = "1.7.3"
正式编码
先定义好传参，其实就是接收输入的目录，所以用派生宏包一下结构体 Args
#[derive(Parser, Debug)]
#[clap(author, version, about, long_about = None)]
struct Args {
#[clap(value_parser)]
input: String,
}
然后在 main 函数写上下面的代码，主要是获取待处理的目录，当它不是一个文件夹时直接 panic
fn main() {
let args: Args = Args::parse();
let p = Path::new(&args.input);
if !p.is_dir() {
panic!("The input is not directory!");
}
read_file(p);
}
我们还得把 read_file 函数补上，这一步就是基本逻辑就是使用 std::fs::read_dir 读取文件夹内的子路径，判断一下读取到的是否是文件或文件夹，如果是文件夹就递归继续读取，直到读到文件为止，读取到文件就用 process_file 处理它，出错了不管直接 unwrap 死就死吧，小项目是这样的捏。
fn read_file(dir: &Path) {
for sub_path in read_dir(dir).expect("Failed to read file path.").flatten() {
let p = sub_path.path();
if p.is_file() {
process_file(&p).unwrap();
} else {
read_file(&p);
}
}
}
个么很自然地就是实现 process_file 函数，先取到文件名（其实是取得完整的文件路径），然后通过正则过滤一下副档名是 mp4 的文件，读取文件大小给进度条结构用，判断下原文件开头三个字节是 0xFF 下继续进行下一步操作，主要是为了防止处理错文件。然后创建个空文件，循环读取原文件内容写入到新文件，后面再把原文件删除，把新文件修改一下名字，至此处理完成。
const BUFFER_SIZE: usize = 1024;
fn process_file(p: &Path) -> Result<(), anyhow::Error> {
let filename = p
.file_name()
.expect("Failed to get filename.")
.to_str()
.expect("Failed to unwrap path str.");
let mut buffer = [0u8; BUFFER_SIZE];
let file_regex = Regex::new(r"\.mp4$").expect("Failed to new file regex.");
if file_regex.is_match(filename) {
let mut original_file = File::open(p)?;
let file_size = original_file.metadata()?.len();
let pb = ProgressBar::new(file_size);
let mut count = original_file.read(&mut buffer)?;
if buffer[0] == 0xFF && buffer[1] == 0xFF && buffer[2] == 0xFF {
let new_filename = String::from("new_") + filename;
let new_filename = p.with_file_name(new_filename);
let new_file_path = Path::new(&new_filename);
let mut new_file = File::create(new_file_path)?;
let mut is_first = true;
while count != 0 {
if is_first {
new_file.write_all(&buffer[3..count])?;
pb.inc((count - 3) as u64);
is_first = false;
} else {
new_file.write_all(&buffer[..count])?;
pb.inc(count as u64);
}
count = original_file.read(&mut buffer)?;
}
remove_file(p)?;
rename(new_file_path, p)?;
pb.finish_with_message("done");
}
}
Ok(())
}
然后就可以通过一般的视频播放器打开文件啦。