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

use std::{convert::TryFrom, io};

use crate::ast::verified as ast;

mod c;
mod rust;

#[derive(Debug, Clone, Copy)]
pub enum Language {
    C,
    Rust,
}

pub(super) trait LanguageGenerator {
    fn generate<W: io::Write>(writer: &mut W, ast: &ast::Ast) -> io::Result<()>;
}

impl TryFrom<&str> for Language {
    type Error = String;
    fn try_from(value: &str) -> Result<Self, Self::Error> {
        match value {
            "c" => Ok(Language::C),
            "rust" => Ok(Language::Rust),
            lang => Err(format!("unsupport language: [{}]", lang)),
        }
    }
}

impl Language {
    pub(crate) fn extension(&self) -> &str {
        match *self {
            Language::C => "h",
            Language::Rust => "rs",
        }
    }

    pub(crate) fn generate<W: io::Write>(self, writer: &mut W, ast: &ast::Ast) -> io::Result<()> {
        match self {
            Language::C => c::Generator::generate(writer, ast),
            Language::Rust => rust::Generator::generate(writer, ast),
        }
    }
}