use serde::Serialize;
use crate::{
datatype::{DataFrame, DataPoint},
element::{ColorBy, CoordinateSystem},
};
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct Candlestick {
#[serde(rename = "type")]
type_: String,
#[serde(skip_serializing_if = "Option::is_none")]
id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
coordiate_system: Option<CoordinateSystem>,
#[serde(skip_serializing_if = "Option::is_none")]
color_by: Option<ColorBy>,
#[serde(skip_serializing_if = "Option::is_none")]
legend_hover_link: Option<bool>,
#[serde(skip_serializing_if = "Vec::is_empty")]
data: DataFrame,
}
impl Candlestick {
pub fn new() -> Self {
Self {
type_: "candlestick".to_string(),
id: None,
name: None,
coordiate_system: None,
color_by: None,
legend_hover_link: None,
data: vec![],
}
}
pub fn id<S: Into<String>>(mut self, id: S) -> Self {
self.id = Some(id.into());
self
}
pub fn name<S: Into<String>>(mut self, name: S) -> Self {
self.name = Some(name.into());
self
}
pub fn coordiate_system<C: Into<CoordinateSystem>>(mut self, coordiate_system: C) -> Self {
self.coordiate_system = Some(coordiate_system.into());
self
}
pub fn color_by(mut self, color_by: ColorBy) -> Self {
self.color_by = Some(color_by);
self
}
pub fn legend_hover_link(mut self, legend_hover_link: bool) -> Self {
self.legend_hover_link = Some(legend_hover_link);
self
}
pub fn data<D: Into<DataPoint>>(mut self, data: Vec<D>) -> Self {
self.data = data.into_iter().map(|d| d.into()).collect();
self
}
}