Merge pull request #216 from mingzhe09088/fix-tc-lang-fb

Fix tc_lang runtime error in FBcode

Author: prigoyal

CMakeLists.txt

@@ -27,6 +27,8 @@ message(STATUS "CMAKE_INSTALL_PREFIX is ${CMAKE_INSTALL_PREFIX}")
 
 add_subdirectory(third-party/pybind11)
 
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DTC_DIR=\"\\\"$ENV{TC_DIR}\\\"\" ")
+
 set(LLVM_DIR ${CLANG_PREFIX}/lib/cmake/llvm)
 message(STATUS "Looking for LLVM in ${LLVM_DIR}")
 find_package(LLVM REQUIRED LLVM)

include/tc/lang/tree.h

@@ -17,6 +17,7 @@
 
 #include <functional>
 #include <memory>
+#include <sstream>
 #include <vector>
 
 #include "tc/lang/lexer.h"

test/test_lang.cc

@@ -29,7 +29,12 @@
 
 using namespace lang;
 
-const std::string expected_file_path = "src/lang/test_expected/";
+#ifndef TC_DIR
+#error "TC_DIR must be defined"
+#endif
+
+const std::string expected_file_path =
+    std::string(TC_DIR) + "/src/lang/test_expected/";
 
 static inline void barf(const char* fmt, ...) {
   char msg[2048];