github/crossdesk-server
1
0
Fork 0
mirror of https://github.com/kunkundi/crossdesk-server.git synced 2026-03-20 22:48:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use the one who creates the transmission as host

Browse Source
This commit is contained in:
dijunkun
2024-06-07 15:22:53 +08:00
parent 1e4dc4fcec
commit 9ddef55896
3 changed files with 58 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/transmission_manager.cpp
View File

@@ -40,6 +40,22 @@ bool TransmissionManager::BindUserIdToTransmission(
return true;
}
bool TransmissionManager::BindHostIdToTransmission(
const std::string& host_id, const std::string& transmission_id) {
if (transmission_host_id_list_.find(transmission_id) ==
transmission_host_id_list_.end()) {
transmission_host_id_list_[transmission_id] = host_id;
LOG_INFO("Bind host id [{}] to transmission [{}]", host_id,
transmission_id);
return true;
} else {
LOG_WARN("Host id [{}] already bind to transmission [{}]", host_id,
transmission_id);
return false;
}
return true;
}
bool TransmissionManager::BindPasswordToTransmission(
const std::string& password, const std::string& transmission_id) {
if (transmission_password_list_.find(transmission_id) ==
@@ -71,6 +87,15 @@ bool TransmissionManager::BindUserIdToWsHandle(
return true;
}
bool TransmissionManager::IsHostOfTransmission(
const std::string& user_id, const std::string& transmission_id) {
if (transmission_host_id_list_.find(transmission_id) ==
transmission_host_id_list_.end()) {
return false;
}
return transmission_host_id_list_[transmission_id] == user_id;
}
std::string TransmissionManager::ReleaseUserIdFromTransmission(
websocketpp::connection_hdl hdl) {
for (auto it = user_id_ws_hdl_list_.begin(); it != user_id_ws_hdl_list_.end();
@@ -126,6 +151,19 @@ bool TransmissionManager::ReleasePasswordFromTransmission(
return true;
}
bool TransmissionManager::ReleaseHostIdFromTransmission(
const std::string& transmission_id) {
if (transmission_host_id_list_.find(transmission_id) ==
transmission_host_id_list_.end()) {
LOG_ERROR("No transmission with id [{}]", transmission_id);
return false;
}
transmission_host_id_list_.erase(transmission_id);
return true;
}
websocketpp::connection_hdl TransmissionManager::GetWsHandle(
const std::string& user_id) {
if (user_id_ws_hdl_list_.find(user_id) != user_id_ws_hdl_list_.end()) {