From dcd8b0a4b4aeefda574c9365f0e3c1212ac3be72 Mon Sep 17 00:00:00 2001
From: Stephen Ball <stephen@aardvarklondon.com>
Date: Fri, 15 May 2015 17:29:43 +0100
Subject: [PATCH] Cleaning up failed deploys

---
 app/Commands/DeployProject.php | 86 +++++++++++++++++++++++++++++-----
 1 file changed, 74 insertions(+), 12 deletions(-)

diff --git a/app/Commands/DeployProject.php b/app/Commands/DeployProject.php
index 2d74ccf34..a7d50c439 100644
--- a/app/Commands/DeployProject.php
+++ b/app/Commands/DeployProject.php
@@ -75,6 +75,7 @@ public function handle()
             $project->status = Project::FAILED;
 
             $this->cancelPendingSteps($this->deployment->steps);
+            $this->cleanupDeployment();
         }
 
         $this->deployment->finished_at = date('Y-m-d H:i:s');
@@ -157,6 +158,52 @@ private function updateRepoInfo()
         $this->deployment->save();
     }
 
+    /**
+     * Removed left over artifacts from a failed deploy on each server
+     *
+     * @return void
+     * @todo Clean this up as there is some duplication with getScript()
+     */
+    private function cleanupDeployment()
+    {
+        $project = $this->deployment->project;
+
+        $release_id = date('YmdHis', strtotime($this->deployment->started_at));
+
+        foreach ($project->servers as $server) {
+
+            if (!$server->deploy_code) {
+                continue;
+            }
+
+            $root_dir = preg_replace('#/$#', '', $server->path);
+
+            if (empty($root_dir)) {
+                continue;
+            }
+
+            $releases_dir = $root_dir . '/releases';
+            $latest_release_dir = $releases_dir . '/' . $release_id;
+
+            $remote_key_file = $root_dir . '/id_rsa';
+            $remote_wrapper_file = $root_dir . '/wrapper.sh';
+
+            $commands = [
+                sprintf('cd %s', $root_dir),
+                sprintf('[ -f %s ] && rm %s', $remote_key_file, $remote_key_file),
+                sprintf('[ -f %s ] && rm %s', $remote_wrapper_file, $remote_wrapper_file),
+                sprintf('[ -d %s ] && rm -rf %s', $latest_release_dir, $latest_release_dir)
+            ];
+
+            $script = implode(PHP_EOL, $commands);
+
+            $process = new Process($this->sshCommand($server, $script));
+            $process->setTimeout(null);
+
+            $process->run();
+        }
+    }
+
     /**
      * Finds all pending steps and marks them as cancelled
      *
@@ -207,18 +254,7 @@ private function runStep(DeployStep $step)
             $failed = false;
 
             if (!empty($script)) {
-                $script = 'set -e' . PHP_EOL . $script;
-                $process = new Process(
-                    'ssh -o CheckHostIP=no \
-                         -o IdentitiesOnly=yes \
-                         -o StrictHostKeyChecking=no \
-                         -o PasswordAuthentication=no \
-                         -o IdentityFile=' . $this->private_key . ' \
-                         -p ' . $server->port . ' \
-                         ' . $user . '@' . $server->ip_address . ' \'bash -s\' << EOF
-                         '.$script.'
-EOF'
-                );
+                $process = new Process($this->sshCommand($server, $script, $user));
                 $process->setTimeout(null);
 
                 $output = '';
@@ -369,6 +405,32 @@ private function logSuccess($message)
         return '<info>' . $message . '</info>';
     }
 
+    /**
+     * Generates the SSH command for running the script on a server
+     *
+     * @param Server $server
+     * @param string $script The script to run
+     * @param string $user
+     * @return string
+     */
+    private function sshCommand(Server $server, $script, $user = null) {
+        if (is_null($user)) {
+            $user = $server->user;
+        }
+
+        $script = 'set -e' . PHP_EOL . $script;
+        return 'ssh -o CheckHostIP=no \
+                 -o IdentitiesOnly=yes \
+                 -o StrictHostKeyChecking=no \
+                 -o PasswordAuthentication=no \
+                 -o IdentityFile=' . $this->private_key . ' \
+                 -p ' . $server->port . ' \
+                 ' . $user . '@' . $server->ip_address . ' \'bash -s\' << EOF
+                 '.$script.'
+EOF';
+
+    }
+
     /**
      * Generates the content of a git bash script
      *