std-mel 0.10.1

use melodium_core::*;
use melodium_macro::{check, mel_function, mel_treatment};

/// Convert stream of chars into stream of strings.
#[mel_treatment(
    input chars Stream<char>
    output text Stream<string>
)]
pub async fn to_string() {
    while let Ok(chars) = chars
        .recv_many()
        .await
        .map(|values| TryInto::<Vec<char>>::try_into(values).unwrap())
    {
        check!(
            text.send_one(chars.into_iter().collect::<String>().into())
                .await
        );
    }
}

/// Convert stream of string into stream of chars.
#[mel_treatment(
    input text Stream<string>
    output chars Stream<char>
)]
pub async fn from_string() {
    while let Ok(text) = text
        .recv_many()
        .await
        .map(|values| TryInto::<Vec<string>>::try_into(values).unwrap())
    {
        let mut output = Vec::new();
        for text in text {
            output.extend(text.chars());
        }

        check!(chars.send_many(output.into()).await);
    }
}

/// Converts stream of chars into UTF-8 encoded stream of bytes.
#[mel_treatment(
    input text Stream<char>
    output encoded Stream<byte>
)]
pub async fn to_utf8() {
    while let Ok(text) = text
        .recv_many()
        .await
        .map(|values| TryInto::<Vec<char>>::try_into(values).unwrap())
    {
        let mut output = Vec::new();
        for text in text {
            output.extend(text.to_string().as_bytes());
        }

        check!(
            encoded
                .send_many(TransmissionValue::Byte(output.into()))
                .await
        );
    }
}

/// Convert char into UTF-8 encoded vector of bytes.
#[mel_function]
pub fn to_utf8(char: char) -> Vec<byte> {
    char.to_string().as_bytes().into()
}

/// Converts vector of bytes into vector of char according to UTF-8 encoding.
///
/// If any sequence of bytes doesn't follow UTF-8 encoding, it is replaced by the `U+FFFD REPLACEMENT CHARACTER` (�).
#[mel_function]
pub fn from_utf8(encoded: Vec<byte>) -> Vec<char> {
    String::from_utf8_lossy(&encoded).chars().collect()
}