/* Copyright (c) 2014 Microsoft Corporation. All rights reserved. Released under Apache 2.0 license as described in the file LICENSE. Author: Leonardo de Moura */ #include #include #include #include "util/interrupt.h" #include "util/script_exception.h" #include "util/sstream.h" #include "util/flet.h" #include "util/lean_path.h" #include "kernel/for_each_fn.h" #include "kernel/replace_fn.h" #include "kernel/abstract.h" #include "kernel/instantiate.h" #include "library/parser_nested_exception.h" #include "library/aliases.h" #include "library/private.h" #include "library/choice.h" #include "library/placeholder.h" #include "library/deep_copy.h" #include "library/module.h" #include "library/error_handling/error_handling.h" #include "frontends/lean/parser.h" #include "frontends/lean/notation_cmd.h" #ifndef LEAN_DEFAULT_PARSER_SHOW_ERRORS #define LEAN_DEFAULT_PARSER_SHOW_ERRORS true #endif namespace lean { // ========================================== // Parser configuration options static name g_parser_show_errors {"lean", "parser", "show_errors"}; RegisterBoolOption(g_parser_show_errors, LEAN_DEFAULT_PARSER_SHOW_ERRORS, "(lean parser) display error messages in the regular output channel"); bool get_parser_show_errors(options const & opts) { return opts.get_bool(g_parser_show_errors, LEAN_DEFAULT_PARSER_SHOW_ERRORS); } // ========================================== parser::local_scope::local_scope(parser & p): m_p(p), m_env(p.env()) { p.push_local_scope(); } parser::local_scope::~local_scope() { m_p.pop_local_scope(); m_p.m_env = m_env; } parser::param_universe_scope::param_universe_scope(parser & p):m_p(p), m_old(m_p.m_type_use_placeholder) { m_p.m_type_use_placeholder = false; } parser::param_universe_scope::~param_universe_scope() { m_p.m_type_use_placeholder = m_old; } parser::parser(environment const & env, io_state const & ios, std::istream & strm, char const * strm_name, script_state * ss, bool use_exceptions, unsigned num_threads, local_level_decls const & lds, local_expr_decls const & eds, unsigned line): m_env(env), m_ios(ios), m_ss(ss), m_verbose(true), m_use_exceptions(use_exceptions), m_scanner(strm, strm_name), m_local_level_decls(lds), m_local_decls(eds), m_pos_table(std::make_shared()) { m_scanner.set_line(line); m_num_threads = num_threads; m_type_use_placeholder = true; m_found_errors = false; updt_options(); m_next_tag_idx = 0; m_curr = scanner::token_kind::Identifier; protected_call([&]() { scan(); }, [&]() { sync_command(); }); } void parser::updt_options() { m_verbose = get_verbose(m_ios.get_options()); m_show_errors = get_parser_show_errors(m_ios.get_options()); } void parser::display_error_pos(unsigned line, unsigned pos) { regular_stream() << get_stream_name() << ":" << line << ":"; if (pos != static_cast(-1)) regular_stream() << pos << ":"; regular_stream() << " error:"; } void parser::display_error_pos(pos_info p) { display_error_pos(p.first, p.second); } void parser::display_error(char const * msg, unsigned line, unsigned pos) { display_error_pos(line, pos); regular_stream() << " " << msg << endl; } void parser::display_error(char const * msg, pos_info p) { display_error(msg, p.first, p.second); } void parser::display_error(exception const & ex) { parser_pos_provider pos_provider(m_pos_table, get_stream_name(), m_last_cmd_pos); ::lean::display_error(regular_stream(), &pos_provider, ex); } void parser::throw_parser_exception(char const * msg, pos_info p) { throw parser_exception(msg, get_stream_name().c_str(), p.first, p.second); } void parser::throw_nested_exception(exception & ex, pos_info p) { throw parser_nested_exception(std::shared_ptr(ex.clone()), std::make_shared(m_pos_table, get_stream_name(), p)); } #define CATCH(ShowError, ThrowError) \ m_found_errors = true; \ if (!m_use_exceptions && m_show_errors) { ShowError ; } \ sync(); \ if (m_use_exceptions) { ThrowError ; } void parser::protected_call(std::function && f, std::function && sync) { try { f(); // } catch (tactic_cmd_error & ex) { // CATCH(display_error(ex), // throw parser_exception(ex.what(), m_strm_name.c_str(), ex.m_pos.first, ex.m_pos.second)); } catch (parser_exception & ex) { CATCH(regular_stream() << ex.what() << endl, throw); } catch (parser_error & ex) { CATCH(display_error(ex.what(), ex.m_pos), throw_parser_exception(ex.what(), ex.m_pos)); } catch (interrupted & ex) { reset_interrupt(); if (m_verbose) regular_stream() << "!!!Interrupted!!!" << endl; sync(); if (m_use_exceptions) throw; } catch (script_exception & ex) { reset_interrupt(); CATCH(display_error(ex), throw_nested_exception(ex, m_last_script_pos)); } catch (exception & ex) { reset_interrupt(); CATCH(display_error(ex), throw_nested_exception(ex, m_last_cmd_pos)); } } void parser::sync_command() { // Keep consuming tokens until we find a Command or End-of-file while (curr() != scanner::token_kind::CommandKeyword && curr() != scanner::token_kind::Eof) next(); } tag parser::get_tag(expr e) { tag t = e.get_tag(); if (t == nulltag) { t = m_next_tag_idx; e.set_tag(t); m_next_tag_idx++; } return t; } expr parser::save_pos(expr e, pos_info p) { m_pos_table->insert(mk_pair(get_tag(e), p)); return e; } expr parser::rec_save_pos(expr const & e, pos_info p) { unsigned m = std::numeric_limits::max(); pos_info dummy(m, 0); for_each(e, [&](expr const & e, unsigned) { if (pos_of(e, dummy).first == m) { save_pos(e, p); return true; } else { return false; } }); return e; } /** \brief Create a copy of \c e, and the position of new expression with p */ expr parser::copy_with_new_pos(expr const & e, pos_info p) { switch (e.kind()) { case expr_kind::Sort: case expr_kind::Constant: case expr_kind::Meta: case expr_kind::Local: case expr_kind::Var: return save_pos(copy(e), p); case expr_kind::App: return save_pos(::lean::mk_app(copy_with_new_pos(app_fn(e), p), copy_with_new_pos(app_arg(e), p)), p); case expr_kind::Lambda: case expr_kind::Pi: return save_pos(update_binding(e, copy_with_new_pos(binding_domain(e), p), copy_with_new_pos(binding_body(e), p)), p); case expr_kind::Let: return save_pos(update_let(e, copy_with_new_pos(let_type(e), p), copy_with_new_pos(let_value(e), p), copy_with_new_pos(let_body(e), p)), p); case expr_kind::Macro: { buffer args; for (unsigned i = 0; i < macro_num_args(e); i++) args.push_back(copy_with_new_pos(macro_arg(e, i), p)); return save_pos(update_macro(e, args.size(), args.data()), p); }} lean_unreachable(); // LCOV_EXCL_LINE } /** \brief Add \c ls to constants occurring in \c e. */ expr parser::propagate_levels(expr const & e, levels const & ls) { if (is_nil(ls)) { return e; } else { return replace(e, [&](expr const & e, unsigned) { if (is_constant(e)) { auto it = m_env.find(const_name(e)); if (it) { level_param_names const & univ_ps = it->get_univ_params(); levels const & old_ls = const_levels(e); if (length(old_ls) + length(ls) <= length(univ_ps)) return some_expr(update_constant(e, append(old_ls, ls))); } } return none_expr(); }); } } pos_info parser::pos_of(expr const & e, pos_info default_pos) { auto it = m_pos_table->find(get_tag(e)); if (it == m_pos_table->end()) return default_pos; else return it->second; } bool parser::curr_is_token(name const & tk) const { return (curr() == scanner::token_kind::Keyword || curr() == scanner::token_kind::CommandKeyword) && get_token_info().value() == tk; } bool parser::curr_is_token_or_id(name const & tk) const { if (curr() == scanner::token_kind::Keyword || curr() == scanner::token_kind::CommandKeyword) return get_token_info().value() == tk; else if (curr() == scanner::token_kind::Identifier) return get_name_val() == tk; else return false; } void parser::check_token_next(name const & tk, char const * msg) { if (!curr_is_token(tk)) throw parser_error(msg, pos()); next(); } name parser::check_id_next(char const * msg) { if (!curr_is_identifier()) throw parser_error(msg, pos()); name r = get_name_val(); next(); return r; } expr parser::mk_app(expr fn, expr arg, pos_info const & p) { return save_pos(::lean::mk_app(fn, arg), p); } void parser::push_local_scope() { if (m_type_use_placeholder) m_local_level_decls.push(); m_local_decls.push(); } void parser::pop_local_scope() { if (m_type_use_placeholder) m_local_level_decls.pop(); m_local_decls.pop(); } void parser::add_local_level(name const & n, level const & l) { if (m_env.is_universe(n)) throw exception(sstream() << "invalid universe declaration, '" << n << "' shadows a global universe"); if (m_local_level_decls.contains(n)) throw exception(sstream() << "invalid universe declaration, '" << n << "' shadows a local universe"); m_local_level_decls.insert(n, l); } void parser::add_local_expr(name const & n, expr const & e, binder_info const & bi) { m_local_decls.insert(n, parameter(pos(), e, bi)); } void parser::add_local(expr const & e) { lean_assert(is_local(e)); add_local_expr(local_pp_name(e), e); } unsigned parser::get_local_level_index(name const & n) const { return m_local_level_decls.find_idx(n); } unsigned parser::get_local_index(name const & n) const { return m_local_decls.find_idx(n); } optional parser::get_local(name const & n) const { if (auto it = m_local_decls.find(n)) return optional(*it); else return optional(); } /** \brief Parse a sequence of identifiers ID*. Store the result in \c result. */ void parser::parse_names(buffer> & result) { while (curr_is_identifier()) { result.emplace_back(pos(), get_name_val()); next(); } } static name g_period("."); static name g_colon(":"); static name g_lparen("("); static name g_rparen(")"); static name g_llevel_curly(".{"); static name g_lcurly("{"); static name g_rcurly("}"); static name g_lbracket("["); static name g_rbracket("]"); static name g_add("+"); static name g_max("max"); static name g_imax("imax"); static name g_cup("\u2294"); static name g_infix("infix"); static name g_infixl("infixl"); static name g_infixr("infixr"); static name g_postfix("postfix"); static name g_notation("notation"); static name g_import("import"); static unsigned g_level_add_prec = 10; static unsigned g_level_cup_prec = 5; unsigned parser::get_small_nat() { mpz val = get_num_val().get_numerator(); lean_assert(val >= 0); if (!val.is_unsigned_int()) throw parser_error("invalid level expression, value does not fit in a machine integer", pos()); return val.get_unsigned_int(); } unsigned parser::parse_small_nat() { unsigned r = get_small_nat(); next(); return r; } static level lift(level l, unsigned k) { while (k > 0) { k--; l = mk_succ(l); } return l; } unsigned parser::curr_level_lbp() const { if (curr_is_token(g_cup)) return g_level_cup_prec; else if (curr_is_token(g_add)) return g_level_add_prec; else return 0; } level parser::parse_max_imax(bool is_max) { auto p = pos(); next(); buffer lvls; while (curr_is_identifier() || curr_is_numeral() || curr_is_token(g_lparen)) { lvls.push_back(parse_level()); } if (lvls.size() < 2) throw parser_error("invalid level expression, max must have at least two arguments", p); unsigned i = lvls.size() - 1; level r = lvls[i]; while (i > 0) { --i; if (is_max) r = mk_max(lvls[i], r); else r = mk_imax(lvls[i], r); } return r; } level parser::parse_level_id() { auto p = pos(); name id = get_name_val(); next(); if (auto it = m_local_level_decls.find(id)) return *it; if (m_env.is_universe(id)) return mk_global_univ(id); if (auto it = get_alias_level(m_env, id)) return *it; throw parser_error(sstream() << "unknown universe '" << id << "'", p); } level parser::parse_level_nud() { if (curr_is_token_or_id(g_max)) { return parse_max_imax(true); } else if (curr_is_token_or_id(g_imax)) { return parse_max_imax(false); } else if (curr_is_token(g_lparen)) { next(); level l = parse_level(); check_token_next(g_rparen, "invalid level expression, ')' expected"); return l; } else if (curr_is_numeral()) { unsigned k = parse_small_nat(); return lift(level(), k); } else if (curr_is_identifier()) { return parse_level_id(); } else { throw parser_error("invalid level expression", pos()); } } level parser::parse_level_led(level left) { auto p = pos(); if (curr_is_token(g_cup)) { next(); level right = parse_level(g_level_cup_prec); return mk_max(left, right); } else if (curr_is_token(g_add)) { next(); if (curr_is_numeral()) { unsigned k = parse_small_nat(); return lift(left, k); } else { throw parser_error("invalid level expression, right hand side of '+' (aka universe lift operator) must be a numeral", p); } } else { throw parser_error("invalid level expression", p); } } level parser::parse_level(unsigned rbp) { level left = parse_level_nud(); while (rbp < curr_level_lbp()) { left = parse_level_led(left); } return left; } expr parser::mk_Type() { if (m_type_use_placeholder) { return mk_sort(mk_level_placeholder()); } else { unsigned i = 1; name l("l"); name r = l.append_after(i); while (m_local_level_decls.contains(r) || m_env.is_universe(r)) { i++; r = l.append_after(i); } level lvl = mk_param_univ(r); add_local_level(r, lvl); return mk_sort(lvl); } } expr parser::elaborate(expr const & e, level_param_names const &) { // TODO(Leo): return e; } std::pair parser::elaborate(expr const & t, expr const & v, level_param_names const &) { return mk_pair(t, v); } /** \brief Parse ID ':' expr, where the expression represents the type of the identifier. */ parameter parser::parse_binder_core(binder_info const & bi) { auto p = pos(); if (!curr_is_identifier()) throw parser_error("invalid binder, identifier expected", p); name id = get_name_val(); next(); expr type; if (curr_is_token(g_colon)) { next(); type = parse_expr(); } else { type = save_pos(mk_expr_placeholder(), p); } return parameter(p, save_pos(mk_local(id, type), p), bi); } parameter parser::parse_binder() { if (curr_is_identifier()) { return parse_binder_core(binder_info()); } else if (curr_is_token(g_lparen)) { next(); auto p = parse_binder_core(binder_info()); check_token_next(g_rparen, "invalid binder, ')' expected"); return p; } else if (curr_is_token(g_lcurly)) { next(); auto p = parse_binder_core(mk_implicit_binder_info()); check_token_next(g_rcurly, "invalid binder, '}' expected"); return p; } else if (curr_is_token(g_lbracket)) { next(); auto p = parse_binder_core(mk_cast_binder_info()); check_token_next(g_rbracket, "invalid binder, ']' expected"); return p; } else { throw parser_error("invalid binder, '(', '{', '[' or identifier expected", pos()); } } /** \brief Parse ID ... ID ':' expr, where the expression represents the type of the identifiers. */ void parser::parse_binder_block(buffer & r, binder_info const & bi) { buffer> names; parse_names(names); if (names.empty()) throw parser_error("invalid binder, identifier expected", pos()); optional type; if (curr_is_token(g_colon)) { next(); type = parse_expr(); } for (auto p : names) { expr arg_type = type ? *type : save_pos(mk_expr_placeholder(), p.first); expr local = save_pos(mk_local(p.second, arg_type), p.first); add_local(local); r.push_back(parameter(p.first, local, bi)); } } void parser::parse_binders_core(buffer & r) { if (curr_is_identifier()) { parse_binder_block(r, binder_info()); } else if (curr_is_token(g_lparen)) { next(); if (!parse_local_notation_decl()) parse_binder_block(r, binder_info()); check_token_next(g_rparen, "invalid binder, ')' expected"); } else if (curr_is_token(g_lcurly)) { next(); parse_binder_block(r, mk_implicit_binder_info()); check_token_next(g_rcurly, "invalid binder, '}' expected"); } else if (curr_is_token(g_lbracket)) { next(); parse_binder_block(r, mk_cast_binder_info()); check_token_next(g_rbracket, "invalid binder, ']' expected"); } else { return; } parse_binders_core(r); } local_environment parser::parse_binders(buffer & r) { flet save(m_env, m_env); // save environment local_expr_decls::mk_scope scope(m_local_decls); unsigned old_sz = r.size(); parse_binders_core(r); if (old_sz == r.size()) throw parser_error("invalid binder, '(', '{', '[' or identifier expected", pos()); return local_environment(m_env); } bool parser::parse_local_notation_decl() { if (curr_is_token(g_infix) || curr_is_token(g_infixl)) { next(); m_env = infixl_cmd_core(*this, false); return true; } else if (curr_is_token(g_infixr)) { next(); m_env = infixr_cmd_core(*this, false); return true; } else if (curr_is_token(g_postfix)) { next(); m_env = postfix_cmd_core(*this, false); return true; } else if (curr_is_token(g_notation)) { next(); m_env = notation_cmd_core(*this, false); return true; } else { return false; } } expr parser::parse_notation(parse_table t, expr * left) { lean_assert(curr() == scanner::token_kind::Keyword); auto p = pos(); buffer args; buffer ps; local_environment lenv(m_env); pos_info binder_pos; if (left) args.push_back(*left); while (true) { if (curr() != scanner::token_kind::Keyword) break; auto r = t.find(get_token_info().value()); if (!r) break; next(); notation::action const & a = r->first; switch (a.kind()) { case notation::action_kind::Skip: break; case notation::action_kind::Expr: args.push_back(parse_expr(a.rbp())); break; case notation::action_kind::Exprs: { buffer r_args; r_args.push_back(parse_expr(a.rbp())); while (curr_is_token(a.get_sep())) { next(); r_args.push_back(parse_expr(a.rbp())); } expr r = instantiate_rev(a.get_initial(), args.size(), args.data()); if (a.is_fold_right()) { unsigned i = r_args.size(); while (i > 0) { --i; args.push_back(r_args[i]); args.push_back(r); r = instantiate_rev(a.get_rec(), args.size(), args.data()); args.pop_back(); args.pop_back(); } } else { for (unsigned i = 0; i < r_args.size(); i++) { args.push_back(r_args[i]); args.push_back(r); r = instantiate_rev(a.get_rec(), args.size(), args.data()); args.pop_back(); args.pop_back(); } } args.push_back(r); break; } case notation::action_kind::Binder: binder_pos = pos(); ps.push_back(parse_binder()); break; case notation::action_kind::Binders: binder_pos = pos(); lenv = parse_binders(ps); break; case notation::action_kind::ScopedExpr: { expr r = parse_scoped_expr(ps, lenv, a.rbp()); if (is_var(a.get_rec(), 0)) { r = abstract(ps, r, a.use_lambda_abstraction(), binder_pos); } else { unsigned i = ps.size(); while (i > 0) { --i; expr const & l = ps[i].m_local; if (a.use_lambda_abstraction()) r = save_pos(Fun(l, r, ps[i].m_bi), binder_pos); else r = save_pos(Pi(l, r, ps[i].m_bi), binder_pos); args.push_back(r); r = instantiate_rev(a.get_rec(), args.size(), args.data()); args.pop_back(); } } args.push_back(r); break; } case notation::action_kind::Ext: args.push_back(a.get_parse_fn()(*this, args.size(), args.data(), p)); break; } t = r->second; } list const & as = t.is_accepting(); if (is_nil(as)) { if (p == pos()) throw parser_error(sstream() << "invalid expression", pos()); else throw parser_error(sstream() << "invalid expression starting at " << p.first << ":" << p.second, pos()); } buffer cs; for (expr const & a : as) { expr r = instantiate_rev(copy_with_new_pos(a, p), args.size(), args.data()); cs.push_back(r); } return mk_choice(cs.size(), cs.data()); } expr parser::parse_nud_notation() { return parse_notation(nud(), nullptr); } expr parser::parse_led_notation(expr left) { if (led().find(get_token_info().value())) return parse_notation(led(), &left); else return mk_app(left, parse_nud_notation(), pos_of(left)); } expr parser::parse_id() { auto p = pos(); name id = get_name_val(); next(); buffer lvl_buffer; levels ls; if (curr_is_token(g_llevel_curly)) { next(); while (!curr_is_token(g_rcurly)) { lvl_buffer.push_back(parse_level()); } next(); ls = to_list(lvl_buffer.begin(), lvl_buffer.end()); } // locals if (auto it1 = m_local_decls.find(id)) return copy_with_new_pos(propagate_levels(it1->m_local, ls), p); optional r; // globals if (m_env.find(id)) r = save_pos(mk_constant(id, ls), p); // aliases auto as = get_alias_exprs(m_env, id); if (!is_nil(as)) { buffer new_as; if (r) new_as.push_back(*r); for (auto const & e : as) { new_as.push_back(copy_with_new_pos(propagate_levels(e, ls), p)); } r = mk_choice(new_as.size(), new_as.data()); } if (!r) throw parser_error(sstream() << "unknown identifier '" << id << "'", p); return *r; } expr parser::parse_numeral_expr() { // TODO(Leo) return expr(); } expr parser::parse_decimal_expr() { // TODO(Leo) return expr(); } expr parser::parse_string_expr() { // TODO(Leo) return expr(); } expr parser::parse_nud() { switch (curr()) { case scanner::token_kind::Keyword: return parse_nud_notation(); case scanner::token_kind::Identifier: return parse_id(); case scanner::token_kind::Numeral: return parse_numeral_expr(); case scanner::token_kind::Decimal: return parse_decimal_expr(); case scanner::token_kind::String: return parse_string_expr(); default: throw parser_error("invalid expression, unexpected token", pos()); } } expr parser::parse_led(expr left) { switch (curr()) { case scanner::token_kind::Keyword: return parse_led_notation(left); case scanner::token_kind::Identifier: return mk_app(left, parse_id(), pos_of(left)); case scanner::token_kind::Numeral: return mk_app(left, parse_numeral_expr(), pos_of(left)); case scanner::token_kind::Decimal: return mk_app(left, parse_decimal_expr(), pos_of(left)); case scanner::token_kind::String: return mk_app(left, parse_string_expr(), pos_of(left)); default: return left; } } unsigned parser::curr_lbp() const { switch (curr()) { case scanner::token_kind::Keyword: return get_token_info().precedence(); case scanner::token_kind::CommandKeyword: case scanner::token_kind::Eof: case scanner::token_kind::ScriptBlock: return 0; case scanner::token_kind::Identifier: case scanner::token_kind::Numeral: case scanner::token_kind::Decimal: case scanner::token_kind::String: case scanner::token_kind::QuotedSymbol: return std::numeric_limits::max(); } lean_unreachable(); // LCOV_EXCL_LINE } expr parser::parse_expr(unsigned rbp) { expr left = parse_nud(); while (rbp < curr_lbp()) { left = parse_led(left); } return left; } expr parser::parse_scoped_expr(unsigned num_params, parameter const * ps, local_environment const & lenv, unsigned rbp) { if (num_params == 0) { return parse_expr(rbp); } else { local_scope scope(*this); m_env = lenv; for (unsigned i = 0; i < num_params; i++) add_local(ps[i].m_local); return parse_expr(rbp); } } expr parser::abstract(unsigned num_params, parameter const * ps, expr const & e, bool lambda, pos_info const & p) { buffer locals; for (unsigned i = 0; i < num_params; i++) locals.push_back(ps[i].m_local); expr r = ::lean::abstract(e, locals.size(), locals.data()); unsigned i = num_params; while (i > 0) { --i; expr const & l = ps[i].m_local; name const & n = local_pp_name(l); expr type = ::lean::abstract(mlocal_type(l), i, locals.data()); if (lambda) r = mk_lambda(n, type, r, ps[i].m_bi); else r = mk_pi(n, type, r, ps[i].m_bi); r = save_pos(r, p); } return r; } tactic parser::parse_tactic(unsigned /* rbp */) { if (curr_is_token(g_lparen)) { next(); auto r = parse_tactic(); check_token_next(g_rparen, "invalid tactic, ')' expected"); return r; } else if (curr_is_identifier()) { auto p = pos(); name id = get_name_val(); next(); if (auto it = tactic_cmds().find(id)) { return it->get_fn()(*this); } else { throw parser_error(sstream() << "unknown tactic '" << id << "'", p); } } else { throw parser_error("invalid tactic, '(' or tactic name expected", pos()); } } void parser::save_hint(expr const & e, tactic const & t) { m_hints.insert(mk_pair(get_tag(e), t)); } void parser::parse_command() { lean_assert(curr() == scanner::token_kind::CommandKeyword); m_last_cmd_pos = pos(); name const & cmd_name = get_token_info().value(); if (auto it = cmds().find(cmd_name)) { next(); m_env = it->get_fn()(*this); } else { auto p = pos(); next(); throw parser_error(sstream() << "unknown command '" << cmd_name << "'", p); } } void parser::parse_script(bool as_expr) { m_last_script_pos = pos(); if (!m_ss) throw exception("failed to execute Lua script, parser does not have a Lua interpreter"); std::string script_code = m_scanner.get_str_val(); if (as_expr) script_code = "return " + script_code; next(); using_script([&](lua_State * L) { dostring(L, script_code.c_str()); }); } static optional find_file(name const & f, char const * ext) { try { return optional(find_file(f, {ext})); } catch (...) { return optional(); } } void parser::parse_imports() { buffer olean_files; buffer lua_files; while (curr_is_token(g_import)) { m_last_cmd_pos = pos(); next(); while (curr_is_identifier()) { name f = get_name_val(); if (auto it = find_file(f, ".lua")) { if (!m_ss) throw parser_error("invalid import, Lua interpreter is not available", pos()); lua_files.push_back(*it); } else if (auto it = find_file(f, ".olean")) { olean_files.push_back(*it); } else { throw parser_error(sstream() << "invalid import, unknow module '" << f << "'", pos()); } next(); } } m_env = import_modules(m_env, olean_files.size(), olean_files.data(), m_num_threads, true, m_ios); using_script([&](lua_State *) { m_ss->exec_unprotected([&]() { for (auto const & f : lua_files) { m_ss->import_explicit(f.c_str()); }}); }); } bool parser::parse_commands() { // We disable hash-consing while parsing to make sure the pos-info are correct. scoped_expr_caching disable(false); try { bool done = false; parse_imports(); while (!done) { protected_call([&]() { check_interrupted(); switch (curr()) { case scanner::token_kind::CommandKeyword: parse_command(); break; case scanner::token_kind::ScriptBlock: parse_script(); break; case scanner::token_kind::Eof: done = true; break; case scanner::token_kind::Keyword: if (curr_is_token(g_period)) { next(); break; } default: throw parser_error("command expected", pos()); } }, [&]() { sync_command(); }); } } catch (interrupt_parser) {} return !m_found_errors; } bool parse_commands(environment & env, io_state & ios, std::istream & in, char const * strm_name, script_state * S, bool use_exceptions, unsigned num_threads) { parser p(env, ios, in, strm_name, S, use_exceptions, num_threads); bool r = p(); ios = p.ios(); env = p.env(); return r; } bool parse_commands(environment & env, io_state & ios, char const * fname, script_state * S, bool use_exceptions, unsigned num_threads) { std::ifstream in(fname); if (in.bad() || in.fail()) throw exception(sstream() << "failed to open file '" << fname << "'"); return parse_commands(env, ios, in, fname, S, use_exceptions, num_threads); } }