dump.rs
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

dump.rs
View Options

	use crate::{
	file::{self, Exists},
	helpers, log,
	process::{id::Id, Runner},
	};

	use colored::Colorize;
	use global_placeholders::global;
	use macros_rs::{crashln, fmtstr, string};
	use reqwest::blocking::Client;
	use reqwest::header::{HeaderMap, HeaderValue};
	use std::{collections::BTreeMap, fs};

	pub fn from(address: &str, token: Option<&str>) -> Result<Runner, anyhow::Error> {
	let client = Client::new();
	let mut headers = HeaderMap::new();

	if let Some(token) = token {
	headers.insert("token", HeaderValue::from_static(Box::leak(Box::from(token))));
	}

	let response = client.get(fmtstr!("{address}/daemon/dump")).headers(headers).send()?;
	let bytes = response.bytes()?;

	Ok(file::from_object(&bytes))
	}

	pub fn read() -> Runner {
	if !Exists::check(&global!("pmc.dump")).file() {
	let runner = Runner {
	id: Id::new(0),
	list: BTreeMap::new(),
	remote: None,
	};

	write(&runner);
	log!("created dump file");
	}

	file::read_object(global!("pmc.dump"))
	}

	pub fn raw() -> Vec<u8> {
	if !Exists::check(&global!("pmc.dump")).file() {
	let runner = Runner {
	id: Id::new(0),
	list: BTreeMap::new(),
	remote: None,
	};

	write(&runner);
	log!("created dump file");
	}

	file::raw(global!("pmc.dump"))
	}

	pub fn write(dump: &Runner) {
	let encoded = match ron::ser::to_string(&dump) {
	Ok(contents) => contents,
	Err(err) => crashln!("{} Cannot encode dump.\n{}", *helpers::FAIL, string!(err).white()),
	};

	if let Err(err) = fs::write(global!("pmc.dump"), encoded) {
	crashln!("{} Error writing dumpfile.\n{}", *helpers::FAIL, string!(err).white())
	}
	}