aboutsummaryrefslogtreecommitdiff
path: root/asm/preprocesser.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'asm/preprocesser.cpp')
-rw-r--r--asm/preprocesser.cpp52
1 files changed, 52 insertions, 0 deletions
diff --git a/asm/preprocesser.cpp b/asm/preprocesser.cpp
index 615cf93..d429bf1 100644
--- a/asm/preprocesser.cpp
+++ b/asm/preprocesser.cpp
@@ -11,3 +11,55 @@
*/
#include "./preprocesser.hpp"
+#include "./base.hpp"
+
+#include <algorithm>
+#include <unordered_map>
+
+using std::pair, std::vector, std::make_pair, std::string, std::string_view;
+
+#define ERR(E) std::make_pair(tokens, (E))
+#define VAL(E) std::make_pair(E, pp_err_t{pp_err_type_t::OK})
+
+pair<vector<token_t *>, pp_err_t>
+preprocess_use_blocks(vector<token_t *> tokens)
+{
+ vector<token_t *> new_tokens;
+ for (size_t i = 0; i < tokens.size(); ++i)
+ {
+ token_t *t = tokens[i];
+ if (t->type == token_type_t::PP_USE)
+ {
+ if (i + 1 >= tokens.size() ||
+ tokens[i + 1]->type != token_type_t::LITERAL_STRING)
+ {
+ new_tokens.clear();
+ return ERR(pp_err_t(pp_err_type_t::EXPECTED_STRING, t));
+ }
+
+ token_t *name = tokens[i + 1];
+ auto source = read_file(name->content.c_str());
+ if (!source)
+ {
+ new_tokens.clear();
+ return ERR(pp_err_t(pp_err_type_t::FILE_NONEXISTENT, name));
+ }
+
+ std::vector<token_t *> ftokens;
+ lerr_t lerr = tokenise_buffer(source.value(), ftokens);
+ if (lerr != lerr_t::OK)
+ {
+ new_tokens.clear();
+ return ERR(pp_err_t(pp_err_type_t::FILE_PARSE_ERROR, name, lerr));
+ }
+
+ new_tokens.insert(new_tokens.end(), ftokens.begin(), ftokens.end());
+
+ i += 2;
+ }
+ else
+ new_tokens.push_back(new token_t{*t});
+ }
+ return VAL(new_tokens);
+}
+