pylyzer/crates/py2erg/gen_decl.rs

use std::io::Write;

use erg_compiler::hir::{HIR, Expr};
use erg_compiler::ty::HasType;

pub struct DeclFile {
    pub filename: String,
    pub code: String,
}

fn escape_type(typ: String) -> String {
    typ.replace('%', "Type_")
}

pub fn gen_decl_er(hir: HIR) -> DeclFile {
    let mut code = "".to_string();
    for chunk in hir.module.into_iter() {
        match chunk {
            Expr::Def(def) => {
                let typ = def.sig.ident().ref_t().to_string();
                let typ = escape_type(typ);
                let decl = format!(".{}: {typ}", def.sig.ident().inspect());
                code += &decl;
            }
            Expr::ClassDef(def) => {
                let decl = format!(".{}: ClassType", def.sig.ident().inspect());
                code += &decl;
            }
            _ => {}
        }
        code.push('\n');
    }
    let filename = hir.name.replace(".py", ".d.er");
    DeclFile { filename, code }
}

pub fn dump_decl_er(hir: HIR) {
    let file = gen_decl_er(hir);
    if !std::path::Path::new("__pycache__").exists() {
        std::fs::create_dir("__pycache__").unwrap();
    }
    let f = std::fs::File::create(format!("__pycache__/{}", file.filename)).unwrap();
    let mut f = std::io::BufWriter::new(f);
    f.write_all(file.code.as_bytes()).unwrap();
}