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

extern crate hidapi;

use std::vec::Vec;

use hidapi::HidApi;
use hidapi::HidError;
use hidapi::HidResult;

const BLINKSTICK_VENDOR_ID: u16 = 0x20A0;
const BLINKSTICK_PRODUCT_ID: u16 = 0x41E5;

const BLINKSTICK_INDEXED_LED_MSG_PACKET_SIZE: usize = 6;

pub struct BlinkStickDevice {
    device: hidapi::HidDevice,
}

impl BlinkStickDevice {
    pub fn open_first() -> Result<BlinkStickDevice, HidError> {
        match HidApi::new() {
            Ok(api) => {
                // Connect to device using its VID and PID
                match api.open(BLINKSTICK_VENDOR_ID, BLINKSTICK_PRODUCT_ID) {
                    Ok(device) => Ok(BlinkStickDevice { device }),
                    Err(e) => return Err(e),
                }
            },
            Err(e) => return Err(e),
        }
    }

    pub fn open(serial: &str) -> Result<BlinkStickDevice, HidError> {
        match HidApi::new() {
            Ok(api) => {
                // Connect to device using its VID and PID
                match api.open_serial(BLINKSTICK_VENDOR_ID, BLINKSTICK_PRODUCT_ID, serial) {
                    Ok(device) => Ok(BlinkStickDevice { device }),
                    Err(e) => return Err(e),
                }
            },
            Err(e) => return Err(e),
        }
    }

    pub fn get_serials() -> Result<Vec<String>, HidError> {
        match HidApi::new() {
            Ok(api) => {
                let mut vec: Vec<String> = Vec::new();

                for device in api.devices() {
                    if device.product_id == BLINKSTICK_PRODUCT_ID && device.vendor_id == BLINKSTICK_VENDOR_ID {
                        if let Some(serial) = &device.serial_number {
                            vec.push(serial.to_owned());
                        }
                    }
                }

                return Ok(vec);
            },
            Err(e) => return Err(e),
        }
    }

    pub fn off(&self, channel: u8, index: u8) -> HidResult<()> {
        return self.set_color(channel, index, 0, 0, 0);
    }

    pub fn set_color(&self, channel: u8, index: u8, r: u8, g: u8, b: u8) -> HidResult<()> {
        let mut buf: [u8; BLINKSTICK_INDEXED_LED_MSG_PACKET_SIZE] =
            [0; BLINKSTICK_INDEXED_LED_MSG_PACKET_SIZE];

        buf[0] = 0x05;
        buf[1] = channel;
        buf[2] = index;
        buf[3] = r;
        buf[4] = g;
        buf[5] = b;

        return self.device.send_feature_report(&buf);
    }
}