You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.0 KiB
86 lines
2.0 KiB
package controllers
|
|
|
|
import (
|
|
"log"
|
|
"fmt"
|
|
"time"
|
|
"net/http"
|
|
"encoding/json"
|
|
lib "yaagobackend/lib"
|
|
"github.com/gorilla/mux"
|
|
|
|
)
|
|
|
|
// User represents a user in the system
|
|
type User struct {
|
|
UserID uint `gorm:"primaryKey"`
|
|
Username string `gorm:"size:50;not null"`
|
|
Email string `gorm:"size:100;not null"`
|
|
Password string `gorm:"size:100;not null"`
|
|
CreatedAt time.Time `gorm:"autoCreateTime"`
|
|
}
|
|
|
|
// GetUsers handles the GET /users route
|
|
func GetUsers(w http.ResponseWriter, r *http.Request) {
|
|
db, err := lib.Connector();
|
|
if(err!=nil){
|
|
http.Error(w, "Error retrieving users", http.StatusNotFound)
|
|
return
|
|
}
|
|
var users []User
|
|
|
|
response := db.Find(&users) // As it passes a pointer of the variable, the variable itself is populated with the data in the db
|
|
|
|
if(response.Error!=nil){
|
|
http.Error(w, "Error retrieving users, no users found.", http.StatusNotFound)
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
|
|
json.NewEncoder(w).Encode(users)
|
|
}
|
|
|
|
// GetUser handles the GET /users/{id} route
|
|
func GetUser(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
id := vars["id"]
|
|
|
|
user := User{UserID: 1, Username: "John Doe"} // Placeholder user
|
|
|
|
if id == "1" {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(user)
|
|
} else {
|
|
http.Error(w, "User not found", http.StatusNotFound)
|
|
}
|
|
}
|
|
|
|
// CreateUser handles the POST /users route
|
|
func CreateUser(w http.ResponseWriter, r *http.Request) {
|
|
user := User{Username: "john_doe", Email: "john@example.com", Password: "securepassword"}
|
|
|
|
err := json.NewDecoder(r.Body).Decode(&user)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
db, err:= lib.Connector()
|
|
|
|
|
|
// Auto-migrate the schema
|
|
db.AutoMigrate(&User{})
|
|
|
|
// Example of inserting a user
|
|
|
|
db.Create(&user)
|
|
|
|
// At this point, user.UserID will be set to the ID generated by the database
|
|
log.Println("New user ID:", user.UserID)
|
|
|
|
|
|
|
|
fmt.Fprintf(w, "Received user: %+v", user)
|
|
}
|
|
|