aboutsummaryrefslogtreecommitdiff
path: root/minecraft-chat/src/text_component.rs
blob: a5030fa12e1bab7caf54ed959e545fc39f462d7b (plain)
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
use std::fmt;

use crate::{
    base_component::BaseComponent,
    component::Component,
    style::{ChatFormatting, Style, TextColor},
};

#[derive(Clone, Debug)]
pub struct TextComponent {
    pub base: BaseComponent,
    pub text: String,
}

const LEGACY_FORMATTING_CODE_SYMBOL: char = '§';

/// Convert a legacy color code string into a Component
/// Technically in Minecraft this is done when displaying the text, but AFAIK it's the same as just doing it in TextComponent
pub fn legacy_color_code_to_text_component(legacy_color_code: &str) -> TextComponent {
    let mut components: Vec<TextComponent> = Vec::with_capacity(1);
    // iterate over legacy_color_code, if it starts with LEGACY_COLOR_CODE_SYMBOL then read the next character and get the style from that
    // otherwise, add the character to the text

    // we don't use a normal for loop since we need to be able to skip after reading the formatter code symbol
    let mut i = 0;
    while i < legacy_color_code.chars().count() {
        if legacy_color_code.chars().nth(i).unwrap() == LEGACY_FORMATTING_CODE_SYMBOL {
            let formatting_code = legacy_color_code.chars().nth(i + 1).unwrap();
            if let Ok(formatter) = ChatFormatting::from_code(formatting_code) {
                if components.is_empty() {
                    components.push(TextComponent::new("".to_string()));
                } else if !components.last().unwrap().text.is_empty() {
                    components.push(TextComponent::new("".to_string()));
                }

                let style = &mut components.last_mut().unwrap().base.style;
                // if the formatter is a reset, then we need to reset the style to the default
                style.apply_formatting(formatter);
            }
            i += 1;
        } else {
            if components.is_empty() {
                components.push(TextComponent::new("".to_string()));
            }
            components
                .last_mut()
                .unwrap()
                .text
                .push(legacy_color_code.chars().nth(i).unwrap());
        };
        i += 1;
    }

    // create the final component by using the first one as the base, and then adding the rest as siblings
    let mut final_component = components.remove(0);
    for component in components {
        final_component.base.siblings.push(component.get());
    }

    final_component
}

impl<'a> TextComponent {
    pub fn new(text: String) -> Self {
        // if it contains a LEGACY_FORMATTING_CODE_SYMBOL, format it
        if text.contains(LEGACY_FORMATTING_CODE_SYMBOL) {
            legacy_color_code_to_text_component(&text)
        } else {
            Self {
                base: BaseComponent::new(),
                text,
            }
        }
    }

    fn get(self) -> Component {
        Component::Text(self)
    }
}

impl fmt::Display for TextComponent {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}", self.text.clone())
    }
}

#[cfg(test)]
mod tests {
    use crate::style::Ansi;

    use super::*;

    #[test]
    fn test_hypixel_motd() {
        let component =
            TextComponent::new("§aHypixel Network  §c[1.8-1.18]\n§b§lHAPPY HOLIDAYS".to_string())
                .get();
        assert_eq!(
            component.to_ansi(None),
            format!(
                "{GREEN}Hypixel Network  {RED}[1.8-1.18]\n{BOLD}{AQUA}HAPPY HOLIDAYS{RESET}",
                GREEN = Ansi::rgb(ChatFormatting::GREEN.color.unwrap()),
                RED = Ansi::rgb(ChatFormatting::RED.color.unwrap()),
                AQUA = Ansi::rgb(ChatFormatting::AQUA.color.unwrap()),
                BOLD = Ansi::BOLD,
                RESET = Ansi::RESET
            )
        );
    }

    #[test]
    fn test_legacy_color_code_to_component() {
        let component = TextComponent::new("§lHello §r§1w§2o§3r§4l§5d".to_string()).get();
        assert_eq!(
            component.to_ansi(None),
            format!(
                "{BOLD}Hello {RESET}{DARK_BLUE}w{DARK_GREEN}o{DARK_AQUA}r{DARK_RED}l{DARK_PURPLE}d{RESET}",
                BOLD = Ansi::BOLD,
                RESET = Ansi::RESET,
                DARK_BLUE = Ansi::rgb(ChatFormatting::DARK_BLUE.color.unwrap()),
                DARK_GREEN = Ansi::rgb(ChatFormatting::DARK_GREEN.color.unwrap()),
                DARK_AQUA = Ansi::rgb(ChatFormatting::DARK_AQUA.color.unwrap()),
                DARK_RED = Ansi::rgb(ChatFormatting::DARK_RED.color.unwrap()),
                DARK_PURPLE = Ansi::rgb(ChatFormatting::DARK_PURPLE.color.unwrap())
            )
        );
    }
}