//! IGD async API example.
//!
//! It demonstrates how to:
//! * get external IP
//! * add port mappings
//! * remove port mappings
//!
//! If everything works fine, 2 port mappings are added, 1 removed and we're left with single
//! port mapping: External 1234 ---> 4321 Internal

use std::{env, net::SocketAddrV4};

use igd_async_std::{aio::search_gateway, PortMappingProtocol};
use simplelog::{Config as LogConfig, LevelFilter, SimpleLogger};

#[async_std::main]
async fn main() {
  let ip = match env::args().nth(1) {
    Some(ip) => ip,
    None => {
      println!("Local socket address is missing!");
      println!("This example requires a socket address representing the local machine and the port to bind to as an argument");
      println!("Example: target/debug/examples/io 192.168.0.198:4321");
      println!("Example: cargo run --features aio --example aio -- 192.168.0.198:4321");
      return;
    }
  };
  let ip: SocketAddrV4 = ip.parse().expect("Invalid socket address");

  let _ = SimpleLogger::init(LevelFilter::Debug, LogConfig::default());

  let gateway = match search_gateway(Default::default()).await {
    Ok(g) => g,
    Err(err) => return println!("Faild to find IGD: {}", err),
  };
  let pub_ip = match gateway.get_external_ip().await {
    Ok(ip) => ip,
    Err(err) => return println!("Failed to get external IP: {}", err),
  };
  println!("Our public IP is {}", pub_ip);
  if let Err(e) = gateway
    .add_port(
      PortMappingProtocol::TCP,
      1234,
      ip,
      120,
      "rust-igd_async_std-async-example",
    )
    .await
  {
    println!("Failed to add port mapping: {}", e);
  }
  println!("New port mapping was successfully added.");

  if let Err(e) = gateway
    .add_port(
      PortMappingProtocol::TCP,
      2345,
      ip,
      120,
      "rust-igd_async_std-async-example",
    )
    .await
  {
    println!("Failed to add port mapping: {}", e);
  }
  println!("New port mapping was successfully added.");

  if gateway
    .remove_port(PortMappingProtocol::TCP, 2345)
    .await
    .is_err()
  {
    println!("Port mapping was not successfully removed");
  } else {
    println!("Port was removed.");
  }
}