use std::io::BufRead;
use std::option::Option;
use std::path::{Path, PathBuf};
use std::process::{Command, Stdio};
#[cfg(feature = "els")]
use erg_common::config::ErgMode;
use erg_common::env::erg_pystd_path;
use erg_common::levenshtein::get_similar_name;
use erg_common::python_util::BUILTIN_PYTHON_MODS;
use erg_common::set::Set;
use erg_common::traits::{Locational, Stream};
use erg_common::vis::Visibility;
use erg_common::{enum_unwrap, get_hash, log, set};
use erg_common::{fn_name, Str};
use ast::{Decorator, DefId, Identifier, OperationKind, SimpleTypeSpec, VarName};
use erg_parser::ast::{self, ConstIdentifier};
use crate::ty::constructors::{free_var, func, func0, func1, proc, ref_, ref_mut, v_enum};
use crate::ty::free::{Constraint, FreeKind, HasLevel};
use crate::ty::value::{GenTypeObj, TypeObj, ValueObj};
use crate::ty::{HasType, ParamTy, SubrType, Type};
use crate::build_hir::HIRBuilder;
use crate::context::{
ClassDefType, Context, ContextKind, DefaultInfo, MethodInfo, RegistrationMode, TraitImpl,
};
use crate::error::readable_name;
use crate::error::{
CompileResult, SingleTyCheckResult, TyCheckError, TyCheckErrors, TyCheckResult,
};
use crate::hir;
use crate::hir::Literal;
use crate::varinfo::{AbsLocation, Mutability, VarInfo, VarKind};
use Mutability::*;
use RegistrationMode::*;
use Visibility::*;
use super::instantiate::TyVarCache;
const UBAR: &Str = &Str::ever("_");
impl Context {
pub(crate) fn registered_info(
&self,
name: &str,
is_const: bool,
) -> Option<(&VarName, &VarInfo)> {
if let Some((name, vi)) = self.params.iter().find(|(maybe_name, _)| {
maybe_name
.as_ref()
.map(|n| &n.inspect()[..] == name)
.unwrap_or(false)
}) {
return Some((name.as_ref().unwrap(), vi));
} else if let Some((name, vi)) = self.locals.get_key_value(name) {
return Some((name, vi));
}
if is_const {
let outer = self.get_outer().or_else(|| self.get_builtins())?;
outer.registered_info(name, is_const)
} else {
None
}
}
fn pre_define_var(
&mut self,
sig: &ast::VarSignature,
opt_t: Option<Type>,
id: Option<DefId>,
) -> TyCheckResult<()> {
let muty = Mutability::from(&sig.inspect().unwrap_or(UBAR)[..]);
let ident = match &sig.pat {
ast::VarPattern::Ident(ident) => ident,
ast::VarPattern::Discard(_) => {
return Ok(());
}
other => unreachable!("{other}"),
};
let vis = ident.vis();
let kind = id.map_or(VarKind::Declared, VarKind::Defined);
let sig_t = self.instantiate_var_sig_t(sig.t_spec.as_ref(), opt_t, PreRegister)?;
let py_name = if let ContextKind::PatchMethodDefs(_base) = &self.kind {
Some(Str::from(format!("::{}{}", self.name, ident)))
} else {
None
};
if let Some(_decl) = self.decls.remove(&ident.name) {
Err(TyCheckErrors::from(TyCheckError::duplicate_decl_error(
self.cfg.input.clone(),
line!() as usize,
sig.loc(),
self.caused_by(),
ident.name.inspect(),
)))
} else {
let vi = VarInfo::new(
sig_t,
muty,
vis,
kind,
None,
self.impl_of(),
py_name,
self.absolutize(ident.name.loc()),
);
self.future_defined_locals.insert(ident.name.clone(), vi);
Ok(())
}
}
pub(crate) fn declare_sub(
&mut self,
sig: &ast::SubrSignature,
id: Option<DefId>,
) -> TyCheckResult<()> {
let name = sig.ident.inspect();
let vis = sig.ident.vis();
let muty = Mutability::from(&name[..]);
let kind = id.map_or(VarKind::Declared, VarKind::Defined);
let comptime_decos = sig
.decorators
.iter()
.filter_map(|deco| match &deco.0 {
ast::Expr::Accessor(ast::Accessor::Ident(local)) if local.is_const() => {
Some(local.inspect().clone())
}
_ => None,
})
.collect::<Set<_>>();
let default_ts =
vec![free_var(self.level, Constraint::new_type_of(Type::Type)); sig.params.len()];
let (errs, t) = match self.instantiate_sub_sig_t(sig, default_ts, PreRegister) {
Ok(t) => (TyCheckErrors::empty(), t),
Err((errs, t)) => (errs, t),
};
let py_name = if let ContextKind::PatchMethodDefs(_base) = &self.kind {
Some(Str::from(format!("::{}{}", self.name, sig.ident)))
} else {
None
};
let vi = VarInfo::new(
t,
muty,
vis,
kind,
Some(comptime_decos),
self.impl_of(),
py_name,
self.absolutize(sig.ident.name.loc()),
);
if let Some(_decl) = self.decls.remove(name) {
Err(TyCheckErrors::from(TyCheckError::duplicate_decl_error(
self.cfg.input.clone(),
line!() as usize,
sig.loc(),
self.caused_by(),
name,
)))
} else {
self.decls.insert(sig.ident.name.clone(), vi);
if errs.is_empty() {
Ok(())
} else {
Err(errs)
}
}
}
pub(crate) fn assign_var_sig(
&mut self,
sig: &ast::VarSignature,
body_t: &Type,
id: DefId,
py_name: Option<Str>,
) -> TyCheckResult<VarInfo> {
let ident = match &sig.pat {
ast::VarPattern::Ident(ident) => ident,
ast::VarPattern::Discard(_) => {
return Ok(VarInfo {
t: body_t.clone(),
..VarInfo::const_default()
});
}
_ => todo!(),
};
if sig.is_const() {
let vi = self.decls.remove(ident.inspect()).unwrap_or_else(|| {
VarInfo::new(
body_t.clone(),
Mutability::Const,
sig.vis(),
VarKind::Declared,
None,
self.impl_of(),
py_name,
self.absolutize(ident.name.loc()),
)
});
self.locals.insert(ident.name.clone(), vi.clone());
return Ok(vi);
}
let muty = Mutability::from(&ident.inspect()[..]);
let py_name = if let Some(vi) = self
.decls
.remove(ident.inspect())
.or_else(|| self.future_defined_locals.remove(ident.inspect()))
{
vi.py_name
} else {
py_name
};
let vis = ident.vis();
let kind = if id.0 == 0 {
VarKind::Declared
} else {
VarKind::Defined(id)
};
let vi = VarInfo::new(
body_t.clone(),
muty,
vis,
kind,
None,
self.impl_of(),
py_name,
self.absolutize(ident.name.loc()),
);
log!(info "Registered {}::{}: {}", self.name, ident.name, vi);
self.locals.insert(ident.name.clone(), vi.clone());
Ok(vi)
}
fn assign_param(
&mut self,
sig: &ast::NonDefaultParamSignature,
default_val_exists: bool,
opt_decl_t: Option<&ParamTy>,
) -> TyCheckResult<()> {
let vis = if cfg!(feature = "py_compatible") {
Public
} else {
Private
};
match &sig.pat {
ast::ParamPattern::Lit(_) => unreachable!(),
ast::ParamPattern::Discard(token) => {
let spec_t = self.instantiate_param_sig_t(
sig,
opt_decl_t,
&mut TyVarCache::new(self.level, self),
Normal,
)?;
let def_id = DefId(get_hash(&(&self.name, "_")));
let kind = VarKind::parameter(def_id, DefaultInfo::NonDefault);
let vi = VarInfo::new(
spec_t,
Immutable,
vis,
kind,
None,
None,
None,
self.absolutize(token.loc()),
);
self.params.push((Some(VarName::from_static("_")), vi));
Ok(())
}
ast::ParamPattern::VarName(name) => {
if self
.registered_info(name.inspect(), name.is_const())
.is_some()
{
Err(TyCheckErrors::from(TyCheckError::reassign_error(
self.cfg.input.clone(),
line!() as usize,
name.loc(),
self.caused_by(),
name.inspect(),
)))
} else {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let spec_t =
self.instantiate_param_sig_t(sig, opt_decl_t, &mut dummy_tv_cache, Normal)?;
if &name.inspect()[..] == "self" {
if let Some(self_t) = self.rec_get_self_t() {
self.sub_unify(&spec_t, &self_t, name.loc(), Some(name.inspect()))?;
} else {
log!(err "self_t is None");
}
}
let default = if default_val_exists {
DefaultInfo::WithDefault
} else {
DefaultInfo::NonDefault
};
let def_id = DefId(get_hash(&(&self.name, name)));
let kind = VarKind::parameter(def_id, default);
let muty = Mutability::from(&name.inspect()[..]);
let vi = VarInfo::new(
spec_t,
muty,
vis,
kind,
None,
None,
None,
self.absolutize(name.loc()),
);
self.params.push((Some(name.clone()), vi));
Ok(())
}
}
ast::ParamPattern::Ref(name) => {
if self
.registered_info(name.inspect(), name.is_const())
.is_some()
{
Err(TyCheckErrors::from(TyCheckError::reassign_error(
self.cfg.input.clone(),
line!() as usize,
name.loc(),
self.caused_by(),
name.inspect(),
)))
} else {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let spec_t =
self.instantiate_param_sig_t(sig, opt_decl_t, &mut dummy_tv_cache, Normal)?;
if &name.inspect()[..] == "self" {
if let Some(self_t) = self.rec_get_self_t() {
self.sub_unify(&spec_t, &self_t, name.loc(), Some(name.inspect()))?;
} else {
log!(err "self_t is None");
}
}
let spec_t = ref_(spec_t);
let default = if default_val_exists {
DefaultInfo::WithDefault
} else {
DefaultInfo::NonDefault
};
let kind = VarKind::parameter(DefId(get_hash(&(&self.name, name))), default);
let vi = VarInfo::new(
spec_t,
Immutable,
vis,
kind,
None,
None,
None,
self.absolutize(name.loc()),
);
self.params.push((Some(name.clone()), vi));
Ok(())
}
}
ast::ParamPattern::RefMut(name) => {
if self
.registered_info(name.inspect(), name.is_const())
.is_some()
{
Err(TyCheckErrors::from(TyCheckError::reassign_error(
self.cfg.input.clone(),
line!() as usize,
name.loc(),
self.caused_by(),
name.inspect(),
)))
} else {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let spec_t =
self.instantiate_param_sig_t(sig, opt_decl_t, &mut dummy_tv_cache, Normal)?;
if &name.inspect()[..] == "self" {
if let Some(self_t) = self.rec_get_self_t() {
self.sub_unify(&spec_t, &self_t, name.loc(), Some(name.inspect()))?;
} else {
log!(err "self_t is None");
}
}
let spec_t = ref_mut(spec_t.clone(), Some(spec_t));
let default = if default_val_exists {
DefaultInfo::WithDefault
} else {
DefaultInfo::NonDefault
};
let kind = VarKind::parameter(DefId(get_hash(&(&self.name, name))), default);
let vi = VarInfo::new(
spec_t,
Immutable,
vis,
kind,
None,
None,
None,
self.absolutize(name.loc()),
);
self.params.push((Some(name.clone()), vi));
Ok(())
}
}
other => {
log!(err "{other}");
unreachable!("{other}")
}
}
}
pub(crate) fn assign_params(
&mut self,
params: &hir::Params,
opt_decl_subr_t: Option<SubrType>,
) -> TyCheckResult<()> {
let mut errs = TyCheckErrors::empty();
if let Some(decl_subr_t) = opt_decl_subr_t {
debug_assert_eq!(
params.non_defaults.len(),
decl_subr_t.non_default_params.len()
);
debug_assert_eq!(params.defaults.len(), decl_subr_t.default_params.len());
for (sig, pt) in params
.non_defaults
.iter()
.zip(decl_subr_t.non_default_params.iter())
{
if let Err(es) = self.assign_param(sig, false, Some(pt)) {
errs.extend(es);
}
}
for (sig, pt) in params
.defaults
.iter()
.zip(decl_subr_t.default_params.iter())
{
if let Err(es) = self.assign_param(&sig.sig, true, Some(pt)) {
errs.extend(es);
}
}
} else {
for sig in params.non_defaults.iter() {
if let Err(es) = self.assign_param(sig, false, None) {
errs.extend(es);
}
}
for sig in params.defaults.iter() {
if let Err(es) = self.assign_param(&sig.sig, true, None) {
errs.extend(es);
}
}
}
if errs.is_empty() {
Ok(())
} else {
Err(errs)
}
}
pub(crate) fn assign_subr(
&mut self,
sig: &ast::SubrSignature,
id: DefId,
body_t: &Type,
) -> TyCheckResult<VarInfo> {
if sig.ident.is_const() {
let vi = self.decls.remove(sig.ident.inspect()).unwrap();
self.locals.insert(sig.ident.name.clone(), vi.clone());
return Ok(vi);
}
let muty = if sig.ident.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let name = &sig.ident.name;
let t = self.get_current_scope_var(name).map(|v| &v.t).unwrap();
let non_default_params = t.non_default_params().unwrap();
let var_args = t.var_params();
let default_params = t.default_params().unwrap();
let mut errs = if let Some(spec_ret_t) = t.return_t() {
let return_t_loc = sig
.return_t_spec
.as_ref()
.map(|t_spec| t_spec.loc())
.unwrap_or_else(|| sig.loc());
self.sub_unify(body_t, spec_ret_t, return_t_loc, None)
.map_err(|errs| {
TyCheckErrors::new(
errs.into_iter()
.map(|e| {
TyCheckError::return_type_error(
self.cfg.input.clone(),
line!() as usize,
e.core.get_loc_with_fallback(),
e.caused_by,
readable_name(name.inspect()),
spec_ret_t,
body_t,
)
})
.collect(),
)
})
} else {
Ok(())
};
let return_t = if errs.is_err() {
Type::Failure
} else {
t.return_t().unwrap().clone()
};
let sub_t = if sig.ident.is_procedural() {
proc(
non_default_params.clone(),
var_args.cloned(),
default_params.clone(),
return_t,
)
} else {
func(
non_default_params.clone(),
var_args.cloned(),
default_params.clone(),
return_t,
)
};
sub_t.lift();
let found_t = self.generalize_t(sub_t);
let py_name = if let Some(vi) = self.decls.remove(name) {
if !self.supertype_of(&vi.t, &found_t) {
let err = TyCheckError::violate_decl_error(
self.cfg.input.clone(),
line!() as usize,
sig.ident.loc(),
self.caused_by(),
name.inspect(),
&vi.t,
&found_t,
);
match errs {
Ok(()) => {
errs = Err(TyCheckErrors::from(err));
}
Err(ref mut es) => {
es.push(err);
}
}
}
vi.py_name
} else {
None
};
let comptime_decos = sig
.decorators
.iter()
.filter_map(|deco| match &deco.0 {
ast::Expr::Accessor(ast::Accessor::Ident(local)) if local.is_const() => {
Some(local.inspect().clone())
}
_ => None,
})
.collect();
let vi = VarInfo::new(
found_t,
muty,
sig.ident.vis(),
VarKind::Defined(id),
Some(comptime_decos),
self.impl_of(),
py_name,
self.absolutize(name.loc()),
);
log!(info "Registered {}::{name}: {}", self.name, &vi.t);
self.locals.insert(name.clone(), vi.clone());
errs?;
Ok(vi)
}
pub(crate) fn fake_subr_assign(
&mut self,
ident: &Identifier,
decorators: &Set<Decorator>,
failure_t: Type,
) -> TyCheckResult<()> {
if ident.is_const() {
let Some(vi) = self.decls.remove(ident.inspect()) else {
return Err(TyCheckErrors::from(TyCheckError::unreachable(
self.cfg.input.clone(),
fn_name!(),
line!(),
)));
};
self.locals.insert(ident.name.clone(), vi);
}
let muty = if ident.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let name = &ident.name;
self.decls.remove(name);
let comptime_decos = decorators
.iter()
.filter_map(|deco| match &deco.0 {
ast::Expr::Accessor(ast::Accessor::Ident(local)) if local.is_const() => {
Some(local.inspect().clone())
}
_ => None,
})
.collect();
let vi = VarInfo::new(
failure_t,
muty,
ident.vis(),
VarKind::DoesNotExist,
Some(comptime_decos),
self.impl_of(),
None,
self.absolutize(name.loc()),
);
log!(info "Registered {}::{name}: {}", self.name, &vi.t);
self.locals.insert(name.clone(), vi);
Ok(())
}
pub(crate) fn preregister(&mut self, block: &ast::Block) -> TyCheckResult<()> {
let mut total_errs = TyCheckErrors::empty();
for expr in block.iter() {
match expr {
ast::Expr::Def(def) => {
if let Err(errs) = self.preregister_def(def) {
total_errs.extend(errs.into_iter());
}
}
ast::Expr::ClassDef(class_def) => {
if let Err(errs) = self.preregister_def(&class_def.def) {
total_errs.extend(errs.into_iter());
}
}
ast::Expr::PatchDef(patch_def) => {
if let Err(errs) = self.preregister_def(&patch_def.def) {
total_errs.extend(errs.into_iter());
}
}
_ => {}
}
}
if total_errs.is_empty() {
Ok(())
} else {
Err(total_errs)
}
}
pub(crate) fn preregister_def(&mut self, def: &ast::Def) -> TyCheckResult<()> {
let id = Some(def.body.id);
let __name__ = def.sig.ident().map(|i| i.inspect()).unwrap_or(UBAR);
match &def.sig {
ast::Signature::Subr(sig) => {
if sig.is_const() {
let tv_cache = self.instantiate_ty_bounds(&sig.bounds, PreRegister)?;
let vis = def.sig.vis();
self.grow(__name__, ContextKind::Proc, vis, Some(tv_cache));
let (obj, const_t) = match self.eval_const_block(&def.body.block) {
Ok(obj) => (obj.clone(), v_enum(set! {obj})),
Err(errs) => {
self.pop();
return Err(errs);
}
};
if let Some(spec) = sig.return_t_spec.as_ref() {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let spec_t = self
.instantiate_typespec(
spec,
None,
&mut dummy_tv_cache,
PreRegister,
false,
)
.map_err(|errs| {
self.pop();
errs
})?;
self.sub_unify(&const_t, &spec_t, def.body.loc(), None)
.map_err(|errs| {
self.pop();
errs
})?;
}
self.pop();
self.register_gen_const(def.sig.ident().unwrap(), obj)?;
} else {
self.declare_sub(sig, id)?;
}
}
ast::Signature::Var(sig) => {
if sig.is_const() {
let kind = ContextKind::from(def.def_kind());
self.grow(__name__, kind, sig.vis(), None);
let (obj, const_t) = match self.eval_const_block(&def.body.block) {
Ok(obj) => (obj.clone(), v_enum(set! {obj})),
Err(errs) => {
self.pop();
return Err(errs);
}
};
if let Some(spec) = sig.t_spec.as_ref() {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let spec_t = self
.instantiate_typespec(
spec,
None,
&mut dummy_tv_cache,
PreRegister,
false,
)
.map_err(|errs| {
self.pop();
errs
})?;
self.sub_unify(&const_t, &spec_t, def.body.loc(), None)
.map_err(|errs| {
self.pop();
errs
})?;
}
self.pop();
if let Some(ident) = sig.ident() {
self.register_gen_const(ident, obj)?;
}
} else {
let opt_t = self
.eval_const_block(&def.body.block)
.map(|o| v_enum(set! {o}))
.ok();
self.pre_define_var(sig, opt_t, id)?;
}
}
}
Ok(())
}
fn register_auto_impl(
&mut self,
name: &'static str,
t: Type,
muty: Mutability,
vis: Visibility,
py_name: Option<Str>,
) {
let name = VarName::from_static(name);
if self.locals.get(&name).is_some() {
panic!("already registered: {name}");
} else {
let vi = VarInfo::new(
t,
muty,
vis,
VarKind::Auto,
None,
self.impl_of(),
py_name,
AbsLocation::unknown(),
);
self.locals.insert(name, vi);
}
}
fn register_fixed_auto_impl(
&mut self,
name: &'static str,
t: Type,
muty: Mutability,
vis: Visibility,
py_name: Option<Str>,
) {
let name = VarName::from_static(name);
if self.locals.get(&name).is_some() {
panic!("already registered: {name}");
} else {
self.locals.insert(
name,
VarInfo::new(
t,
muty,
vis,
VarKind::FixedAuto,
None,
self.impl_of(),
py_name,
AbsLocation::unknown(),
),
);
}
}
fn _register_gen_decl(
&mut self,
name: VarName,
t: Type,
vis: Visibility,
impl_of: Option<Type>,
py_name: Option<Str>,
) {
if self.decls.get(&name).is_some() {
panic!("already registered: {name}");
} else {
let vi = VarInfo::new(
t,
Immutable,
vis,
VarKind::Declared,
None,
impl_of,
py_name,
self.absolutize(name.loc()),
);
self.decls.insert(name, vi);
}
}
fn _register_gen_impl(
&mut self,
name: VarName,
t: Type,
muty: Mutability,
vis: Visibility,
impl_of: Option<Type>,
py_name: Option<Str>,
) {
if self.locals.get(&name).is_some() {
panic!("already registered: {name}");
} else {
let id = DefId(get_hash(&(&self.name, &name)));
let vi = VarInfo::new(
t,
muty,
vis,
VarKind::Defined(id),
None,
impl_of,
py_name,
self.absolutize(name.loc()),
);
self.locals.insert(name, vi);
}
}
pub(crate) fn register_trait(&mut self, class: Type, methods: Self) {
let trait_ = if let ContextKind::MethodDefs(Some(tr)) = &methods.kind {
tr.clone()
} else {
todo!()
};
self.super_traits.push(trait_.clone());
self.methods_list
.push((ClassDefType::impl_trait(class, trait_), methods));
}
pub(crate) fn register_marker_trait(&mut self, trait_: Type) {
self.super_traits.push(trait_);
}
pub(crate) fn register_gen_const(
&mut self,
ident: &Identifier,
obj: ValueObj,
) -> SingleTyCheckResult<()> {
if self.rec_get_const_obj(ident.inspect()).is_some() && ident.vis().is_private() {
Err(TyCheckError::reassign_error(
self.cfg.input.clone(),
line!() as usize,
ident.loc(),
self.caused_by(),
ident.inspect(),
))
} else {
match obj {
ValueObj::Type(t) => match t {
TypeObj::Generated(gen) => {
self.register_gen_type(ident, gen);
}
TypeObj::Builtin(t) => {
self.register_type_alias(ident, t);
}
},
other => {
let id = DefId(get_hash(ident));
let vi = VarInfo::new(
v_enum(set! {other.clone()}),
Const,
ident.vis(),
VarKind::Defined(id),
None,
self.impl_of(),
None,
AbsLocation::unknown(),
);
self.decls.insert(ident.name.clone(), vi);
self.consts.insert(ident.name.clone(), other);
}
}
Ok(())
}
}
pub(crate) fn register_gen_type(&mut self, ident: &Identifier, gen: GenTypeObj) {
match gen {
GenTypeObj::Class(_) => {
if gen.typ().is_monomorphic() {
let mut ctx = Self::mono_class(
gen.typ().qual_name(),
self.cfg.clone(),
self.shared.clone(),
2,
self.level,
);
let mut methods =
Self::methods(None, self.cfg.clone(), self.shared.clone(), 2, self.level);
let new_t = if let Some(base) = gen.base_or_sup() {
match base {
TypeObj::Builtin(Type::Record(_)) => {}
other => {
methods.register_fixed_auto_impl(
"base",
other.typ().clone(),
Immutable,
Private,
None,
);
}
}
func1(base.typ().clone(), gen.typ().clone())
} else {
func0(gen.typ().clone())
};
methods.register_fixed_auto_impl(
"__new__",
new_t.clone(),
Immutable,
Private,
Some("__call__".into()),
);
methods.register_auto_impl("new", new_t, Immutable, Public, None);
ctx.methods_list
.push((ClassDefType::Simple(gen.typ().clone()), methods));
self.register_gen_mono_type(ident, gen, ctx, Const);
} else {
todo!("polymorphic type definition is not supported yet");
}
}
GenTypeObj::Subclass(_) => {
if gen.typ().is_monomorphic() {
let super_classes = vec![gen.base_or_sup().unwrap().typ().clone()];
let mut ctx = Self::mono_class(
gen.typ().qual_name(),
self.cfg.clone(),
self.shared.clone(),
2,
self.level,
);
for sup in super_classes.into_iter() {
let (_, sup_ctx) = self
.get_nominal_type_ctx(&sup)
.unwrap_or_else(|| todo!("{sup} not found"));
ctx.register_superclass(sup, sup_ctx);
}
let mut methods =
Self::methods(None, self.cfg.clone(), self.shared.clone(), 2, self.level);
if let Some(sup) =
self.rec_get_const_obj(&gen.base_or_sup().unwrap().typ().local_name())
{
let sup = enum_unwrap!(sup, ValueObj::Type);
let param_t = match sup {
TypeObj::Builtin(t) => t,
TypeObj::Generated(t) => t.base_or_sup().unwrap().typ(),
};
let param_t = if let Some(additional) = gen.additional() {
self.intersection(param_t, additional.typ())
} else {
param_t.clone()
};
let new_t = func1(param_t, gen.typ().clone());
methods.register_fixed_auto_impl(
"__new__",
new_t.clone(),
Immutable,
Private,
Some("__call__".into()),
);
methods.register_auto_impl("new", new_t, Immutable, Public, None);
ctx.methods_list
.push((ClassDefType::Simple(gen.typ().clone()), methods));
self.register_gen_mono_type(ident, gen, ctx, Const);
} else {
todo!("super class not found")
}
} else {
todo!("polymorphic type definition is not supported yet");
}
}
GenTypeObj::Trait(_) => {
if gen.typ().is_monomorphic() {
let mut ctx = Self::mono_trait(
gen.typ().qual_name(),
self.cfg.clone(),
self.shared.clone(),
2,
self.level,
);
let req = enum_unwrap!(gen.base_or_sup().unwrap(), TypeObj::Builtin:(Type::Record:(_)));
for (field, t) in req.iter() {
let muty = if field.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let vi = VarInfo::new(
t.clone(),
muty,
field.vis,
VarKind::Declared,
None,
self.impl_of(),
None,
AbsLocation::unknown(),
);
ctx.decls
.insert(VarName::from_str(field.symbol.clone()), vi);
}
self.register_gen_mono_type(ident, gen, ctx, Const);
} else {
todo!("polymorphic type definition is not supported yet");
}
}
GenTypeObj::Subtrait(_) => {
if gen.typ().is_monomorphic() {
let super_classes = vec![gen.base_or_sup().unwrap().typ().clone()];
let mut ctx = Self::mono_trait(
gen.typ().qual_name(),
self.cfg.clone(),
self.shared.clone(),
2,
self.level,
);
let additional = gen.additional().map(
|additional| enum_unwrap!(additional, TypeObj::Builtin:(Type::Record:(_))),
);
if let Some(additional) = additional {
for (field, t) in additional.iter() {
let muty = if field.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let vi = VarInfo::new(
t.clone(),
muty,
field.vis,
VarKind::Declared,
None,
self.impl_of(),
None,
AbsLocation::unknown(),
);
ctx.decls
.insert(VarName::from_str(field.symbol.clone()), vi);
}
}
for sup in super_classes.into_iter() {
if let Some((_, sup_ctx)) = self.get_nominal_type_ctx(&sup) {
ctx.register_supertrait(sup, sup_ctx);
} else {
log!(err "{sup} not found");
}
}
self.register_gen_mono_type(ident, gen, ctx, Const);
} else {
todo!("polymorphic type definition is not supported yet");
}
}
GenTypeObj::Patch(_) => {
if gen.typ().is_monomorphic() {
let base = enum_unwrap!(gen.base_or_sup().unwrap(), TypeObj::Builtin);
let ctx = Self::mono_patch(
gen.typ().qual_name(),
base.clone(),
self.cfg.clone(),
self.shared.clone(),
2,
self.level,
);
self.register_gen_mono_patch(ident, gen, ctx, Const);
} else {
todo!("polymorphic patch definition is not supported yet");
}
}
other => todo!("{other:?}"),
}
}
pub(crate) fn register_type_alias(&mut self, ident: &Identifier, t: Type) {
if self.mono_types.contains_key(ident.inspect()) {
panic!("{ident} has already been registered");
} else if self.rec_get_const_obj(ident.inspect()).is_some() && ident.vis().is_private() {
panic!("{ident} has already been registered as const");
} else {
let name = &ident.name;
let muty = Mutability::from(&ident.inspect()[..]);
let id = DefId(get_hash(&(&self.name, &name)));
self.decls.insert(
name.clone(),
VarInfo::new(
Type::Type,
muty,
ident.vis(),
VarKind::Defined(id),
None,
self.impl_of(),
None,
self.absolutize(name.loc()),
),
);
self.consts
.insert(name.clone(), ValueObj::Type(TypeObj::Builtin(t)));
}
}
fn register_gen_mono_type(
&mut self,
ident: &Identifier,
gen: GenTypeObj,
ctx: Self,
muty: Mutability,
) {
if self.mono_types.contains_key(ident.inspect()) {
panic!("{ident} has already been registered");
} else if self.rec_get_const_obj(ident.inspect()).is_some() && ident.vis().is_private() {
panic!("{ident} has already been registered as const");
} else {
let t = gen.typ().clone();
let meta_t = gen.meta_type();
let name = &ident.name;
let id = DefId(get_hash(&(&self.name, &name)));
self.decls.insert(
name.clone(),
VarInfo::new(
meta_t,
muty,
ident.vis(),
VarKind::Defined(id),
None,
self.impl_of(),
None,
self.absolutize(name.loc()),
),
);
self.consts
.insert(name.clone(), ValueObj::Type(TypeObj::Generated(gen)));
for impl_trait in ctx.super_traits.iter() {
if let Some(impls) = self.trait_impls.get_mut(&impl_trait.qual_name()) {
impls.insert(TraitImpl::new(t.clone(), impl_trait.clone()));
} else {
self.trait_impls.insert(
impl_trait.qual_name(),
set![TraitImpl::new(t.clone(), impl_trait.clone())],
);
}
}
for (trait_method, vi) in ctx.decls.iter() {
if let Some(types) = self.method_to_traits.get_mut(trait_method.inspect()) {
types.push(MethodInfo::new(t.clone(), vi.clone()));
} else {
self.method_to_traits.insert(
trait_method.inspect().clone(),
vec![MethodInfo::new(t.clone(), vi.clone())],
);
}
}
for (class_method, vi) in ctx.locals.iter() {
if let Some(types) = self.method_to_classes.get_mut(class_method.inspect()) {
types.push(MethodInfo::new(t.clone(), vi.clone()));
} else {
self.method_to_classes.insert(
class_method.inspect().clone(),
vec![MethodInfo::new(t.clone(), vi.clone())],
);
}
}
self.mono_types.insert(name.clone(), (t, ctx));
}
}
fn register_gen_mono_patch(
&mut self,
ident: &Identifier,
gen: GenTypeObj,
ctx: Self,
muty: Mutability,
) {
if self.patches.contains_key(ident.inspect()) {
panic!("{ident} has already been registered");
} else if self.rec_get_const_obj(ident.inspect()).is_some() && ident.vis().is_private() {
panic!("{ident} has already been registered as const");
} else {
let t = gen.typ().clone();
let meta_t = gen.meta_type();
let name = &ident.name;
let id = DefId(get_hash(&(&self.name, &name)));
self.decls.insert(
name.clone(),
VarInfo::new(
meta_t,
muty,
ident.vis(),
VarKind::Defined(id),
None,
self.impl_of(),
None,
self.absolutize(name.loc()),
),
);
self.consts
.insert(name.clone(), ValueObj::Type(TypeObj::Generated(gen)));
for impl_trait in ctx.super_traits.iter() {
if let Some(impls) = self.trait_impls.get_mut(&impl_trait.qual_name()) {
impls.insert(TraitImpl::new(t.clone(), impl_trait.clone()));
} else {
self.trait_impls.insert(
impl_trait.qual_name(),
set![TraitImpl::new(t.clone(), impl_trait.clone())],
);
}
}
for (trait_method, vi) in ctx.decls.iter() {
if let Some(types) = self.method_to_traits.get_mut(trait_method.inspect()) {
types.push(MethodInfo::new(t.clone(), vi.clone()));
} else {
self.method_to_traits.insert(
trait_method.inspect().clone(),
vec![MethodInfo::new(t.clone(), vi.clone())],
);
}
}
for (class_method, vi) in ctx.locals.iter() {
if let Some(types) = self.method_to_classes.get_mut(class_method.inspect()) {
types.push(MethodInfo::new(t.clone(), vi.clone()));
} else {
self.method_to_classes.insert(
class_method.inspect().clone(),
vec![MethodInfo::new(t.clone(), vi.clone())],
);
}
}
self.patches.insert(name.clone(), ctx);
}
}
pub(crate) fn import_mod(
&mut self,
kind: OperationKind,
mod_name: &Literal,
) -> CompileResult<PathBuf> {
if kind.is_erg_import() {
self.import_erg_mod(mod_name)
} else {
self.import_py_mod(mod_name)
}
}
fn import_erg_mod(&self, mod_name: &Literal) -> CompileResult<PathBuf> {
let __name__ = enum_unwrap!(mod_name.value.clone(), ValueObj::Str);
let mod_cache = self.mod_cache().unwrap();
let py_mod_cache = self.py_mod_cache().unwrap();
let path = match Self::resolve_real_path(&self.cfg, Path::new(&__name__[..])) {
Some(path) => path,
None => {
let err = TyCheckErrors::from(TyCheckError::import_error(
self.cfg.input.clone(),
line!() as usize,
format!("module {__name__} not found"),
mod_name.loc(),
self.caused_by(),
mod_cache.get_similar_name(&__name__),
self.similar_builtin_py_mod_name(&__name__)
.or_else(|| py_mod_cache.get_similar_name(&__name__)),
));
return Err(err);
}
};
if let Some(referrer) = self.cfg.input.path() {
let graph = &self.shared.as_ref().unwrap().graph;
graph.inc_ref(referrer, path.clone());
}
if mod_cache.get(&path).is_some() {
return Ok(path);
}
let mut cfg = self.cfg.inherit(path.clone());
let src = cfg.input.read();
let mut builder =
HIRBuilder::new_with_cache(cfg, __name__, self.shared.as_ref().unwrap().clone());
match builder.build(src, "exec") {
Ok(artifact) => {
mod_cache.register(
path.clone(),
Some(artifact.object),
builder.pop_mod_ctx().unwrap(),
);
}
Err(artifact) => {
if let Some(hir) = artifact.object {
mod_cache.register(path, Some(hir), builder.pop_mod_ctx().unwrap());
}
return Err(artifact.errors);
}
}
Ok(path)
}
fn similar_builtin_py_mod_name(&self, name: &Str) -> Option<Str> {
get_similar_name(BUILTIN_PYTHON_MODS.into_iter(), name).map(Str::rc)
}
fn is_pystd_main_module(&self, path: &Path) -> bool {
let mut path = PathBuf::from(path);
if path.ends_with("__init__.d.er") {
path.pop();
path.pop();
} else {
path.pop();
}
let pystd_path = erg_pystd_path();
path == pystd_path
}
fn mod_name(&self, path: &Path) -> Str {
let mut name = path
.file_name()
.unwrap()
.to_str()
.unwrap()
.trim_end_matches(".d.er")
.to_string();
for parent in path.components().rev().skip(1) {
let parent = parent.as_os_str().to_str().unwrap();
if parent.ends_with(".d") {
name = parent.trim_end_matches(".d").to_string() + "." + &name;
} else {
break;
}
}
Str::from(name)
}
fn get_path(&self, mod_name: &Literal, __name__: Str) -> CompileResult<PathBuf> {
match Self::resolve_decl_path(&self.cfg, Path::new(&__name__[..])) {
Some(path) => {
if let Ok(first_line) = std::fs::File::open(&path).and_then(|f| {
let mut line = "".to_string();
std::io::BufReader::new(f).read_line(&mut line)?;
Ok(line)
}) {
if first_line.starts_with("# failed") {
let _ = self.try_gen_py_decl_file(&__name__);
}
}
if self.is_pystd_main_module(path.as_path())
&& !BUILTIN_PYTHON_MODS.contains(&&__name__[..])
{
let err = TyCheckError::module_env_error(
self.cfg.input.clone(),
line!() as usize,
&__name__,
mod_name.loc(),
self.caused_by(),
);
return Err(TyCheckErrors::from(err));
}
Ok(path)
}
None => {
if let Ok(path) = self.try_gen_py_decl_file(&__name__) {
return Ok(path);
}
let err = TyCheckError::import_error(
self.cfg.input.clone(),
line!() as usize,
format!("module {__name__} not found"),
mod_name.loc(),
self.caused_by(),
self.mod_cache().unwrap().get_similar_name(&__name__),
self.similar_builtin_py_mod_name(&__name__)
.or_else(|| self.py_mod_cache().unwrap().get_similar_name(&__name__)),
);
Err(TyCheckErrors::from(err))
}
}
}
fn try_gen_py_decl_file(&self, __name__: &Str) -> Result<PathBuf, ()> {
if let Ok(path) = self.cfg.input.local_py_resolve(Path::new(&__name__[..])) {
let (out, err) = if self.cfg.quiet_repl {
(Stdio::null(), Stdio::null())
} else {
(Stdio::inherit(), Stdio::inherit())
};
if let Ok(_status) = Command::new("pylyzer")
.arg("--dump-decl")
.arg(path.to_str().unwrap())
.stdout(out)
.stderr(err)
.spawn()
.and_then(|mut child| child.wait())
{
if let Some(path) = Self::resolve_decl_path(&self.cfg, Path::new(&__name__[..])) {
return Ok(path);
}
}
}
Err(())
}
fn import_py_mod(&self, mod_name: &Literal) -> CompileResult<PathBuf> {
let __name__ = enum_unwrap!(mod_name.value.clone(), ValueObj::Str);
let py_mod_cache = self.py_mod_cache().unwrap();
let path = self.get_path(mod_name, __name__)?;
if py_mod_cache.get(&path).is_some() {
return Ok(path);
}
let mut cfg = self.cfg.inherit(path.clone());
let src = cfg.input.read();
let mut builder = HIRBuilder::new_with_cache(
cfg,
self.mod_name(&path),
self.shared.as_ref().unwrap().clone(),
);
match builder.build(src, "declare") {
Ok(artifact) => {
let ctx = builder.pop_mod_ctx().unwrap();
py_mod_cache.register(path.clone(), Some(artifact.object), ctx);
}
Err(artifact) => {
if let Some(hir) = artifact.object {
py_mod_cache.register(path, Some(hir), builder.pop_mod_ctx().unwrap());
}
return Err(artifact.errors);
}
}
Ok(path)
}
pub fn del(&mut self, ident: &hir::Identifier) -> CompileResult<()> {
let is_const = self.rec_get_const_obj(ident.inspect()).is_some();
let is_builtin = self
.get_builtins()
.unwrap()
.get_var_kv(ident.inspect())
.is_some();
if is_const || is_builtin {
Err(TyCheckErrors::from(TyCheckError::del_error(
self.cfg.input.clone(),
line!() as usize,
ident,
is_const,
self.caused_by(),
)))
} else if self.locals.get(ident.inspect()).is_some() {
let vi = self.locals.remove(ident.inspect()).unwrap();
self.deleted_locals.insert(ident.name.clone(), vi);
Ok(())
} else {
Err(TyCheckErrors::from(TyCheckError::no_var_error(
self.cfg.input.clone(),
line!() as usize,
ident.loc(),
self.caused_by(),
ident.inspect(),
self.get_similar_name(ident.inspect()),
)))
}
}
pub(crate) fn cast(
&mut self,
type_spec: ast::TypeSpec,
call: &mut hir::Call,
) -> TyCheckResult<()> {
let mut dummy_tv_cache = TyVarCache::new(self.level, self);
let cast_to = self.instantiate_typespec(
&type_spec,
None,
&mut dummy_tv_cache,
RegistrationMode::Normal,
false,
)?;
let lhs = enum_unwrap!(
call.args.get_mut_left_or_key("pred").unwrap(),
hir::Expr::BinOp
)
.lhs
.as_mut();
match (
self.supertype_of(lhs.ref_t(), &cast_to),
self.subtype_of(lhs.ref_t(), &cast_to),
) {
(true, true) => Ok(()),
(false, true) => Ok(()), (true, false) => {
if let hir::Expr::Accessor(ref acc) = lhs {
self.change_var_type(acc, cast_to.clone())?;
}
match lhs.ref_t() {
Type::FreeVar(fv) if fv.is_linked() => {
let constraint = Constraint::new_subtype_of(cast_to);
fv.replace(FreeKind::new_unbound(self.level, constraint));
}
Type::FreeVar(fv) => {
let new_constraint = Constraint::new_subtype_of(cast_to);
fv.update_constraint(new_constraint, false);
}
_ => {
*lhs.ref_mut_t() = cast_to;
}
}
Ok(())
}
(false, false) => Err(TyCheckErrors::from(TyCheckError::invalid_type_cast_error(
self.cfg.input.clone(),
line!() as usize,
lhs.loc(),
self.caused_by(),
&lhs.to_string(),
&cast_to,
None,
))),
}
}
fn change_var_type(&mut self, acc: &hir::Accessor, t: Type) -> TyCheckResult<()> {
#[allow(clippy::single_match)]
match acc {
hir::Accessor::Ident(ident) => {
if let Some(vi) = self.get_mut_current_scope_var(&ident.name) {
vi.t = t;
} else {
return Err(TyCheckErrors::from(TyCheckError::feature_error(
self.cfg.input.clone(),
acc.loc(),
&format!("casting {acc}"),
self.caused_by(),
)));
}
}
_ => {
}
}
Ok(())
}
#[cfg(feature = "els")]
pub(crate) fn inc_ref_simple_typespec(&self, simple: &SimpleTypeSpec) {
if self.cfg.mode == ErgMode::LanguageServer {
if let Ok(vi) = self.rec_get_var_info(
&simple.ident,
crate::compile::AccessKind::Name,
&self.cfg.input,
&self.name,
) {
self.inc_ref(&vi, &simple.ident.name);
}
}
}
#[cfg(not(feature = "els"))]
pub(crate) fn inc_ref_simple_typespec(&self, _simple: &SimpleTypeSpec) {}
#[cfg(feature = "els")]
pub(crate) fn inc_ref_const_local(&self, local: &ConstIdentifier) {
if self.cfg.mode == ErgMode::LanguageServer {
if let Ok(vi) = self.rec_get_var_info(
local,
crate::compile::AccessKind::Name,
&self.cfg.input,
&self.name,
) {
self.inc_ref(&vi, &local.name);
}
}
}
#[cfg(not(feature = "els"))]
pub(crate) fn inc_ref_const_local(&self, _local: &ConstIdentifier) {}
#[cfg(feature = "els")]
pub fn inc_ref<L: Locational>(&self, vi: &VarInfo, name: &L) {
if self.cfg.mode == ErgMode::LanguageServer {
self.index()
.unwrap()
.add_ref(vi.def_loc.clone(), self.absolutize(name.loc()));
}
}
#[cfg(not(feature = "els"))]
pub fn inc_ref<L: Locational>(&self, _vi: &VarInfo, _name: &L) {}
}