Skip to content

Commit

Permalink
arg: add dbpath argument
Browse files Browse the repository at this point in the history
This is great for testing without using an existing DB

Signed-off-by: William Casarin <[email protected]>
  • Loading branch information
jb55 committed Aug 3, 2024
1 parent 3862652 commit abd529e
Showing 1 changed file with 18 additions and 1 deletion.
19 changes: 18 additions & 1 deletion src/app.rs
Original file line number Diff line number Diff line change
Expand Up @@ -529,6 +529,7 @@ struct Args {
is_mobile: Option<bool>,
keys: Vec<Keypair>,
light: bool,
dbpath: Option<String>,
}

fn parse_args(args: &[String]) -> Args {
Expand All @@ -538,6 +539,7 @@ fn parse_args(args: &[String]) -> Args {
is_mobile: None,
keys: vec![],
light: false,
dbpath: None,
};

let mut i = 0;
Expand Down Expand Up @@ -584,6 +586,15 @@ fn parse_args(args: &[String]) -> Args {
} else {
error!("failed to parse filter '{}'", filter);
}
} else if arg == "--dbpath" {
i += 1;
let path = if let Some(next_arg) = args.get(i) {
next_arg
} else {
error!("dbpath argument missing?");
continue;
};
res.dbpath = Some(path.clone());
} else if arg == "-r" || arg == "--relay" {
i += 1;
let relay = if let Some(next_arg) = args.get(i) {
Expand Down Expand Up @@ -659,6 +670,12 @@ impl Damus {

setup_cc(cc, is_mobile, parsed_args.light);

let dbpath = parsed_args
.dbpath
.unwrap_or(data_path.as_ref().to_str().expect("db path ok").to_string());

let _ = std::fs::create_dir_all(dbpath.clone());

let imgcache_dir = data_path.as_ref().join(ImageCache::rel_datadir());
let _ = std::fs::create_dir_all(imgcache_dir.clone());

Expand Down Expand Up @@ -712,7 +729,7 @@ impl Damus {
selected_timeline: 0,
timelines: parsed_args.timelines,
textmode: false,
ndb: Ndb::new(data_path.as_ref().to_str().expect("db path ok"), &config).expect("ndb"),
ndb: Ndb::new(&dbpath, &config).expect("ndb"),
account_manager,
//compose: "".to_string(),
frame_history: FrameHistory::default(),
Expand Down

0 comments on commit abd529e

Please sign in to comment.