forky_cli 0.2.0

Rust Utilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use anyhow::Result;
use colorize::AnsiColor;
use forky_core::prelude::*;
use hyper::Uri;
use std::fmt::Display;

#[derive(Debug, Clone, Copy)]
pub struct Address {
	pub host: [u8; 4],
	pub port: u16,
	pub port_tls: u16,
	pub secure: bool,
}

impl Address {
	pub fn to_uri(&self) -> Result<Uri> {
		let uri = self.to_string().parse()?;
		Ok(uri)
	}

	pub fn host_from_str(host: &str) -> anyhow::Result<[u8; 4]> {
		let mut parts = host.split('.');
		let v1 = parts.next().ok()?.parse::<u8>()?;
		let v2 = parts.next().ok()?.parse::<u8>()?;
		let v3 = parts.next().ok()?.parse::<u8>()?;
		let v4 = parts.next().ok()?.parse::<u8>()?;
		Ok([v1, v2, v3, v4])
	}
	pub fn is_all(&self) -> bool {
		self.host[0] == 0
			&& self.host[1] == 0
			&& self.host[2] == 0
			&& self.host[3] == 0
	}
	pub fn to_socket_addr(&self) -> std::net::SocketAddr {
		self.socket_addr(self.port)
	}
	pub fn to_socket_addr_tls(&self) -> std::net::SocketAddr {
		self.socket_addr(self.port_tls)
	}
	fn socket_addr(&self, port: u16) -> std::net::SocketAddr {
		std::net::SocketAddr::new(
			std::net::IpAddr::V4(std::net::Ipv4Addr::new(
				self.host[0],
				self.host[1],
				self.host[2],
				self.host[3],
			)),
			port,
		)
	}
	pub fn to_string_pretty(&self) -> String {
		self.to_string().b_cyan().underlined().bold()
	}
}

impl Default for Address {
	fn default() -> Self {
		Self {
			host: [0, 0, 0, 0],
			port: 7777,
			port_tls: 7778,
			secure: false,
		}
	}
}
impl Display for Address {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		let host = if self.is_all() {
			&[127, 0, 0, 1]
		} else {
			&self.host
		};
		let (prefix, port) = if self.secure {
			("https://", self.port_tls)
		} else {
			("http://", self.port)
		};

		let str = format!(
			"{}{}.{}.{}.{}:{}",
			prefix, host[0], host[1], host[2], host[3], port
		);
		write!(f, "{str}")
	}
}