From 7f730467739634fdde04c8fe45a0398d2b6a0c10 Mon Sep 17 00:00:00 2001
From: Matthew Dillon <matthewrdillon@gmail.com>
Date: Sun, 3 Dec 2017 17:06:50 -0700
Subject: [PATCH] WIP

---
 app/controllers/collections/create.js      |  2 ++
 app/controllers/collections/detail/edit.js |  2 ++
 app/mixins/cleanup-form.js                 | 32 ++++++++++++++++++++++
 app/routes/collections/create.js           |  3 +-
 app/routes/collections/detail/edit.js      |  3 +-
 5 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100644 app/mixins/cleanup-form.js

diff --git a/app/controllers/collections/create.js b/app/controllers/collections/create.js
index 37192de..8b27025 100644
--- a/app/controllers/collections/create.js
+++ b/app/controllers/collections/create.js
@@ -11,6 +11,8 @@ export default Controller.extend(ValidationMixin, {
   CollectionSpeciesValidations,
   DatasheetValidations,
 
+  hasMany: ['collectionSpecies', 'datasheets'],
+
   options: computed('projectOptions', 'studyLocationOptions',
                     'collectionTypeOptions', 'collectionMethodOptions',
                     'speciesOptions', 'adfgPermitOptions', function() {
diff --git a/app/controllers/collections/detail/edit.js b/app/controllers/collections/detail/edit.js
index 72c586c..205a4aa 100644
--- a/app/controllers/collections/detail/edit.js
+++ b/app/controllers/collections/detail/edit.js
@@ -11,6 +11,8 @@ export default Controller.extend(ValidationMixin, {
   CollectionSpeciesValidations,
   DatasheetValidations,
 
+  hasMany: ['collectionSpecies', 'datasheets'],
+
   options: computed('projectOptions', 'studyLocationOptions',
                     'collectionTypeOptions', 'collectionMethodOptions',
                     'speciesOptions', 'adfgPermitOptions', function() {
diff --git a/app/mixins/cleanup-form.js b/app/mixins/cleanup-form.js
new file mode 100644
index 0000000..ec83827
--- /dev/null
+++ b/app/mixins/cleanup-form.js
@@ -0,0 +1,32 @@
+import Ember from 'ember';
+
+const { Mixin, run: { once } } = Ember;
+
+export default Mixin.create({
+  actions: {
+    willTransition(transition) {
+      if (confirm('Any unsaved changes will be discarded.')) {
+        let model = this.get('controller.model');
+        let hasMany = this.get('controller.hasMany');
+
+        hasMany.forEach((relationship) => {
+          model.get(relationship).forEach((r) => {
+            once(this, () => {
+              if (r.get('isNew')) {
+                r.deleteRecord();
+              } else {
+                r.rollbackAttributes();
+              }
+            }, this);
+          });
+        });
+
+        if (model.get('isNew')) {
+          model.deleteRecord();
+        }
+      } else {
+        return false;
+      }
+    },
+  },
+});
diff --git a/app/routes/collections/create.js b/app/routes/collections/create.js
index 2ecd267..cd56f73 100644
--- a/app/routes/collections/create.js
+++ b/app/routes/collections/create.js
@@ -1,8 +1,9 @@
 import Ember from 'ember';
+import CleanupFormMixin from 'ccdb-web/mixins/cleanup-form';
 
 const { Route, RSVP } = Ember;
 
-export default Route.extend({
+export default Route.extend(CleanupFormMixin, {
   model() {
     const store = this.get('store');
     return RSVP.hash({
diff --git a/app/routes/collections/detail/edit.js b/app/routes/collections/detail/edit.js
index fc99b16..6237884 100644
--- a/app/routes/collections/detail/edit.js
+++ b/app/routes/collections/detail/edit.js
@@ -1,8 +1,9 @@
 import Ember from 'ember';
+import CleanupFormMixin from 'ccdb-web/mixins/cleanup-form';
 
 const { Route, RSVP } = Ember;
 
-export default Route.extend({
+export default Route.extend(CleanupFormMixin, {
   model() {
     const store = this.get('store');
     const model = this.modelFor('collections.detail');