// Copyright 2015 The Rust Project Developers. See the COPYRIGHT // file at the top-level directory of this distribution and at // http://rust-lang.org/COPYRIGHT. // // Licensed under the Apache License, Version 2.0 or the MIT license // , at your // option. This file may not be copied, modified, or distributed // except according to those terms. use utils; use std::path::{Path, PathBuf}; use std::collections::BTreeMap; use syntax::ast; use syntax::codemap; use syntax::parse::parser; /// List all the files containing modules of a crate. /// If a file is used twice in a crate, it appears only once. pub fn list_files<'a>( krate: &'a ast::Crate, codemap: &codemap::CodeMap, ) -> BTreeMap { let mut result = BTreeMap::new(); // Enforce file order determinism let root_filename: PathBuf = codemap.span_to_filename(krate.span).into(); list_submodules( &krate.module, root_filename.parent().unwrap(), codemap, &mut result, ); result.insert(root_filename, &krate.module); result } /// Recursively list all external modules included in a module. fn list_submodules<'a>( module: &'a ast::Mod, search_dir: &Path, codemap: &codemap::CodeMap, result: &mut BTreeMap, ) { debug!("list_submodules: search_dir: {:?}", search_dir); for item in &module.items { if let ast::ItemKind::Mod(ref sub_mod) = item.node { if !utils::contains_skip(&item.attrs) { let is_internal = codemap.span_to_filename(item.span) == codemap.span_to_filename(sub_mod.inner); let dir_path = if is_internal { search_dir.join(&item.ident.to_string()) } else { let mod_path = module_file(item.ident, &item.attrs, search_dir, codemap); let dir_path = mod_path.parent().unwrap().to_owned(); result.insert(mod_path, sub_mod); dir_path }; list_submodules(sub_mod, &dir_path, codemap, result); } } } } /// Find the file corresponding to an external mod fn module_file( id: ast::Ident, attrs: &[ast::Attribute], dir_path: &Path, codemap: &codemap::CodeMap, ) -> PathBuf { if let Some(path) = parser::Parser::submod_path_from_attr(attrs, dir_path) { return path; } match parser::Parser::default_submod_path(id, dir_path, codemap).result { Ok(parser::ModulePathSuccess { path, .. }) => path, Err(_) => panic!("Couldn't find module {}", id), } }