From 268f20c6fdc7b91823e6bc202b8abdf9f2e85fc9 Mon Sep 17 00:00:00 2001
From: Markus Olsson <j.markus.olsson@gmail.com>
Date: Tue, 16 May 2017 11:25:45 +0200
Subject: [PATCH] Let removeRepository accept a repository to remove

---
 app/src/ui/app.tsx | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/app/src/ui/app.tsx b/app/src/ui/app.tsx
index b9dd6e7c815..81f49670ead 100644
--- a/app/src/ui/app.tsx
+++ b/app/src/ui/app.tsx
@@ -170,7 +170,7 @@ export class App extends React.Component<IAppProps, IAppState> {
       case 'add-local-repository': return this.showAddLocalRepo()
       case 'create-branch': return this.showCreateBranch()
       case 'show-branches': return this.showBranches()
-      case 'remove-repository': return this.removeRepository()
+      case 'remove-repository': return this.removeRepository(this.getRepository())
       case 'create-repository': return this.showCreateRepository()
       case 'rename-branch': return this.renameBranch()
       case 'delete-branch': return this.deleteBranch()
@@ -453,10 +453,14 @@ export class App extends React.Component<IAppProps, IAppState> {
     this.addRepositories(paths)
   }
 
-  private removeRepository() {
-    const repository = this.getRepository()
+  private removeRepository(repository: Repository | CloningRepository | null) {
 
-    if (!repository || repository instanceof CloningRepository) {
+    if (!repository) {
+      return
+    }
+
+    if (repository instanceof CloningRepository) {
+      this.props.dispatcher.removeRepositories([ repository ])
       return
     }