fix many bug

Author: stepbystepcode

main.cc

@@ -28,6 +28,9 @@ 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/avatar", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
+        avatar(req, std::move(callback));
     });
     drogon::app().run();
     return 0;

mysql.cc

@@ -440,6 +440,7 @@ bool sql_check(std::string user, std::string passwd)
     return result;
 }
 Json::Value sql_find_my_msg(std::string me)
+
 {
     std::cout << "login user: " << me << std::endl;
     try
@@ -507,4 +508,15 @@ Json::Value sql_find_my_msg(std::string me)
         std::cerr << "SQL Exception: " << e.what() << std::endl;
         return "error";
     }
+}
+void set_avatar(std::string person, int avatar)
+{
+    sql::mysql::MySQL_Driver *driver = sql::mysql::get_mysql_driver_instance();
+    sql::Connection *con = driver->connect("tcp://8.130.48.157:3306", "root", "abc.123");
+    con->setSchema("flypen");
+    std::string update_sql = "UPDATE users SET avatar = ? WHERE username = ?";
+    sql::PreparedStatement *updateStatement = con->prepareStatement(update_sql);
+    updateStatement->setInt(1, avatar);
+    updateStatement->setString(2, person);
+    updateStatement->execute();
 }

mysql.h

@@ -16,5 +16,6 @@ Json::Value sql_find_my_msg(std::string);
 Json::Value get_chat_info(std::string,std::string="");
 
 void sql_delete_operation(std::string,std::string);
+void set_avatar(std::string person, int avatar);
 
 #endif

user_controller.cc

@@ -17,11 +17,10 @@ void Handle(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr
     Json::FastWriter writer;
     std::string msg = handler(req_json);
     res_json["msg"] = msg;
-    if (msg=="Login Success"){
+    if (msg.find("Success")!= std::string::npos){
         res_json["token"] = jwtGen(req_json);
         res_json["username"] = req_json["username"].asString();
-    }else
-        res_json["token"] = "";
+    }
     auto output = writer.write(res_json);
     auto res = HttpResponse::newHttpResponse();
     res->addHeader("Access-Control-Allow-Origin", "*");
@@ -42,4 +41,16 @@ std::string loginUser(const Json::Value& req_json) {
         return "Login Success";
     }else
         return "Login Failed";
+}
+void avatar(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
+    auto res = HttpResponse::newHttpResponse();
+    res->addHeader("Access-Control-Allow-Origin", "*");
+    if (jwtVerify(req)) {
+        std::string receiver = jwtDecrypt(req->getHeader("Authorization").substr(7));
+        set_avatar(receiver, stoi(req->getParameter("avatar")));
+        res->setBody("Success");
+    } else {
+        res->setBody("No Authorization");
+    }
+    callback(res);
 }

user_controller.h

@@ -6,4 +6,5 @@ typedef std::string (*HandlerFunc)(const Json::Value&);
 void Handle(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback, HandlerFunc handler);
 std::string registerUser(const Json::Value& req_json);
 std::string loginUser(const Json::Value& req_json);
+void avatar(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
 #endif