From f8d2bf386dbc29cc61184157e1c3880de4febeff Mon Sep 17 00:00:00 2001
From: Richard Patel <terorie@alphakevin.club>
Date: Sat, 17 Nov 2018 04:54:29 +0100
Subject: [PATCH] Fix FTP error ignore

---
 main.go | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index dc070e2..f962f34 100644
--- a/main.go
+++ b/main.go
@@ -62,11 +62,6 @@ func cmdBase(clic *cli.Context) error {
 		case <-ticker.C:
 			t, err := FetchTask()
 			if err != nil {
-				if err == fasturl.ErrUnknownScheme {
-					// Not an error
-					err = nil
-					continue
-				}
 				logrus.WithError(err).
 					Error("Failed getting new task")
 				time.Sleep(30 * time.Second)
@@ -82,7 +77,11 @@ func cmdBase(clic *cli.Context) error {
 
 			var baseUri fasturl.URL
 			err = baseUri.Parse(t.Url)
-			if err != nil {
+			if urlErr, ok := err.(*fasturl.Error); ok && urlErr.Err == fasturl.ErrUnknownScheme {
+				// Not an error
+				err = nil
+				continue
+			} else if err != nil {
 				logrus.WithError(err).
 					Error("Failed getting new task")
 				time.Sleep(30 * time.Second)