bool CQueueItem::TryRemoveAll()
{
const int oldVisibleOffspring = m_visibleOffspring;
std::vector<CQueueItem*>::iterator iter;
std::vector<CQueueItem*> keepChildren;
m_visibleOffspring = 0;
for (iter = m_children.begin() + m_removed_at_front; iter != m_children.end(); iter++)
{
CQueueItem* pItem = *iter;
if (pItem->TryRemoveAll())
delete pItem;
else
{
keepChildren.push_back(pItem);
m_visibleOffspring++;
m_visibleOffspring += pItem->GetChildrenCount(true);
}
}
m_children = keepChildren;
m_maxCachedIndex = -1;
m_removed_at_front = 0;
CQueueItem* parent = GetParent();
while (parent)
{
parent->m_maxCachedIndex = -1;
parent->m_visibleOffspring -= oldVisibleOffspring - m_visibleOffspring;
parent = parent->GetParent();
}
return m_children.empty();
}
bool CServerItem::TryRemoveAll()
{
wxASSERT(!GetParent());
const int oldVisibleOffspring = m_visibleOffspring;
std::vector<CQueueItem*>::iterator iter;
std::vector<CQueueItem*> keepChildren;
m_visibleOffspring = 0;
for (iter = m_children.begin() + m_removed_at_front; iter != m_children.end(); ++iter) {
CQueueItem* pItem = *iter;
if (pItem->TryRemoveAll()) {
if (pItem->GetType() == QueueItemType::File || pItem->GetType() == QueueItemType::Folder) {
CFileItem* pFileItem = static_cast<CFileItem*>(pItem);
RemoveFileItemFromList(pFileItem, true);
}
delete pItem;
}
else {
keepChildren.push_back(pItem);
m_visibleOffspring++;
m_visibleOffspring += pItem->GetChildrenCount(true);
}
}
std::swap(m_children, keepChildren);
m_removed_at_front = 0;
m_maxCachedIndex = -1;
wxASSERT(oldVisibleOffspring >= m_visibleOffspring);
wxASSERT(m_visibleOffspring >= static_cast<int>(m_children.size()));
(void)oldVisibleOffspring;
return m_children.empty();
}
bool CServerItem::TryRemoveAll()
{
const int oldVisibleOffspring = m_visibleOffspring;
std::vector<CQueueItem*>::iterator iter;
std::vector<CQueueItem*> keepChildren;
m_visibleOffspring = 0;
for (iter = m_children.begin() + m_removed_at_front; iter != m_children.end(); iter++)
{
CQueueItem* pItem = *iter;
if (pItem->TryRemoveAll())
{
if (pItem->GetType() == QueueItemType_File || pItem->GetType() == QueueItemType_Folder)
{
CFileItem* pFileItem = reinterpret_cast<CFileItem*>(pItem);
RemoveFileItemFromList(pFileItem);
}
delete pItem;
}
else
{
keepChildren.push_back(pItem);
m_visibleOffspring++;
m_visibleOffspring += pItem->GetChildrenCount(true);
}
}
m_children = keepChildren;
m_removed_at_front = 0;
CQueueItem* parent = GetParent();
while (parent)
{
parent->m_visibleOffspring -= oldVisibleOffspring - m_visibleOffspring;
parent = parent->GetParent();
}
return m_children.empty();
}
void CQueueViewFailed::OnRequeueSelected(wxCommandEvent& event)
{
#ifndef __WXMSW__
// GetNextItem is O(n) if nothing is selected, GetSelectedItemCount() is O(1)
if (!GetSelectedItemCount())
return;
#endif
bool failedToRequeueAll = false;
std::list<CQueueItem*> selectedItems;
long item = -1;
long skipTo = -1;
while (true)
{
item = GetNextItem(item, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
if (item == -1)
break;
SetItemState(item, 0, wxLIST_STATE_SELECTED);
if (item < skipTo)
continue;
CQueueItem* pItem = GetQueueItem(item);
if (pItem->GetType() == QueueItemType_Server)
skipTo = item + pItem->GetChildrenCount(true) + 1;
selectedItems.push_back(GetQueueItem(item));
}
if (selectedItems.empty())
return;
CQueueView* pQueueView = m_pQueue->GetQueueView();
while (!selectedItems.empty())
{
CQueueItem* pItem = selectedItems.front();
selectedItems.pop_front();
if (pItem->GetType() == QueueItemType_Server)
{
CServerItem* pOldServerItem = (CServerItem*)pItem;
CServerItem* pServerItem = pQueueView->CreateServerItem(pOldServerItem->GetServer());
unsigned int childrenCount = pOldServerItem->GetChildrenCount(false);
for (unsigned int i = 0; i < childrenCount; i++)
{
CFileItem* pFileItem = (CFileItem*)pItem->GetChild(i, false);
pFileItem->m_errorCount = 0;
pFileItem->m_statusMessage.Clear();
if (!pFileItem->Download() && !wxFileName::FileExists(pFileItem->GetLocalFile()))
{
failedToRequeueAll = true;
RemoveItem(pItem, true, false, false);
continue;
}
if (pFileItem->m_edit == CEditHandler::remote)
{
CEditHandler* pEditHandler = CEditHandler::Get();
if (!pEditHandler)
{
failedToRequeueAll = true;
delete pFileItem;
continue;
}
enum CEditHandler::fileState state = pEditHandler->GetFileState(pFileItem->GetRemoteFile(), pFileItem->GetRemotePath(), pServerItem->GetServer());
if (state == CEditHandler::unknown)
{
wxASSERT(pFileItem->Download());
wxString file = pFileItem->GetRemoteFile();
if (!pEditHandler->AddFile(CEditHandler::remote, file, pFileItem->GetRemotePath(), pServerItem->GetServer()))
{
failedToRequeueAll = true;
delete pFileItem;
continue;
}
pFileItem->SetLocalFile(file);
}
else if (state == CEditHandler::upload_and_remove_failed)
{
wxASSERT(!pFileItem->Download());
if (!pEditHandler->UploadFile(pFileItem->GetRemoteFile(), pFileItem->GetRemotePath(), pServerItem->GetServer(), true))
failedToRequeueAll = true;
delete pFileItem;
continue;
}
else
{
failedToRequeueAll = true;
delete pFileItem;
continue;
}
}
pFileItem->SetParent(pServerItem);
pQueueView->InsertItem(pServerItem, pFileItem);
}
m_fileCount -= childrenCount;
m_itemCount -= childrenCount + 1;
pOldServerItem->DetachChildren();
delete pOldServerItem;
std::vector<CServerItem*>::iterator iter;
for (iter = m_serverList.begin(); iter != m_serverList.end(); iter++)
{
if (*iter == pOldServerItem)
break;
}
if (iter != m_serverList.end())
m_serverList.erase(iter);
if (!pServerItem->GetChildrenCount(false))
{
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
}
else
{
CFileItem* pFileItem = (CFileItem*)pItem;
pFileItem->m_errorCount = 0;
pFileItem->m_statusMessage.Clear();
if (!pFileItem->Download() && !wxFileName::FileExists(pFileItem->GetLocalFile()))
{
failedToRequeueAll = true;
RemoveItem(pItem, true, false, false);
continue;
}
CServerItem* pOldServerItem = (CServerItem*)pItem->GetTopLevelItem();
CServerItem* pServerItem = pQueueView->CreateServerItem(pOldServerItem->GetServer());
RemoveItem(pItem, false, false, false);
if (pFileItem->m_edit == CEditHandler::remote)
{
CEditHandler* pEditHandler = CEditHandler::Get();
if (!pEditHandler)
{
if (!pServerItem->GetChildrenCount(false))
{
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
failedToRequeueAll = true;
delete pItem;
continue;
}
enum CEditHandler::fileState state = pEditHandler->GetFileState(pFileItem->GetRemoteFile(), pFileItem->GetRemotePath(), pServerItem->GetServer());
if (state == CEditHandler::unknown)
{
wxASSERT(pFileItem->Download());
wxString file = pFileItem->GetRemoteFile();
if (!pEditHandler->AddFile(CEditHandler::remote, file, pFileItem->GetRemotePath(), pServerItem->GetServer()))
{
if (!pServerItem->GetChildrenCount(false))
{
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
failedToRequeueAll = true;
delete pItem;
continue;
}
pFileItem->SetLocalFile(file);
}
else if (state == CEditHandler::upload_and_remove_failed)
{
wxASSERT(!pFileItem->Download());
if (!pEditHandler->UploadFile(pFileItem->GetRemoteFile(), pFileItem->GetRemotePath(), pServerItem->GetServer(), true))
failedToRequeueAll = true;
if (!pServerItem->GetChildrenCount(false))
{
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
delete pItem;
continue;
}
else
{
if (!pServerItem->GetChildrenCount(false))
{
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
failedToRequeueAll = true;
delete pItem;
continue;
}
}
pItem->SetParent(pServerItem);
pQueueView->InsertItem(pServerItem, pItem);
}
}
m_fileCountChanged = true;
pQueueView->CommitChanges();
if (pQueueView->IsActive())
pQueueView->AdvanceQueue(false);
DisplayNumberQueuedFiles();
SaveSetItemCount(m_itemCount);
RefreshListOnly();
if (!m_itemCount && m_pQueue->GetQueueView()->GetItemCount())
m_pQueue->SetSelection(0);
if (failedToRequeueAll)
wxMessageBox(_("Not all items could be requeued for viewing/editing."));
}
bool CQueueViewBase::RemoveItem(CQueueItem* pItem, bool destroy, bool updateItemCount /*=true*/, bool updateSelections /*=true*/)
{
if (pItem->GetType() == QueueItemType_File || pItem->GetType() == QueueItemType_Folder)
{
wxASSERT(m_fileCount > 0);
m_fileCount--;
m_fileCountChanged = true;
}
else if (pItem->GetType() == QueueItemType_FolderScan)
{
wxASSERT(m_folderScanCount > 0);
m_folderScanCount--;
m_folderScanCountChanged = true;
}
int index = 0;
if (updateSelections)
index = GetItemIndex(pItem);
CQueueItem* topLevelItem = pItem->GetTopLevelItem();
int count = topLevelItem->GetChildrenCount(true);
topLevelItem->RemoveChild(pItem, destroy);
bool didRemoveParent;
int oldCount = m_itemCount;
if (!topLevelItem->GetChild(0))
{
std::vector<CServerItem*>::iterator iter;
for (iter = m_serverList.begin(); iter != m_serverList.end(); iter++)
{
if (*iter == topLevelItem)
break;
}
if (iter != m_serverList.end())
m_serverList.erase(iter);
UpdateSelections_ItemRangeRemoved(GetItemIndex(topLevelItem), count + 1);
delete topLevelItem;
m_itemCount -= count + 1;
if (updateItemCount)
SaveSetItemCount(m_itemCount);
didRemoveParent = true;
}
else
{
count -= topLevelItem->GetChildrenCount(true);
if (updateSelections)
UpdateSelections_ItemRangeRemoved(index, count);
m_itemCount -= count;
if (updateItemCount)
SaveSetItemCount(m_itemCount);
didRemoveParent = false;
}
if (updateItemCount)
{
if (m_fileCountChanged || m_folderScanCountChanged)
DisplayNumberQueuedFiles();
if (oldCount > m_itemCount)
{
bool eraseBackground = GetTopItem() + GetCountPerPage() + 1 >= m_itemCount;
RefreshListOnly(eraseBackground);
if (eraseBackground)
Update();
}
}
return didRemoveParent;
}
void CQueueViewFailed::OnRequeueSelected(wxCommandEvent&)
{
#ifndef __WXMSW__
// GetNextItem is O(n) if nothing is selected, GetSelectedItemCount() is O(1)
if (!GetSelectedItemCount())
return;
#endif
bool failedToRequeueAll = false;
std::list<CQueueItem*> selectedItems;
long item = -1;
long skipTo = -1;
for (;;) {
item = GetNextItem(item, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
if (item == -1)
break;
SetItemState(item, 0, wxLIST_STATE_SELECTED);
if (item < skipTo)
continue;
CQueueItem* pItem = GetQueueItem(item);
if (pItem->GetType() == QueueItemType::Server)
skipTo = item + pItem->GetChildrenCount(true) + 1;
selectedItems.push_back(GetQueueItem(item));
}
if (selectedItems.empty())
return;
CQueueView* pQueueView = m_pQueue->GetQueueView();
while (!selectedItems.empty()) {
CQueueItem* pItem = selectedItems.front();
selectedItems.pop_front();
if (pItem->GetType() == QueueItemType::Server) {
CServerItem* pServerItem = (CServerItem*)pItem;
failedToRequeueAll |= !RequeueServerItem(pServerItem);
}
else {
CFileItem* pFileItem = (CFileItem*)pItem;
CServerItem* pOldServerItem = (CServerItem*)pItem->GetTopLevelItem();
CServerItem* pServerItem = pQueueView->CreateServerItem(pOldServerItem->GetServer());
RemoveItem(pItem, false, false, false);
failedToRequeueAll |= !RequeueFileItem(pFileItem, pServerItem);
if (!pServerItem->GetChildrenCount(false)) {
pQueueView->CommitChanges();
pQueueView->RemoveItem(pServerItem, true, true, true);
}
}
}
m_fileCountChanged = true;
pQueueView->CommitChanges();
if (pQueueView->IsActive())
pQueueView->AdvanceQueue(false);
DisplayNumberQueuedFiles();
SaveSetItemCount(m_itemCount);
RefreshListOnly();
if (!m_itemCount && m_pQueue->GetQueueView()->GetItemCount())
m_pQueue->SetSelection(0);
if (failedToRequeueAll)
wxMessageBoxEx(_("Not all items could be requeued for transfer."));
}