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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
use std::path::Path;
use std::path::PathBuf;
use std::str::FromStr;

use super::serial::SerialNumber;


pub const DEVICE_SERIAL_ENV: &str = "PLAYDATE_SERIAL_DEVICE";


/// Device query. Contains 4 options:
/// - None: query all devices
/// - Serial: query by serial number
/// - Path: query by path/name of serial port
/// - Com: query by COM port number (windows only)
#[derive(Clone)]
#[cfg_attr(feature = "clap", derive(clap::Parser))]
#[cfg_attr(feature = "clap", command(author, version, about, long_about = None, name = "device"))]
pub struct Query {
	/// Serial number of usb device or absolute path to socket.
	/// Format: 'PDUN-XNNNNNN'
	#[cfg_attr(unix, doc = "or '/dev/cu.usbmodemPDUN_XNNNNNN(N)'.")]
	#[cfg_attr(windows, doc = "or 'COM{X}', where {X} is a number of port, e.g.: COM3.")]
	#[cfg_attr(feature = "clap", arg(env = DEVICE_SERIAL_ENV, name = "device"))]
	pub value: Option<Value>,
}

impl Default for Query {
	fn default() -> Self {
		Self { value: std::env::var(DEVICE_SERIAL_ENV).map(|s| Value::from_str(&s).ok())
		                                              .ok()
		                                              .flatten() }
	}
}

impl std::fmt::Display for Query {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		match self.value {
			Some(ref value) => value.fmt(f),
			None => write!(f, "None"),
		}
	}
}

impl std::fmt::Debug for Query {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		match self.value.as_ref() {
			Some(value) => f.debug_tuple("Query").field(&value.to_string()).finish(),
			None => f.debug_tuple("Query").field(&None::<()>).finish(),
		}
	}
}

impl std::fmt::Display for Value {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		match self {
			Value::Serial(sn) => write!(f, "sn:{sn}"),
			Value::Path(path) => write!(f, "serial:{}", path.display()),
			Value::Com(port) => write!(f, "serial:COM{port}"),
		}
	}
}


#[derive(Clone, Debug)]
pub enum Value {
	/// Serial number of usb device.
	Serial(SerialNumber),
	/// Absolute path of serial/modem/telnet-socket.
	///
	/// In case of unmounting or installing it also can by mount-point.
	#[cfg_attr(not(unix), doc = "Use only on Unix.")]
	Path(PathBuf),
	/// COM port.
	#[cfg_attr(not(windows), doc = "Use only on Windows.")]
	Com(u16),
}

type ParseError = <SerialNumber as FromStr>::Err;
impl FromStr for Value {
	type Err = crate::error::Error;

	fn from_str(dev: &str) -> Result<Self, Self::Err> {
		let name = dev.trim();
		if name.is_empty() {
			return Err(ParseError::from(name).into());
		}

		#[cfg(windows)]
		match name.strip_prefix("COM").map(|s| s.parse::<u16>()) {
			Some(Ok(com)) => return Ok(Value::Com(com)),
			Some(Err(err)) => {
				use std::io::{Error, ErrorKind};

				return Err(Error::new(
					ErrorKind::InvalidInput,
					format!("Invalid format, seems to COM port, but {err}."),
				).into());
			},
			None => { /* nothing there */ },
		}

		let serial = SerialNumber::try_from(name);
		let path = Path::new(name);
		let is_direct = path.is_absolute() && path.exists();

		match serial {
			Ok(serial) => {
				if is_direct {
					Ok(Value::Path(path.to_owned()))
				} else {
					Ok(Value::Serial(serial))
				}
			},
			Err(err) => {
				if is_direct {
					Ok(Value::Path(path.to_owned()))
				} else {
					Err(err.into())
				}
			},
		}
	}
}

impl<'s> TryFrom<&'s str> for Value {
	type Error = crate::error::Error;
	fn try_from(dev: &'s str) -> Result<Self, Self::Error> { Self::from_str(dev) }
}

impl Value {
	pub fn to_printable_string(&self) -> String {
		match self {
			Self::Serial(sn) => sn.to_string(),
			Self::Path(p) => p.display().to_string(),
			Self::Com(n) => format!("COM{n}"),
		}
	}
}