update/better-logic

This commit is contained in:
SowinskiBraeden committed 2022-02-07 14:12:32 -08:00
1 parent 592cedc276
commit 325c842230
4 files changed
+59 -26

No files matched your search

+36 -22
View File
@@ -39,8 +39,7 @@ func AuthAdmin(c *fiber.Ctx) bool {
claims := token.Claims.(*jwt.StandardClaims)
var admin models.Admin
findErr := adminCollection.FindOne(context.TODO(), bson.M{"aid": claims.Issuer}).Decode(&admin)
findErr := adminCollection.FindOne(context.TODO(), bson.M{"aid": claims.Issuer})
if findErr != nil {
return false
}
@@ -60,8 +59,7 @@ func AuthStudent(c *fiber.Ctx) (verified bool, sid string) {
claims := token.Claims.(*jwt.StandardClaims)
var student models.Admin
findErr := adminCollection.FindOne(context.TODO(), bson.M{"sid": claims.Issuer}).Decode(&student)
findErr := studentCollection.FindOne(context.TODO(), bson.M{"sid": claims.Issuer})
if findErr != nil {
return false, ""
}
@@ -70,7 +68,7 @@ func AuthStudent(c *fiber.Ctx) (verified bool, sid string) {
}
func Enroll(c *fiber.Ctx) error {
var data map[string]string
var data map[string]interface{}
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
if err := c.BodyParser(&data); err != nil {
@@ -92,7 +90,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"] == "" {
if data["firstname"] == "" || data["lastname"] == "" || data["age"] == "" || data["gradelevel"] == "" || data["dob"] == "" || data["email"] == "" || data["province"] == "" || data["city"] == "" || data["address"] == "" || data["postal"] == "" {
cancel()
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
"success": false,
@@ -101,22 +99,31 @@ func Enroll(c *fiber.Ctx) error {
}
var student models.Student
student.PersonalData.FirstName = data["firstname"]
student.PersonalData.MiddleName = data["middlename"]
student.PersonalData.LastName = data["lastname"]
student.PersonalData.Age, _ = strconv.Atoi(data["age"])
student.SchoolData.GradeLevel, _ = strconv.Atoi(data["gradelevel"])
student.PersonalData.DOB = data["dob"]
student.PersonalData.Email = data["email"]
student.PersonalData.Province = data["province"]
student.PersonalData.City = data["city"]
student.PersonalData.Address = data["address"]
student.PersonalData.Postal = data["postal"]
student.PersonalData.FirstName = data["firstname"].(string)
student.PersonalData.MiddleName = data["middlename"].(string)
student.PersonalData.LastName = data["lastname"].(string)
student.PersonalData.Age = data["age"].(int)
student.SchoolData.GradeLevel = data["gradelevel"].(int)
student.PersonalData.DOB = data["dob"].(string)
student.PersonalData.Email = data["email"].(string)
student.PersonalData.Province = data["province"].(string)
student.PersonalData.City = data["city"].(string)
student.PersonalData.Address = data["address"].(string)
student.PersonalData.Postal = data["postal"].(string)
student.PersonalData.Contacts = []string{}
student.SchoolData.YOG = ((12 - student.SchoolData.GradeLevel) + time.Now().Year()) + 1
student.AccountData.SchoolEmail = student.GenerateSchoolEmail()
var schoolEmail string = ""
offset := 0
for {
schoolEmail = student.GenerateSchoolEmail(offset, schoolEmail)
if student.EmailExists(schoolEmail) == true {
break
}
offset++
}
student.AccountData.SchoolEmail = schoolEmail
// Disable login block
student.AccountData.AccountDisabled = false
@@ -176,7 +183,7 @@ func Enroll(c *fiber.Ctx) error {
}
func RegisterTeacher(c *fiber.Ctx) error {
var data map[string]string
var data map[string]interface{}
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
if err := c.BodyParser(&data); err != nil {
@@ -207,9 +214,14 @@ func RegisterTeacher(c *fiber.Ctx) error {
}
var teacher models.Teacher
teacher.FirstName = data["firstname"]
teacher.LastName = data["lastname"]
teacher.Email = data["email"]
teacher.FirstName = data["firstname"].(string)
teacher.MiddleName = data["middlename"].(string)
teacher.LastName = data["lastname"].(string)
teacher.Email = data["email"].(string)
teacher.Province = data["province"].(string)
teacher.City = data["city"].(string)
teacher.Postal = data["postal"].(string)
teacher.DOB = data["postal"].(string)
teacher.SchoolEmail = teacher.GenerateSchoolEmail()
@@ -303,6 +315,8 @@ func CreateAdmin(c *fiber.Ctx) error {
admin.LastName = data["lastname"]
admin.Email = data["email"]
admin.SchoolEmail = admin.GenerateSchoolEmail()
tempPass := admin.GeneratePassword(12, 1, 1, 1)
admin.Password = admin.HashPassword(tempPass)
admin.TempPassword = true
+7
View File
@@ -16,6 +16,7 @@ type Admin struct {
FirstName string `json:"firstname" validate:"required"`
LastName string `json:"lastname" validate:"required"`
Email string `json:"email" validate:"required"`
SchoolEmail string `json:"schoolemail"`
Password string `json:"-" validate:"min=10,max=32"`
TempPassword bool `json:"temppassword"`
AID string `json:"aid"`
@@ -23,6 +24,12 @@ type Admin struct {
Updated_at time.Time `json:"updated_at"`
}
func (a *Admin) GenerateSchoolEmail() string {
var email string = strings.ToLower(a.LastName) + "_" + strings.ToLower(string(a.FirstName[0])) + "@surreyschools.ca"
// Add check to see if email already exists
return email
}
func (s *Admin) HashPassword(password string) string {
hash, _ := bcrypt.GenerateFromPassword([]byte(password), 14)
return string(hash)
+15 -2
View File
@@ -1,6 +1,7 @@
package models
import (
"context"
"math/rand"
"school-management/database"
"strings"
@@ -9,11 +10,13 @@ import (
"golang.org/x/crypto/bcrypt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
)
var lockerCollection *mongo.Collection = database.OpenCollection(database.Client, "lockers")
var studentCollection *mongo.Collection = database.OpenCollection(database.Client, "students")
type Student struct {
ID primitive.ObjectID `bson:"_id"`
@@ -56,9 +59,19 @@ func (s *Student) HashPassword(password string) string {
return string(hash)
}
func (s *Student) GenerateSchoolEmail() string {
func (s *Student) EmailExists(email string) bool {
findErr := studentCollection.FindOne(context.TODO(), bson.M{"schooldata.schoolemail": email})
if findErr != nil {
return false
}
return true
}
func (s *Student) GenerateSchoolEmail(offset int, lastEmail string) string {
var email string = strings.ToLower(string(s.PersonalData.FirstName[0])) + "." + strings.ToLower(s.PersonalData.LastName) + "@surreyschools.ca"
// Add check to see if email already exists
if offset > 0 {
email = lastEmail[:offset] + string(s.PersonalData.FirstName[offset]) + lastEmail[offset:]
}
return email
}
+1 -2
View File
@@ -45,8 +45,7 @@ func (t *Teacher) HashPassword(password string) string {
}
func (t *Teacher) GenerateSchoolEmail() string {
var email string = strings.ToLower(string(t.FirstName[0])) + "_" + strings.ToLower(t.LastName) + "@surreyschools.ca"
// Add check to see if email already exists
var email string = strings.ToLower(t.LastName) + "_" + strings.ToLower(string(t.FirstName[0])) + "@surreyschools.ca"
return email
}