1 files changed, 86 insertions, 0 deletions
diff --git a/src/context.cc b/src/context.cc
new file mode 100644
index 0000000..1c1dce2
--- /dev/null
+++ b/src/context.cc
@@ -0,0 +1,86 @@
+// This file is part of Cait <https://github.com/Fuwn/cait>.
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 3.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// SPDX-License-Identifier: GPL-3.0-only
+
+#include <iostream>
+#include <optional>
+#include <thread>
+
+#include "context.hh"
+
+namespace cait {
+
+context::context(const parser &parser) {
+  this->jobs = std::thread::hardware_concurrency() + 2;
+
+  for (auto &parse_node : parser.nodes()) {
+    std::visit(
+        [this](auto &node) {
+          using T = std::decay_t<decltype(node)>;
+
+          if constexpr (std::is_same_v<T, node::variable>) {
+            this->identifiers.insert(
+                {std::make_pair(node.name().word,
+                                token_type::variable_declaration),
+                 node});
+          } else if constexpr (std::is_same_v<T, node::rule>) {
+            this->identifiers.insert(
+                {std::make_pair(node.name().word, token_type::rule_declaration),
+                 node});
+          }
+        },
+        parse_node);
+  }
+}
+
+[[maybe_unused]] auto context::variable(const std::string &variable)
+    -> std::optional<node::variable> {
+  auto *it = std::get_if<node::variable>(&this->identifiers.at(
+      std::make_pair(variable, token_type::variable_declaration)));
+
+  if (it == nullptr) {
+    return std::nullopt;
+  } else {
+    return *it;
+  }
+}
+
+[[maybe_unused]] auto context::rule(const std::string &variable)
+    -> std::optional<node::rule> {
+  auto *it = std::get_if<node::rule>(&this->identifiers.at(
+      std::make_pair(variable, token_type::rule_declaration)));
+
+  if (it == nullptr) {
+    return std::nullopt;
+  } else {
+    return *it;
+  }
+}
+
+[[maybe_unused]] auto context::build(const std::string &variable)
+    -> std::optional<node::build> {
+  auto *it = std::get_if<node::build>(&this->identifiers.at(
+      std::make_pair(variable, token_type::build_declaration)));
+
+  if (it == nullptr) {
+    return std::nullopt;
+  } else {
+    return *it;
+  }
+}
+
+} // namespace cait