X-Git-Url: https://git.jacobcasper.com/?p=brackets.git;a=blobdiff_plain;f=main.go;h=f0fb3acca9526a40bcbbab5518649fd2cb75a5f6;hp=94fbd814e4ddf63a8b399dcd69bbae56b2c1e66f;hb=2f5373e70d68008c2fbd0deeef305b624434f59a;hpb=1f562ce3e66992c3a556370b8d8b94bbe55c8434

diff --git a/main.go b/main.go
index 94fbd81..f0fb3ac 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,7 @@
 package main
 
 import (
-	"git.jacobcasper.com/brackets/client"
-	"git.jacobcasper.com/brackets/db"
+	"git.jacobcasper.com/brackets/env"
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/zmb3/spotify"
 	"log"
@@ -12,25 +11,20 @@ import (
 type handler func(http.ResponseWriter, *http.Request)
 
 func main() {
-	lockedDb, err := db.New()
+	env, err := env.New()
 	if err != nil {
-		log.Fatal("Could not open db: ", err.Error())
-	}
-
-	client, err := client.Get()
-	if err != nil {
-		log.Fatal("Could not get client: ", err.Error())
+		log.Fatal("Could not set up Env: ", err.Error())
 	}
 
 	http.HandleFunc(
 		"/artist/add",
-		addArtistHandler(lockedDb, client),
+		artistAddHandler(env),
 	)
 
 	log.Fatal(http.ListenAndServe(":8080", nil))
 }
 
-func addArtistHandler(db *db.DB, c *spotify.Client) handler {
+func artistAddHandler(env *env.Env) handler {
 	return func(w http.ResponseWriter, r *http.Request) {
 		if r.Method != "POST" {
 			http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
@@ -45,19 +39,19 @@ func addArtistHandler(db *db.DB, c *spotify.Client) handler {
 			return
 		}
 
-		artist, err := c.GetArtist(spotify.ID(artistId))
+		artist, err := env.C.GetArtist(spotify.ID(artistId))
 		if err != nil {
 			log.Printf("Failed to retrieve artist %s: %s", artistId, err.Error())
 			http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
 			return
 		}
 
-		db.Mu.Lock()
-		defer db.Mu.Unlock()
-		db.Db.Exec("INSERT INTO ARTIST (ID, NAME) VALUES (?, ?)", artist.ID, artist.Name)
+		env.Db.Mu.Lock()
+		defer env.Db.Mu.Unlock()
+		env.Db.Db.Exec("INSERT INTO ARTIST (ID, NAME) VALUES (?, ?)", artist.ID, artist.Name)
 
 		for _, genre := range artist.Genres {
-			result, err := db.Db.Exec("REPLACE INTO GENRE (NAME) VALUES (?)", genre)
+			result, err := env.Db.Db.Exec("REPLACE INTO GENRE (NAME) VALUES (?)", genre)
 			if err != nil {
 				log.Printf("Failed to insert genre %s: %s", genre, err.Error())
 				http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
@@ -71,7 +65,7 @@ func addArtistHandler(db *db.DB, c *spotify.Client) handler {
 				return
 			}
 
-			db.Db.Exec("INSERT INTO ARTIST_GENRE_XREF (ARTIST_ID, GENRE_ID) VALUES (?, ?)", artist.ID, genreId)
+			env.Db.Db.Exec("INSERT INTO ARTIST_GENRE_XREF (ARTIST_ID, GENRE_ID) VALUES (?, ?)", artist.ID, genreId)
 		}
 		w.WriteHeader(http.StatusCreated)
 	}