Cargo.toml

@@ -39,4 +39,3 @@ url = { version = "2", features = [ "serde" ] }
 [profile.release]
 lto = true
 codegen-units = 1
-debug = 1

src/cache/disk_cache.rs

@@ -8,7 +8,7 @@ use std::sync::Arc;
 use async_trait::async_trait;
 use bytes::Bytes;
 use futures::StreamExt;
-use log::{error, warn, LevelFilter};
+use log::{warn, LevelFilter};
 use sqlx::sqlite::SqliteConnectOptions;
 use sqlx::{ConnectOptions, SqlitePool};
 use tokio::fs::remove_file;
@@ -36,7 +36,7 @@ impl DiskCache {
     pub async fn new(disk_max_size: u64, disk_path: PathBuf) -> Arc<Box<dyn Cache>> {
         let (db_tx, db_rx) = channel(128);
         let db_pool = {
-            let db_url = format!("sqlite:{}/metadata.sqlite", disk_path.to_string_lossy());
+            let db_url = format!("sqlite:{}/metadata.sqlite", disk_path.to_str().unwrap());
             let mut options = SqliteConnectOptions::from_str(&db_url)
                 .unwrap()
                 .create_if_missing(true);
@@ -80,13 +80,7 @@ async fn db_listener(
     let mut recv_stream = ReceiverStream::new(db_rx).ready_chunks(128);
     while let Some(messages) = recv_stream.next().await {
         let now = chrono::Utc::now();
-        let mut transaction = match db_pool.begin().await {
+        let mut transaction = db_pool.begin().await.unwrap();
-            Ok(transaction) => transaction,
-            Err(e) => {
-                error!("Failed to start a transaction to DB, cannot update DB. Disk cache may be losing track of files! {}", e);
-                continue;
-            }
-        };
         for message in messages {
             match message {
                 DbMessage::Get(entry) => {
@@ -117,42 +111,15 @@ async fn db_listener(
                 }
             }
         }
-
+        transaction.commit().await.unwrap();
-        if let Err(e) = transaction.commit().await {
-            error!(
-                "Failed to commit transaction to DB. Disk cache may be losing track of files! {}",
-                e
-            );
-        }
 
         if cache.on_disk_size() >= max_on_disk_size {
-            let mut conn = match db_pool.acquire().await {
+            let mut conn = db_pool.acquire().await.unwrap();
-                Ok(conn) => conn,
+            let items =
-                Err(e) => {
+                sqlx::query!("select id, size from Images order by accessed asc limit 1000")
-                    error!(
+                    .fetch_all(&mut conn)
-                        "Failed to get a DB connection and cannot prune disk cache: {}",
+                    .await
-                        e
+                    .unwrap();
-                    );
-                    continue;
-                }
-            };
-
-            let items = {
-                let request =
-                    sqlx::query!("select id, size from Images order by accessed asc limit 1000")
-                        .fetch_all(&mut conn)
-                        .await;
-                match request {
-                    Ok(items) => items,
-                    Err(e) => {
-                        error!(
-                            "Failed to fetch oldest images and cannot prune disk cache: {}",
-                            e
-                        );
-                        continue;
-                    }
-                }
-            };
 
             let mut size_freed = 0;
             for item in items {

src/cache/fs.rs

@@ -84,14 +84,14 @@ pub async fn write_file<
 
     let mut file = {
         let mut write_lock = WRITING_STATUS.write().await;
-        let parent = path.parent().expect("The path to have a parent");
+        let parent = path.parent().unwrap();
         create_dir_all(parent).await?;
         let file = File::create(path).await?; // we need to make sure the file exists and is truncated.
         write_lock.insert(path.to_path_buf(), rx.clone());
         file
     };
 
-    let metadata_string = serde_json::to_string(&metadata).expect("serialization to work");
+    let metadata_string = serde_json::to_string(&metadata).unwrap();
     let metadata_size = metadata_string.len();
     // need owned variant because async lifetime
     let path_buf = path.to_path_buf();
@@ -151,8 +151,9 @@ pub async fn write_file<
         }
 
         tokio::spawn(db_callback(bytes_written));
-        if let Some(sender) = on_complete {
+        if accumulate {
             tokio::spawn(async move {
+                let sender = on_complete.unwrap();
                 sender
                     .send((
                         cache_key,
@@ -243,7 +244,7 @@ impl Stream for ConcurrentFsStream {
             if let Poll::Ready(Some(WritingStatus::Done(n))) =
                 self.receiver.as_mut().poll_next_unpin(cx)
             {
-                self.bytes_total = Some(NonZeroU32::new(n).expect("Stored a 0 byte image?"))
+                self.bytes_total = Some(NonZeroU32::new(n).unwrap())
             }
 
             // Okay, now we know if we've read enough bytes or not. If the

src/cache/mem_cache.rs

@@ -13,6 +13,7 @@ use tokio::sync::mpsc::{channel, Sender};
 use tokio::sync::Mutex;
 
 /// Memory accelerated disk cache. Uses an LRU in memory to speed up reads.
+///
 pub struct MemoryLruCache {
     inner: Arc<Box<dyn Cache>>,
     cur_mem_size: AtomicU64,

src/main.rs

@@ -3,12 +3,11 @@
 #![allow(clippy::module_name_repetitions)]
 
 use std::env::{self, VarError};
-use std::hint::unreachable_unchecked;
-use std::num::ParseIntError;
 use std::process;
-use std::sync::atomic::{AtomicBool, Ordering};
+use std::sync::atomic::AtomicBool;
 use std::sync::Arc;
 use std::time::Duration;
+use std::{num::ParseIntError, sync::atomic::Ordering};
 
 use actix_web::rt::{spawn, time, System};
 use actix_web::web::{self, Data};
@@ -49,7 +48,7 @@ enum ServerError {
 }
 
 #[actix_web::main]
-async fn main() -> Result<(), Box<dyn std::error::Error>> {
+async fn main() -> Result<(), std::io::Error> {
     // It's ok to fail early here, it would imply we have a invalid config.
     dotenv::dotenv().ok();
     let cli_args = CliArgs::parse();
@@ -67,10 +66,10 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         (0, 0) => LevelFilter::Info,
         (_, 1) => LevelFilter::Debug,
         (_, n) if n > 1 => LevelFilter::Trace,
-        _ => unsafe { unreachable_unchecked() },
+        _ => unreachable!(),
     };
 
-    SimpleLogger::new().with_level(log_level).init()?;
+    SimpleLogger::new().with_level(log_level).init().unwrap();
 
     print_preamble_and_warnings();
 
@@ -82,7 +81,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     };
     let client_secret_1 = client_secret.clone();
 
-    let server = ServerState::init(&client_secret, &cli_args).await?;
+    let server = ServerState::init(&client_secret, &cli_args).await.unwrap();
     let data_0 = Arc::new(RwLockServerState(RwLock::new(server)));
     let data_1 = Arc::clone(&data_0);
 

src/ping.rs

@@ -39,9 +39,7 @@ impl<'a> Request<'a> {
             port: config.port,
             disk_space: config.disk_quota,
             network_speed: config.network_speed,
-            build_version: client_api_version!()
+            build_version: client_api_version!().parse().unwrap(),
-                .parse()
-                .expect("to parse the build version"),
             tls_created_at: Some(state.0.read().tls_config.created_at.clone()),
         }
     }
@@ -55,9 +53,7 @@ impl<'a> From<(&'a str, &CliArgs)> for Request<'a> {
             port: config.port,
             disk_space: config.disk_quota,
             network_speed: config.network_speed,
-            build_version: client_api_version!()
+            build_version: client_api_version!().parse().unwrap(),
-                .parse()
-                .expect("to parse the build version"),
             tls_created_at: None,
         }
     }

src/state.rs

@@ -8,7 +8,6 @@ use parking_lot::RwLock;
 use rustls::sign::CertifiedKey;
 use rustls::ResolvesServerCert;
 use sodiumoxide::crypto::box_::PrecomputedKey;
-use thiserror::Error;
 use url::Url;
 
 pub struct ServerState {
@@ -22,22 +21,8 @@ pub struct ServerState {
 pub static PREVIOUSLY_PAUSED: AtomicBool = AtomicBool::new(false);
 pub static PREVIOUSLY_COMPROMISED: AtomicBool = AtomicBool::new(false);
 
-#[derive(Error, Debug)]
-pub enum ServerInitError {
-    #[error(transparent)]
-    MalformedResponse(reqwest::Error),
-    #[error(transparent)]
-    Timeout(reqwest::Error),
-    #[error(transparent)]
-    SendFailure(reqwest::Error),
-    #[error("Failed to parse token key")]
-    KeyParseError(String),
-    #[error("Token key was not provided in initial request")]
-    MissingTokenKey,
-}
-
 impl ServerState {
-    pub async fn init(secret: &str, config: &CliArgs) -> Result<Self, ServerInitError> {
+    pub async fn init(secret: &str, config: &CliArgs) -> Result<Self, ()> {
         let resp = reqwest::Client::new()
             .post(CONTROL_CENTER_PING_URL)
             .json(&Request::from((secret, config)))
@@ -54,18 +39,18 @@ impl ServerState {
                 Ok(mut resp) => {
                     let key = resp
                         .token_key
-                        .ok_or(ServerInitError::MissingTokenKey)
                         .and_then(|key| {
                             if let Some(key) = base64::decode(&key)
                                 .ok()
                                 .and_then(|k| PrecomputedKey::from_slice(&k))
                             {
-                                Ok(key)
+                                Some(key)
                             } else {
                                 error!("Failed to parse token key: got {}", key);
-                                Err(ServerInitError::KeyParseError(key))
+                                None
                             }
-                        })?;
+                        })
+                        .unwrap();
 
                     PREVIOUSLY_COMPROMISED.store(resp.paused, Ordering::Release);
                     if resp.compromised {
@@ -106,17 +91,17 @@ impl ServerState {
                 }
                 Err(e) => {
                     warn!("Got malformed response: {}", e);
-                    Err(ServerInitError::MalformedResponse(e))
+                    Err(())
                 }
             },
             Err(e) => match e {
                 e if e.is_timeout() => {
                     error!("Response timed out to control server. Is MangaDex down?");
-                    Err(ServerInitError::Timeout(e))
+                    Err(())
                 }
                 e => {
                     warn!("Failed to send request: {}", e);
-                    Err(ServerInitError::SendFailure(e))
+                    Err(())
                 }
             },
         }