remove unnecessary unwrap

dns-config/src/lib.rs

@@ -1,5 +1,5 @@
-use std::fs::read_to_string;
-use toml::Table;
+use std::{error::Error, fs::read_to_string, io};
+use toml::{Table, Value};
 
 pub struct ZoneConnectionData {
     pub conection_str: String,
@@ -13,34 +13,54 @@ pub struct ServerZonesConnection {
 }
 
 impl ZoneConnectionData {
-    pub fn get_config(file_name: &String) -> Vec<Self> {
+    pub fn get_config(file_name: &String) -> Result<Vec<Self>, Box<dyn Error>> {
         let mut ret = Vec::new();
-        let key_toml = read_to_string(file_name).unwrap().parse::<Table>().unwrap();
-        for servers in key_toml["server"].as_array().unwrap() {
-            let conection_str: String = servers["connection_str"].as_str().unwrap().to_string();
-            let key_name: String = servers["key"]["name"].as_str().unwrap().to_string();
-            let key: String = servers["key"]["value"].as_str().unwrap().to_string();
+        let key_toml = read_to_string(file_name)?.parse::<Table>()?;
+
+        for servers in key_toml["server"]
+            .as_array()
+            .ok_or(generate_toml_parser_error_in_field("server"))?
+        {
+            let conection_str: String = servers["connection_str"]
+                .as_str()
+                .ok_or(generate_toml_parser_error_in_field("connection_str"))?
+                .to_string();
+            let key_name: String = servers["key"]["name"]
+                .as_str()
+                .ok_or(generate_toml_parser_error_in_field("key name"))?
+                .to_string();
+            let key: String = servers["key"]["value"]
+                .as_str()
+                .ok_or(generate_toml_parser_error_in_field("key value"))?
+                .to_string();
             ret.push(Self {
                 conection_str,
                 key_name,
                 key,
             });
         }
-        ret
+        Ok(ret)
     }
 }
 
 impl ServerZonesConnection {
-    pub fn get_config(file_name: &String) -> Self {
-        let key_toml = read_to_string(file_name).unwrap().parse::<Table>().unwrap();
-        let root_domain: String = key_toml["zone"]["root_domain"]
-            .as_str()
-            .unwrap()
-            .to_string();
+    pub fn get_config(file_name: &String) -> Result<Self, Box<dyn Error>> {
+        let root_domain: String =
+            read_to_string(file_name)?.parse::<Table>()?["zone"]["root_domain"]
+                .as_str()
+                .ok_or(generate_toml_parser_error_in_field("zone root_domain"))?
+                .to_string();
 
-        Self {
+        Ok(Self {
             root_domain,
-            servers_data: ZoneConnectionData::get_config(file_name),
-        }
+            servers_data: ZoneConnectionData::get_config(file_name)?,
+        })
     }
 }
+
+fn generate_toml_parser_error_in_field(field: &str) -> io::Error {
+    io::Error::new(
+        io::ErrorKind::InvalidData,
+        format!("Invalid format for config.toml in {} field", field),
+    )
+}