tibrs 0.1.3

Provides functionality to compile and decompile tibasic source for TI graphing calculators
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

use anyhow::{Context, Result};
use itertools::Itertools;
use pest::{iterators::Pair, Parser};
use pest_derive::Parser;

use crate::{TibToken, TokenMap};

#[derive(Parser)]
#[grammar = "../grammar.peg"]
pub struct TibParser;

static RAW_TOKS: &str = include_str!("../toks.json");

pub fn pair_to_str<'lt>(p: &Pair<'lt, Rule>) -> &'lt str {
    match p.as_str() {
        "\n" => "@@NEWLINE",
        s => s,
    }
}

pub fn get_tok_map() -> TokenMap {
    serde_json::from_str(RAW_TOKS).unwrap()
}

pub fn parse_str(tib: &str) -> Result<Vec<TibToken>> {
    let preprocessed_source = tib.split('\n').map(|line| line.trim_start()).join("\n");
    let token_map = get_tok_map();
    let pairs = TibParser::parse(Rule::program, &preprocessed_source)?.next()
        .unwrap();

    let tokens: Result<Vec<TibToken>> = pairs
        .into_inner()
        .filter_map(|pair| {
            if let Rule::WHITESPACE | Rule::EOI = pair.as_rule() {
                return None; // skip
            }

            Some(
                token_map
                    .get(&*format!("{:?}", pair.as_rule()))
                    .with_context(|| format!("Failed to look up token: '{:?}'", pair.as_rule()))
                    .map(|borrowed| borrowed.clone()),
            )
        })
        .collect();

    tokens
}