start adding updatereplaydata function

etternagame · Dec 1, 2018 · aeb0ae6 · aeb0ae6
1 parent d1c403d
commit aeb0ae6
Show file tree

Hide file tree

Showing 2 changed files with 139 additions and 3 deletions.
diff --git a/src/DownloadManager.cpp b/src/DownloadManager.cpp
@@ -1142,6 +1142,106 @@ DownloadManager::UploadScoreWithReplayDataFromDisk(string sk,
 	HTTPRequests.push_back(req);
 	return;
 }
+void // for online replay viewing accuracy -mina
+DownloadManager::UpdateOnlineScoreReplayData(string& sk,
+												   function<void()> callback)
+{
+	if (!LoggedIn())
+		return;
+
+	auto hs = SCOREMAN->GetScoresByKey().at(sk);
+	CURL* curlHandle = initCURLHandle(true);
+	string url = serverURL.Get() + "/updatereplaydata";
+	curl_httppost* form = nullptr;
+	curl_httppost* lastPtr = nullptr;
+	SetCURLFormPostField(
+	  curlHandle, form, lastPtr, "scorekey", hs->GetScoreKey());
+	string replayString;
+	vector<float> offsets = hs->GetOffsetVector();
+	vector<int> columns = hs->GetTrackVector();
+	vector<TapNoteType> types = hs->GetTapNoteTypeVector();
+	auto& rows = hs->GetNoteRowVector();
+
+	if (!offsets.empty()) {
+		replayString = "[";
+		auto steps = SONGMAN->GetStepsByChartkey(hs->GetChartKey());
+		if (steps == nullptr)
+			return;
+		vector<float> timestamps =
+		  steps->GetTimingData()->ConvertReplayNoteRowsToTimestamps(
+			rows, hs->GetMusicRate());
+		for (size_t i = 0; i < offsets.size(); i++) {
+			replayString += "[";
+			replayString += to_string(timestamps[i]) + ",";
+			replayString += to_string(1000.f * offsets[i]) + ",";
+			if (hs->GetReplayType() == 2) {
+				replayString += to_string(columns[i]) + ",";
+				replayString += to_string(types[i]) + ",";
+			}
+			replayString += to_string(rows[i]);
+			replayString += "],";
+		}
+		replayString =
+		  replayString.substr(0, replayString.size() - 1); // remove ","
+		replayString += "]";
+		hs->UnloadReplayData();
+	} else
+		replayString = "";
+	SetCURLFormPostField(
+	  curlHandle, form, lastPtr, "replay_data", replayString);
+	SetCURLPostToURL(curlHandle, url);
+	curl_easy_setopt(curlHandle, CURLOPT_HTTPPOST, form);
+	auto done = [this, hs, callback](HTTPRequest& req, CURLMsg*) {
+		long response_code;
+		curl_easy_getinfo(req.handle, CURLINFO_RESPONSE_CODE, &response_code);
+		json j;
+		try {
+			j = json::parse(req.result);
+			bool delay = false;
+			try {
+				auto errors = j["errors"];
+				for (auto error : errors) {
+					int status = error["status"];
+					if (status == 22) {
+						delay = true;
+						DLMAN->StartSession(
+						  DLMAN->sessionUser,
+						  DLMAN->sessionPass,
+						  [hs](bool logged) {
+							  if (logged) {
+								  DLMAN->UploadScoreWithReplayDataFromDisk(
+									hs->GetScoreKey());
+							  }
+						  });
+					} else if (status == 404 || status == 405 ||
+							   status == 406) {
+						hs->AddUploadedServer(serverURL.Get());
+					}
+				}
+			} catch (exception e) {
+			}
+			if (!delay && j["data"]["type"] == "ssrResults") {
+				auto diffs = j["data"]["attributes"]["diff"];
+				FOREACH_ENUM(Skillset, ss)
+				if (ss != Skill_Overall)
+					(DLMAN->sessionRatings)[ss] +=
+					  diffs.value(SkillsetToString(ss), 0.0);
+				(DLMAN->sessionRatings)[Skill_Overall] +=
+				  diffs.value("Rating", 0.0);
+				hs->AddUploadedServer(serverURL.Get());
+				HTTPRunning = response_code;
+			}
+		} catch (exception e) {
+		}
+		if (callback)
+			callback();
+	};
+	HTTPRequest* req = new HTTPRequest(curlHandle, done);
+	SetCURLResultsString(curlHandle, &(req->result));
+	curl_multi_add_handle(mHTTPHandle, req->handle);
+	HTTPRequests.push_back(req);
+	return;
+}
 void
 DownloadManager::EndSessionIfExists()
 {
@@ -1887,6 +1987,41 @@ DownloadManager::StartSession(string user,
 	HTTPRequests.push_back(req);
 }
 
+void
+UpdateReplayDataSequentially(deque<HighScore*> toUpload)
+{
+	auto it = toUpload.begin();
+	if (it != toUpload.end()) {
+		toUpload.pop_front();
+		auto& hs = (*it);
+		DLMAN->UpdateOnlineScoreReplayData(
+		  hs->GetScoreKey(), [hs, toUpload]() {
+			  hs->AddUploadedServer(serverURL.Get());
+			  UpdateReplayDataSequentially(toUpload);
+		  });
+	}
+	return;
+}
+bool
+DownloadManager::UpdateOnlineScoreReplayData()
+{
+	if (!LoggedIn())
+		return false;
+	auto scores = SCOREMAN->GetAllPBPtrs();
+	deque<HighScore*> toUpload;
+	for (auto& vec : scores) {
+		for (auto& scorePtr : vec) {
+			auto ts = scorePtr->GetTopScore();
+			if ((ts == 1 || ts == 2) &&
+				!scorePtr->IsUploadedToServer(serverURL.Get())) {
+				if (scorePtr->HasReplayData())
+					toUpload.emplace_back(scorePtr);
+			}
+		}
+	}
+	UpdateReplayDataSequentially(toUpload);
+	return true;
+}
 void
 uploadSequentially(deque<HighScore*> toUpload)
 {
@@ -1903,7 +2038,7 @@ uploadSequentially(deque<HighScore*> toUpload)
 	return;
 }
 bool
-DownloadManager::UploadScores()
+DownloadManager::UploadScores()		// keeping in case we want to use this later -mina
 {
 	if (!LoggedIn())
 		return false;
@@ -1914,15 +2049,14 @@ DownloadManager::UploadScores()
 			auto ts = scorePtr->GetTopScore();
 			if ((ts == 1 || ts == 2) &&
 				!scorePtr->IsUploadedToServer(serverURL.Get())) {
-				if (scorePtr->HasReplayData())
+				if (scorePtr->HasReplayData())	// we should remove this check maybe -mina
 					toUpload.emplace_back(scorePtr);
 			}
 		}
 	}
 	uploadSequentially(toUpload);
 	return true;
 }
-
 int
 DownloadManager::GetSkillsetRank(Skillset ss)
 {

diff --git a/src/DownloadManager.h b/src/DownloadManager.h
@@ -226,6 +226,7 @@ class DownloadManager
 						done); // Sends login request if not already logging in
 	void OnLogin();
 	bool UploadScores(); // Uploads all scores not yet uploaded to current
+	bool UpdateOnlineScoreReplayData();	// attempts updates existing replaydata
 						 // server (Async, 1 request per score)
 	void RefreshPackList(string url);
 
@@ -249,6 +250,7 @@ class DownloadManager
 	void UploadScoreWithReplayDataFromDisk(
 	  string sk,
 	  function<void()> callback = function<void()>());
+	void UpdateOnlineScoreReplayData(string& sk, function<void()> callback);
 	void UploadScore(HighScore* hs);
 
 	bool ShouldUploadScores();