ik-rs 0.3.6

use crate::core::char_util::utf8_len;
use crate::core::char_util::CharType;
use crate::core::lexeme::{Lexeme, LexemeType};
use crate::core::ordered_linked_list::OrderedLinkedList;
use crate::core::segmentor::Segmenter;

const SEGMENTER_NAME: &str = "LETTER_SEGMENTER";

const LETTER_CONNECTOR: [char; 7] = ['#', '&', '+', '-', '.', '@', '_'];

const NUM_CONNECTOR: [char; 2] = [',', '.'];

#[derive(Debug)]
pub struct LetterSegmenter {
    start: Option<usize>,
    end: Option<usize>,

    english_start: Option<usize>,
    english_end: Option<usize>,

    arabic_start: Option<usize>,
    arabic_end: Option<usize>,
}

impl Segmenter for LetterSegmenter {
    fn analyze(
        &mut self,
        input: &str,
        cursor: usize,
        curr_char_type: &CharType,
        origin_lexemes: &mut OrderedLinkedList<Lexeme>,
    ) {
        self.process_english_letter(input, cursor, curr_char_type, origin_lexemes);
        self.process_arabic_letter(input, cursor, curr_char_type, origin_lexemes);
        self.process_mix_letter(input, cursor, curr_char_type, origin_lexemes);
    }
    fn name(&self) -> &str {
        return SEGMENTER_NAME;
    }
}

impl Default for LetterSegmenter {
    fn default() -> Self {
        LetterSegmenter {
            start: None,
            end: None,
            english_start: None,
            english_end: None,
            arabic_start: None,
            arabic_end: None,
        }
    }
}
impl LetterSegmenter {
    /// mix letter
    /// windows2000 | zhiyi.shen@gmail.com
    fn process_mix_letter(
        &mut self,
        input: &str,
        cursor: usize,
        curr_char_type: &CharType,
        origin_lexemes: &mut OrderedLinkedList<Lexeme>,
    ) {
        let curr_char = input.chars().nth(cursor).unwrap();
        let char_count = utf8_len(input);
        match self.start {
            None => match curr_char_type {
                CharType::ARABIC | CharType::ENGLISH => {
                    self.start = Some(cursor);
                    self.end = Some(cursor);
                }
                _ => {}
            },
            Some(start) => match curr_char_type {
                CharType::ARABIC | CharType::ENGLISH => {
                    self.end = Some(cursor);
                }
                CharType::USELESS if self.is_letter_connector(&curr_char) => {
                    self.end = Some(cursor);
                }
                _ => {
                    let new_lexeme =
                        Lexeme::new(start..(self.end.unwrap() + 1), LexemeType::LETTER);
                    origin_lexemes.insert(new_lexeme);
                    self.reset_mix_state();
                }
            },
        }
        self.end.zip(self.start).map(|(end, start)| {
            if end == (char_count - 1) {
                let new_lexeme = Lexeme::new(start..end + 1, LexemeType::LETTER);
                origin_lexemes.insert(new_lexeme);
                self.reset_mix_state();
            }
        });
    }

    // english
    fn process_english_letter(
        &mut self,
        input: &str,
        cursor: usize,
        curr_char_type: &CharType,
        origin_lexemes: &mut OrderedLinkedList<Lexeme>,
    ) {
        let char_count = utf8_len(input);
        match self.english_start {
            None => match curr_char_type {
                CharType::ENGLISH => {
                    self.english_start = Some(cursor);
                    self.english_end = Some(cursor);
                }
                _ => {}
            },
            Some(start) => match curr_char_type {
                CharType::ENGLISH => {
                    self.english_end = Some(cursor);
                }
                _ => {
                    let new_lexeme =
                        Lexeme::new(start..(self.english_end.unwrap() + 1), LexemeType::ENGLISH);
                    origin_lexemes.insert(new_lexeme);
                    self.reset_english_state();
                }
            },
        }
        self.english_end
            .zip(self.english_start)
            .map(|(end, start)| {
                if end == (char_count - 1) {
                    let new_lexeme = Lexeme::new(start..end + 1, LexemeType::ENGLISH);
                    origin_lexemes.insert(new_lexeme);
                    self.reset_english_state();
                }
            });
    }

    // arabic
    fn process_arabic_letter(
        &mut self,
        input: &str,
        cursor: usize,
        curr_char_type: &CharType,
        origin_lexemes: &mut OrderedLinkedList<Lexeme>,
    ) {
        let curr_char = input.chars().nth(cursor).unwrap();
        match self.arabic_start {
            None => match curr_char_type {
                CharType::ARABIC => {
                    self.arabic_start = Some(cursor);
                    self.arabic_end = Some(cursor);
                }
                _ => {}
            },
            Some(start) => match curr_char_type {
                CharType::ARABIC => {
                    self.arabic_end = Some(cursor);
                }
                CharType::USELESS if self.is_num_connector(&curr_char) => {}
                _ => {
                    let new_lexeme =
                        Lexeme::new(start..(self.arabic_end.unwrap() + 1), LexemeType::ARABIC);
                    origin_lexemes.insert(new_lexeme);
                    self.reset_arabic_state();
                }
            },
        }
        let char_count = utf8_len(input);
        self.arabic_end.zip(self.arabic_start).map(|(end, start)| {
            if end == (char_count - 1) {
                let new_lexeme = Lexeme::new(start..end + 1, LexemeType::ARABIC);
                origin_lexemes.insert(new_lexeme);
                self.reset_arabic_state();
            }
        });
    }
    fn reset_mix_state(&mut self) {
        self.start = None;
        self.end = None;
    }

    fn reset_english_state(&mut self) {
        self.english_start = None;
        self.english_end = None;
    }

    fn reset_arabic_state(&mut self) {
        self.arabic_start = None;
        self.arabic_end = None;
    }

    fn is_letter_connector(&self, input: &char) -> bool {
        LETTER_CONNECTOR.contains(input)
    }

    fn is_num_connector(&self, input: &char) -> bool {
        NUM_CONNECTOR.contains(input)
    }
}