cnsprcy 0.3.2

WIP
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
87
88
89

use std::net::{
	SocketAddr,
	SocketAddrV4,
	SocketAddrV6,
	AddrParseError
};

use liter::column::{
	Affinity,
	Column
};
use rusqlite::Result as SqlResult;
use rusqlite::types::{
	FromSql,
	FromSqlResult,
	FromSqlError,
	ToSql,
	ToSqlOutput,
	ValueRef
};
use serde::{Serialize, Deserialize};

const DUMMY_PREFIX: &str = "dummy_addr/";

#[derive(Copy, Clone, Debug, Serialize, Deserialize, PartialEq, Eq, Hash)]
pub enum Address {
	V4UdpChaCha20(SocketAddrV4),
	V6UdpChaCha20(SocketAddrV6),
	Dummy(usize)
}

impl std::fmt::Display for Address {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		match &self {
			Self::V4UdpChaCha20(addr) =>
				write!(f, "IPv4-UDP-ChaCha20/{addr}"),
			Self::V6UdpChaCha20(addr) =>
				write!(f, "IPv6-UDP-ChaCha20/{addr}"),
			Self::Dummy(addr) => write!(f, "{DUMMY_PREFIX}{addr}")
		}
	}
}

impl std::str::FromStr for Address {
	type Err = AddrParseError;

	fn from_str(s: &str) -> Result<Self, Self::Err> {
		match SocketAddr::from_str(s) {
			Ok(SocketAddr::V4(addr)) => Ok(Self::V4UdpChaCha20(addr)),
			Ok(SocketAddr::V6(addr)) => Ok(Self::V6UdpChaCha20(addr)),
			Err(e) => Err(e)
		}
	}
}

/*	SQL */
impl Column for Address {
	const AFFINITY: Affinity = Affinity::Text;
}

impl FromSql for Address {
	fn column_result(value: ValueRef<'_>) -> FromSqlResult<Self> {
		let val_as_str = value.as_str()?;
		#[cfg(test)]
		if let Some(dummy_addr) =  val_as_str.strip_prefix(DUMMY_PREFIX) {
			return dummy_addr.parse()
				.map(Self::Dummy)
				.map_err(Into::into)
				.map_err(FromSqlError::Other)
		}

		val_as_str.parse()
			.map_err(Into::into)
			.map_err(FromSqlError::Other)
	}
}

impl ToSql for Address {
	fn to_sql(&self) -> SqlResult<ToSqlOutput<'_>> {
		let as_string = match &self {
			Self::V4UdpChaCha20(addr) => addr.to_string(),
			Self::V6UdpChaCha20(addr) => addr.to_string(),
			Self::Dummy(addr) => format!("{DUMMY_PREFIX}{addr}")
		};
		Ok(ToSqlOutput::Owned(as_string.into()))
	}
}

liter::types::impl_from_to_sql_2!(Address);