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
use crate::error::ErrorKind;
use crate::mesparse::{CommandType, Message, MessageParser};
use crate::{Payload, Result};
use log::{debug, info};
use std::io::prelude::*;
use std::net::{IpAddr, Shutdown, SocketAddr, TcpStream};
use std::time::Duration;
pub struct TuyaDevice {
mp: MessageParser,
addr: SocketAddr,
}
impl TuyaDevice {
pub fn create(ver: &str, key: Option<&str>, addr: IpAddr) -> Result<TuyaDevice> {
let mp = MessageParser::create(ver, key)?;
Ok(TuyaDevice::create_with_mp(mp, addr))
}
pub fn create_with_mp(mp: MessageParser, addr: IpAddr) -> TuyaDevice {
TuyaDevice {
mp,
addr: SocketAddr::new(addr, 6668),
}
}
pub fn set(&self, tuya_payload: Payload, seq_id: u32) -> Result<()> {
let mes = Message::new(tuya_payload, CommandType::Control, Some(seq_id));
let replies = self.send(&mes, seq_id)?;
replies
.iter()
.for_each(|mes| info!("Decoded response ({}):\n{}", seq_id, mes));
Ok(())
}
pub fn get(&self, tuya_payload: Payload, seq_id: u32) -> Result<Vec<Message>> {
let mes = Message::new(tuya_payload, CommandType::DpQuery, Some(seq_id));
let replies = self.send(&mes, seq_id)?;
replies
.iter()
.for_each(|mes| info!("Decoded response ({}):\n{}", seq_id, mes));
Ok(replies)
}
fn send(&self, mes: &Message, seq_id: u32) -> Result<Vec<Message>> {
let mut tcpstream = TcpStream::connect(&self.addr)?;
tcpstream.set_nodelay(true)?;
tcpstream.set_read_timeout(Some(Duration::new(2, 0)))?;
tcpstream.set_read_timeout(Some(Duration::new(2, 0)))?;
info!("Writing message to {} ({}):\n{}", self.addr, seq_id, &mes);
let bts = tcpstream.write(self.mp.encode(&mes, true)?.as_ref())?;
info!("Wrote {} bytes ({})", bts, seq_id);
let mut buf = [0; 256];
let bts = tcpstream.read(&mut buf)?;
info!("Received {} bytes ({})", bts, seq_id);
if bts == 0 {
return Err(ErrorKind::BadTcpRead);
} else {
debug!(
"Received response ({}):\n{}",
seq_id,
hex::encode(&buf[..bts])
);
}
debug!("Shutting down connection ({})", seq_id);
tcpstream.shutdown(Shutdown::Both)?;
self.mp.parse(&buf[..bts])
}
}