nil-client 0.5.6

// Copyright (C) Call of Nil contributors
// SPDX-License-Identifier: AGPL-3.0-only

use crate::error::Result;
use serde::{Deserialize, Serialize};
use std::env;
use std::ffi::{OsStr, OsString};
use std::net::SocketAddrV4;
use std::sync::{Arc, LazyLock};
use strum::EnumIs;
use tap::Pipe;
use url::Url;

const URL: &str = "https://tsukilabs.dev.br/nil/";

static REMOTE_SERVER_ADDR: LazyLock<Url> = LazyLock::new(|| {
  env::var("NIL_REMOTE_SERVER_ADDR")
    .unwrap_or_else(|_| URL.to_owned())
    .pipe_deref(Url::parse)
    .expect("Failed to parse remote server address")
});

#[derive(Clone, Copy, Debug, Default, EnumIs, PartialEq, Eq, Hash, Deserialize, Serialize)]
#[serde(tag = "kind", rename_all = "kebab-case")]
pub enum ServerAddr {
  #[default]
  Remote,
  Local {
    addr: SocketAddrV4,
  },
}

impl ServerAddr {
  #[inline]
  pub fn url(&self, route: &str) -> Result<Url> {
    match self {
      Self::Remote => Ok(REMOTE_SERVER_ADDR.join(route)?),
      Self::Local { addr } => {
        let ip = addr.ip();
        let port = addr.port();
        Ok(Url::parse(&format!("http://{ip}:{port}/{route}"))?)
      }
    }
  }
}

impl From<SocketAddrV4> for ServerAddr {
  fn from(addr: SocketAddrV4) -> Self {
    Self::Local { addr }
  }
}

impl From<&[u8]> for ServerAddr {
  fn from(bytes: &[u8]) -> Self {
    if let Ok(addr) = SocketAddrV4::parse_ascii(bytes) {
      Self::Local { addr }
    } else {
      Self::Remote
    }
  }
}

impl From<&OsStr> for ServerAddr {
  fn from(value: &OsStr) -> Self {
    Self::from(value.as_encoded_bytes())
  }
}

impl From<OsString> for ServerAddr {
  fn from(value: OsString) -> Self {
    Self::from(value.as_os_str())
  }
}

macro_rules! from_bytes {
  ($($type_:ty),+ $(,)?) => {
    $(
      impl From<$type_> for ServerAddr {
        fn from(value: $type_) -> Self {
          Self::from(value.as_bytes())
        }
      }
    )+
  };
}

from_bytes!(&str, String, &String, Arc<str>, Box<str>);