diff --git a/src/client/Client.cpp b/src/client/Client.cpp index 511e659f4..b35c29219 100644 --- a/src/client/Client.cpp +++ b/src/client/Client.cpp @@ -850,6 +850,8 @@ void Client::DeleteStamp(std::string stampID) return; } } + + updateStamps(); } std::string Client::AddStamp(GameSave * saveData) @@ -911,7 +913,7 @@ void Client::updateStamps() return; } -void Client::UnDeleteStamps() +void Client::RescanStamps() { DIR * directory; struct dirent * entry; @@ -921,7 +923,7 @@ void Client::UnDeleteStamps() stampIDs.clear(); while (entry = readdir(directory)) { - if(strncmp(entry->d_name, "..", 3) && strncmp(entry->d_name, ".", 2) && strstr(entry->d_name, ".stm")) + if(strncmp(entry->d_name, "..", 3) && strncmp(entry->d_name, ".", 2) && strstr(entry->d_name, ".stm") && strlen(entry->d_name) == 14) { char stampname[11]; strncpy(stampname, entry->d_name, 10); diff --git a/src/client/Client.h b/src/client/Client.h index 1cda795b7..4f657886b 100644 --- a/src/client/Client.h +++ b/src/client/Client.h @@ -112,7 +112,7 @@ public: void DeleteStamp(std::string stampID); std::string AddStamp(GameSave * saveData); std::vector GetStamps(int start, int count); - void Client::UnDeleteStamps(); + void RescanStamps(); int GetStampsCount(); SaveFile * GetFirstStamp(); diff --git a/src/localbrowser/LocalBrowserController.cpp b/src/localbrowser/LocalBrowserController.cpp index 6d1b42dec..0f6ec760c 100644 --- a/src/localbrowser/LocalBrowserController.cpp +++ b/src/localbrowser/LocalBrowserController.cpp @@ -53,10 +53,10 @@ void LocalBrowserController::RemoveSelected() }; std::stringstream desc; - desc << "Are you sure you want to delete " << browserModel->GetSelected().size() << " saves"; + desc << "Are you sure you want to delete " << browserModel->GetSelected().size() << " stamp"; if(browserModel->GetSelected().size()>1) desc << "s"; - new ConfirmPrompt("Delete saves", desc.str(), new RemoveSelectedConfirmation(this)); + new ConfirmPrompt("Delete stamps", desc.str(), new RemoveSelectedConfirmation(this)); } void LocalBrowserController::removeSelectedC() @@ -72,7 +72,7 @@ void LocalBrowserController::removeSelectedC() for(int i = 0; i < saves.size(); i++) { std::stringstream saveName; - saveName << "Deleting save [" << saves[i] << "] ..."; + saveName << "Deleting stamp [" << saves[i] << "] ..."; notifyStatus(saveName.str()); Client::Ref().DeleteStamp(saves[i]); notifyProgress((float(i+1)/float(saves.size())*100)); @@ -89,9 +89,27 @@ void LocalBrowserController::removeSelectedC() new TaskWindow("Removing saves", new RemoveSavesTask(this, selected)); } -void LocalBrowserController::UnDeleteStamps() +void LocalBrowserController::RescanStamps() { - browserModel->UnDeleteStamps(); + class RescanConfirmation: public ConfirmDialogueCallback { + public: + LocalBrowserController * c; + RescanConfirmation(LocalBrowserController * c_) { c = c_; } + virtual void ConfirmCallback(ConfirmPrompt::DialogueResult result) { + if (result == ConfirmPrompt::ResultOkay) + c->rescanStampsC(); + } + virtual ~RescanConfirmation() { } + }; + + std::stringstream desc; + desc << "Rescanning the stamps folder can find stamps added to the stamps folder or recover stamps when the stamps.def file has been lost or damaged. However, be warned that this may mess up the current sorting order"; + new ConfirmPrompt("Rescan", desc.str(), new RescanConfirmation(this)); +} + +void LocalBrowserController::rescanStampsC() +{ + browserModel->RescanStamps(); browserModel->UpdateSavesList(browserModel->GetPageNum()); } diff --git a/src/localbrowser/LocalBrowserController.h b/src/localbrowser/LocalBrowserController.h index 1ba3bf637..efa9c9308 100644 --- a/src/localbrowser/LocalBrowserController.h +++ b/src/localbrowser/LocalBrowserController.h @@ -27,7 +27,8 @@ public: void removeSelectedC(); void ClearSelection(); void Selected(std::string stampID, bool selected); - void UnDeleteStamps(); + void RescanStamps(); + void rescanStampsC(); void RefreshSavesList(); void OpenSave(SaveFile * stamp); void SetStamp(); diff --git a/src/localbrowser/LocalBrowserModel.cpp b/src/localbrowser/LocalBrowserModel.cpp index 4a7e780d8..9e869ccfb 100644 --- a/src/localbrowser/LocalBrowserModel.cpp +++ b/src/localbrowser/LocalBrowserModel.cpp @@ -87,9 +87,9 @@ void LocalBrowserModel::UpdateSavesList(int pageNumber) notifySavesListChanged(); } -void LocalBrowserModel::UnDeleteStamps() +void LocalBrowserModel::RescanStamps() { - Client::Ref().UnDeleteStamps(); + Client::Ref().RescanStamps(); } int LocalBrowserModel::GetPageCount() diff --git a/src/localbrowser/LocalBrowserModel.h b/src/localbrowser/LocalBrowserModel.h index 6bd19eb62..c2a6f9b9d 100644 --- a/src/localbrowser/LocalBrowserModel.h +++ b/src/localbrowser/LocalBrowserModel.h @@ -32,7 +32,7 @@ public: void AddObserver(LocalBrowserView * observer); std::vector GetSavesList(); void UpdateSavesList(int pageNumber); - void UnDeleteStamps(); + void RescanStamps(); SaveFile * GetSave(); void SetSave(SaveFile * newStamp); std::vector GetSelected() { return selected; } diff --git a/src/localbrowser/LocalBrowserView.cpp b/src/localbrowser/LocalBrowserView.cpp index daf623c0a..a6efc4cb6 100644 --- a/src/localbrowser/LocalBrowserView.cpp +++ b/src/localbrowser/LocalBrowserView.cpp @@ -26,7 +26,7 @@ LocalBrowserView::LocalBrowserView(): { nextButton = new ui::Button(ui::Point(XRES+BARSIZE-52, YRES+MENUSIZE-18), ui::Point(50, 16), "Next \x95"); previousButton = new ui::Button(ui::Point(1, YRES+MENUSIZE-18), ui::Point(50, 16), "\x96 Prev"); - undeleteButton = new ui::Button(ui::Point(XRES+BARSIZE-122, YRES+MENUSIZE-18), ui::Point(60, 16), "Stamp Fix"); + undeleteButton = new ui::Button(ui::Point(XRES+BARSIZE-122, YRES+MENUSIZE-18), ui::Point(60, 16), "Rescan"); infoLabel = new ui::Label(ui::Point(51, YRES+MENUSIZE-18), ui::Point(XRES+BARSIZE-102, 16), "Loading..."); AddComponent(infoLabel); AddComponent(nextButton); @@ -68,8 +68,7 @@ LocalBrowserView::LocalBrowserView(): UndeleteAction(LocalBrowserView * _v) { v = _v; } void ActionCallback(ui::Button * sender) { - if(ConfirmPrompt::Blocking("Bring back deleted stamps", "Versions 83.1-83.8 had a bug where stamps past page 3 were deleted from the list. Click continue to bring these back and read every stamp in the stamps/ folder.\n\nWARNING:\nThis will also bring back all stamps deleted before version 74.2 (March 2012)", "Continue")) - v->c->UnDeleteStamps(); + v->c->RescanStamps(); } }; undeleteButton->SetActionCallback(new UndeleteAction(this));