luau-lexer 0.2.6

//! The [`State`] struct.

use smol_str::SmolStr;

use crate::{
    position::{Position, PositionComponent},
    token::Trivia,
};

/// A struct representing the state of a lexer at a specific time.
#[derive(Clone, Debug, Default, Hash, PartialEq, Eq, PartialOrd, Ord)]
#[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
pub struct State {
    /// The current character position.
    pub(crate) position: usize,

    /// The current byte position.
    pub(crate) byte_position: usize,

    /// The current [`position`](Position) in the file.
    pub(crate) lexer_position: Position,

    /// The spaces after the last parsed token.
    pub(crate) last_trivia: Vec<Trivia>,
}

impl State {
    /// Move the state by the passed character.
    pub fn increment_position_by_char(&mut self, character: char) {
        self.position += 1;
        self.byte_position += character.len_utf8();

        match character {
            '\n' => {
                self.lexer_position.character = 0;
                self.lexer_position.line += 1;
            }
            _ => self.lexer_position.character += 1,
        }
    }

    /// Move th state ahead by the passed amount of characters.
    pub fn increment_position(&mut self, amount: PositionComponent) {
        self.position += amount as usize;
        self.lexer_position.character += amount;
    }

    /// Get the current file [`position`](Position).
    #[inline]
    pub fn lexer_position(&self) -> Position {
        self.lexer_position
    }
}