json QueueFileUtils::serializeFileProperty(const QueueItemPtr& aFile, int aPropertyName) noexcept {
switch (aPropertyName) {
case PROP_SOURCES:
{
auto c = QueueManager::getInstance()->getSourceCount(aFile);
return Serializer::serializeSourceCount(c);
}
case PROP_STATUS:
{
return {
{ "finished", aFile->isSet(QueueItem::FLAG_FINISHED) },
{ "str", formatDisplayStatus(aFile) },
};
}
case PROP_PRIORITY: {
return Serializer::serializePriority(*aFile.get());
}
case PROP_TYPE:
{
return Serializer::serializeFileType(aFile->getTarget());
}
}
dcassert(0);
return nullptr;
}
int FileQueue::isFileQueued(const TTHValue& aTTH, const string& fileName) noexcept {
QueueItemPtr qi = getQueuedFile(aTTH, fileName);
if (qi) {
return (qi->isFinished() ? 2 : 1);
}
return 0;
}
void ViewFileManager::on(QueueManagerListener::ItemFinished, const QueueItemPtr& aQI, const string&, const HintedUser&, int64_t /*aSpeed*/) noexcept {
if (!isViewedItem(aQI))
return;
auto file = getFile(aQI->getTTH());
if (file) {
file->onRemovedQueue(aQI->getTarget(), true);
fire(ViewFileManagerListener::FileFinished(), file);
}
}
void ViewFileManager::on(QueueManagerListener::ItemStatusUpdated, const QueueItemPtr& aQI) noexcept {
if (!isViewedItem(aQI)) {
return;
}
auto file = getFile(aQI->getTTH());
if (file) {
file->onProgress(aQI->getTarget(), aQI->getDownloadedBytes());
}
}
void ViewFileManager::on(QueueManagerListener::ItemAdded, const QueueItemPtr& aQI) noexcept {
if (!isViewedItem(aQI)) {
return;
}
auto file = createFile(aQI->getTarget(), aQI->getTTH(), aQI->isSet(QueueItem::FLAG_TEXT), false);
if (file) {
file->onAddedQueue(aQI->getTarget(), aQI->getSize());
}
}
void BundleQueue::removeBundleItem(QueueItemPtr& aQI, bool aDownloadFinished) noexcept {
dcassert(aQI->getBundle());
aQI->getBundle()->removeQueue(aQI, aDownloadFinished);
if (!aQI->getBundle()->isFileBundle()) {
forEachPath(aQI->getBundle(), aQI->getTarget(), [&](PathInfo& aInfo) {
if (aQI->isDownloaded()) {
aInfo.finishedFiles--;
} else {
aInfo.queuedFiles--;
}
if (!aDownloadFinished) {
aInfo.size -= aQI->getSize();
}
#ifdef _DEBUG
if (aInfo.queuedFiles == 0 && aInfo.finishedFiles == 0) {
dcassert(aInfo.size == 0);
}
dcassert(aInfo.size >= 0 && aInfo.finishedFiles >= 0 && aInfo.queuedFiles >= 0);
#endif
});
}
if (aDownloadFinished || !aQI->isDownloaded()) {
queueSize -= aQI->getSize();
dcassert(queueSize >= 0);
}
}
std::string QueueFileUtils::getStringInfo(const QueueItemPtr& aItem, int aPropertyName) noexcept {
switch (aPropertyName) {
case PROP_NAME: return getDisplayName(aItem);
case PROP_TARGET: return aItem->getTarget();
case PROP_TYPE: return Format::formatFileType(aItem->getTarget());
case PROP_STATUS: return formatDisplayStatus(aItem);
case PROP_PRIORITY: return AirUtil::getPrioText(aItem->getPriority());
case PROP_SOURCES: return formatFileSources(aItem);
case PROP_TTH: return aItem->getTTH().toBase32();
default: dcassert(0); return Util::emptyString;
}
}
std::string QueueFileUtils::formatDisplayStatus(const QueueItemPtr& aItem) noexcept {
if (aItem->isSet(QueueItem::FLAG_FINISHED)) {
return STRING(FINISHED);
}
auto percentage = aItem->getPercentage(QueueManager::getInstance()->getDownloadedBytes(aItem));
if (aItem->isPausedPrio()) {
return STRING_F(PAUSED_PCT, percentage);
} else if (QueueManager::getInstance()->isWaiting(aItem)) {
return STRING_F(WAITING_PCT, percentage);
} else {
return STRING_F(RUNNING_PCT, percentage);
}
}
int QueueFileUtils::compareFiles(const QueueItemPtr& a, const QueueItemPtr& b, int aPropertyName) noexcept {
switch (aPropertyName) {
case PROP_NAME: {
return Util::pathSort(a->getTarget(), b->getTarget());
}
case PROP_TYPE: {
return Util::stricmp(Util::getFileExt(a->getTarget()), Util::getFileExt(b->getTarget()));
}
case PROP_PRIORITY: {
COMPARE_FINISHED(a, b);
return compare(static_cast<int>(a->getPriority()), static_cast<int>(b->getPriority()));
}
case PROP_STATUS: {
COMPARE_FINISHED(a, b);
return compare(
a->getPercentage(QueueManager::getInstance()->getDownloadedBytes(a)),
b->getPercentage(QueueManager::getInstance()->getDownloadedBytes(b))
);
}
case PROP_SOURCES: {
COMPARE_FINISHED(a, b);
auto countsA = QueueManager::getInstance()->getSourceCount(a);
auto countsB = QueueManager::getInstance()->getSourceCount(b);
return QueueItemBase::SourceCount::compare(countsA, countsB);
}
default:
dcassert(0);
}
return 0;
}
void DirectoryListingManager::on(QueueManagerListener::ItemRemoved, const QueueItemPtr& qi, bool aFinished) noexcept {
if (!qi->isSet(QueueItem::FLAG_USER_LIST))
return;
auto u = qi->getSources()[0].getUser();
if (qi->isSet(QueueItem::FLAG_DIRECTORY_DOWNLOAD) && !aFinished)
removeDirectoryDownload(u, qi->getTempTarget(), qi->isSet(QueueItem::FLAG_PARTIAL_LIST));
if (qi->isSet(QueueItem::FLAG_CLIENT_VIEW)) {
DirectoryListingPtr dl = nullptr;
{
RLock l(cs);
auto p = viewedLists.find(u);
if (p == viewedLists.end()) {
dcassert(0);
return;
}
dl = p->second;
}
dl->onListRemovedQueue(qi->getTarget(), qi->getTempTarget(), aFinished);
bool closing = (dl->getClosing() || !dl->hasCompletedDownloads());
if (!aFinished && !dl->hasDownloads() && closing) {
removeList(u);
}
}
}
void ViewFileManager::on(QueueManagerListener::ItemRemoved, const QueueItemPtr& aQI, bool finished) noexcept {
if (finished || !isViewedItem(aQI)) {
return;
}
removeFile(aQI->getTTH());
}
void UserQueue::addQI(QueueItemPtr& qi, const HintedUser& aUser, bool aIsBadSource /*false*/) noexcept{
if (qi->getPriority() == QueueItem::HIGHEST) {
auto& l = userPrioQueue[aUser.user];
l.insert(upper_bound(l.begin(), l.end(), qi, QueueItem::SizeSortOrder()), qi);
}
BundlePtr bundle = qi->getBundle();
if (bundle) {
aUser.user->addQueued(qi->getSize());
if (bundle->addUserQueue(qi, aUser, aIsBadSource)) {
addBundle(bundle, aUser);
} else {
dcassert(userBundleQueue.find(aUser.user) != userBundleQueue.end());
}
}
}
void UserQueue::addQI(QueueItemPtr& qi, const HintedUser& aUser, bool newBundle /*false*/, bool isBadSource /*false*/) {
if (qi->getPriority() == QueueItem::HIGHEST) {
auto& l = userPrioQueue[aUser.user];
l.insert(upper_bound(l.begin(), l.end(), qi, QueueItem::SizeSortOrder()), qi);
}
BundlePtr bundle = qi->getBundle();
if (bundle) {
if (bundle->addUserQueue(qi, aUser, isBadSource)) {
addBundle(bundle, aUser);
if (!newBundle) {
QueueManager::getInstance()->fire(QueueManagerListener::BundleSources(), bundle);
}
} else {
dcassert(userBundleQueue.find(aUser.user) != userBundleQueue.end());
}
}
}
void UserQueue::removeQI(QueueItemPtr& qi, const UserPtr& aUser, bool removeRunning /*true*/, bool addBad /*false*/, bool fireSources /*false*/) {
if(removeRunning) {
qi->removeDownloads(aUser);
}
dcassert(qi->isSource(aUser));
BundlePtr bundle = qi->getBundle();
if (bundle) {
if (!bundle->isSource(aUser)) {
return;
}
if (qi->getBundle()->removeUserQueue(qi, aUser, addBad)) {
removeBundle(bundle, aUser);
if (fireSources) {
QueueManager::getInstance()->fire(QueueManagerListener::BundleSources(), bundle);
}
} else {
dcassert(userBundleQueue.find(aUser) != userBundleQueue.end());
}
}
if (qi->getPriority() == QueueItem::HIGHEST) {
auto j = userPrioQueue.find(aUser);
dcassert(j != userPrioQueue.end());
if (j == userPrioQueue.end()) {
return;
}
auto& l = j->second;
auto i = find(l.begin(), l.end(), qi);
dcassert(i != l.end());
if (i == l.end()) {
return;
}
l.erase(i);
if(l.empty()) {
userPrioQueue.erase(j);
}
}
}
string QueueFileUtils::getDisplayName(const QueueItemPtr& aItem) noexcept {
if (aItem->getBundle() && !aItem->getBundle()->isFileBundle()) {
return aItem->getTarget().substr(aItem->getBundle()->getTarget().size(), aItem->getTarget().size());
}
return aItem->getTargetFileName();
}
void FinishedManager::on(QueueManagerListener::Finished, const QueueItemPtr& qi, const string&, const HintedUser& aUser, int64_t aSpeed) noexcept {
if(!qi->isSet(QueueItem::FLAG_USER_LIST) || SETTING(LOG_FILELIST_TRANSFERS)) {
FinishedItemPtr item = new FinishedItem(qi->getTarget(), aUser, qi->getSize(), static_cast<int64_t>(aSpeed), GET_TIME(), qi->getTTH().toBase32());
{
Lock l(cs);
downloads.push_back(item);
}
fire(FinishedManagerListener::AddedDl(), item);
if(SETTING(SYSTEM_SHOW_DOWNLOADS)) {
size_t BUF_SIZE = STRING(FINISHED_DOWNLOAD).size() + UNC_MAX_PATH + 128;
char* buf = new char[BUF_SIZE];
snprintf(buf, BUF_SIZE, CSTRING(FINISHED_DOWNLOAD), Util::getFileName(qi->getTarget()).c_str(),
Util::toString(ClientManager::getInstance()->getNicks(aUser)).c_str());
LogManager::getInstance()->message(buf, LogManager::LOG_INFO);
delete[] buf;
}
}
}
void FileQueue::add(QueueItemPtr& qi) noexcept {
if (!qi->isSet(QueueItem::FLAG_USER_LIST) && !qi->isSet(QueueItem::FLAG_CLIENT_VIEW) && !qi->isSet(QueueItem::FLAG_FINISHED)) {
dcassert(qi->getSize() >= 0);
queueSize += qi->getSize();
}
dcassert(queueSize >= 0);
queue.emplace(const_cast<string*>(&qi->getTarget()), qi);
tthIndex.emplace(const_cast<TTHValue*>(&qi->getTTH()), qi);
}
void DirectoryListingManager::on(QueueManagerListener::ItemFinished, const QueueItemPtr& qi, const string& dir, const HintedUser& aUser, int64_t /*aSpeed*/) noexcept {
if (!qi->isSet(QueueItem::FLAG_CLIENT_VIEW) || !qi->isSet(QueueItem::FLAG_USER_LIST))
return;
DirectoryListingPtr dl;
{
RLock l(cs);
auto p = viewedLists.find(aUser.user);
if (p == viewedLists.end()) {
return;
}
dl = p->second;
}
if (dl) {
dl->setFileName(qi->getListName());
if (dl->hasCompletedDownloads()) {
dl->addFullListTask(dir);
} else {
fire(DirectoryListingManagerListener::OpenListing(), dl, dir, Util::emptyString);
}
}
}
pair<QueueItem::StringMap::const_iterator, bool> FileQueue::add(QueueItemPtr& qi) noexcept {
dcassert(queueSize >= 0);
auto ret = pathQueue.emplace(const_cast<string*>(&qi->getTarget()), qi);
if (ret.second) {
tthIndex.emplace(const_cast<TTHValue*>(&qi->getTTH()), qi);
if (!qi->isSet(QueueItem::FLAG_USER_LIST) && !qi->isSet(QueueItem::FLAG_CLIENT_VIEW) && !qi->isSet(QueueItem::FLAG_FINISHED)) {
dcassert(qi->getSize() >= 0);
queueSize += qi->getSize();
}
tokenQueue.emplace(qi->getToken(), qi);
}
return ret;
}
double QueueFileUtils::getNumericInfo(const QueueItemPtr& aItem, int aPropertyName) noexcept {
switch (aPropertyName) {
case PROP_SIZE: return (double)aItem->getSize();
case PROP_BYTES_DOWNLOADED: return (double)QueueManager::getInstance()->getDownloadedBytes(aItem);
case PROP_PRIORITY: return (double)aItem->getPriority();
case PROP_TIME_ADDED: return (double)aItem->getTimeAdded();
case PROP_TIME_FINISHED: return (double)aItem->getTimeFinished();
case PROP_SPEED: return (double)QueueManager::getInstance()->getAverageSpeed(aItem);
case PROP_SECONDS_LEFT: return (double)QueueManager::getInstance()->getSecondsLeft(aItem);
case PROP_BUNDLE: return (double)(aItem->getBundle() ? aItem->getBundle()->getToken() : -1);
default: dcassert(0); return 0;
}
}
void DirectoryListingManager::on(QueueManagerListener::ItemAdded, const QueueItemPtr& aQI) noexcept {
if (!aQI->isSet(QueueItem::FLAG_CLIENT_VIEW) || !aQI->isSet(QueueItem::FLAG_USER_LIST))
return;
auto user = aQI->getSources()[0].getUser();
auto dl = hasList(user);
if (dl) {
dl->onAddedQueue(aQI->getTarget());
return;
}
if (!aQI->isSet(QueueItem::FLAG_PARTIAL_LIST)) {
dl = createList(user, false, aQI->getListName(), false);
} else {
dl = createList(user, true, Util::emptyString, false);
}
dl->onAddedQueue(aQI->getTarget());
}
void BundleQueue::addBundleItem(QueueItemPtr& aQI, BundlePtr& aBundle) noexcept {
dcassert(!aQI->getBundle());
aBundle->addQueue(aQI);
aQI->setBundle(aBundle);
if (!aBundle->isFileBundle()) {
forEachPath(aBundle, aQI->getTarget(), [&](PathInfo& aInfo) {
if (aQI->isDownloaded()) {
aInfo.finishedFiles++;
} else {
aInfo.queuedFiles++;
}
aInfo.size += aQI->getSize();
});
}
if (!aQI->isDownloaded()) {
queueSize += aQI->getSize();
}
}
void UserQueue::removeQI(QueueItemPtr& qi, const UserPtr& aUser, bool removeRunning /*true*/, Flags::MaskType reason) noexcept{
if(removeRunning) {
qi->removeDownloads(aUser);
}
dcassert(qi->isSource(aUser));
BundlePtr bundle = qi->getBundle();
if (bundle) {
if (!bundle->isSource(aUser)) {
return;
}
aUser->removeQueued(qi->getSize());
if (qi->getBundle()->removeUserQueue(qi, aUser, reason)) {
removeBundle(bundle, aUser);
} else {
dcassert(userBundleQueue.find(aUser) != userBundleQueue.end());
}
}
if (qi->getPriority() == QueueItem::HIGHEST) {
auto j = userPrioQueue.find(aUser);
dcassert(j != userPrioQueue.end());
if (j == userPrioQueue.end()) {
return;
}
auto& l = j->second;
auto i = find(l.begin(), l.end(), qi);
dcassert(i != l.end());
if (i == l.end()) {
return;
}
l.erase(i);
if(l.empty()) {
userPrioQueue.erase(j);
}
}
}
void FileQueue::remove(QueueItemPtr& qi) noexcept {
//TargetMap
auto f = pathQueue.find(const_cast<string*>(&qi->getTarget()));
if (f != pathQueue.end()) {
pathQueue.erase(f);
if (!qi->isSet(QueueItem::FLAG_USER_LIST) && (!qi->isSet(QueueItem::FLAG_FINISHED) || !qi->getBundle()) && !qi->isSet(QueueItem::FLAG_CLIENT_VIEW)) {
dcassert(qi->getSize() >= 0);
queueSize -= qi->getSize();
}
}
dcassert(queueSize >= 0);
//TTHIndex
auto s = tthIndex.equal_range(const_cast<TTHValue*>(&qi->getTTH()));
dcassert(s.first != s.second);
auto k = find(s | map_values, qi);
if (k.base() != s.second) {
tthIndex.erase(k.base());
}
// Tokens
tokenQueue.erase(qi->getToken());
}
void FinishedManager::on(QueueManagerListener::Finished, const QueueItemPtr& qi, const string&, const DownloadPtr& p_download) noexcept
{
if (!ClientManager::isShutdown())
{
const bool isFile = !qi->isAnySet(QueueItem::FLAG_USER_LIST | QueueItem::FLAG_DCLST_LIST | QueueItem::FLAG_USER_GET_IP);
if (isFile)
{
PLAY_SOUND(SOUND_FINISHFILE);
}
if (isFile || (qi->isAnySet(QueueItem::FLAG_USER_LIST | QueueItem::FLAG_DCLST_LIST) && BOOLSETTING(LOG_FILELIST_TRANSFERS)))
{
auto item = std::make_shared<FinishedItem>(qi->getTarget(), p_download->getHintedUser(),
qi->getSize(), p_download->getRunningAverage(),
GET_TIME(), qi->getTTH(), p_download->getActual(), p_download->getUser()->getIPAsString());
if (SETTING(DB_LOG_FINISHED_DOWNLOADS))
{
CFlylinkDBManager::getInstance()->save_transfer_history(false, e_TransferDownload, item);
}
rotation_items(item, e_Download);
fly_fire2(FinishedManagerListener::AddedDl(), item, false);
log(p_download->getUser()->getCID(), qi->getTarget(), STRING(FINISHED_DOWNLOAD));
}
}
}
bool ViewFileManager::isViewedItem(const QueueItemPtr& aQI) noexcept {
return aQI->isSet(QueueItem::FLAG_CLIENT_VIEW) && !aQI->isSet(QueueItem::FLAG_USER_LIST) && !aQI->isSet(QueueItem::FLAG_OPEN);
}
void UserQueue::addQI(QueueItemPtr& qi, bool newBundle /*false*/) noexcept {
for(const auto& i: qi->getSources()) {
addQI(qi, i.getUser(), newBundle);
}
}
void UserQueue::removeQI(QueueItemPtr& qi, bool removeRunning /*true*/) noexcept{
for(const auto& i: qi->getSources()) {
removeQI(qi, i.getUser(), removeRunning, 0);
}
}
void UserQueue::setQIPriority(QueueItemPtr& qi, QueueItemBase::Priority p) noexcept {
removeQI(qi, false);
qi->setPriority(p);
addQI(qi);
}
void UserQueue::removeDownload(QueueItemPtr& qi, const string& aToken) noexcept {
qi->removeDownload(aToken);
}
