edward
/
omegaupload
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use fstrings where possible

master
Edward Shen 2022-01-16 00:49:42 -08:00
parent 4e2cfcfac6
commit 732e2bd2f3
Signed by: edward
GPG Key ID: 19182661E818369F
4 changed files with 23 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cli/src/main.rs
View File

@ -162,14 +162,14 @@ fn handle_upload(
url.set_fragment(Some(&fragment.build().expose_secret()));
println!("{}", url);
println!("{url}");
Ok(())
}
fn handle_download(mut url: ParsedUrl) -> Result<()> {
url.sanitized_url
.set_path(&format!("{}{}", API_ENDPOINT, url.sanitized_url.path()));
.set_path(&format!("{API_ENDPOINT}{}", url.sanitized_url.path()));
let res = Client::new()
.get(url.sanitized_url)
.send()
@ -212,7 +212,7 @@ fn handle_download(mut url: ParsedUrl) -> Result<()> {
std::io::stdout().write_all(&data)?;
}
eprintln!("{}", expiration_text);
eprintln!("{expiration_text}");
Ok(())
}

26
server/src/main.rs
View File

@ -97,7 +97,7 @@ async fn main() -> Result<()> {
.route("/:code", get(|| async { INDEX_PAGE }))
.nest("/static", root_service)
.route(
&format!("{}{}", API_ENDPOINT, "/:code"),
&format!("{API_ENDPOINT}/:code"),
get(paste::<SHORT_CODE_SIZE>).delete(delete::<SHORT_CODE_SIZE>),
)
.layer(AddExtensionLayer::new(db))
@ -163,11 +163,11 @@ fn set_up_expirations<const N: usize>(db: &Arc<DB>) {
if corrupted == 0 {
info!("No corrupted pastes found.");
} else {
warn!("Found {} corrupted pastes.", corrupted);
warn!("Found {corrupted} corrupted pastes.");
}
info!("Found {} expired pastes.", expired);
info!("Found {} active pastes.", pending);
info!("Found {expired} expired pastes.");
info!("Found {pending} active pastes.");
info!("Cleanup timers have been initialized.");
}
@ -196,7 +196,7 @@ async fn upload<const N: usize>(
if let Some(header) = maybe_expires {
if let Expiration::UnixTime(time) = header.0 {
if (time - Utc::now()) > *MAX_PASTE_AGE {
warn!("{} exceeds allowed paste lifetime", time);
warn!("{time} exceeds allowed paste lifetime");
return Err(StatusCode::BAD_REQUEST);
}
}
@ -223,7 +223,7 @@ async fn upload<const N: usize>(
.await;
if matches!(query, Ok(false)) {
new_key = Some(key);
trace!("Found new key after {} attempts.", i);
trace!("Found new key after {i} attempts.");
break;
}
}
@ -271,7 +271,7 @@ async fn upload<const N: usize>(
}
}
e => {
error!("Failed to insert paste into db: {:?}", e);
error!("Failed to insert paste into db: {e:?}");
return Err(StatusCode::INTERNAL_SERVER_ERROR);
}
}
@ -289,7 +289,7 @@ async fn paste<const N: usize>(
let metadata: Expiration = {
let meta_cf = db.cf_handle(META_CF_NAME).unwrap();
let query_result = db.get_cf(meta_cf, key).map_err(|e| {
error!("Failed to fetch initial query: {}", e);
error!("Failed to fetch initial query: {e}");
StatusCode::INTERNAL_SERVER_ERROR
})?;
@ -308,7 +308,7 @@ async fn paste<const N: usize>(
if let Expiration::UnixTime(expires) = metadata {
if expires < Utc::now() {
delete_entry(db, url.as_bytes()).await.map_err(|e| {
error!("Failed to join handle: {}", e);
error!("Failed to join handle: {e}");
StatusCode::INTERNAL_SERVER_ERROR
})??;
return Err(StatusCode::NOT_FOUND);
@ -319,7 +319,7 @@ async fn paste<const N: usize>(
// not sure if perf of get_pinned is better than spawn_blocking
let blob_cf = db.cf_handle(BLOB_CF_NAME).unwrap();
let query_result = db.get_pinned_cf(blob_cf, key).map_err(|e| {
error!("Failed to fetch initial query: {}", e);
error!("Failed to fetch initial query: {e}");
StatusCode::INTERNAL_SERVER_ERROR
})?;
@ -340,7 +340,7 @@ async fn paste<const N: usize>(
Expiration::BurnAfterReading | Expiration::BurnAfterReadingWithDeadline(_)
) {
delete_entry(db, key).await.map_err(|e| {
error!("Failed to join handle: {}", e);
error!("Failed to join handle: {e}");
StatusCode::INTERNAL_SERVER_ERROR
})??;
}
@ -367,11 +367,11 @@ fn delete_entry<const N: usize>(db: Arc<DB>, key: [u8; N]) -> JoinHandle<Result<
let blob_cf = db.cf_handle(BLOB_CF_NAME).unwrap();
let meta_cf = db.cf_handle(META_CF_NAME).unwrap();
if let Err(e) = db.delete_cf(blob_cf, &key) {
warn!("{}", e);
warn!("{e}");
return Err(StatusCode::INTERNAL_SERVER_ERROR);
}
if let Err(e) = db.delete_cf(meta_cf, &key) {
warn!("{}", e);
warn!("{e}");
return Err(StatusCode::INTERNAL_SERVER_ERROR);
}
Ok(())

2
web/src/decrypt.rs
View File

@ -113,7 +113,7 @@ fn handle_zip_archive(blob: Arc<Blob>, container: Vec<u8>) -> DecryptedData {
log!("Unsupported: ", s.to_string());
}
_ => {
log!(format!("Error: {}", err));
log!(format!("Error: {err}"));
}
},
}

13
web/src/main.rs
View File

@ -114,7 +114,7 @@ fn main() {
Ok(partial_parsed_url) => partial_parsed_url,
Err(e) => {
error!("Failed to parse text fragment; bailing.");
render_message(format!("Invalid paste link: {}", e).into());
render_message(format!("Invalid paste link: {e}").into());
return;
}
};
@ -148,7 +148,7 @@ fn main() {
}
e => {
render_message("Internal error occurred.".into());
error!(format!("Error occurred at pw prompt: {:?}", e));
error!(format!("Error occurred at pw prompt: {e:?}"));
return;
}
}
@ -191,8 +191,7 @@ async fn fetch_resources(
"Network failure: Failed to completely read encryption paste.".into(),
);
bail!(format!(
"JsFuture returned an error while fetching resp buffer: {:?}",
e
"JsFuture returned an error while fetching resp buffer: {e:?}",
));
}
};
@ -204,14 +203,14 @@ async fn fetch_resources(
return Ok(());
}
let (decrypted, mimetype) = match decrypt(data, &key, password) {
let (decrypted, mimetype) = match decrypt(data, &key, password, name.as_deref()) {
Ok(data) => data,
Err(e) => {
let msg = match e {
CryptoError::Password => "The provided password was incorrect.",
CryptoError::SecretKey => "The secret key in the URL was incorrect.",
ref e => {
log!(format!("Bad kdf or corrupted blob: {}", e));
log!(format!("Bad kdf or corrupted blob: {e}"));
"An internal error occurred."
}
};
@ -248,7 +247,7 @@ async fn fetch_resources(
render_message(err.status_text().into());
}
Err(err) => {
render_message(format!("{}", err).into());
render_message(format!("{err}").into());
}
}