SowinskiBraeden/school-management-api
1
0 Fork 0
Code Issues Packages Projects Releases Wiki Activity

Login and main router with jwt tokens

Browse Source
This commit is contained in:
SowinskiBraeden committed 2021-10-07 22:21:47 -07:00
1 parent 7049abf5d8
commit 0a357a7c2b
6 files changed
+219 -26
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

controllers/authController.go
+199 -20
View File
@@ -8,6 +8,7 @@ import (
"strconv"
"github.com/dgrijalva/jwt-go"
"github.com/gofiber/fiber/v2"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
@@ -17,6 +18,8 @@ import (
var teacherCollection *mongo.Collection = database.OpenCollection(database.Client, "teachers")
var studentCollection *mongo.Collection = database.OpenCollection(database.Client, "students")
const SecretKey = "secret"
func Enroll(c *fiber.Ctx) error {
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
@@ -24,7 +27,7 @@ func Enroll(c *fiber.Ctx) error {
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
@@ -34,7 +37,7 @@ func Enroll(c *fiber.Ctx) error {
// Check minimum enroll field requirements are met
if data["firstname"] == "" || data["lastname"] == "" || data["age"] == "" || data["gradelevel"] == "" || data["dob"] == "" || data["email"] == "" {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
@@ -73,7 +76,7 @@ func Enroll(c *fiber.Ctx) error {
result, insertErr := studentCollection.InsertOne(ctx, student)
if insertErr != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "the student could not be inserted",
"error": insertErr,
@@ -81,7 +84,11 @@ func Enroll(c *fiber.Ctx) error {
}
defer cancel()
return c.Status(201).JSON(result)
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": true,
"message": "successfully inserted student",
"result": result,
})
}
func RegisterTeacher(c *fiber.Ctx) error {
@@ -91,7 +98,7 @@ func RegisterTeacher(c *fiber.Ctx) error {
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
@@ -101,7 +108,7 @@ func RegisterTeacher(c *fiber.Ctx) error {
// Check minimum register teacher field requirements are met
if data["firstname"] == "" || data["lastname"] == "" || data["dob"] == "" || data["email"] == "" {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
@@ -126,7 +133,7 @@ func RegisterTeacher(c *fiber.Ctx) error {
result, insertErr := teacherCollection.InsertOne(ctx, teacher)
if insertErr != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "the teacher could not be inserted",
"error": insertErr,
@@ -134,7 +141,116 @@ func RegisterTeacher(c *fiber.Ctx) error {
}
defer cancel()
return c.Status(201).JSON(result)
return c.Status(fiber.StatusOK).JSON(fiber.Map{
"success": true,
"message": "successfully inserted teacher",
"result": result,
})
}
func StudentLogin(c *fiber.Ctx) error {
var data map[string]string
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
})
}
// Check required fields are included
if data["sid"] == "" || data["password"] == "" {
cancel()
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
}
var student models.Student
err := studentCollection.FindOne(ctx, bson.M{"sid": data["sid"]}).Decode(&student)
defer cancel()
if err != nil {
cancel()
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "student not found",
"error": err,
})
}
defer cancel()
var verified bool = student.ComparePasswords(data["password"])
if verified == false {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "incorrect password",
})
}
claims := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.StandardClaims{
Issuer: student.SID,
ExpiresAt: time.Now().Add(time.Hour * 24).Unix(), // 1 Day
})
token, err := claims.SignedString([]byte(SecretKey))
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "could not log in",
})
}
cookie := fiber.Cookie{
Name: "jwt",
Value: token,
Expires: time.Now().Add(time.Hour * 24),
HTTPOnly: true,
}
c.Cookie(&cookie)
return c.Status(200).JSON(fiber.Map{
"success": true,
"message": "correct password",
})
}
func TeacherLogin(c *fiber.Ctx) error {
return c.JSON("status:ok")
}
func Student(c *fiber.Ctx) error {
cookie := c.Cookies("jwt")
token, err := jwt.ParseWithClaims(cookie, &jwt.StandardClaims{}, func(token *jwt.Token) (interface{}, error) {
return []byte(SecretKey), nil
})
if err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
"success": false,
"message": "student not found",
})
}
claims := token.Claims.(*jwt.StandardClaims)
var student models.Student
findErr := studentCollection.FindOne(context.TODO(), bson.M{"sid": claims.Issuer}).Decode(&student)
if findErr != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
"success": false,
"message": "student not found",
})
}
return c.JSON(student)
}
func Teacher(c *fiber.Ctx) error {
return c.JSON("status:ok")
}
func UpdateStudentName(c *fiber.Ctx) error {
@@ -143,7 +259,7 @@ func UpdateStudentName(c *fiber.Ctx) error {
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
@@ -153,7 +269,7 @@ func UpdateStudentName(c *fiber.Ctx) error {
// Check id and names are included
if data["_id"] == "" || data["firstname"] == "" || data["middlename"] == "" || data["lastname"] == "" {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
@@ -164,7 +280,7 @@ func UpdateStudentName(c *fiber.Ctx) error {
cancel()
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
"success": false,
"message": "Catchphrase not found",
"message": "student not found",
"error": err,
})
}
@@ -183,7 +299,7 @@ func UpdateStudentName(c *fiber.Ctx) error {
)
if updateErr != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "the student could not be updated",
"error": updateErr,
@@ -191,11 +307,70 @@ func UpdateStudentName(c *fiber.Ctx) error {
}
defer cancel()
return c.Status(200).JSON(result)
return c.Status(fiber.StatusOK).JSON(fiber.Map{
"success": true,
"message": "successfully updated student",
"result": result,
})
}
func UpdateStudentGradeLevel(c *fiber.Ctx) error {
return c.JSON("status:ok")
var data map[string]string
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
})
}
// Check required fields are included
if data["_id"] == "" || data["gradelevel"] == "" {
cancel()
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
}
studentObjectId, err := primitive.ObjectIDFromHex(data["_id"])
if err != nil {
cancel()
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
"success": false,
"message": "student not found",
"error": err,
})
}
update := bson.M{
"$set": bson.M{
"gradelevel": data["gradelevel"],
},
}
result, updateErr := studentCollection.UpdateOne(
ctx,
bson.M{"_id": studentObjectId},
update,
)
if updateErr != nil {
cancel()
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "the student could not be updated",
"error": updateErr,
})
}
defer cancel()
return c.Status(fiber.StatusOK).JSON(fiber.Map{
"success": true,
"message": "successfully updated student",
"result": result,
})
}
func UpdateStudentHomeroom(c *fiber.Ctx) error {
@@ -256,7 +431,7 @@ func UpdateTeacherName(c *fiber.Ctx) error {
if err := c.BodyParser(&data); err != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "Failed to parse body",
"error": err,
@@ -266,7 +441,7 @@ func UpdateTeacherName(c *fiber.Ctx) error {
// Check id and names are included
if data["_id"] == "" || data["firstname"] == "" || data["middlename"] == "" || data["lastname"] == "" {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
"message": "missing required fields",
})
@@ -277,7 +452,7 @@ func UpdateTeacherName(c *fiber.Ctx) error {
cancel()
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
"success": false,
"message": "Catchphrase not found",
"message": "teacher not found",
"error": err,
})
}
@@ -296,13 +471,17 @@ func UpdateTeacherName(c *fiber.Ctx) error {
)
if updateErr != nil {
cancel()
return c.Status(400).JSON(fiber.Map{
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
"success": false,
"message": "the student could not be updated",
"message": "the teacher could not be updated",
"error": updateErr,
})
}
defer cancel()
return c.Status(200).JSON(result)
return c.Status(fiber.StatusOK).JSON(fiber.Map{
"success": true,
"message": "successfully updated teacher",
"result": result,
})
}