From 588d402c631d0f62a91a4fe628d4778a6fe347c5 Mon Sep 17 00:00:00 2001
From: bytedream <bytedream@protonmail.com>
Date: Mon, 21 Mar 2022 14:30:04 +0100
Subject: [PATCH] Just some style corrections

---
 downloader.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/downloader.go b/downloader.go
index d455f89..48c0295 100644
--- a/downloader.go
+++ b/downloader.go
@@ -89,7 +89,7 @@ func (d Downloader) download(format *Format) error {
 	}
 
 	if _, err := os.Stat(d.TempDir); os.IsNotExist(err) {
-		if err := os.Mkdir(d.TempDir, 0700); err != nil {
+		if err = os.Mkdir(d.TempDir, 0700); err != nil {
 			return err
 		}
 	}
@@ -267,12 +267,14 @@ func (d Downloader) downloadSegments(format *Format) ([]string, error) {
 							break
 						}
 						if k == 2 {
+							file.Close()
 							cancel()
 							return
 						}
 						select {
 						case <-d.Context.Done():
 						case <-ctx.Done():
+							file.Close()
 							return
 						case <-time.After(5 * time.Duration(k) * time.Second):
 							// sleep if an error occurs. very useful because sometimes the connection times out