oak-nix 0.0.11

Nix package manager language parser with support for declarative package configuration and reproducible builds.
Documentation 
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

/// Element types for the Nix language syntax tree.
pub mod element_type;

use crate::{
    language::NixLanguage,
    lexer::{NixLexer, token_type::NixTokenType},
};
use oak_core::{
    parser::{ParseCache, ParseOutput, Parser, ParserState, parse_with_lexer},
    source::{Source, TextEdit},
};

pub(crate) type State<'a, S> = ParserState<'a, NixLanguage, S>;

/// Parser for the Nix language.
pub struct NixParser<'a> {
    /// The language configuration for this parser.
    pub language: &'a NixLanguage,
}

impl<'a> NixParser<'a> {
    /// Creates a new `NixParser` with the given configuration.
    pub fn new(language: &'a NixLanguage) -> Self {
        Self { language }
    }

    fn parse_expr<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        match state.peek_kind() {
            Some(NixTokenType::LeftBrace) => self.parse_set(state),
            Some(NixTokenType::LeftBracket) => self.parse_list(state),
            Some(NixTokenType::Let) => self.parse_let_in(state),
            Some(NixTokenType::If) => self.parse_if_then_else(state),
            Some(NixTokenType::Identifier) => {
                state.bump();
                if state.at(NixTokenType::Colon) {
                    // This might be a lambda, but for now just bump
                    state.bump();
                    self.parse_expr(state)
                }
            }
            _ => state.bump(),
        }
    }

    fn parse_set<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        let cp = state.checkpoint();
        state.expect(NixTokenType::LeftBrace).ok();
        while state.not_at_end() && !state.at(NixTokenType::RightBrace) {
            self.parse_binding(state)
        }
        state.expect(NixTokenType::RightBrace).ok();
        state.finish_at(cp, crate::parser::element_type::NixElementType::Set);
    }

    fn parse_list<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        let cp = state.checkpoint();
        state.expect(NixTokenType::LeftBracket).ok();
        while state.not_at_end() && !state.at(NixTokenType::RightBracket) {
            self.parse_expr(state)
        }
        state.expect(NixTokenType::RightBracket).ok();
        state.finish_at(cp, crate::parser::element_type::NixElementType::List);
    }

    fn parse_let_in<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        let cp = state.checkpoint();
        state.expect(NixTokenType::Let).ok();
        while state.not_at_end() && !state.at(NixTokenType::In) {
            self.parse_binding(state)
        }
        if state.at(NixTokenType::In) {
            state.bump();
            self.parse_expr(state)
        }
        state.finish_at(cp, crate::parser::element_type::NixElementType::LetIn);
    }

    fn parse_if_then_else<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        let cp = state.checkpoint();
        state.expect(NixTokenType::If).ok();
        self.parse_expr(state);
        if state.at(NixTokenType::Then) {
            state.bump();
            self.parse_expr(state)
        }
        if state.at(NixTokenType::Else) {
            state.bump();
            self.parse_expr(state)
        }
        state.finish_at(cp, crate::parser::element_type::NixElementType::IfThenElse);
    }

    fn parse_binding<'b, S: Source + ?Sized>(&self, state: &mut State<'b, S>) {
        let cp = state.checkpoint();
        state.expect(NixTokenType::Identifier).ok();
        if state.at(NixTokenType::Assign) {
            state.bump();
            self.parse_expr(state);
            state.expect(NixTokenType::Semicolon).ok();
        }
        state.finish_at(cp, crate::parser::element_type::NixElementType::Binding);
    }
}

impl<'config> Parser<NixLanguage> for NixParser<'config> {
    fn parse<'a, S: Source + ?Sized>(&self, text: &'a S, edits: &[TextEdit], cache: &'a mut impl ParseCache<NixLanguage>) -> ParseOutput<'a, NixLanguage> {
        let lexer = NixLexer::new(self.language);
        parse_with_lexer(&lexer, text, edits, cache, |state| {
            let cp = state.checkpoint();

            while state.not_at_end() {
                self.parse_expr(state);
            }

            Ok(state.finish_at(cp, crate::parser::element_type::NixElementType::Root))
        })
    }
}