Allow deleting of stamps

6 files changed, 167 insertions, 3 deletions

diff --git a/src/stamps/StampsController.cpp b/src/stamps/StampsController.cpp
index 07c1efd..fde93a5 100644
--- a/src/stamps/StampsController.cpp
+++ b/src/stamps/StampsController.cpp
@@ -4,9 +4,15 @@
  *  Created on: Mar 29, 2012
  *      Author: Simon
  */
+#include <sstream>
+#include <unistd.h>
 
+#include "client/Client.h"
 #include "StampsController.h"
 #include "interface/Engine.h"
+#include "dialogues/ConfirmPrompt.h"
+#include "tasks/TaskWindow.h"
+#include "tasks/Task.h"
 
 #include "StampsModel.h"
 #include "StampsView.h"
@@ -34,6 +40,58 @@ Save * StampsController::GetStamp()
 	return stampsModel->GetStamp();
 }
 
+void StampsController::RemoveSelected()
+{
+	class RemoveSelectedConfirmation: public ConfirmDialogueCallback {
+	public:
+		StampsController * c;
+		RemoveSelectedConfirmation(StampsController * c_) {	c = c_;	}
+		virtual void ConfirmCallback(ConfirmPrompt::DialogueResult result) {
+			if (result == ConfirmPrompt::ResultOkay)
+				c->removeSelectedC();
+		}
+		virtual ~RemoveSelectedConfirmation() { }
+	};
+
+	std::stringstream desc;
+	desc << "Are you sure you want to delete " << stampsModel->GetSelected().size() << " stamp";
+	if(stampsModel->GetSelected().size()>1)
+		desc << "s";
+	new ConfirmPrompt("Delete stamps", desc.str(), new RemoveSelectedConfirmation(this));
+}
+
+void StampsController::removeSelectedC()
+{
+	class RemoveSavesTask : public Task
+	{
+		std::vector<std::string> stamps;
+	public:
+		RemoveSavesTask(std::vector<std::string> stamps_) { stamps = stamps_; }
+		virtual void doWork()
+		{
+			for(int i = 0; i < stamps.size(); i++)
+			{
+				std::stringstream stampID;
+				stampID << "Deleting stamp [" << stamps[i] << "] ...";
+ 				notifyStatus(stampID.str());
+ 				Client::Ref().DeleteStamp(stamps[i]);
+				usleep(100*1000);
+				notifyProgress((float(i+1)/float(stamps.size())*100));
+			}
+		}
+	};
+
+	std::vector<std::string> selected = stampsModel->GetSelected();
+	new TaskWindow("Removing saves", new RemoveSavesTask(selected));
+	ClearSelection();
+	stampsModel->UpdateStampsList(stampsModel->GetPageNum());
+}
+
+void StampsController::ClearSelection()
+{
+	stampsModel->ClearSelected();
+}
+
 void StampsController::NextPage()
 {
 	if(stampsModel->GetPageNum()>1)
@@ -54,6 +112,14 @@ void StampsController::Update()
 	}
 }
 
+void StampsController::Selected(std::string stampID, bool selected)
+{
+	if(selected)
+		stampsModel->SelectStamp(stampID);
+	else
+		stampsModel->DeselectStamp(stampID);
+}
+
 void StampsController::Exit()
 {
 	if(ui::Engine::Ref().GetWindow() == stampsView)
diff --git a/src/stamps/StampsController.h b/src/stamps/StampsController.h
index 6b1c401..7053714 100644
--- a/src/stamps/StampsController.h
+++ b/src/stamps/StampsController.h
@@ -23,6 +23,10 @@ public:
 	StampsController(ControllerCallback * callback);
 	StampsView * GetView() {return stampsView;}
 	Save * GetStamp();
+	void RemoveSelected();
+	void removeSelectedC();
+	void ClearSelection();
+	void Selected(std::string stampID, bool selected);
 	void OpenStamp(Save * stamp);
 	void SetStamp();
 	void NextPage();
diff --git a/src/stamps/StampsModel.cpp b/src/stamps/StampsModel.cpp
index b679d13..cfdc72c 100644
--- a/src/stamps/StampsModel.cpp
+++ b/src/stamps/StampsModel.cpp
@@ -65,6 +65,7 @@ void StampsModel::UpdateStampsList(int pageNumber)
 	stampsList.clear();
 	currentPage = pageNumber;
 	notifyPageChanged();
+	notifyStampsListChanged();
 	/*notifyStampsListChanged();
 	for(int i = 0; i < tempStampsList.size(); i++)
 	{
@@ -76,11 +77,53 @@ void StampsModel::UpdateStampsList(int pageNumber)
 	for(int i = stampsEnd-20; i<stampIDs.size() && i<stampsEnd; i++)
 	{
 		Save * tempSave = Client::Ref().GetStamp(stampIDs[i]);
-		stampsList.push_back(tempSave);
+		if(tempSave)
+		{
+			stampsList.push_back(tempSave);
+		}
 	}
 	notifyStampsListChanged();
 }
 
+void StampsModel::SelectStamp(std::string stampID)
+{
+	for(int i = 0; i < selected.size(); i++)
+	{
+		if(selected[i]==stampID)
+		{
+			return;
+		}
+	}
+	selected.push_back(stampID);
+	notifySelectedChanged();
+}
+
+void StampsModel::DeselectStamp(std::string stampID)
+{
+	bool changed = false;
+restart:
+	for(int i = 0; i < selected.size(); i++)
+	{
+		if(selected[i]==stampID)
+		{
+			selected.erase(selected.begin()+i);
+			changed = true;
+			goto restart; //Just ensure all cases are removed.
+		}
+	}
+	if(changed)
+		notifySelectedChanged();
+}
+
+void StampsModel::notifySelectedChanged()
+{
+	for(int i = 0; i < observers.size(); i++)
+	{
+		StampsView* cObserver = observers[i];
+		cObserver->NotifySelectedChanged(this);
+	}
+}
+
 StampsModel::~StampsModel() {
 	if(stamp)
 		delete stamp;
diff --git a/src/stamps/StampsModel.h b/src/stamps/StampsModel.h
index cee158c..0356349 100644
--- a/src/stamps/StampsModel.h
+++ b/src/stamps/StampsModel.h
@@ -9,11 +9,13 @@
 #define STAMPSMODEL_H_
 
 #include <vector>
+#include <string>
 #include <math.h>
 #include "search/Save.h"
 
 class StampsView;
 class StampsModel {
+	vector<std::string> selected;
 	Save * stamp;
 	std::vector<std::string> stampIDs;
 	std::vector<Save*> stampsList;
@@ -21,6 +23,7 @@ class StampsModel {
 	int currentPage;
 	void notifyStampsListChanged();
 	void notifyPageChanged();
+	void notifySelectedChanged();
 public:
 	StampsModel();
 	int GetPageCount() { return max(1, (int)(ceil(stampIDs.size()/16))); }
@@ -30,6 +33,10 @@ public:
 	void UpdateStampsList(int pageNumber);
 	Save * GetStamp();
 	void SetStamp(Save * newStamp);
+	vector<std::string> GetSelected() { return selected; }
+	void ClearSelected() { selected.clear(); notifySelectedChanged(); }
+	void SelectStamp(std::string stampID);
+	void DeselectStamp(std::string stampID);
 	virtual ~StampsModel();
 };
 
diff --git a/src/stamps/StampsView.cpp b/src/stamps/StampsView.cpp
index c0d5981..883cabd 100644
--- a/src/stamps/StampsView.cpp
+++ b/src/stamps/StampsView.cpp
@@ -9,7 +9,6 @@
 #include "client/Client.h"
 #include "StampsView.h"
 
-#include "interface/SaveButton.h"
 #include "dialogues/ErrorMessage.h"
 #include "StampsController.h"
 #include "StampsModel.h"
@@ -50,6 +49,22 @@ StampsView::StampsView():
 	};
 	previousButton->SetActionCallback(new PrevPageAction(this));
 	previousButton->SetAlignment(AlignLeft, AlignBottom);
+
+	class RemoveSelectedAction : public ui::ButtonAction
+	{
+		StampsView * v;
+	public:
+		RemoveSelectedAction(StampsView * _v) { v = _v; }
+		void ActionCallback(ui::Button * sender)
+		{
+			v->c->RemoveSelected();
+		}
+	};
+
+	removeSelected = new ui::Button(ui::Point((((XRES+BARSIZE)-100)/2), YRES+MENUSIZE-18), ui::Point(100, 16), "Delete");
+	removeSelected->Visible = false;
+	removeSelected->SetActionCallback(new RemoveSelectedAction(this));
+	AddComponent(removeSelected);
 }
 
 void StampsView::OnTick(float dt)
@@ -108,6 +123,10 @@ void StampsView::NotifyStampsListChanged(StampsModel * sender)
 		{
 			v->c->OpenStamp(sender->GetSave());
 		}
+		virtual void SelectedCallback(ui::SaveButton * sender)
+		{
+			v->c->Selected(sender->GetSave()->GetName(), sender->GetSelected());
+		}
 	};
 	for(i = 0; i < saves.size(); i++)
 	{
@@ -126,6 +145,7 @@ void StampsView::NotifyStampsListChanged(StampsModel * sender)
 						),
 					ui::Point(buttonWidth, buttonHeight),
 					saves[i]);
+		saveButton->SetSelectable(true);
 		saveButton->SetActionCallback(new SaveOpenAction(this));
 		stampButtons.push_back(saveButton);
 		AddComponent(saveButton);
@@ -133,6 +153,27 @@ void StampsView::NotifyStampsListChanged(StampsModel * sender)
 	}
 }
 
+void StampsView::NotifySelectedChanged(StampsModel * sender)
+{
+	vector<std::string> selected = sender->GetSelected();
+	for(int j = 0; j < stampButtons.size(); j++)
+	{
+		stampButtons[j]->SetSelected(false);
+		for(int i = 0; i < selected.size(); i++)
+		{
+			if(stampButtons[j]->GetSave()->GetName()==selected[i])
+				stampButtons[j]->SetSelected(true);
+		}
+	}
+
+	if(selected.size())
+	{
+		removeSelected->Visible = true;
+	}
+	else
+		removeSelected->Visible = false;
+}
+
 void StampsView::OnMouseWheel(int x, int y, int d)
 {
 	if(!d)
diff --git a/src/stamps/StampsView.h b/src/stamps/StampsView.h
index 02e5569..0264611 100644
--- a/src/stamps/StampsView.h
+++ b/src/stamps/StampsView.h
@@ -13,15 +13,17 @@
 #include "interface/Button.h"
 #include "interface/Textbox.h"
 #include "interface/Label.h"
+#include "interface/SaveButton.h"
 
 class StampsController;
 class StampsModel;
 class StampsView: public ui::Window {
 	StampsController * c;
-	std::vector<ui::Component*> stampButtons;
+	std::vector<ui::SaveButton*> stampButtons;
 	ui::Button * previousButton;
 	ui::Button * nextButton;
 	ui::Label * infoLabel;
+	ui::Button * removeSelected;
 public:
 	StampsView();
 	//virtual void OnDraw();
@@ -29,6 +31,7 @@ public:
 	void AttachController(StampsController * c_) { c = c_; };
 	void NotifyPageChanged(StampsModel * sender);
 	void NotifyStampsListChanged(StampsModel * sender);
+	void NotifySelectedChanged(StampsModel * sender);
 	virtual void OnMouseWheel(int x, int y, int d);
 	virtual void OnKeyPress(int key, Uint16 character, bool shift, bool ctrl, bool alt);
 	virtual ~StampsView();