libclang

drogonframework · Aug 8, 2021 · e1a0266 · e1a0266
1 parent 74343db
commit e1a0266
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/drogon_ctl/CMakeLists.txt b/drogon_ctl/CMakeLists.txt
@@ -79,6 +79,17 @@ else(WIN32)
   install(FILES "${CMAKE_CURRENT_BINARY_DIR}/dg_ctl"
           DESTINATION ${INSTALL_BIN_DIR})
 endif(WIN32)
+find_package(Clang REQUIRED)
+include_directories(${LLVM_INCLUDE_DIRS})
+separate_arguments(LLVM_DEFINITIONS_LIST NATIVE_COMMAND ${LLVM_DEFINITIONS})
+add_definitions(${LLVM_DEFINITIONS_LIST})
+# Find the libraries that correspond to the LLVM components
+# that we wish to use
+llvm_map_components_to_libnames(llvm_libs support core irreader)
+
+# Link against LLVM libraries
+target_link_libraries(drogon_ctl PRIVATE ${llvm_libs})
+
 set(ctl_targets _drogon_ctl drogon_ctl)
 set_property(TARGET ${ctl_targets} PROPERTY CXX_STANDARD ${DROGON_CXX_STANDARD})
 set_property(TARGET ${ctl_targets} PROPERTY CXX_STANDARD_REQUIRED ON)

diff --git a/drogon_ctl/create_swagger.cc b/drogon_ctl/create_swagger.cc
@@ -30,6 +30,8 @@
 #endif
 #include <fstream>
 
+#include <clang-c/Index.h>
+
 using namespace drogon_ctl;
 
 static void forEachControllerHeaderIn(
@@ -78,11 +80,20 @@ static void forEachControllerHeaderIn(
 static void parseControllerHeader(const std::string &headerFile,
                                   Json::Value &docs)
 {
-    std::ifstream infile(headerFile);
-
-    for (std::string buffer; std::getline(infile, buffer);)
+    CXIndex index = clang_createIndex(0, 0);
+    CXTranslationUnit unit = clang_parseTranslationUnit(
+        index,
+        headerFile.c_str(), nullptr, 0,
+        nullptr, 0,
+        CXTranslationUnit_None);
+    if (unit == nullptr)
     {
+        std::cerr << "Unable to parse translation unit. Quitting." << std::endl;
+        exit(-1);
     }
+
+    clang_disposeTranslationUnit(unit);
+    clang_disposeIndex(index);
 }
 static std::string makeSwaggerDocument(const Json::Value &config)
 {