blob: d283dcb96efe02d3a90c1631def5690e17029cd3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
/*
This file is part of jellything (https://codeberg.org/metamuffin/jellything)
which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
Copyright (C) 2026 metamuffin <metamuffin.org>
*/
use anyhow::{Result, bail};
use rocksdb::DB;
use std::{env::args, fs::File, io::Read, path::Path};
fn main() -> Result<()> {
let in_path = args().nth(1).unwrap();
let out_path = args().nth(2).unwrap();
let db = DB::open_default(out_path)?;
if !in_path.ends_with("/") {
bail!("path needs to end with /")
}
traverse(&db, &in_path, in_path.as_ref())?;
db.flush()?;
Ok(())
}
fn traverse(db: &DB, prefix: &str, path: &Path) -> Result<()> {
if path.is_dir() {
for e in path.read_dir()? {
traverse(db, prefix, &e?.path())?;
}
}
if path.is_file() {
let key = path.to_string_lossy();
let key = key.strip_prefix(prefix).unwrap();
let mut value = Vec::new();
File::open(path)?.read_to_end(&mut value)?;
println!("{key}");
db.put(key, value)?;
}
Ok(())
}
|
