prepare user system

api/handlers/api.go

@@ -1,14 +1,20 @@
 package handlers
 
 import (
+	"context"
 	"encoding/json"
+	"fmt"
 	"io"
+	"log"
 	"net/http"
+	"time"
 
 	"github.com/SowinskiBraeden/DeviceBookingAPI/api"
 	"github.com/SowinskiBraeden/DeviceBookingAPI/models"
+	"github.com/SowinskiBraeden/DeviceBookingAPI/util"
 	"github.com/go-playground/validator/v10"
 	"github.com/gorilla/mux"
+	"go.mongodb.org/mongo-driver/bson"
 	"go.uber.org/zap"
 
 	"github.com/SowinskiBraeden/DeviceBookingAPI/config"
@@ -27,9 +33,14 @@ type App struct {
 
 // New creates a new mux router and all the routes
 func (a *App) New() *mux.Router {
+
+	// Detect if system is new and needs default admin
+	a.newSystem()
+
 	r := mux.NewRouter()
 	cow := Cow{DB: databases.NewCowDatabase(a.dbHelper)}
 	device := Device{DB: databases.NewDeviceDatabase(a.dbHelper)}
+	// user := User{DB: databases.NewUserDatabase(a.dbHelper)}
 
 	// healthcheck
 	r.HandleFunc("/health", healthCheckHandler)
@@ -37,7 +48,7 @@ func (a *App) New() *mux.Router {
 	apiCreate := r.PathPrefix("/api/v1").Subrouter()
 
 	// Data handlers, create, delete, update etc.
-	apiCreate.Handle("/cow/{cow_id}", api.Middleware(http.HandlerFunc(cow.CowByIDHandler))).Methods("GET")                   // By Object ID not Cow Name
+	apiCreate.Handle("/cow/{cow_id}", api.Middleware(http.HandlerFunc(cow.CowByObjectIDHandler))).Methods("GET")             // By Object ID not Cow Name
 	apiCreate.Handle("/cows", api.Middleware(http.HandlerFunc(cow.CowHandler))).Methods("GET")                               // Returns all cows
 	apiCreate.Handle("/cows", api.Middleware(http.HandlerFunc(cow.CowHandlerQuery))).Methods("POST")                         // Returns list of cows based of name query
 	apiCreate.Handle("/cows/new", api.Middleware(http.HandlerFunc(cow.NewCowHandler))).Methods("POST")                       // Create new cow
@@ -46,7 +57,7 @@ func (a *App) New() *mux.Router {
 	apiCreate.Handle("/cows/get_devices/{cow_id}", api.Middleware(http.HandlerFunc(device.GetChildDevices))).Methods("POST") // Returns a list of devices from a given Cow obj
 	apiCreate.Handle("cows/bookings/{cow_id}", api.Middleware(http.HandlerFunc(cow.GetBookingsHandler))).Methods("GET")      // Returns all bookings for a given cow
 
-	apiCreate.Handle("/device/{device_id}", api.Middleware(http.HandlerFunc(device.DeviceByIDHandler))).Methods("GET")            // By Object ID not Device Name
+	apiCreate.Handle("/device/{device_id}", api.Middleware(http.HandlerFunc(device.DeviceByObjectIDHandler))).Methods("GET")      // By Object ID not Device Name
 	apiCreate.Handle("/devices", api.Middleware(http.HandlerFunc(device.DeviceHandler))).Methods("GET")                           // Returns all devices
 	apiCreate.Handle("/devices", api.Middleware(http.HandlerFunc(device.DeviceHandlerQuery))).Methods("POST")                     // Returns list of devices based of name query
 	apiCreate.Handle("/devices/new", api.Middleware(http.HandlerFunc(device.NewDeviceHandler))).Methods("POST")                   // create new device
@@ -93,3 +104,34 @@ func healthCheckHandler(w http.ResponseWriter, r *http.Request) {
 	})
 	_, _ = io.WriteString(w, string(b))
 }
+
+func (a *App) newSystem() {
+	var DB databases.UserDatabase = databases.NewUserDatabase(a.dbHelper)
+	dbResp, err := DB.Find(context.TODO(), bson.M{})
+	if err != nil {
+		zap.S().With(err).Error("Unable to detect new system: failed to get users")
+	}
+
+	if len(dbResp) == 0 {
+		fmt.Println("Admin account setup...")
+
+		for {
+			defaultAdmin := util.CreateDefaultAdmin(DB)
+
+			if util.Confirm("Are the above credentials correct?") {
+				ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+				defer cancel()
+
+				_, err := DB.InsertOne(ctx, defaultAdmin)
+				if err != nil {
+					log.Printf("Failed to create an admin\n")
+					break
+				}
+
+				log.Printf("Successfully created default admin")
+				log.Printf("Your default admin ID is %s", defaultAdmin.Details.UID)
+				break
+			}
+		}
+	}
+}