From ce1360c733ad8f78eb19fcdce23258c8793ff173 Mon Sep 17 00:00:00 2001
From: Travis Tomka <underoathosx@gmail.com>
Date: Thu, 30 Mar 2017 19:58:52 -0600
Subject: [PATCH] Add docker-compose restart command.

---
 src/Task/Restart.php   | 33 +++++++++++++++++++++++++++++++++
 src/Task/loadTasks.php |  8 ++++++++
 2 files changed, 41 insertions(+)
 create mode 100755 src/Task/Restart.php

diff --git a/src/Task/Restart.php b/src/Task/Restart.php
new file mode 100755
index 0000000..75f024e
--- /dev/null
+++ b/src/Task/Restart.php
@@ -0,0 +1,33 @@
+<?php
+
+namespace Droath\RoboDockerCompose\Task;
+
+use Droath\RoboDockerCompose\DockerServicesTrait;
+
+/**
+ * Define docker compose restart command.
+ */
+class Restart extends Base
+{
+    use DockerServicesTrait;
+
+    /**
+     * {@inheritdoc}
+     */
+    protected $action = 'restart';
+
+    /**
+     * Specify a shutdown timeout in seconds.
+     *
+     * @param int $timeout
+     *   The timeout in seconds.
+     *
+     * @return self
+     */
+    public function timeout($timeout = 10)
+    {
+        $this->option('timeout', $timeout);
+
+        return $this;
+    }
+}
diff --git a/src/Task/loadTasks.php b/src/Task/loadTasks.php
index 863de21..9e4c538 100644
--- a/src/Task/loadTasks.php
+++ b/src/Task/loadTasks.php
@@ -22,4 +22,12 @@ trait loadTasks
     {
         return $this->task(Down::class, $pathToDockerCompose);
     }
+
+    /**
+     * Docker compose restart task.
+     */
+    protected function taskDockerComposeRestart($pathToDockerCompose = null)
+    {
+        return $this->task(Restart::class, $pathToDockerCompose);
+    }
 }