86 lines
2.6 KiB
Rust
86 lines
2.6 KiB
Rust
//! This module defines Concrete Syntax Tree (CST), used by rust-analyzer.
|
|
//!
|
|
//! The CST includes comments and whitespace, provides a single node type,
|
|
//! `SyntaxNode`, and a basic traversal API (parent, children, siblings).
|
|
//!
|
|
//! The *real* implementation is in the (language-agnostic) `rowan` crate, this
|
|
//! modules just wraps its API.
|
|
|
|
use ra_parser::ParseError;
|
|
use rowan::{GreenNodeBuilder, Language};
|
|
|
|
use crate::{
|
|
syntax_error::{SyntaxError, SyntaxErrorKind},
|
|
Parse, SmolStr, SyntaxKind, TextUnit,
|
|
};
|
|
|
|
pub(crate) use rowan::{GreenNode, GreenToken};
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
|
|
pub enum RustLanguage {}
|
|
impl Language for RustLanguage {
|
|
type Kind = SyntaxKind;
|
|
|
|
fn kind_from_raw(raw: rowan::SyntaxKind) -> SyntaxKind {
|
|
SyntaxKind::from(raw.0)
|
|
}
|
|
|
|
fn kind_to_raw(kind: SyntaxKind) -> rowan::SyntaxKind {
|
|
rowan::SyntaxKind(kind.into())
|
|
}
|
|
}
|
|
|
|
pub type SyntaxNode = rowan::SyntaxNode<RustLanguage>;
|
|
pub type SyntaxToken = rowan::SyntaxToken<RustLanguage>;
|
|
pub type SyntaxElement = rowan::SyntaxElement<RustLanguage>;
|
|
pub type SyntaxNodeChildren = rowan::SyntaxNodeChildren<RustLanguage>;
|
|
pub type SyntaxElementChildren = rowan::SyntaxElementChildren<RustLanguage>;
|
|
|
|
pub use rowan::{Direction, NodeOrToken};
|
|
|
|
pub struct SyntaxTreeBuilder {
|
|
errors: Vec<SyntaxError>,
|
|
inner: GreenNodeBuilder<'static>,
|
|
}
|
|
|
|
impl Default for SyntaxTreeBuilder {
|
|
fn default() -> SyntaxTreeBuilder {
|
|
SyntaxTreeBuilder { errors: Vec::new(), inner: GreenNodeBuilder::new() }
|
|
}
|
|
}
|
|
|
|
impl SyntaxTreeBuilder {
|
|
pub(crate) fn finish_raw(self) -> (GreenNode, Vec<SyntaxError>) {
|
|
let green = self.inner.finish();
|
|
(green, self.errors)
|
|
}
|
|
|
|
pub fn finish(self) -> Parse<SyntaxNode> {
|
|
let (green, errors) = self.finish_raw();
|
|
let node = SyntaxNode::new_root(green);
|
|
if cfg!(debug_assertions) {
|
|
crate::validation::validate_block_structure(&node);
|
|
}
|
|
Parse::new(node.green().clone(), errors)
|
|
}
|
|
|
|
pub fn token(&mut self, kind: SyntaxKind, text: SmolStr) {
|
|
let kind = RustLanguage::kind_to_raw(kind);
|
|
self.inner.token(kind, text)
|
|
}
|
|
|
|
pub fn start_node(&mut self, kind: SyntaxKind) {
|
|
let kind = RustLanguage::kind_to_raw(kind);
|
|
self.inner.start_node(kind)
|
|
}
|
|
|
|
pub fn finish_node(&mut self) {
|
|
self.inner.finish_node()
|
|
}
|
|
|
|
pub fn error(&mut self, error: ParseError, text_pos: TextUnit) {
|
|
let error = SyntaxError::new(SyntaxErrorKind::ParseError(error), text_pos);
|
|
self.errors.push(error)
|
|
}
|
|
}
|