From 9abfd9f864a2b4f9e03f6253d0c02716b08024f7 Mon Sep 17 00:00:00 2001
From: Redume <Danilivasenko09@gmail.com>
Date: Thu, 15 Dec 2022 21:04:53 +0300
Subject: [PATCH] Refactoring the code and add new table, for favorite img

---
 functions/database.go | 46 +++++++++++++++++++++++++++++++++++--------
 1 file changed, 38 insertions(+), 8 deletions(-)

diff --git a/functions/database.go b/functions/database.go
index 52dd11c..f24e5e9 100644
--- a/functions/database.go
+++ b/functions/database.go
@@ -11,14 +11,7 @@ func Database() {
 		Logger(err.Error())
 	}
 
-	var existsSettings bool
-	err = db.QueryRow("SELECT EXISTS(SELECT name FROM sqlite_master WHERE type='table' AND name='settings')").Scan(&existsSettings)
-
-	if err != nil {
-		Logger(err.Error())
-	}
-
-	if existsSettings == false {
+	if !TableExists(db, "settings") {
 		sqlTable := `
 			CREATE TABLE IF NOT EXISTS settings (
 			    startup INTEGER DEFAULT 0,
@@ -42,4 +35,41 @@ func Database() {
 			Logger(err.Error())
 		}
 	}
+
+	if !TableExists(db, "favorite") {
+		sqlTable := `
+			CREATE TABLE IF NOT EXISTS favorite (
+			    title TEXT,
+			    explanation TEXT,
+			    copyright TEXT,
+			    date TEXT,
+			    url TEXT,
+			    hdurl TEXT,
+			    media_type TEXT
+			);`
+
+		_, err = db.Exec(sqlTable)
+		if err != nil {
+			Logger(err.Error())
+		}
+	}
+}
+
+func GetDatabase() *sql.DB {
+	db, err := sql.Open("sqlite3", "EveryNasa.db")
+	if err != nil {
+		Logger(err.Error())
+	}
+
+	return db
+}
+
+func TableExists(db *sql.DB, name string) bool {
+	var exists bool
+	err := db.QueryRow("SELECT EXISTS(SELECT name FROM sqlite_master WHERE type='table' AND name=?)", name).Scan(&exists)
+	if err != nil {
+		Logger(err.Error())
+	}
+
+	return exists
 }