treehouse/crates/treehouse-format/src/lib.rs

use std::ops::Range;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum BranchKind {
    /// Expanded by default.
    Expanded,
    /// Folded by default.
    Collapsed,
}

impl BranchKind {
    pub fn char(&self) -> char {
        match self {
            BranchKind::Expanded => '-',
            BranchKind::Collapsed => '+',
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Branch {
    pub indent_level: usize,
    pub config: Range<usize>,
    pub kind: BranchKind,
    pub content: Range<usize>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Parser<'a> {
    pub input: &'a str,
    pub position: usize,
}

#[derive(Debug, thiserror::Error, Clone, PartialEq, Eq)]
pub enum ParseErrorKind {
    #[error("branch kind (`+` or `-`) expected")]
    BranchKindExpected,
}

#[derive(Debug, Clone, PartialEq, Eq, thiserror::Error)]
#[error("{range:?}: {kind}")]
pub struct ParseError {
    pub kind: ParseErrorKind,
    pub range: Range<usize>,
}

impl<'a> Parser<'a> {
    fn current(&self) -> Option<char> {
        self.input[self.position..].chars().next()
    }

    fn advance(&mut self) {
        self.position += self.current().map(|c| c.len_utf8()).unwrap_or(0);
    }

    fn eat_as_long_as(&mut self, c: char) -> usize {
        let mut count = 0;
        while self.current() == Some(c) {
            count += 1;
            self.advance();
        }
        count
    }

    fn eat_until(&mut self, c: char) {
        while self.current() != Some(c) {
            self.advance();
        }
        self.advance();
    }

    pub fn next_branch(&mut self) -> Result<Option<Branch>, ParseError> {
        if self.current().is_none() {
            return Ok(None);
        }

        let indent_level = self.eat_as_long_as(' ');

        // TODO: Configs
        let config_start = self.position;
        let config_end = self.position;

        let branch_kind_start = self.position;
        let branch_kind = match self.current() {
            Some('-') => BranchKind::Expanded,
            Some('+') => BranchKind::Collapsed,
            _ => {
                return Err(ParseError {
                    kind: ParseErrorKind::BranchKindExpected,
                    range: branch_kind_start..branch_kind_start + 1,
                })
            }
        };
        self.advance();

        let content_start = self.position;
        loop {
            self.eat_until('\n');
            if let Some('\n') | None = self.current() {
                self.advance();
                break;
            }
        }
        let content_end = self.position;

        Ok(Some(Branch {
            indent_level,
            config: config_start..config_end,
            kind: branch_kind,
            content: content_start..content_end,
        }))
    }
}