Introduce MaybeError type

src/data_provider/web_api.rs

@@ -1,18 +1,20 @@
-use std::io;
+use crate::fs::*;
+use crate::notification::NotificationData::{CompressError, PermissionDenied, WebApiUnavailable};
+use crate::notification::{MaybeError, NotificationData};
 use log::info;
+use std::io;
 use tokio::fs::{File, OpenOptions};
 use tokio::io::{AsyncReadExt, AsyncWriteExt};
-use zstd::{DEFAULT_COMPRESSION_LEVEL, Decoder, Encoder};
-use crate::fs::*;
-use crate::notification::NotificationData;
-use crate::notification::NotificationData::{CompressError, PermissionDenied, WebApiUnavailable};
+use zstd::{Decoder, Encoder, DEFAULT_COMPRESSION_LEVEL};
 
 // const API_URL: &str = "http://127.0.0.1:8089/";
 const API_URL: &str = "https://learning.micialware.ru/";
 
-pub async fn send_data(id: String) -> Result<(), NotificationData> {
+pub async fn send_data(id: String) -> MaybeError {
     let db_file = db_path();
-    let data = tokio::fs::read(db_file).await.map_err(|_| WebApiUnavailable)?;
+    let data = tokio::fs::read(db_file)
+        .await
+        .map_err(|_| WebApiUnavailable)?;
     let data = compress(data)?;
 
     let api_url = API_URL.to_string();
@@ -23,30 +25,35 @@ pub async fn send_data(id: String) -> Result<(), NotificationData> {
         .post(&id_url)
         .body(data)
         .send()
-        .await.map_err(|_| WebApiUnavailable)?
+        .await
+        .map_err(|_| WebApiUnavailable)?
         .text()
-        .await.map_err(|_| WebApiUnavailable)?;
+        .await
+        .map_err(|_| WebApiUnavailable)?;
     set_local_version(new_version.parse::<u32>().unwrap_or_default()).await;
     Ok(())
 }
 
 fn compress(data: Vec<u8>) -> Result<Vec<u8>, NotificationData> {
-    let mut encoder = Encoder::new(Vec::new(), DEFAULT_COMPRESSION_LEVEL).map_err(|_| CompressError)?;
+    let mut encoder =
+        Encoder::new(Vec::new(), DEFAULT_COMPRESSION_LEVEL).map_err(|_| CompressError)?;
     io::copy(&mut &data[..], &mut encoder).map_err(|_| PermissionDenied)?;
     encoder.finish().map_err(|_| CompressError)?;
     Ok(data)
 }
 
-pub async fn load_data(id: String, temp: bool) -> Result<(), NotificationData> {
+pub async fn load_data(id: String, temp: bool) -> MaybeError {
     let api_url = API_URL.to_string();
     let id_url = format!("{api_url}download/{id}");
     let client = reqwest::Client::new();
     let data = client
         .get(&id_url)
         .send()
-        .await.map_err(|_| WebApiUnavailable)?
+        .await
+        .map_err(|_| WebApiUnavailable)?
         .bytes()
-        .await.map_err(|_| WebApiUnavailable)?
+        .await
+        .map_err(|_| WebApiUnavailable)?
         .to_vec();
     let data = decompress(data)?;
 
@@ -54,15 +61,19 @@ pub async fn load_data(id: String, temp: bool) -> Result<(), NotificationData> {
     let file_name = if temp { "data.db.tmp" } else { "data.db" };
     let db_file = path.join(file_name);
 
-    tokio::fs::write(db_file, data).await.map_err(|_| PermissionDenied)?;
+    tokio::fs::write(db_file, data)
+        .await
+        .map_err(|_| PermissionDenied)?;
 
     let version_url = format!("{API_URL}{id}/version");
     let version = client
         .get(&version_url)
         .send()
-        .await.map_err(|_| WebApiUnavailable)?
+        .await
+        .map_err(|_| WebApiUnavailable)?
         .text()
-        .await.map_err(|_| WebApiUnavailable)?;
+        .await
+        .map_err(|_| WebApiUnavailable)?;
     set_local_version(version.parse::<u32>().unwrap_or_default()).await;
     Ok(())
 }
@@ -93,22 +104,29 @@ pub async fn first_sync() -> Result<String, NotificationData> {
 pub async fn get_web_version(key: &str) -> Result<u32, NotificationData> {
     let id_url = format!("{API_URL}{key}/version");
     let client = reqwest::Client::new();
-    let version = client.get(&id_url).send().await.map_err(|_| WebApiUnavailable)?.text().await.map_err(|_| WebApiUnavailable)?;
+    let version = client
+        .get(&id_url)
+        .send()
+        .await
+        .map_err(|_| WebApiUnavailable)?
+        .text()
+        .await
+        .map_err(|_| WebApiUnavailable)?;
     info!("version: {}", version);
     Ok(version.parse::<u32>().unwrap_or_default())
 }
 
-pub async fn get_local_version() -> u32 {
+pub async fn get_local_version() -> Result<u32, NotificationData> {
     let file = db_meta_path();
     if file.exists() {
         let mut data = String::new();
         File::open(file.clone())
             .await
-            .unwrap()
+            .map_err(|_| PermissionDenied)?
             .read_to_string(&mut data)
             .await
-            .unwrap();
-        return data.parse::<u32>().unwrap_or_default();
+            .map_err(|_| PermissionDenied)?;
+        return Ok(data.parse::<u32>().unwrap_or_default());
     }
 
     let mut file = OpenOptions::new()
@@ -117,12 +135,14 @@ pub async fn get_local_version() -> u32 {
         .truncate(true)
         .open(file)
         .await
-        .unwrap();
-    file.write_all("0".as_bytes()).await.unwrap();
-    0
+        .map_err(|_| PermissionDenied)?;
+    file.write_all("0".as_bytes())
+        .await
+        .map_err(|_| PermissionDenied)?;
+    Ok(0)
 }
 
-pub async fn set_local_version(version: u32) {
+pub async fn set_local_version(version: u32) -> Result<(), NotificationData> {
     let file_path = db_meta_path();
     let mut file = OpenOptions::new()
         .write(true)
@@ -130,8 +150,9 @@ pub async fn set_local_version(version: u32) {
         .truncate(true)
         .open(file_path)
         .await
-        .unwrap();
+        .map_err(|_| PermissionDenied)?;
     file.write_all(version.to_string().as_bytes())
         .await
-        .unwrap();
+        .map_err(|_| PermissionDenied)?;
+    Ok(())
 }