-use rustc_ast as ast;
+use rustc_ast::{self as ast, token};
use rustc_data_structures::fx::{FxHashMap, FxHashSet};
use rustc_data_structures::sync::Lrc;
use rustc_errors::{ColorConfig, ErrorReported, FatalError};
use rustc_errors::emitter::{Emitter, EmitterWriter};
use rustc_errors::Handler;
use rustc_parse::maybe_new_parser_from_source_str;
- use rustc_parse::parser::ForceCollect;
use rustc_session::parse::ParseSess;
use rustc_span::source_map::FilePathMapping;
}
};
- loop {
- match parser.parse_item(ForceCollect::No) {
- Ok(Some(item)) => {
+ match parser.parse_mod(&token::Eof) {
+ Ok((_attrs, items, _span)) => {
+ for item in items {
if !found_main {
if let ast::ItemKind::Fn(..) = item.kind {
if item.ident.name == sym::main {
break;
}
}
- Ok(None) => break,
- Err(mut e) => {
- e.cancel();
- break;
- }
+ }
+ Err(mut e) => {
+ e.cancel();
}
}