1 files changed, 31 insertions, 0 deletions
diff --git a/desktop-widgets/command_divesite.h b/desktop-widgets/command_divesite.h
new file mode 100644
index 000000000..9c4f93058
--- /dev/null
+++ b/desktop-widgets/command_divesite.h
@@ -0,0 +1,31 @@
+// SPDX-License-Identifier: GPL-2.0
+// Note: this header file is used by the undo-machinery and should not be included elsewhere.
+
+#ifndef COMMAND_DIVESITE_H
+#define COMMAND_DIVESITE_H
+
+#include "command_base.h"
+
+#include <QVector>
+
+// We put everything in a namespace, so that we can shorten names without polluting the global namespace
+namespace Command {
+
+class DeleteDiveSites : public Base {
+public:
+	DeleteDiveSites(const QVector<dive_site *> &sites);
+private:
+	bool workToBeDone() override;
+
+	// For redo
+	std::vector<dive_site *> sitesToRemove;
+
+	// For undo
+	std::vector<OwningDiveSitePtr> sitesToAdd;
+	void undo() override;
+	void redo() override;
+};
+
+} // namespace Command
+
+#endif // COMMAND_DIVESITE_H