rust_tuyapi/
tuyadevice.rs

1//! # TuyaDevice
2//! The TuyaDevice represents a communication channel with a Tuya compatible device. It
3//! encapsulates the device key, version and ip address. By supplying a Payload to either set() or
4//! get() functions the framework takes care of sending and receiving the reply from the device.
5//!
6//! The TuyaDevice is the high level device communication API. To get in to the nitty gritty
7//! details, create a MessageParser.
8use crate::mesparse::{CommandType, Message, MessageParser};
9use crate::transports::{Transport, TuyaTransport};
10use crate::{Payload, Result};
11use log::{debug, info};
12use std::net::{IpAddr, SocketAddr, TcpStream, UdpSocket};
13
14pub struct TuyaDevice {
15    mp: MessageParser,
16    addr: SocketAddr,
17    transport: Transport,
18}
19
20impl TuyaDevice {
21    pub fn create(ver: &str, key: Option<&str>, addr: IpAddr) -> Result<TuyaDevice> {
22        let mp = MessageParser::create(ver, key)?;
23        Ok(TuyaDevice::create_with_mp(mp, addr, Transport::TCP(6668)))
24    }
25
26    pub fn create_with_transport(
27        ver: &str,
28        key: Option<&str>,
29        addr: IpAddr,
30        transport: Transport,
31    ) -> Result<TuyaDevice> {
32        let mp = MessageParser::create(ver, key)?;
33        Ok(TuyaDevice::create_with_mp(mp, addr, transport))
34    }
35
36    pub fn create_with_mp(mp: MessageParser, addr: IpAddr, transport: Transport) -> TuyaDevice {
37        match transport {
38            Transport::TCP(port) | Transport::UDP(port) => TuyaDevice {
39                mp,
40                addr: SocketAddr::new(addr, port),
41                transport,
42            },
43        }
44    }
45
46    pub fn set(&self, tuya_payload: Payload, seq_id: u32) -> Result<()> {
47        let mes = Message::new(tuya_payload, CommandType::Control, Some(seq_id));
48        let replies = self.send(&mes, seq_id)?;
49        replies
50            .iter()
51            .for_each(|mes| info!("Decoded response ({}):\n{}", seq_id, mes));
52        Ok(())
53    }
54
55    pub fn get(&self, tuya_payload: Payload, seq_id: u32) -> Result<Vec<Message>> {
56        let mes = Message::new(tuya_payload, CommandType::DpQuery, Some(seq_id));
57        let replies = self.send(&mes, seq_id)?;
58        replies
59            .iter()
60            .for_each(|mes| info!("Decoded response ({}):\n{}", seq_id, mes));
61        Ok(replies)
62    }
63
64    pub fn refresh(&self, tuya_payload: Payload, seq_id: u32) -> Result<Vec<Message>> {
65        let mes = Message::new(tuya_payload, CommandType::DpRefresh, Some(seq_id));
66        let replies = self.send(&mes, seq_id)?;
67        replies
68            .iter()
69            .for_each(|mes| info!("Decoded response ({}):\n{}", seq_id, mes));
70        Ok(replies)
71    }
72
73    fn send(&self, mes: &Message, seq_id: u32) -> Result<Vec<Message>> {
74        let mut transport: Box<dyn TuyaTransport> = match self.transport {
75            Transport::TCP(_) => Box::new(TcpStream::connect(self.addr)?),
76            Transport::UDP(port) => Box::new(UdpSocket::bind(format!("0.0.0.0:{}", port))?),
77        };
78        transport.setup(self.addr)?;
79        info!("Writing message to {} ({}):\n{}", self.addr, seq_id, &mes);
80        let bts = transport.do_send(self.mp.encode(mes, true)?.as_ref())?;
81        info!("Wrote {} bytes ({})", bts, seq_id);
82        let mut buf = [0; 256];
83        let bts = transport.do_read(&mut buf)?;
84        info!("Received {} bytes ({})", bts, seq_id);
85        if bts == 0 {
86            return Err(transport.error());
87        } else {
88            debug!(
89                "Received response ({}):\n{}",
90                seq_id,
91                hex::encode(&buf[..bts])
92            );
93        }
94        debug!("Shutting down connection ({})", seq_id);
95        transport.teardown()?;
96        self.mp.parse(&buf[..bts])
97    }
98}
rust_tuyapi/tuyadevice.rs

rust_tuyapi/
tuyadevice.rs
