X-Git-Url: https://gitweb.ps.run/toc/blobdiff_plain/dbc4a22d3c8c4189459f0361cb9da06415ec2dc9..be56e29997e17685eebf8bd7cb745183c60de7db:/src/find.h

diff --git a/src/find.h b/src/find.h
new file mode 100644
index 0000000..31400ab
--- /dev/null
+++ b/src/find.h
@@ -0,0 +1,121 @@
+#pragma once
+
+#include "repr.h"
+#include "typeInfo.h"
+
+#include <functional>
+#include <optional>
+
+template<typename T>
+using opt = std::optional<T>;
+
+template<typename T>
+opt<T> find(const std::vector<T> & ts, std::function<bool(T)> f)
+{
+  for (auto t : ts)
+    if (f(t))
+      return t;
+  return nullopt;
+}
+
+opt<Function> findFunction(
+  const Program & p,
+  const std::string & name,
+  const std::vector<std::string> & namespacePrefixes)
+{
+  if (namespacePrefixes.empty())
+  {
+    return find<Function>(p.functions, [&](Function f) { return f.name == name; });
+  }
+  
+  auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+
+  if (!n.has_value())
+    return nullopt;
+
+  for (int i = 1; i < namespacePrefixes.size(); i++)
+  {
+    n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+    
+    if (!n.has_value())
+      return nullopt;
+  }
+
+  return find<Function>(n.value().functions, [&](Function f) { return f.name == name; });
+}
+
+opt<Struct> findStruct(
+  const Program & p,
+  const std::string & name,
+  const std::vector<std::string> & namespacePrefixes)
+{
+  if (namespacePrefixes.empty())
+  {
+    return find<Struct>(p.structs, [&](Struct s) { return s.name == name; });
+  }
+  
+  auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+  
+  if (!n.has_value())
+    return nullopt;
+    
+  for (int i = 1; i < namespacePrefixes.size(); i++)
+  {
+    n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+
+    if (!n.has_value())
+      return nullopt;
+  }
+  return find<Struct>(n.value().structs, [&](Struct s) { return s.name == name; });
+}
+
+opt<Variable> findVariable(
+  const Program & p,
+  const std::string & name,
+  const std::vector<std::string> & namespacePrefixes)
+{
+  for (auto n : namespacePrefixes)
+    std::cout << n << std::endl;
+  if (namespacePrefixes.empty())
+  {
+    return find<Variable>(p.variables, [&](Variable v) { return v.name == name; });
+  }
+  
+  auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+  
+  if (!n.has_value())
+    return nullopt;
+      
+  for (int i = 1; i < namespacePrefixes.size(); i++)
+  {
+    n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+    
+    if (!n.has_value())
+      return nullopt;
+  }
+  return find<Variable>(n.value().variables, [&](Variable v) { return v.name == name; });
+}
+
+opt<Function> findStructMethod(
+  const Program & p,
+  const std::string & name,
+  TypeInfo ti)
+{
+  if (!ti.isStruct)
+    return nullopt;
+  auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);
+  if (!s.has_value())
+    return nullopt;
+  return find<StructMember<Function>>(s.value().methods, [&](Function f) { return f.name == name; });
+}
+
+opt<Variable> findStructMember(
+  const Program & p,
+  TypeInfo ti,
+  const std::string & name)
+{
+  auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);
+  if (!s.has_value())
+    return nullopt;
+  return find<StructMember<Variable>>(s.value().members, [&](Variable v) { return v.name == name; });
+}
\ No newline at end of file