erg/compiler/erg_compiler/check.rs

use erg_common::config::ErgConfig;
use erg_common::error::MultiErrorDisplay;
use erg_common::traits::Runnable;

use erg_parser::ast::AST;
use erg_parser::builder::ASTBuilder;

use crate::context::Context;
use crate::effectcheck::SideEffectChecker;
use crate::error::{TyCheckError, TyCheckErrors};
use crate::hir::HIR;
use crate::lower::ASTLowerer;
use crate::mod_cache::SharedModuleCache;
use crate::ownercheck::OwnershipChecker;

/// Summarize lowering, side-effect checking, and ownership checking
#[derive(Debug)]
pub struct Checker {
    pub cfg: ErgConfig,
    lowerer: ASTLowerer,
    ownership_checker: OwnershipChecker,
}

impl Checker {
    pub fn new(cfg: ErgConfig, mod_cache: SharedModuleCache) -> Self {
        Self {
            cfg,
            lowerer: ASTLowerer::new(mod_cache),
            ownership_checker: OwnershipChecker::new(),
        }
    }

    pub fn check(&mut self, ast: AST, mode: &str) -> Result<(HIR, Context), TyCheckErrors> {
        let (hir, ctx, warns) = self.lowerer.lower(ast, mode)?;
        if self.cfg.verbose >= 2 {
            warns.fmt_all_stderr();
        }
        let effect_checker = SideEffectChecker::new();
        let hir = effect_checker.check(hir)?;
        let hir = self.ownership_checker.check(hir)?;
        Ok((hir, ctx))
    }
}

pub struct CheckerRunner {
    checker: Checker,
}

impl Runnable for CheckerRunner {
    type Err = TyCheckError;
    type Errs = TyCheckErrors;
    const NAME: &'static str = "Erg type-checker";

    fn new(cfg: ErgConfig) -> Self {
        Self {
            checker: Checker::new(cfg, SharedModuleCache::new()),
        }
    }

    #[inline]
    fn cfg(&self) -> &ErgConfig {
        &self.checker.cfg
    }

    #[inline]
    fn finish(&mut self) {}

    fn clear(&mut self) {}

    fn exec(&mut self) -> Result<(), Self::Errs> {
        let mut builder = ASTBuilder::new(self.cfg().copy());
        let ast = builder.build()?;
        let (hir, _) = self.checker.check(ast, "exec")?;
        println!("{hir}");
        Ok(())
    }

    fn eval(&mut self, src: String) -> Result<String, TyCheckErrors> {
        let mut builder = ASTBuilder::new(self.cfg().copy());
        let ast = builder.build_with_str(src)?;
        let (hir, _) = self.checker.check(ast, "eval")?;
        Ok(hir.to_string())
    }
}