rofaxul/src/webhook.rs

/*!
Implementation of the Device trait for webhook connected lights.

*/

use crate::{Device, Pattern, SolidColor, Wave};
use reqwest::blocking::Client;

// ------------------------------------------------------------------------------------------------
// Public Types
// ------------------------------------------------------------------------------------------------

///
/// The device implementation for a webhook connected light.
///
#[derive(Clone, Debug)]
pub struct WebhookDevice {
    id: String,
}

// ------------------------------------------------------------------------------------------------
// Public Functions
// ------------------------------------------------------------------------------------------------

///
/// Return a device implementation for a webhook connected light.
///
pub fn new_device_for(device_id: &str) -> crate::error::Result<impl Device> {
    if !device_id.is_empty() && device_id.chars().all(|c| c.is_ascii_hexdigit()) {
        Ok(WebhookDevice {
            id: device_id.to_string(),
        })
    } else {
        Err(crate::error::ErrorKind::InvalidDeviceID.into())
    }
}

// ------------------------------------------------------------------------------------------------
// Private Types
// ------------------------------------------------------------------------------------------------

const API_V1: &str = "https://api.luxafor.com/webhook/v1/actions";

// ------------------------------------------------------------------------------------------------
// Implementations
// ------------------------------------------------------------------------------------------------

impl Device for WebhookDevice {
    fn id(&self) -> String {
        self.id.clone()
    }

    fn turn_off(&self) -> crate::error::Result<()> {
        self.set_solid_color(SolidColor::Custom {
            red: 00,
            green: 00,
            blue: 00,
        })
    }

    fn set_solid_color(&self, color: SolidColor) -> crate::error::Result<()> {
        info!("Setting the color of device '{}' to {}", self.id, color);

        let body = if let SolidColor::Custom {
            red: _,
            green: _,
            blue: _,
        } = color
        {
            r#"{
  "userId": "DID",
  "actionFields":{
    "color": "custom",
    "custom_color": "COLOR"
  }
}"#
            .replace("DID", &self.id.to_string())
            .replace("COLOR", &color.to_string())
        } else {
            r#"{
  "userId": "DID",
  "actionFields":{
    "color": "COLOR"
  }
}"#
            .replace("DID", &self.id.to_string())
            .replace("COLOR", &color.to_string())
        };

        let url = &format!("{}/{}", API_V1, "solid_color");

        send_request(url, body)
    }

    fn set_fade_to_color(
        &self,
        _color: SolidColor,
        _fade_duration: u8,
    ) -> crate::error::Result<()> {
        Err(crate::error::ErrorKind::UnsupportedCommand.into())
    }

    fn set_color_strobe(
        &self,
        color: SolidColor,
        _strobe_speed: u8,
        repeat_count: u8,
    ) -> crate::error::Result<()> {
        info!(
            "Setting the strobe color of device '{}' to {}",
            self.id, color
        );

        let body = if let SolidColor::Custom {
            red: _,
            green: _,
            blue: _,
        } = color
        {
            r#"{
  "userId": "DID",
  "actionFields":{
    "repeat": RPT,
    "color": "custom",
    "custom_color": "COLOR"
  }
}"#
            .replace("DID", &self.id.to_string())
            .replace("COLOR", &color.to_string())
            .replace("RPT", &repeat_count.to_string())
        } else {
            r#"{
  "userId": "DID",
  "actionFields":{
    "repeat": RPT,
    "color": "COLOR"
  }
}"#
            .replace("DID", &self.id.to_string())
            .replace("COLOR", &color.to_string())
            .replace("RPT", &repeat_count.to_string())
        };

        let url = &format!("{}/{}", API_V1, "blink");

        send_request(url, body)
    }

    fn set_color_wave(
        &self,
        _color: SolidColor,
        _wave_pattern: Wave,
        _wave_speed: u8,
        _repeat_count: u8,
    ) -> crate::error::Result<()> {
        Err(crate::error::ErrorKind::UnsupportedCommand.into())
    }

    fn set_pattern(&self, pattern: Pattern, repeat_count: u8) -> crate::error::Result<()> {
        info!("Setting the pattern of device '{}' to {}", self.id, pattern);
        warn!(
            "Ignoring repeat count {}, not supported in the webhook API",
            repeat_count
        );

        let body = r#"{
  "userId": "DID",
  "actionFields":{
    "pattern": "PATTERN"
  }
}"#
        .replace("DID", &self.id.to_string())
        .replace("PATTERN", &pattern.to_string());

        let url = &format!("{}/{}", API_V1, "pattern");

        send_request(url, body)
    }
}

// ------------------------------------------------------------------------------------------------
// Private Functions
// ------------------------------------------------------------------------------------------------

fn send_request(api: &str, body: String) -> crate::error::Result<()> {
    debug!("Sending to: {}", api);
    debug!("Sending data: {:?}", body);

    let client = Client::new();
    let result = client
        .post(api)
        .header("Content-Type", "application/json")
        .body(body)
        .send()?;

    if result.status().is_success() {
        info!("call successful");
        Ok(())
    } else {
        let status_code = result.status().as_u16();
        error!("call failed");
        error!("{:?}", result.text());
        Err(crate::error::ErrorKind::UnexpectedError(status_code).into())
    }
}