rust_minecraft_server/src/server/data/text_component.rs

use std::io::Read;

use palette::{Hsl, IntoColor, Srgb};
use rust_mc_proto::{DataReader, Packet};
use serde::{Deserialize, Serialize};
use serde_with::skip_serializing_none;

use crate::server::ServerError;




#[derive(Debug, Serialize, Deserialize, Clone)]
#[skip_serializing_none]
pub struct TextComponent {
    pub text: String,
    pub color: Option<String>,
    pub bold: Option<bool>,
    pub italic: Option<bool>,
    pub underlined: Option<bool>,
    pub strikethrough: Option<bool>,
    pub obfuscated: Option<bool>,
    pub extra: Option<Vec<TextComponent>>,
}

impl TextComponent {
    pub fn new(text: String) -> Self {
        Self {
            text,
            color: None,
            bold: None,
            italic: None,
            underlined: None,
            strikethrough: None,
            obfuscated: None,
            extra: None
        }
    }

    pub fn rainbow(text: String) -> TextComponent {
        if text.is_empty() {
            return TextComponent::new(text);
        }

        let children = text.char_indices()
            .map(|(i, c)| {
                let hue = (i as f32) / (text.chars().count() as f32) * 360.0;
                let hsl = Hsl::new(hue, 1.0, 0.5);
                let rgb: Srgb = hsl.into_color();
                let r = (rgb.red * 255.0).round() as u8;
                let g = (rgb.green * 255.0).round() as u8;
                let b = (rgb.blue * 255.0).round() as u8;
                let mut component = TextComponent::new(c.to_string());
                component.color = Some(format!("#{:02X}{:02X}{:02X}", r, g, b));
                component
            })
            .collect::<Vec<TextComponent>>();
        
        let mut parent = children[0].clone();
        parent.extra = Some(children[1..].to_vec());
        parent
    }

    pub fn builder() -> TextComponentBuilder {
        TextComponentBuilder::new()
    }

    pub fn as_json(self) -> Result<String, ServerError> {
        serde_json::to_string(&self)
            .map_err(|_| ServerError::SerTextComponent)
    }

    pub fn from_json(text: &str) -> Result<TextComponent, ServerError> {
        serde_json::from_str(text)
            .map_err(|_| ServerError::DeTextComponent)
    }
}

impl Default for TextComponent {
    fn default() -> Self {
        Self::new(String::new())
    }
}

pub struct TextComponentBuilder {
    text: String,
    color: Option<String>,
    bold: Option<bool>,
    italic: Option<bool>,
    underlined: Option<bool>,
    strikethrough: Option<bool>,
    obfuscated: Option<bool>,
    extra: Option<Vec<TextComponent>>,
}

impl TextComponentBuilder {
    pub fn new() -> Self {
        Self {
            text: String::new(),
            color: None,
            bold: None,
            italic: None,
            underlined: None,
            strikethrough: None,
            obfuscated: None,
            extra: None,
        }
    }

    pub fn text(mut self, text: &str) -> Self {
        self.text = text.to_string();
        self
    }

    pub fn color(mut self, color: &str) -> Self {
        self.color = Some(color.to_string());
        self
    }

    pub fn bold(mut self, bold: bool) -> Self {
        self.bold = Some(bold);
        self
    }

    pub fn italic(mut self, italic: bool) -> Self {
        self.italic = Some(italic);
        self
    }

    pub fn underlined(mut self, underlined: bool) -> Self {
        self.underlined = Some(underlined);
        self
    }

    pub fn strikethrough(mut self, strikethrough: bool) -> Self {
        self.strikethrough = Some(strikethrough);
        self
    }

    pub fn obfuscated(mut self, obfuscated: bool) -> Self {
        self.obfuscated = Some(obfuscated);
        self
    }

    pub fn extra(mut self, extra: Vec<TextComponent>) -> Self {
        self.extra = Some(extra);
        self
    }

    pub fn build(self) -> TextComponent {
        TextComponent { 
            text: self.text, 
            color: self.color, 
            bold: self.bold, 
            italic: self.italic, 
            underlined: self.underlined, 
            strikethrough: self.strikethrough, 
            obfuscated: self.obfuscated, 
            extra: self.extra
        }
    }
}

pub trait ReadWriteNBT<T>: DataReader {
    fn read_nbt(&mut self) -> Result<T, ServerError>;
    fn write_nbt(&mut self, val: &T) -> Result<(), ServerError>;
}

impl ReadWriteNBT<TextComponent> for Packet {
    fn read_nbt(&mut self) -> Result<TextComponent, ServerError> {
        let mut data = Vec::new();
        let pos = self.get_ref().position();
        self.get_mut().read_to_end(&mut data).map_err(|_| ServerError::DeTextComponent)?;
        let (remaining, value) = craftflow_nbt::from_slice(&data).map_err(|_| ServerError::DeTextComponent)?;
        self.get_mut().set_position(pos + (data.len() - remaining.len()) as u64);
        Ok(value)
    }
    
    fn write_nbt(&mut self, val: &TextComponent) -> Result<(), ServerError> {
        craftflow_nbt::to_writer(self.get_mut(), val)
            .map_err(|_| ServerError::SerTextComponent)?;
        Ok(())
    }
}