From e9ae967962b1723f4ad82133baa4f526568ab131 Mon Sep 17 00:00:00 2001
From: Ivan <developer@iillexial.me>
Date: Tue, 22 Dec 2020 15:10:38 +0200
Subject: [PATCH] Terminate session on fail (#12)

Rollback changes if Start() returns an error
---
 main.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/main.go b/main.go
index afae092..2339d1f 100644
--- a/main.go
+++ b/main.go
@@ -41,6 +41,10 @@ func main() {
 	case "start":
 		fmt.Println("Starting a new session...")
 		err = smug.Start(*config, options.Windows)
+		if err != nil {
+			fmt.Println("Oops, an error occurred! Rolling back...")
+			smug.Stop(*config, options.Windows)
+		}
 	case "stop":
 		if len(options.Windows) == 0 {
 			fmt.Println("Terminating session...")