#8 add view and api to track activities

api/main.go

@@ -3,22 +3,15 @@ package main
 import (
 	"api/context"
 	"api/middleware"
-	"api/utils"
+	"api/workout"
+
 	"database/sql"
 	"log"
 	"net/http"
 
-	"firebase.google.com/go/auth"
 	_ "github.com/mattn/go-sqlite3"
 )
 
-type Person struct {
-	ID   int
-	Name string
-}
-
-type Context context.Context
-
 func main() {
 
 	context.InitializeDB()
@@ -29,16 +22,11 @@ func main() {
 	}
 	defer db.Close()
 
-	var context = Context{
-		DB: db,
-	}
-
-	var router = http.NewServeMux()
-	router.HandleFunc("GET /", context.HelloWorld)
+	var router = getRouter(db)
 
 	var server = http.Server{
 		Addr:    ":8080",
-		Handler: middleware.Logging(middleware.EnsureAuth(router)),
+		Handler: middleware.Logging(middleware.EnableCors(middleware.EnsureAuth(router))),
 	}
 	log.Println("Starting server at", server.Addr)
 
@@ -48,28 +36,9 @@ func main() {
 	}
 }
 
-func (ctx *Context) HelloWorld(w http.ResponseWriter, r *http.Request) {
-	token := r.Context().Value(middleware.TOKEN_KEY).(*auth.Token)
-	log.Println(token.UID)
-
-	sqlStmt := `select COUNT(*) from person;`
-
-	result, err := ctx.DB.Query(sqlStmt)
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	var count int
-	if result.Next() {
-		if result.Scan(&count) != nil {
-			log.Fatal(err)
-		}
-	} else {
-		log.Fatal("No rows found")
-	}
-
-	err = utils.WriteJSON(w, count)
-	if err != nil {
-		http.Error(w, err.Error(), http.StatusInternalServerError)
-	}
+func getRouter(db *sql.DB) *http.ServeMux {
+	var router = http.NewServeMux()
+	router.HandleFunc("POST /workout", workout.NewWorkout(db))
+	router.HandleFunc("GET /workout", workout.GetWorkouts(db))
+	return router
 }