|
#include "common/pch.h"
|
#include<codecvt>
|
#include<list>
|
#include<string>
|
#include"StringUtil.h"
|
#include"ProjectBuildNetworkApi.h"
|
#include "JsonUtil.h"
|
#include <chrono>
|
#include <ctime>
|
#include <random>
|
#include "md5.h"
|
#include <sstream>
|
#include <iostream>
|
#include <iomanip>
|
#include <codecvt>
|
#include <map>
|
#include <curl/curl.h>
|
|
|
string ProjectBuildNetworkApi::get_rquest_id()
|
{
|
string requestIdStr;
|
std::chrono::system_clock::time_point now = std::chrono::system_clock::now();
|
// ½«µ±Ç°Ê±¼äµãת»»Îªtime_tÀàÐÍ
|
std::time_t currentTime = std::chrono::system_clock::to_time_t(now);
|
requestIdStr.append("_").append(std::to_string(currentTime)).append("_");
|
// Éú³ÉËæ»úÊý
|
std::random_device rd;
|
std::mt19937 rng(rd());
|
std::uniform_int_distribution<int> intDist(1, 100000); // Éú³É·¶Î§ÔÚ1-100Ö®¼äµÄÕûÊý
|
int randomInt = intDist(rng);
|
requestIdStr.append(std::to_string(randomInt));
|
return requestIdStr;
|
}
|
|
string ProjectBuildNetworkApi::get_sign(rapidjson::Document& document)
|
{
|
//¼ÆËãÇ©Ãû
|
std::list<string> strList;
|
for (rapidjson::Value::ConstMemberIterator itr = document.MemberBegin(); itr != document.MemberEnd(); ++itr) {
|
std::cout << "¼ü: " << itr->name.GetString() << ", Öµ: ";
|
if (itr->value.IsObject() || itr->value.IsArray()) {
|
//Èç¹ûÊǶÔÏó
|
rapidjson::StringBuffer buffer;
|
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
|
itr->value.Accept(writer);
|
std::string jsonString = buffer.GetString();
|
strList.push_back(string(itr->name.GetString()).append("=").append(jsonString));
|
}
|
else {
|
if (itr->value.IsString()) {
|
strList.push_back(string(itr->name.GetString()).append("=").append(itr->value.GetString()));
|
}
|
else if (itr->value.IsInt()) {
|
strList.push_back(string(itr->name.GetString()).append("=").append(std::to_string(itr->value.GetInt())));
|
}
|
}
|
|
}
|
|
strList.sort();
|
strList.push_back("%Yeshi2014@#.");
|
string src = "";
|
int index = 0;
|
for (std::list<string>::iterator el = strList.begin(); el != strList.end(); el++) {
|
src.append(*el);
|
index++;
|
if (index != strList.size()) {
|
src.append("&");
|
}
|
}
|
//MD5
|
//cout << "¼ÓÃÜÇ°×Ö·û´®£º" <<src << endl;
|
string md5 = MD5(src).toStr();
|
return md5;
|
}
|
|
|
std::string ProjectBuildNetworkApi::load_request_data(std::string data)
|
{
|
rapidjson::Document document;
|
document.Parse(data.c_str());
|
string requestIdStr = (document["type"].IsInt() ? std::to_string(document["type"].GetInt()) : document["type"].GetString()).append(get_rquest_id());
|
rapidjson::Document::AllocatorType& allocator = document.GetAllocator();
|
|
// Ìí¼Ó request_id ×Ö¶Î
|
rapidjson::Value requestId(rapidjson::kStringType);
|
requestId.SetString(requestIdStr.c_str(), allocator);
|
document.AddMember("request_id", requestId, allocator);
|
// Ìí¼ÓÇ©Ãû£¬sign
|
string sign = get_sign(document);
|
rapidjson::Value signV(rapidjson::kStringType);
|
signV.SetString(sign.c_str(), allocator);
|
document.AddMember("sign", signV, allocator);
|
|
rapidjson::StringBuffer buffer;
|
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
|
document.Accept(writer);
|
|
std::string jsonString = buffer.GetString();
|
//·¢ËÍÊý¾Ý
|
std::stringstream ss;
|
ss << std::setw(8) << std::setfill('0') << jsonString.length();
|
std::string length_str = string("##").append(ss.str());
|
std::string fstr = length_str.append(jsonString);
|
return fstr;
|
}
|
|
|
static size_t WriteCallback(void* contents, size_t size, size_t nmemb, void* userp) {
|
((std::string*)userp)->append((char*)contents, size * nmemb);
|
return size * nmemb;
|
}
|
|
|
std::string ProjectBuildNetworkApi::base_request_get(string ip, int port,string path, map<string,string> params)
|
{
|
CURL* curl = curl_easy_init();
|
std::string response;
|
if (curl) {
|
string url = "http://";
|
url.append(ip).append(":").append(to_string(port)).append(path);
|
int index = 0;
|
if (!params.empty()) {
|
url.append("?");
|
for (auto it = params.begin(); it != params.end(); ++it) {
|
url.append(it->first).append("=");
|
char* output = curl_easy_escape(curl, it->second.c_str(), it->second.length());
|
url.append(output);
|
if (index != params.size() - 1) {
|
url.append("&");
|
}
|
index++;
|
}
|
}
|
curl_easy_setopt(curl, CURLOPT_URL, url.c_str()); // ÉèÖÃURL
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback); // ÉèÖûص÷
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);// ´«µÝ½ÓÊÕ»º³åÇø
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 600L);// Á¬½Ó³¬Ê± 600 Ãë
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 600L); // ´«Êä×ܳ¬Ê± 600 Ãë
|
CURLcode res = curl_easy_perform(curl); // Ö´ÐÐÇëÇó
|
if (res != CURLE_OK) {
|
std::cerr << "cURL failed: " << curl_easy_strerror(res) << std::endl;
|
throw;
|
}
|
else {
|
long response_code;
|
curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &response_code);
|
std::cout << "Response code: " << response_code << std::endl;
|
// ´òÓ¡ÏìÓ¦ÄÚÈÝ
|
std::cout << "Response body:\n" << response << std::endl;
|
}
|
|
curl_easy_cleanup(curl); // ÇåÀí×ÊÔ´
|
return response;
|
}
|
throw;
|
}
|
|
|
string ProjectBuildNetworkApi::build(string ip, int port, string dir, string script, string output)
|
{
|
map<string, string> params;
|
params["dir"] = dir;
|
params["script"] = script;
|
params["output"] = output;
|
return base_request_get(ip, port, "/build",params);
|
}
|
