95 lines
2.3 KiB
Rust
95 lines
2.3 KiB
Rust
// Copyright (c) 2022 Marceline Cramer
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#[macro_use]
|
|
extern crate pest_derive;
|
|
|
|
use console::Style;
|
|
|
|
pub mod lexer;
|
|
pub mod namevec;
|
|
pub mod parse;
|
|
pub mod symbols;
|
|
|
|
use lexer::Token;
|
|
|
|
pub struct ColorTheme {
|
|
normal: Style,
|
|
keyword: Style,
|
|
literal: Style,
|
|
comment: Style,
|
|
error: Style,
|
|
}
|
|
|
|
impl ColorTheme {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
normal: Style::default(),
|
|
keyword: Style::new().blue(),
|
|
literal: Style::new().cyan(),
|
|
comment: Style::new().white(),
|
|
error: Style::new().black().on_red().bold(),
|
|
}
|
|
}
|
|
|
|
pub fn token_style(&self, token: &Token) -> &Style {
|
|
use Token::*;
|
|
match token {
|
|
Struct | Function | For | If | Else | In | Let | Mut => &self.keyword,
|
|
BinaryInteger | OctalInteger | DecimalInteger => &self.literal,
|
|
SingleLineComment => &self.comment,
|
|
Error => &self.error,
|
|
_ => &self.normal,
|
|
}
|
|
}
|
|
}
|
|
|
|
mod pest_parser {
|
|
#[derive(Parser)]
|
|
#[grammar = "sprite.pest"]
|
|
pub struct SpriteParser;
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn lex_file() {
|
|
let source = include_str!("test/clock.fae");
|
|
let theme = ColorTheme::new();
|
|
let mut lex = lexer::Lexer::new(source);
|
|
|
|
while let Some(token) = lex.next() {
|
|
let style = theme.token_style(&token);
|
|
let styled = style.apply_to(format!("{:?}: {}", token, lex.slice()));
|
|
println!("{}", styled);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_pest() {
|
|
use pest::Parser;
|
|
use pest_parser::{Rule, SpriteParser};
|
|
let source = include_str!("test/clock.fae");
|
|
let parsed = SpriteParser::parse(Rule::module, source);
|
|
match parsed {
|
|
Ok(ok) => println!("{:#?}", ok),
|
|
Err(err) => panic!("{}", err),
|
|
}
|
|
}
|
|
|
|
// TODO use spans to color-code instead of raw tokens, to show original whitespace
|
|
/*#[test]
|
|
fn color_file() {
|
|
let source = include_str!("test/example.fae");
|
|
let theme = ColorTheme::new();
|
|
let mut lex = Token::lexer(source);
|
|
|
|
while let Some(token) = lex.next() {
|
|
let style = theme.token_style(&token);
|
|
print!("{}", style.apply_to(lex.slice()));
|
|
}
|
|
}*/
|
|
}
|