ethanaobrien/ew
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add ability to disable account exports

Browse source
This commit is contained in:
Ethan O'Brien 2024-05-27 15:40:43 -05:00
parent 0b969eab97
commit a8f1550c94
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/router/webui.rs
View file

@ -13,10 +13,18 @@ use crate::include_file;
use crate::router::{userdata, items}; use crate::router::{userdata, items};
fn get_config() -> JsonValue { fn get_config() -> JsonValue {
let def = object!{
import: true,
export: true
};
let contents = fs::read_to_string("config.json").unwrap_or(String::from("aaaaaaaaaaaaaaaaa")); let contents = fs::read_to_string("config.json").unwrap_or(String::from("aaaaaaaaaaaaaaaaa"));
json::parse(&contents).unwrap_or(object!{ let mut rv = json::parse(&contents).unwrap_or(def.clone());
import: true for val in def.entries() {
}) if rv[val.0] == JsonValue::Null {
rv[val.0] = val.1.clone();
}
}
rv
} }
fn save_config(val: String) { fn save_config(val: String) {
let mut current = get_config(); let mut current = get_config();
@ -68,13 +76,7 @@ pub fn login(_req: HttpRequest, body: String) -> HttpResponse {
pub fn import(_req: HttpRequest, body: String) -> HttpResponse { pub fn import(_req: HttpRequest, body: String) -> HttpResponse {
if get_config()["import"].as_bool().unwrap() == false { if get_config()["import"].as_bool().unwrap() == false {
let resp = object!{ return error("Importing accounts is disabled on this server.");
result: "Err",
message: "Importing accounts is disabled on this server."
};
return HttpResponse::Ok()
.insert_header(ContentType::json())
.body(json::stringify(resp));
} }
let body = json::parse(&body).unwrap(); let body = json::parse(&body).unwrap();
@ -220,6 +222,9 @@ pub fn admin_post(req: HttpRequest, body: String) -> HttpResponse {
} }
pub fn export(req: HttpRequest) -> HttpResponse { pub fn export(req: HttpRequest) -> HttpResponse {
if get_config()["export"].as_bool().unwrap() == false {
return error("Exporting accounts is disabled on this server.");
}
let token = get_login_token(&req); let token = get_login_token(&req);
if token.is_none() { if token.is_none() {
return error("Not logged in"); return error("Not logged in");