rust/src/comp/middle/resolve.rs
Marijn Haverbeke e33af7e0b5 Make let bindings' scope start at their definition
Also, allow let bindings in a block to shadow each other.
2011-08-16 13:45:41 +02:00

1409 lines
45 KiB
Rust

import syntax::ast;
import syntax::ast::*;
import ast::ident;
import ast::fn_ident;
import ast::def;
import ast::def_id;
import ast::node_id;
import ast::local_def;
import metadata::csearch;
import metadata::cstore;
import driver::session::session;
import util::common::*;
import std::map::new_int_hash;
import std::map::new_str_hash;
import syntax::codemap::span;
import syntax::ast::respan;
import middle::ty::constr_table;
import syntax::visit;
import visit::vt;
import std::ivec;
import std::int;
import std::map::hashmap;
import std::list;
import std::list::list;
import std::list::nil;
import std::list::cons;
import std::option;
import std::option::is_none;
import std::option::some;
import std::option::none;
import std::str;
import syntax::print::pprust::*;
export resolve_crate;
export def_map;
// Resolving happens in two passes. The first pass collects defids of all
// (internal) imports and modules, so that they can be looked up when needed,
// and then uses this information to resolve the imports. The second pass
// locates all names (in expressions, types, and alt patterns) and resolves
// them, storing the resulting def in the AST nodes.
// This module internally uses -1 as a def_id for the top_level module in a
// crate. The parser doesn't assign a def_id to this module.
// (FIXME See https://github.com/graydon/rust/issues/358 for the reason this
// isn't a const.)
tag scope {
scope_crate;
scope_item(@ast::item);
scope_fn(ast::fn_decl, ast::proto, [ast::ty_param]);
scope_native_item(@ast::native_item);
scope_loop(@ast::local); // there's only 1 decl per loop.
scope_block(ast::blk, @mutable uint, @mutable uint);
scope_arm(ast::arm);
}
type scopes = list[scope];
tag import_state {
todo(@ast::view_item, scopes); // only used for explicit imports
resolving(span);
resolved(option::t[def],
/* value */
option::t[def],
/* type */
option::t[def]); /* module */
}
type ext_hash = hashmap[{did: def_id, ident: str, ns: namespace}, def];
fn new_ext_hash() -> ext_hash {
type key = {did: def_id, ident: str, ns: namespace};
fn hash(v: &key) -> uint {
ret str::hash(v.ident) + util::common::hash_def(v.did) +
alt v.ns {
ns_value. { 1u }
ns_type. { 2u }
ns_module. { 3u }
};
}
fn eq(v1: &key, v2: &key) -> bool {
ret util::common::def_eq(v1.did, v2.did) &&
str::eq(v1.ident, v2.ident) && v1.ns == v2.ns;
}
ret std::map::mk_hashmap[key, def](hash, eq);
}
tag mod_index_entry {
mie_view_item(@ast::view_item);
mie_item(@ast::item);
mie_native_item(@ast::native_item);
mie_tag_variant(/* tag item */@ast::item, /* variant index */uint);
}
type mod_index = hashmap[ident, list[mod_index_entry]];
// A tuple of an imported def and the import stmt that brung it
type glob_imp_def = {def: def, item: @ast::view_item};
type indexed_mod =
{m: option::t[ast::_mod],
index: mod_index,
mutable glob_imports: [glob_imp_def],
glob_imported_names: hashmap[str, import_state]};
/* native modules can't contain tags, and we don't store their ASTs because we
only need to look at them to determine exports, which they can't control.*/
type def_map = hashmap[node_id, def];
type env =
{cstore: cstore::cstore,
def_map: def_map,
ast_map: ast_map::map,
imports: hashmap[ast::node_id, import_state],
mod_map: hashmap[ast::node_id, @indexed_mod],
ext_map: hashmap[def_id, [ident]],
ext_cache: ext_hash,
mutable reported: [{ident: str, sc: scope}],
sess: session};
// Used to distinguish between lookups from outside and from inside modules,
// since export restrictions should only be applied for the former.
tag dir { inside; outside; }
tag namespace { ns_value; ns_type; ns_module; }
fn resolve_crate(sess: session, amap: &ast_map::map, crate: @ast::crate) ->
def_map {
let e =
@{cstore: sess.get_cstore(),
def_map: new_int_hash[def](),
ast_map: amap,
imports: new_int_hash[import_state](),
mod_map: new_int_hash[@indexed_mod](),
ext_map: new_def_hash[[ident]](),
ext_cache: new_ext_hash(),
mutable reported: ~[],
sess: sess};
map_crate(e, crate);
resolve_imports(*e);
check_for_collisions(e, *crate);
resolve_names(e, crate);
ret e.def_map;
}
// Locate all modules and imports and index them, so that the next passes can
// resolve through them.
fn map_crate(e: &@env, c: &@ast::crate) {
// First, find all the modules, and index the names that they contain
let v_map_mod =
@{visit_view_item: bind index_vi(e, _, _, _),
visit_item: bind index_i(e, _, _, _)
with *visit::default_visitor[scopes]()};
visit::visit_crate(*c, cons(scope_crate, @nil), visit::mk_vt(v_map_mod));
// Register the top-level mod
e.mod_map.insert(-1,
@{m: some(c.node.module),
index: index_mod(c.node.module),
mutable glob_imports: ~[],
glob_imported_names: new_str_hash[import_state]()});
fn index_vi(e: @env, i: &@ast::view_item, sc: &scopes, v: &vt[scopes]) {
alt i.node {
ast::view_item_import(_, ids, id) {
e.imports.insert(id, todo(i, sc));
}
_ { }
}
}
fn index_i(e: @env, i: &@ast::item, sc: &scopes, v: &vt[scopes]) {
visit_item_with_scope(i, sc, v);
alt i.node {
ast::item_mod(md) {
let s = new_str_hash[import_state]();
e.mod_map.insert(i.id,
@{m: some(md),
index: index_mod(md),
mutable glob_imports: ~[],
glob_imported_names: s});
}
ast::item_native_mod(nmd) {
let s = new_str_hash[import_state]();
e.mod_map.insert(i.id,
@{m: none[ast::_mod],
index: index_nmod(nmd),
mutable glob_imports: ~[],
glob_imported_names: s});
}
_ { }
}
}
// Next, assemble the links for globbed imports.
let v_link_glob =
@{visit_view_item: bind link_glob(e, _, _, _),
visit_item: visit_item_with_scope
with *visit::default_visitor[scopes]()};
visit::visit_crate(*c, cons(scope_crate, @nil),
visit::mk_vt(v_link_glob));
fn link_glob(e: @env, vi: &@ast::view_item, sc: &scopes, v: &vt[scopes]) {
fn find_mod(e: @env, sc: scopes) -> @indexed_mod {
alt sc {
cons(scope_item(i), tl) {
alt i.node {
ast::item_mod(_) | ast::item_native_mod(_) {
ret e.mod_map.get(i.id);
}
_ { be find_mod(e, *tl); }
}
}
_ {
ret e.mod_map.get(-1); //top-level
}
}
}
alt vi.node {
//if it really is a glob import, that is
ast::view_item_import_glob(path, _) {
let imp = follow_import(*e, sc, path, vi.span);
if option::is_some(imp) {
find_mod(e, sc).glob_imports +=
~[{def: option::get(imp), item: vi}];
}
}
_ { }
}
}
}
fn resolve_imports(e: &env) {
for each it: @{key: ast::node_id, val: import_state} in e.imports.items()
{
alt it.val {
todo(item, sc) { resolve_import(e, item, sc); }
resolved(_, _, _) { }
}
}
e.sess.abort_if_errors();
}
fn resolve_names(e: &@env, c: &@ast::crate) {
let v =
@{visit_native_item: visit_native_item_with_scope,
visit_item: visit_item_with_scope,
visit_block: visit_block_with_scope,
visit_decl: visit_decl_with_scope,
visit_arm: bind walk_arm(e, _, _, _),
visit_pat: bind walk_pat(e, _, _, _),
visit_expr: bind walk_expr(e, _, _, _),
visit_ty: bind walk_ty(e, _, _, _),
visit_constr: bind walk_constr(e, _, _, _, _, _),
visit_fn: bind visit_fn_with_scope(e, _, _, _, _, _, _, _)
with *visit::default_visitor()};
visit::visit_crate(*c, cons(scope_crate, @nil), visit::mk_vt(v));
e.sess.abort_if_errors();
fn walk_expr(e: @env, exp: &@ast::expr, sc: &scopes, v: &vt[scopes]) {
visit_expr_with_scope(exp, sc, v);
alt exp.node {
ast::expr_path(p) {
maybe_insert(e, exp.id,
lookup_path_strict(*e, sc, exp.span, p.node,
ns_value));
}
_ { }
}
}
fn walk_ty(e: @env, t: &@ast::ty, sc: &scopes, v: &vt[scopes]) {
visit::visit_ty(t, sc, v);
alt t.node {
ast::ty_path(p, id) {
maybe_insert(e, id,
lookup_path_strict(*e, sc, t.span, p.node, ns_type));
}
_ { }
}
}
fn walk_constr(e: @env, p: &ast::path, sp: &span, id: node_id,
sc: &scopes, v: &vt[scopes]) {
maybe_insert(e, id, lookup_path_strict(*e, sc, sp, p.node, ns_value));
}
fn walk_arm(e: @env, a: &ast::arm, sc: &scopes, v: &vt[scopes]) {
visit_arm_with_scope(a, sc, v);
}
fn walk_pat(e: &@env, pat: &@ast::pat, sc: &scopes, v: &vt[scopes]) {
visit::visit_pat(pat, sc, v);
alt pat.node {
ast::pat_tag(p, _) {
let fnd = lookup_path_strict(*e, sc, p.span, p.node, ns_value);
alt option::get(fnd) {
ast::def_variant(did, vid) {
e.def_map.insert(pat.id, option::get(fnd));
}
_ {
e.sess.span_err
(p.span, "not a tag variant: " + ast::path_name(p));
}
}
}
_ { }
}
}
fn maybe_insert(e: @env, id: node_id, def: option::t[def]) {
if option::is_some(def) { e.def_map.insert(id, option::get(def)); }
}
}
// Visit helper functions
fn visit_item_with_scope(i: &@ast::item, sc: &scopes, v: &vt[scopes]) {
visit::visit_item(i, cons(scope_item(i), @sc), v);
}
fn visit_native_item_with_scope(ni: &@ast::native_item, sc: &scopes,
v: &vt[scopes]) {
visit::visit_native_item(ni, cons(scope_native_item(ni), @sc), v);
}
fn visit_fn_with_scope(e: &@env, f: &ast::_fn, tp: &[ast::ty_param],
sp: &span, name: &fn_ident, id: node_id, sc: &scopes,
v: &vt[scopes]) {
// is this a main fn declaration?
alt name {
some(nm) {
if is_main_name(~[nm]) && !e.sess.get_opts().library {
// This is a main function -- set it in the session
// as the main ID
e.sess.set_main_id(id);
}
}
_ { }
}
// here's where we need to set up the mapping
// for f's constrs in the table.
for c: @ast::constr in f.decl.constraints {
resolve_constr(e, id, c, sc, v);
}
visit::visit_fn(f, tp, sp, name, id,
cons(scope_fn(f.decl, f.proto, tp), @sc), v);
}
fn visit_block_with_scope(b: &ast::blk, sc: &scopes, v: &vt[scopes]) {
let pos = @mutable 0u, loc = @mutable 0u;
let block_sc = cons(scope_block(b, pos, loc), @sc);
for stmt in b.node.stmts {
v.visit_stmt(stmt, block_sc, v);
*pos += 1u;
*loc = 0u;
}
visit::visit_expr_opt(b.node.expr, block_sc, v);
}
fn visit_decl_with_scope(d: &@decl, sc: &scopes, v: &vt[scopes]) {
let loc_pos = alt list::car(sc) {
scope_block(_, _, pos) { pos }
_ { @mutable 0u }
};
alt d.node {
decl_local(locs) {
for loc in locs {
v.visit_local(loc, sc, v);
*loc_pos += 1u;
}
}
decl_item(it) { v.visit_item(it, sc, v); }
}
}
fn visit_arm_with_scope(a: &ast::arm, sc: &scopes, v: &vt[scopes]) {
visit::visit_arm(a, cons(scope_arm(a), @sc), v);
}
fn visit_expr_with_scope(x: &@ast::expr, sc: &scopes, v: &vt[scopes]) {
alt x.node {
ast::expr_for(decl, coll, blk) | ast::expr_for_each(decl, coll, blk) {
let new_sc = cons[scope](scope_loop(decl), @sc);
v.visit_expr(coll, sc, v);
v.visit_local(decl, new_sc, v);
v.visit_block(blk, new_sc, v);
}
ast::expr_fn(f) {
visit::visit_expr(x, cons(scope_fn(f.decl, f.proto, ~[]), @sc), v);
}
_ { visit::visit_expr(x, sc, v); }
}
}
fn follow_import(e: &env, sc: &scopes, path: &[ident], sp: &span) ->
option::t[def] {
let path_len = ivec::len(path);
let dcur = lookup_in_scope_strict(e, sc, sp, path.(0), ns_module);
let i = 1u;
while true && option::is_some(dcur) {
if i == path_len { break; }
dcur =
lookup_in_mod_strict(e, sc, option::get(dcur), sp, path.(i),
ns_module, outside);
i += 1u;
}
if i == path_len {
alt option::get(dcur) {
ast::def_mod(_) | ast::def_native_mod(_) { ret dcur; }
_ {
e.sess.span_err(sp,
str::connect(path, "::") +
" does not name a module.");
ret none;
}
}
} else { ret none; }
}
fn resolve_constr(e: @env, id: node_id, c: &@ast::constr, sc: &scopes,
v: &vt[scopes]) {
let new_def =
lookup_path_strict(*e, sc, c.span, c.node.path.node, ns_value);
if option::is_some(new_def) {
alt option::get(new_def) {
ast::def_fn(pred_id, ast::pure_fn.) {
e.def_map.insert(c.node.id, ast::def_fn(pred_id, ast::pure_fn));
}
_ {
e.sess.span_err(c.span,
"Non-predicate in constraint: " +
path_to_str(c.node.path));
}
}
}
}
// Import resolution
fn resolve_import(e: &env, it: &@ast::view_item, sc_in: &scopes) {
let defid;
let ids;
let name;
alt it.node {
ast::view_item_import(_name, _ids, _id) {
defid = local_def(_id);
ids = _ids;
name = _name;
}
}
e.imports.insert(defid.node, resolving(it.span));
let n_idents = ivec::len(ids);
let end_id = ids.(n_idents - 1u);
// Ignore the current scope if this import would shadow itself.
let sc =
if str::eq(name, ids.(0)) { std::list::cdr(sc_in) } else { sc_in };
if n_idents == 1u {
register(e, defid, it.span, end_id, sc_in,
lookup_in_scope(e, sc, it.span, end_id, ns_value),
lookup_in_scope(e, sc, it.span, end_id, ns_type),
lookup_in_scope(e, sc, it.span, end_id, ns_module));
remove_if_unresolved(e.imports, defid.node);
} else {
let // FIXME (issue #521)
dcur =
alt lookup_in_scope(e, sc, it.span, ids.(0), ns_module) {
some(dcur) { dcur }
none. {
unresolved_err(e, sc, it.span, ids.(0), ns_name(ns_module));
remove_if_unresolved(e.imports, defid.node);
ret ()
}
};
let i = 1u;
while true {
if i == n_idents - 1u {
register(e, defid, it.span, end_id, sc_in,
lookup_in_mod(e, dcur, it.span, end_id, ns_value,
outside),
lookup_in_mod(e, dcur, it.span, end_id, ns_type,
outside),
lookup_in_mod(e, dcur, it.span, end_id, ns_module,
outside));
remove_if_unresolved(e.imports, defid.node);
break;
} else {
dcur =
alt lookup_in_mod(e, dcur, it.span, ids.(i), ns_module,
outside) {
some(dcur) { dcur }
none. {
unresolved_err(e, sc, it.span, ids.(i),
ns_name(ns_module));
remove_if_unresolved(e.imports, defid.node);
ret () // FIXME (issue #521)
}
};
i += 1u;
}
}
}
fn register(e: &env, defid: def_id, sp: &span, name: &ident, sc: &scopes,
val: &option::t[def], typ: &option::t[def],
md: &option::t[def]) {
if is_none(val) && is_none(typ) && is_none(md) {
unresolved_err(e, sc, sp, name, "import");
} else { e.imports.insert(defid.node, resolved(val, typ, md)); }
}
fn remove_if_unresolved(imports: hashmap[ast::node_id, import_state],
node_id: ast::node_id) {
// If we couldn't resolve the import, don't leave it in a partially
// resolved state, to avoid having it reported later as a cyclic
// import
if imports.contains_key(node_id) {
alt imports.get(node_id) {
resolving(_) { imports.remove(node_id); }
_ { }
}
}
}
}
// Utilities
fn ns_name(ns: namespace) -> str {
alt ns {
ns_type. { ret "typename"; }
ns_value. { ret "name"; }
ns_module. { ret "modulename"; }
}
}
fn unresolved_err(e: &env, sc: &scopes, sp: &span, name: &ident, kind: &str) {
fn find_fn_or_mod_scope(sc: scopes) -> scope {
while true {
alt sc {
cons(cur, rest) {
alt cur {
scope_crate. | scope_fn(_, _, _) |
scope_item(@{node: ast::item_mod(_), _}) {
ret cur;
}
_ { sc = *rest; }
}
}
}
}
fail;
}
let err_scope = find_fn_or_mod_scope(sc);
for rs: {ident: str, sc: scope} in e.reported {
if str::eq(rs.ident, name) && err_scope == rs.sc { ret; }
}
e.reported += ~[{ident: name, sc: err_scope}];
e.sess.span_err(sp, mk_unresolved_msg(name, kind));
}
fn unresolved_fatal(e: &env, sp: &span, id: &ident, kind: &str) -> ! {
e.sess.span_fatal(sp, mk_unresolved_msg(id, kind));
}
fn mk_unresolved_msg(id: &ident, kind: &str) -> str {
ret #fmt("unresolved %s: %s", kind, id);
}
// Lookup helpers
fn lookup_path_strict(e: &env, sc: &scopes, sp: &span, pth: &ast::path_,
ns: namespace) -> option::t[def] {
let n_idents = ivec::len(pth.idents);
let headns = if n_idents == 1u { ns } else { ns_module };
let first_scope;
if pth.global {
first_scope = list::cons(scope_crate, @list::nil);
} else { first_scope = sc; }
let dcur =
lookup_in_scope_strict(e, first_scope, sp, pth.idents.(0), headns);
let i = 1u;
while i < n_idents && option::is_some(dcur) {
let curns = if n_idents == i + 1u { ns } else { ns_module };
dcur =
lookup_in_mod_strict(e, sc, option::get(dcur), sp, pth.idents.(i),
curns, outside);
i += 1u;
}
ret dcur;
}
fn lookup_in_scope_strict(e: &env, sc: scopes, sp: &span, name: &ident,
ns: namespace) -> option::t[def] {
alt lookup_in_scope(e, sc, sp, name, ns) {
none. { unresolved_err(e, sc, sp, name, ns_name(ns)); ret none; }
some(d) { ret some(d); }
}
}
fn scope_is_fn(sc: &scope) -> bool {
ret alt sc {
scope_fn(_, ast::proto_iter., _) | scope_fn(_, ast::proto_fn., _) |
scope_native_item(_) {
true
}
_ { false }
};
}
fn def_is_local(d: &def) -> bool {
ret alt d {
ast::def_arg(_) | ast::def_local(_) | ast::def_binding(_) { true }
_ { false }
};
}
fn def_is_obj_field(d: &def) -> bool {
ret alt d { ast::def_obj_field(_) { true } _ { false } };
}
fn def_is_ty_arg(d: &def) -> bool {
ret alt d { ast::def_ty_arg(_,_) { true } _ { false } };
}
fn lookup_in_scope(e: &env, sc: scopes, sp: &span, name: &ident,
ns: namespace) -> option::t[def] {
fn in_scope(e: &env, sp: &span, name: &ident, s: &scope, ns: namespace) ->
option::t[def] {
alt s {
scope_crate. {
ret lookup_in_local_mod(e, -1, sp, name, ns, inside);
}
scope_item(it) {
alt it.node {
ast::item_obj(ob, ty_params, _) {
ret lookup_in_obj(name, ob, ty_params, ns);
}
ast::item_tag(_, ty_params) {
if ns == ns_type { ret lookup_in_ty_params(name, ty_params); }
}
ast::item_mod(_) {
ret lookup_in_local_mod(e, it.id, sp, name, ns, inside);
}
ast::item_native_mod(m) {
ret lookup_in_local_native_mod(e, it.id, sp, name, ns);
}
ast::item_ty(_, ty_params) {
if ns == ns_type { ret lookup_in_ty_params(name, ty_params); }
}
_ { }
}
}
scope_native_item(it) {
alt it.node {
ast::native_item_fn(_, decl, ty_params) {
ret lookup_in_fn(name, decl, ty_params, ns);
}
}
}
scope_fn(decl, _, ty_params) {
ret lookup_in_fn(name, decl, ty_params, ns);
}
scope_loop(local) {
if ns == ns_value {
alt lookup_in_pat(name, local.node.pat) {
some(did) { ret some(ast::def_local(did)); }
_ {}
}
}
}
scope_block(b, pos, loc) {
ret lookup_in_block(name, b.node, *pos, *loc, ns);
}
scope_arm(a) {
if ns == ns_value {
ret option::map(ast::def_binding,
lookup_in_pat(name, a.pats.(0)));
}
}
}
ret none[def];
}
let left_fn = false;
// Used to determine whether obj fields are in scope
let left_fn_level2 = false;
while true {
alt { sc } {
nil. { ret none[def]; }
cons(hd, tl) {
let fnd = in_scope(e, sp, name, hd, ns);
if !is_none(fnd) {
let df = option::get(fnd);
if left_fn && def_is_local(df) ||
left_fn_level2 && def_is_obj_field(df) ||
scope_is_fn(hd) && left_fn && def_is_ty_arg(df) {
let msg =
alt ns {
ns_type. {
"Attempt to use a type \
argument out of scope"
}
_ {
"attempted dynamic \
environment-capture"
}
};
e.sess.span_fatal(sp, msg);
}
ret fnd;
}
if left_fn { left_fn_level2 = true; }
if (ns == ns_value || ns == ns_type) && !left_fn {
left_fn = scope_is_fn(hd);
}
sc = *tl;
}
}
}
e.sess.bug("reached unreachable code in lookup_in_scope"); // sigh
}
fn lookup_in_ty_params(name: &ident, ty_params: &[ast::ty_param]) ->
option::t[def] {
let i = 0u;
for tp: ast::ty_param in ty_params {
if str::eq(tp.ident, name) { ret some(ast::def_ty_arg(i,tp.kind)); }
i += 1u;
}
ret none[def];
}
fn lookup_in_pat(name: &ident, pat: &@ast::pat) -> option::t[def_id] {
let found = none;
for each bound in ast::pat_bindings(pat) {
let p_name = alt bound.node { ast::pat_bind(n) { n } };
if str::eq(p_name, name) {
found = some(local_def(bound.id));
}
}
ret found;
}
fn lookup_in_fn(name: &ident, decl: &ast::fn_decl,
ty_params: &[ast::ty_param], ns: namespace) ->
option::t[def] {
alt ns {
ns_value. {
for a: ast::arg in decl.inputs {
if str::eq(a.ident, name) {
ret some(ast::def_arg(local_def(a.id)));
}
}
ret none[def];
}
ns_type. { ret lookup_in_ty_params(name, ty_params); }
_ { ret none[def]; }
}
}
fn lookup_in_obj(name: &ident, ob: &ast::_obj, ty_params: &[ast::ty_param],
ns: namespace) -> option::t[def] {
alt ns {
ns_value. {
for f: ast::obj_field in ob.fields {
if str::eq(f.ident, name) {
ret some(ast::def_obj_field(local_def(f.id)));
}
}
ret none[def];
}
ns_type. { ret lookup_in_ty_params(name, ty_params); }
_ { ret none[def]; }
}
}
fn lookup_in_block(name: &ident, b: &ast::blk_, pos: uint, loc_pos: uint,
ns: namespace) -> option::t[def] {
let i = ivec::len(b.stmts);
while i > 0u {
i -= 1u;
let st = b.stmts.(i);
alt st.node {
ast::stmt_decl(d, _) {
alt d.node {
ast::decl_local(locs) {
if i <= pos {
let j = ivec::len(locs);
while j > 0u {
j -= 1u;
let loc = locs.(j);
if ns == ns_value && (i < pos || j < loc_pos) {
alt lookup_in_pat(name, loc.node.pat) {
some(did) { ret some(ast::def_local(did)); }
_ {}
}
}
}
}
}
ast::decl_item(it) {
alt it.node {
ast::item_tag(variants, _) {
if ns == ns_type {
if str::eq(it.ident, name) {
ret some(ast::def_ty(local_def(it.id)));
}
} else if (ns == ns_value) {
for v: ast::variant in variants {
if str::eq(v.node.name, name) {
let i = v.node.id;
ret some(ast::def_variant(local_def(it.id),
local_def(i)));
}
}
}
}
_ {
if str::eq(it.ident, name) {
let found = found_def_item(it, ns);
if !is_none(found) { ret found; }
}
}
}
}
}
}
_ { }
}
}
ret none[def];
}
fn found_def_item(i: &@ast::item, ns: namespace) -> option::t[def] {
alt i.node {
ast::item_const(_, _) {
if ns == ns_value { ret some(ast::def_const(local_def(i.id))); }
}
ast::item_fn(f, _) {
if ns == ns_value {
ret some(ast::def_fn(local_def(i.id), f.decl.purity));
}
}
ast::item_mod(_) {
if ns == ns_module { ret some(ast::def_mod(local_def(i.id))); }
}
ast::item_native_mod(_) {
if ns == ns_module { ret some(ast::def_native_mod(local_def(i.id))); }
}
ast::item_ty(_, _) {
if ns == ns_type { ret some(ast::def_ty(local_def(i.id))); }
}
ast::item_res(_, _, _, ctor_id) {
alt ns {
ns_value. {
ret some(ast::def_fn(local_def(ctor_id), ast::impure_fn));
}
ns_type. { ret some(ast::def_ty(local_def(i.id))); }
_ { }
}
}
ast::item_tag(_, _) {
if ns == ns_type { ret some(ast::def_ty(local_def(i.id))); }
}
ast::item_obj(_, _, ctor_id) {
alt ns {
ns_value. {
ret some(ast::def_fn(local_def(ctor_id), ast::impure_fn));
}
ns_type. { ret some(ast::def_ty(local_def(i.id))); }
_ { }
}
}
_ { }
}
ret none[def];
}
fn lookup_in_mod_strict(e: &env, sc: &scopes, m: def, sp: &span, name: &ident,
ns: namespace, dr: dir) -> option::t[def] {
alt lookup_in_mod(e, m, sp, name, ns, dr) {
none. { unresolved_err(e, sc, sp, name, ns_name(ns)); ret none; }
some(d) { ret some(d); }
}
}
fn lookup_in_mod(e: &env, m: &def, sp: &span, name: &ident, ns: namespace,
dr: dir) -> option::t[def] {
let defid = ast::def_id_of_def(m);
if defid.crate != ast::local_crate {
// examining a module in an external crate
let cached = e.ext_cache.find({did: defid, ident: name, ns: ns});
if !is_none(cached) { ret cached; }
let path = ~[name];
if defid.node != -1 { path = e.ext_map.get(defid) + path; }
let fnd = lookup_external(e, defid.crate, path, ns);
if !is_none(fnd) {
e.ext_cache.insert({did: defid, ident: name, ns: ns},
option::get(fnd));
}
ret fnd;
}
alt m {
ast::def_mod(defid) {
ret lookup_in_local_mod(e, defid.node, sp, name, ns, dr);
}
ast::def_native_mod(defid) {
ret lookup_in_local_native_mod(e, defid.node, sp, name, ns);
}
}
}
fn found_view_item(e: &env, vi: @ast::view_item, ns: namespace) ->
option::t[def] {
alt vi.node {
ast::view_item_use(_, _, id) {
let cnum = cstore::get_use_stmt_cnum(e.cstore, id);
ret some(ast::def_mod({crate: cnum, node: -1}));
}
ast::view_item_import(_, _, id) {
ret lookup_import(e, local_def(id), ns);
}
ast::view_item_import_glob(_, defid) {
ret none[def]; //will be handled in the fallback glob pass
}
}
}
fn lookup_import(e: &env, defid: def_id, ns: namespace) -> option::t[def] {
alt e.imports.get(defid.node) {
todo(item, sc) {
resolve_import(e, item, sc);
ret lookup_import(e, defid, ns);
}
resolving(sp) { e.sess.span_err(sp, "cyclic import"); ret none; }
resolved(val, typ, md) {
ret alt ns { ns_value. { val } ns_type. { typ } ns_module. { md } };
}
}
}
fn lookup_in_local_native_mod(e: &env, node_id: node_id, sp: &span,
id: &ident, ns: namespace) -> option::t[def] {
ret lookup_in_local_mod(e, node_id, sp, id, ns, inside);
}
fn lookup_in_local_mod(e: &env, node_id: node_id, sp: &span, id: &ident,
ns: namespace, dr: dir) -> option::t[def] {
let info = e.mod_map.get(node_id);
if dr == outside && !ast::is_exported(id, option::get(info.m)) {
// if we're in a native mod, then dr==inside, so info.m is some _mod
ret none[def]; // name is not visible
}
alt info.index.find(id) {
none. { }
some(lst_) {
let lst = lst_;
while true {
alt lst {
nil. { break; }
cons(hd, tl) {
let found = lookup_in_mie(e, hd, ns);
if !is_none(found) { ret found; }
lst = *tl;
}
}
}
}
}
// not local or explicitly imported; try globs:
ret lookup_glob_in_mod(e, info, sp, id, ns, outside);
}
fn lookup_glob_in_mod(e: &env, info: @indexed_mod, sp: &span, id: &ident,
wanted_ns: namespace, dr: dir) -> option::t[def] {
fn per_ns(e: &env, info: @indexed_mod, sp: &span, id: &ident,
ns: namespace, dr: dir) -> option::t[def] {
fn lookup_in_mod_(e: &env, def: &glob_imp_def, sp: &span,
name: &ident, ns: namespace, dr: dir) ->
option::t[glob_imp_def] {
alt lookup_in_mod(e, def.def, sp, name, ns, dr) {
option::some(d) { option::some({def: d, item: def.item}) }
option::none. { option::none }
}
}
let matches =
ivec::filter_map(bind lookup_in_mod_(e, _, sp, id, ns, dr),
{ info.glob_imports });
if ivec::len(matches) == 0u {
ret none;
} else if (ivec::len(matches) == 1u) {
ret some(matches.(0).def);
} else {
for match: glob_imp_def in matches {
let sp = match.item.span;
e.sess.span_note(sp, #fmt("'%s' is imported here", id));
}
e.sess.span_fatal(sp,
"'" + id + "' is glob-imported from" +
" multiple different modules.");
}
}
// since we don't know what names we have in advance,
// absence takes the place of todo()
if !info.glob_imported_names.contains_key(id) {
info.glob_imported_names.insert(id, resolving(sp));
let val = per_ns(e, info, sp, id, ns_value, dr);
let typ = per_ns(e, info, sp, id, ns_type, dr);
let md = per_ns(e, info, sp, id, ns_module, dr);
info.glob_imported_names.insert(id, resolved(val, typ, md));
}
alt info.glob_imported_names.get(id) {
todo(_, _) { e.sess.bug("Shouldn't've put a todo in."); }
resolving(sp) {
ret none[def]; //circularity is okay in import globs
}
resolved(val, typ, md) {
ret alt wanted_ns {
ns_value. { val }
ns_type. { typ }
ns_module. { md }
};
}
}
}
fn lookup_in_mie(e: &env, mie: &mod_index_entry, ns: namespace) ->
option::t[def] {
alt mie {
mie_view_item(view_item) { ret found_view_item(e, view_item, ns); }
mie_item(item) { ret found_def_item(item, ns); }
mie_tag_variant(item, variant_idx) {
alt item.node {
ast::item_tag(variants, _) {
if ns == ns_value {
let vid = variants.(variant_idx).node.id;
ret some(ast::def_variant(local_def(item.id),
local_def(vid)));
} else { ret none[def]; }
}
}
}
mie_native_item(native_item) {
alt native_item.node {
ast::native_item_ty. {
if ns == ns_type {
ret some(ast::def_native_ty(local_def(native_item.id)));
}
}
ast::native_item_fn(_, _, _) {
if ns == ns_value {
ret some(ast::def_native_fn(local_def(native_item.id)));
}
}
}
}
_ { }
}
ret none[def];
}
// Module indexing
fn add_to_index(index: &hashmap[ident, list[mod_index_entry]], id: &ident,
ent: &mod_index_entry) {
alt index.find(id) {
none. { index.insert(id, cons(ent, @nil[mod_index_entry])); }
some(prev) { index.insert(id, cons(ent, @prev)); }
}
}
fn index_mod(md: &ast::_mod) -> mod_index {
let index = new_str_hash[list[mod_index_entry]]();
for it: @ast::view_item in md.view_items {
alt it.node {
ast::view_item_import(ident, _, _) | ast::view_item_use(ident, _, _)
{
add_to_index(index, ident, mie_view_item(it));
}
//globbed imports have to be resolved lazily.
ast::view_item_import_glob(_, _) | ast::view_item_export(_, _) {
}
}
}
for it: @ast::item in md.items {
alt it.node {
ast::item_const(_, _) | ast::item_fn(_, _) | ast::item_mod(_) |
ast::item_native_mod(_) | ast::item_ty(_, _) |
ast::item_res(_, _, _, _) | ast::item_obj(_, _, _) {
add_to_index(index, it.ident, mie_item(it));
}
ast::item_tag(variants, _) {
add_to_index(index, it.ident, mie_item(it));
let variant_idx: uint = 0u;
for v: ast::variant in variants {
add_to_index(index, v.node.name,
mie_tag_variant(it, variant_idx));
variant_idx += 1u;
}
}
}
}
ret index;
}
fn index_nmod(md: &ast::native_mod) -> mod_index {
let index = new_str_hash[list[mod_index_entry]]();
for it: @ast::view_item in md.view_items {
alt it.node {
ast::view_item_use(ident, _, _) | ast::view_item_import(ident, _, _)
{
add_to_index(index, ident, mie_view_item(it));
}
ast::view_item_import_glob(_, _) | ast::view_item_export(_, _) { }
}
}
for it: @ast::native_item in md.items {
add_to_index(index, it.ident, mie_native_item(it));
}
ret index;
}
// External lookups
fn ns_for_def(d: def) -> namespace {
ret alt d {
ast::def_fn(id, _) { ns_value }
ast::def_obj_field(id) { ns_value }
ast::def_mod(id) { ns_module }
ast::def_native_mod(id) { ns_module }
ast::def_const(id) { ns_value }
ast::def_arg(id) { ns_value }
ast::def_local(id) { ns_value }
ast::def_variant(_, id) { ns_value }
ast::def_ty(id) { ns_type }
ast::def_binding(id) { ns_type }
ast::def_use(id) { ns_module }
ast::def_native_ty(id) { ns_type }
ast::def_native_fn(id) { ns_value }
};
}
fn lookup_external(e: &env, cnum: int, ids: &[ident], ns: namespace) ->
option::t[def] {
for d: def in csearch::lookup_defs(e.sess.get_cstore(), cnum, ids) {
e.ext_map.insert(ast::def_id_of_def(d), ids);
if ns == ns_for_def(d) { ret some(d); }
}
ret none[def];
}
// Collision detection
fn check_for_collisions(e: &@env, c: &ast::crate) {
// Module indices make checking those relatively simple -- just check each
// name for multiple entities in the same namespace.
for each m: @{key: ast::node_id, val: @indexed_mod}
in e.mod_map.items() {
for each name: @{key: ident, val: list[mod_index_entry]}
in m.val.index.items() {
check_mod_name(*e, name.key, name.val);
}
}
// Other scopes have to be checked the hard way.
let v = @{visit_item: bind check_item(e, _, _, _),
visit_block: bind check_block(e, _, _, _),
visit_arm: bind check_arm(e, _, _, _),
visit_expr: bind check_expr(e, _, _, _),
visit_ty: bind check_ty(e, _, _, _)
with *visit::default_visitor()};
visit::visit_crate(c, (), visit::mk_vt(v));
}
fn check_mod_name(e: &env, name: &ident, entries: list[mod_index_entry]) {
let saw_mod = false;
let saw_type = false;
let saw_value = false;
fn dup(e: &env, sp: &span, word: &str, name: &ident) {
e.sess.span_fatal(sp, "duplicate definition of " + word + name);
}
while true {
alt entries {
cons(entry, rest) {
if !is_none(lookup_in_mie(e, entry, ns_value)) {
if saw_value {
dup(e, mie_span(entry), "", name);
} else { saw_value = true; }
}
if !is_none(lookup_in_mie(e, entry, ns_type)) {
if saw_type {
dup(e, mie_span(entry), "type ", name);
} else { saw_type = true; }
}
if !is_none(lookup_in_mie(e, entry, ns_module)) {
if saw_mod {
dup(e, mie_span(entry), "module ", name);
} else { saw_mod = true; }
}
entries = *rest;
}
nil. { break; }
}
}
}
fn mie_span(mie: &mod_index_entry) -> span {
ret alt mie {
mie_view_item(item) { item.span }
mie_item(item) { item.span }
mie_tag_variant(item, _) { item.span }
mie_native_item(item) { item.span }
};
}
fn check_item(e: &@env, i: &@ast::item, x: &(), v: &vt[()]) {
fn typaram_names(tps: &[ast::ty_param]) -> [ident] {
let x: [ast::ident] = ~[];
for tp: ast::ty_param in tps { x += ~[tp.ident] }
ret x;
}
visit::visit_item(i, x, v);
alt i.node {
ast::item_fn(f, ty_params) {
check_fn(*e, i.span, f);
ensure_unique(*e, i.span, typaram_names(ty_params),
ident_id, "type parameter");
}
ast::item_obj(ob, ty_params, _) {
fn field_name(field: &ast::obj_field) -> ident { ret field.ident; }
ensure_unique(*e, i.span, ob.fields, field_name, "object field");
for m: @ast::method in ob.methods {
check_fn(*e, m.span, m.node.meth);
}
ensure_unique(*e, i.span, typaram_names(ty_params),
ident_id, "type parameter");
}
ast::item_tag(_, ty_params) {
ensure_unique(*e, i.span, typaram_names(ty_params),
ident_id, "type parameter");
}
_ { }
}
}
fn check_pat(ch: checker, p: &@ast::pat) {
for each p in ast::pat_bindings(p) {
let ident = alt p.node { pat_bind(n) { n } };
add_name(ch, p.span, ident);
}
}
fn check_arm(e: &@env, a: &ast::arm, x: &(), v: &vt[()]) {
visit::visit_arm(a, x, v);
let ch0 = checker(*e, "binding");
check_pat(ch0, a.pats.(0));
let seen0 = ch0.seen;
let i = ivec::len(a.pats);
while i > 1u {
i -= 1u;
let ch = checker(*e, "binding");
check_pat(ch, a.pats.(i));
// Ensure the bindings introduced in this pattern are the same as in
// the first pattern.
if ivec::len(ch.seen) != ivec::len(seen0) {
e.sess.span_err(a.pats.(i).span,
"inconsistent number of bindings");
} else {
for name: ident in ch.seen {
if is_none(ivec::find(bind str::eq(name, _), seen0)) {
// Fight the alias checker
let name_ = name;
e.sess.span_err(a.pats.(i).span,
"binding " + name_ +
" does not occur in first pattern");
}
}
}
}
}
fn check_block(e: &@env, b: &ast::blk, x: &(), v: &vt[()]) {
visit::visit_block(b, x, v);
let values = checker(*e, "value");
let types = checker(*e, "type");
let mods = checker(*e, "module");
for st: @ast::stmt in b.node.stmts {
alt st.node {
ast::stmt_decl(d, _) {
alt d.node {
ast::decl_local(locs) {
let local_values = checker(*e, "value");
for loc in locs {
for each p in ast::pat_bindings(loc.node.pat) {
let ident = alt p.node { pat_bind(n) { n } };
add_name(local_values, p.span, ident);
check_name(values, p.span, ident);
}
}
}
ast::decl_item(it) {
alt it.node {
ast::item_tag(variants, _) {
add_name(types, it.span, it.ident);
for v: ast::variant in variants {
add_name(values, v.span, v.node.name);
}
}
ast::item_mod(_) | ast::item_native_mod(_) {
add_name(mods, it.span, it.ident);
}
ast::item_const(_, _) | ast::item_fn(_, _) {
add_name(values, it.span, it.ident);
}
ast::item_ty(_, _) { add_name(types, it.span, it.ident); }
ast::item_res(_, _, _, _) | ast::item_obj(_, _, _) {
add_name(types, it.span, it.ident);
add_name(values, it.span, it.ident);
}
_ { }
}
}
}
}
_ { }
}
}
}
fn check_fn(e: &env, sp: &span, f: &ast::_fn) {
fn arg_name(a: &ast::arg) -> ident { ret a.ident; }
ensure_unique(e, sp, f.decl.inputs, arg_name, "argument");
}
fn check_expr(e: &@env, ex: &@ast::expr, x: &(), v: &vt[()]) {
alt ex.node {
ast::expr_rec(fields, _) {
fn field_name(f: &ast::field) -> ident { ret f.node.ident; }
ensure_unique(*e, ex.span, fields, field_name, "field");
}
_ { }
}
visit::visit_expr(ex, x, v);
}
fn check_ty(e: &@env, ty: &@ast::ty, x: &(), v: &vt[()]) {
alt ty.node {
ast::ty_rec(fields) {
fn field_name(f: &ast::ty_field) -> ident { ret f.node.ident; }
ensure_unique(*e, ty.span, fields, field_name, "field");
}
_ { }
}
visit::visit_ty(ty, x, v);
}
type checker = @{mutable seen: [ident], kind: str, sess: session};
fn checker(e: &env, kind: str) -> checker {
let seen: [ident] = ~[];
ret @{mutable seen: seen, kind: kind, sess: e.sess};
}
fn check_name(ch: &checker, sp: &span, name: &ident) {
for s: ident in ch.seen {
if str::eq(s, name) {
ch.sess.span_fatal(sp, "duplicate " + ch.kind + " name: " + name);
}
}
}
fn add_name(ch: &checker, sp: &span, name: &ident) {
check_name(ch, sp, name);
ch.seen += ~[name];
}
fn ident_id(i: &ident) -> ident { ret i; }
fn ensure_unique[T](e: &env, sp: &span, elts: &[T], id: fn(&T) -> ident ,
kind: &str) {
let ch = checker(e, kind);
for elt: T in elts { add_name(ch, sp, id(elt)); }
}
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End: