From 8026cef1dfa2631f5b606e77c7f7b77db5ede85e Mon Sep 17 00:00:00 2001
From: Juan Cruz Viotti <jviottidc@gmail.com>
Date: Tue, 19 Jan 2016 09:44:48 -0400
Subject: [PATCH] Allow to re-select the first two step selections

You can click on the selected image/drive label to select them again.

The re-selection is disabled if there is a burning in process.

Fixes: https://github.com/resin-io/resin-etcher/issues/90
---
 build/browser/app.js | 23 +++++++++++++++++++++++
 lib/browser/app.js   | 23 +++++++++++++++++++++++
 lib/index.html       |  4 ++--
 3 files changed, 48 insertions(+), 2 deletions(-)

diff --git a/build/browser/app.js b/build/browser/app.js
index cd95ccf5..b47084c1 100644
--- a/build/browser/app.js
+++ b/build/browser/app.js
@@ -93,6 +93,29 @@ app.controller('AppController', function($q, DriveScannerService, SelectionState
     console.debug('Drive selected: ' + drive.device);
   };
 
+  this.reselectImage = function() {
+    if (self.writer.isBurning()) {
+      return;
+    }
+
+    // Reselecting an image automatically
+    // de-selects the current drive, if any.
+    // This is made so the user effectively
+    // "returns" to the first step.
+    self.selection.clear();
+
+    console.debug('Reselecting image');
+  };
+
+  this.reselectDrive = function() {
+    if (self.writer.isBurning()) {
+      return;
+    }
+
+    self.selection.removeDrive();
+    console.debug('Reselecting drive');
+  };
+
   this.burn = function(image, drive) {
 
     // Stop scanning drives when burning
diff --git a/lib/browser/app.js b/lib/browser/app.js
index d0450ef6..dad82f44 100644
--- a/lib/browser/app.js
+++ b/lib/browser/app.js
@@ -92,6 +92,29 @@ app.controller('AppController', function($q, DriveScannerService, SelectionState
     console.debug('Drive selected: ' + drive.device);
   };
 
+  this.reselectImage = function() {
+    if (self.writer.isBurning()) {
+      return;
+    }
+
+    // Reselecting an image automatically
+    // de-selects the current drive, if any.
+    // This is made so the user effectively
+    // "returns" to the first step.
+    self.selection.clear();
+
+    console.debug('Reselecting image');
+  };
+
+  this.reselectDrive = function() {
+    if (self.writer.isBurning()) {
+      return;
+    }
+
+    self.selection.removeDrive();
+    console.debug('Reselecting drive');
+  };
+
   this.burn = function(image, drive) {
 
     // Stop scanning drives when burning
diff --git a/lib/index.html b/lib/index.html
index b28190e5..ffa9fa03 100644
--- a/lib/index.html
+++ b/lib/index.html
@@ -30,7 +30,7 @@
                   <hero-button ng-click="app.selectImage()">Select image</hero-button>
                 </div>
                 <div ng-show="app.selection.hasImage()">
-                  <span ng-bind="app.selection.getImage() | basename"></span>
+                  <span ng-bind="app.selection.getImage() | basename" ng-click="app.reselectImage()"></span>
                 </div>
 
                 <p class="step-footer tiny">*supported files: .img, .iso</p>
@@ -67,7 +67,7 @@
                   </div>
 
                 </div>
-                <div ng-show="app.selection.hasDrive()" ng-bind="app.selection.getDrive().name"></div>
+                <div ng-show="app.selection.hasDrive()" ng-bind="app.selection.getDrive().name" ng-click="app.reselectDrive()"></div>
               </div>
             </div>
           </div>