lean2/src/frontends/lean/dependencies.cpp

/*
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 <utility>
#include <fstream>
#include <string>
#include "util/sstream.h"
#include "util/lean_path.h"
#include "frontends/lean/scanner.h"

namespace lean {
void display_deps(environment const & env, std::ostream & out, char const * fname) {
    name import("import");
    std::ifstream in(fname);
    if (in.bad() || in.fail())
        throw exception(sstream() << "failed to open file '" << fname << "'");
    scanner s(in, fname);
    bool import_args = false;
    while (true) {
        scanner::token_kind t = scanner::token_kind::Identifier;
        try {
            t = s.scan(env);
        } catch (exception &) {
            continue;
        }
        if (t == scanner::token_kind::Eof) {
            return;
        } else if (t == scanner::token_kind::CommandKeyword && s.get_token_info().value() == import) {
            import_args = true;
        } else if (import_args == true && t == scanner::token_kind::Identifier) {
            std::string m_name = find_file(name_to_file(s.get_name_val()), {".lean", ".olean", ".lua"});
            int last_idx = m_name.find_last_of(".");
            std::string rawname = m_name.substr(0, last_idx);
            std::string ext = m_name.substr(last_idx);
            if (ext == ".lean")
                m_name = rawname + ".olean";
            display_path(out, m_name);
            out << "\n";
        } else {
            import_args = false;
        }
    }
}
}