From 2af6392a523a0e41c77d72924458d70b0b060e85 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafa=C5=82=20Muszy=C5=84ski?= <rmuszynski1@gmail.com>
Date: Tue, 22 Sep 2015 12:36:54 +0200
Subject: [PATCH] remove copied dirs on rollback

---
 src/Updater/Service/UpdateService.php | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/Updater/Service/UpdateService.php b/src/Updater/Service/UpdateService.php
index 512ff19..294b85f 100644
--- a/src/Updater/Service/UpdateService.php
+++ b/src/Updater/Service/UpdateService.php
@@ -86,6 +86,10 @@ private function copyFiles(array $file, $rollback = false)
                 default:
                     break;
             }
+        } else {
+            if ($file['type'] === self::TYPE_ADD) {
+                $fileSystem->remove($this->updater->getWorkingDir().$file['file']);
+            }
         }
     }