50 lines
1.7 KiB
Rust
50 lines
1.7 KiB
Rust
|
use troll_patrol::{
|
||
|
|
extra_info::{self, ExtraInfo},
|
||
|
|
//negative_report::SerializableNegativeReport,
|
||
|
|
//positive_report::SerializablePositiveReport,
|
||
|
|
*,
|
||
|
|
};
|
||
|
|
|
||
|
|
use clap::Parser;
|
||
|
|
use sled::Db;
|
||
|
|
use std::{collections::HashSet, path::PathBuf};
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() {
|
||
|
|
// TODO: Currently, we're processing extra-infos here, but we want to:
|
||
|
|
// 1. Run a server to accept incoming reports
|
||
|
|
// 2. Periodically (daily):
|
||
|
|
// a) download new extra-infos
|
||
|
|
// b) determine whether we think each bridge is blocked or not
|
||
|
|
// c) report these results to the LA
|
||
|
|
// 3. Store all our data
|
||
|
|
|
||
|
|
let db: Db = sled::open(&CONFIG.db.db_path).unwrap();
|
||
|
|
|
||
|
|
// Track which files have been processed. This is slightly redundant
|
||
|
|
// because we're only downloading files we don't already have, but it
|
||
|
|
// might be a good idea to check in case we downloaded a file but didn't
|
||
|
|
// process it for some reason.
|
||
|
|
let mut processed_extra_infos_files = match db.get(b"extra_infos_files").unwrap() {
|
||
|
|
Some(v) => bincode::deserialize(&v).unwrap(),
|
||
|
|
None => HashSet::<String>::new(),
|
||
|
|
};
|
||
|
|
|
||
|
|
let new_files = extra_info::download_extra_infos().await.unwrap();
|
||
|
|
|
||
|
|
let mut new_extra_infos = HashSet::<ExtraInfo>::new();
|
||
|
|
|
||
|
|
// Make set of new extra-infos
|
||
|
|
for extra_info_file in &new_files {
|
||
|
|
extra_info::add_extra_infos(&extra_info_file, &mut new_extra_infos);
|
||
|
|
processed_extra_infos_files.insert(extra_info_file.to_string());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add new extra-infos data to database
|
||
|
|
for extra_info in new_extra_infos {
|
||
|
|
add_extra_info_to_db(&db, extra_info);
|
||
|
|
}
|
||
|
|
|
||
|
|
db.insert(b"extra_infos_files", bincode::serialize(&processed_extra_infos_files).unwrap()).unwrap();
|
||
|
|
}
|