orinfar 0.0.6

A Text Editor For Witches
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

use crate::{c::c_node_to_color, markdown::md_node_to_color};
use crossterm::style::Color;

pub struct OrinLanguage {
    pub extensions: Vec<String>,
    pub highlight: fn(&str, &str, &str) -> Option<Color>,
    pub lang: tree_sitter::Language,
}

impl OrinLanguage {
    pub fn new<'a>(
        extensions: &'a [&'a str],
        lang: tree_sitter::Language,
        highlight: fn(&str, &str, &str) -> Option<Color>,
    ) -> Self {
        Self {
            extensions: extensions.iter().map(|s| (*s).to_string()).collect(),
            lang,
            highlight,
        }
    }

    pub fn from_ext(extension: &str) -> Option<Self> {
        let lang = match extension {
            "c" | "h" => Self::new(&["c", "h"], tree_sitter_c::LANGUAGE.into(), c_node_to_color),
            "md" => Self::new(&["md"], tree_sitter_md::LANGUAGE.into(), md_node_to_color),
            _ => {
                return None;
            }
        };

        Some(lang)
    }
}