ladybird/Userland/Libraries/LibCpp/Preprocessor.h
Itamar 0c4dc00f01 LibCpp: Import definitions from headers while processing
When the preprocessor encounters an #include statement it now adds
the preprocessor definitions that exist in the included header to its
own set of definitions.

We previously only aggregated the definitions from headers after
processing the source, which was less correct. (For example, there
could be an #ifdef that depends on a definition from another header).
2021-08-07 21:24:11 +02:00

72 lines
2 KiB
C++

/*
* Copyright (c) 2021, Itamar S. <itamar8910@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/FlyString.h>
#include <AK/Function.h>
#include <AK/HashMap.h>
#include <AK/Optional.h>
#include <AK/String.h>
#include <AK/StringView.h>
#include <AK/Vector.h>
#include <LibCpp/Token.h>
namespace Cpp {
class Preprocessor {
public:
explicit Preprocessor(const String& filename, const StringView& program);
Vector<Token> process_and_lex();
Vector<StringView> included_paths() const { return m_included_paths; }
struct DefinedValue {
Optional<StringView> value;
FlyString filename;
size_t line { 0 };
size_t column { 0 };
};
using Definitions = HashMap<StringView, DefinedValue>;
const Definitions& definitions() const { return m_definitions; }
void set_ignore_unsupported_keywords(bool ignore) { m_options.ignore_unsupported_keywords = ignore; }
void set_keep_include_statements(bool keep) { m_options.keep_include_statements = keep; }
Function<Definitions(StringView)> definitions_in_header_callback { nullptr };
private:
using PreprocessorKeyword = StringView;
PreprocessorKeyword handle_preprocessor_line(StringView const&);
void handle_preprocessor_keyword(StringView const& keyword, GenericLexer& line_lexer);
Vector<Token> process_line(StringView const& line);
String m_filename;
String m_program;
Definitions m_definitions;
Vector<StringView> m_lines;
size_t m_line_index { 0 };
size_t m_current_depth { 0 };
Vector<size_t> m_depths_of_taken_branches;
Vector<size_t> m_depths_of_not_taken_branches;
enum class State {
Normal,
SkipIfBranch,
SkipElseBranch
};
State m_state { State::Normal };
Vector<StringView> m_included_paths;
String m_processed_text;
struct Options {
bool ignore_unsupported_keywords { false };
bool keep_include_statements { false };
} m_options;
};
}