1 files changed, 33 insertions, 0 deletions

diff --git a/src/add.rs b/src/add.rs
new file mode 100644
index 0000000..8ccf53b
--- /dev/null
+++ b/src/add.rs
@@ -0,0 +1,33 @@
+use std::{net::Ipv4Addr, str::FromStr, fs::File};
+use std::io::Write;
+use anyhow::{anyhow, Result};
+use crate::wg::{config::WireguardConfig, peer::Peer};
+
+pub fn run(wg_conf: &mut WireguardConfig, wg_conf_path: String, peer_name: &String, ip: Option<&String>) -> Result<()> {
+    let ip = match ip {
+        Some(s) => {
+            Ipv4Addr::from_str(s.as_str())?
+        },
+        None => {
+            match wg_conf.next_free_ip() {
+                Ok(i) => i,
+                Err(e) => {
+                    return Err(e);
+                }
+            }
+        }
+    };
+
+    match wg_conf.get_peer(peer_name.as_str()) {
+        Some(_) => { return Err(anyhow!("There is already a peer named {}", peer_name)); },
+        None => {}
+    }
+
+    let p = Peer::new(peer_name.clone(), ip)?;
+    wg_conf.peers.push(p);
+
+    let mut f = File::create(wg_conf_path)?;
+    write!(f, "{}", wg_conf.gen_config()?)?;
+
+    Ok(())
+}
\ No newline at end of file