Merge pull request #27 from lglglglgy/main

🚀 refined file tree system

Author: stepbystepcode

CMakeLists.txt

@@ -24,7 +24,7 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
 set(CMAKE_CXX_EXTENSIONS OFF)
 
 
-add_executable(${PROJECT_NAME} main.cc mysql.cc user_controller.cc jwt_controller.cc msg_controller.cc)
+add_executable(${PROJECT_NAME} main.cc mysql.cc user_controller.cc jwt_controller.cc msg_controller.cc file_controller.cc)
 
 
 # 链接MySQL Connector/C++库
@@ -52,7 +52,7 @@ target_link_libraries(${PROJECT_NAME} PRIVATE -L${MYSQL_CONNECTOR_CPP_ROOT}/lib6
 find_package(OpenSSL REQUIRED)
 
 # 添加 OpenSSL 库到可执行文件链接参数
-target_link_libraries(flypen PRIVATE OpenSSL::Crypto)
+target_link_libraries(${PROJECT_NAME} PRIVATE OpenSSL::Crypto)
 
 #link_directories(/home/lgy/c++kc/mysql-connector/lib64)
 #
@@ -95,4 +95,4 @@ target_sources(${PROJECT_NAME}
 
 # ##############################################################################
 
-add_subdirectory(test)
+#add_subdirectory(test)

file_controller.cc

@@ -0,0 +1,36 @@
+#include "file_controller.h"
+#include <iostream>
+#include <stdexcept>
+#include <stdio.h>
+#include <string>
+#include <json/json.h>
+
+
+std::string shell_commons(const char* cmd) {
+    char buffer[128];
+    std::string result = "";
+    FILE* pipe = popen(cmd, "r");
+    if (!pipe) throw std::runtime_error("popen() failed!");
+    try {
+        while (fgets(buffer, sizeof buffer, pipe) != NULL) {
+            result += buffer;
+        }
+    } catch (...) {
+        pclose(pipe);
+        throw;
+    }
+    pclose(pipe);
+    return result;
+}
+
+std::string genTree(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
+{
+    char *pathvar;
+    pathvar = getenv("PWD");
+    std::string result = shell_commons(("cd "+std::string(pathvar)+"/../root " + "&&"+"tree -J" ).c_str()) ;
+    std::cout << result << std::endl;
+    auto res = HttpResponse::newHttpResponse();
+    res ->setBody(result);  
+    callback(res);
+    return result;
+}

file_controller.h

@@ -0,0 +1,9 @@
+#ifndef _FILE_CONTROLLER_H_
+#define _FILE_CONTROLLER_H_
+#include <string>
+#include <drogon/drogon.h>
+using namespace drogon;
+std::string shell_commons(const char* cmd);
+std::string genTree(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
+//std::string 
+#endif

main.cc

@@ -2,6 +2,7 @@
 #include <drogon/drogon.h>
 #include "msg_controller.h"
 #include "user_controller.h"
+#include "file_controller.h"
 #include <drogon/WebSocketController.h>
 using namespace drogon;
 std::unordered_map<std::string, WebSocketConnectionPtr> clientTable;
@@ -29,8 +30,13 @@ int main() {
       drogon::app().registerHandler("/api/info", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
         info(req, std::move(callback));
     });
+
+    drogon::app().registerHandler("/api/file/tree", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
+        genTree(req, std::move(callback));
+    });  
       drogon::app().registerHandler("/api/avatar", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
         avatar(req, std::move(callback));
+
     });
     drogon::app().run();
     return 0;