language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-04 18:58:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add special variables module/global

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-05-25 16:37:38 +09:00
parent 30eac6dbcf
commit 4af0c4c799
2 changed files with 57 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

21
crates/erg_compiler/link_hir.rs
View file

@ -176,7 +176,15 @@ impl<'a> HIRLinker<'a> {
Accessor::Attr(attr) => {
self.replace_import(&mut attr.obj);
}
Accessor::Ident(_) => {}
Accessor::Ident(ident) => match &ident.inspect()[..] {
"module" => {
*expr = Self::self_module();
}
"global" => {
*expr = Expr::from(Identifier::public("__builtins__"));
}
_ => {}
},
}
}
Expr::Array(array) => match array {
@ -286,6 +294,12 @@ impl<'a> HIRLinker<'a> {
}
}
fn self_module() -> Expr {
let __import__ = Identifier::public("__import__");
let __name__ = Identifier::public("__name__");
Expr::from(__import__).call1(Expr::from(__name__))
}
/// ```erg
/// x = import "mod"
/// ```
@ -311,10 +325,7 @@ impl<'a> HIRLinker<'a> {
// self = __import__(__name__)
if matches!((path.canonicalize(), self.cfg.input.unescaped_path().canonicalize()), (Ok(l), Ok(r)) if l == r)
{
let __import__ = Identifier::public("__import__");
let __name__ = Identifier::public("__name__");
let call = Expr::from(__import__).call1(Expr::from(__name__));
*expr = call;
*expr = Self::self_module();
return;
}
// In the case of REPL, entries cannot be used up