auth roles

Author: krishkumar84

cmd/portal-api/main.go

@@ -37,12 +37,12 @@ func main() {
 
 	router := http.NewServeMux()
 
-	router.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
+	router.HandleFunc("GET/health", func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(http.StatusOK)
 		w.Write([]byte("OK"))
 	})
 
-	router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+	router.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(http.StatusOK)
 		w.Write([]byte("Welcome to BDCOE Portal API server is dockerized up and running"))
 	})

pkg/http/handler/users/users.go

@@ -39,7 +39,7 @@ func New(storage storage.Storage) http.HandlerFunc{
 			return
 		}
 
-		lastId, err := storage.CreateUser(user.Name, user.Email, user.Password, user.StudentId)
+		lastId, err := storage.CreateUser(user.Name, user.Email, user.Password, user.StudentId,types.RoleUser)
 
 		slog.Info("User created sucessfully",slog.String("userId",fmt.Sprint(lastId)))
 		if err != nil {

pkg/storage/mongodb/mongodb.go

@@ -41,7 +41,7 @@ func New(cfg *config.Config) (*MongoDB, error) {
 }
 
 // User operations
-func (m *MongoDB) CreateUser(name, email, password, studentId string) (string, error) {
+func (m *MongoDB) CreateUser(name, email, password, studentId string, role types.Role) (string, error) {
     collection := m.db.Collection("users")
     ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
     defer cancel()
@@ -69,6 +69,7 @@ func (m *MongoDB) CreateUser(name, email, password, studentId string) (string, e
         StudentId: studentId,
         Password:  password,
         CreatedAt: time.Now(),
+        Role:      role,
     }
 
     result, err := collection.InsertOne(ctx, user)

pkg/storage/storage.go

@@ -6,6 +6,6 @@ type Storage interface {
 	// CreateUser(name string, email string, age int)(int64, error)
 	// GetUserById(id int64)(types.User, error) 
 	// GetAllUsers()([]types.User, error)
-	CreateUser(name string, email string, password string, studentId string) (string, error)
+	CreateUser(name string, email string, password string, studentId string, role types.Role) (string, error)
 	GetUserByEmail(email string) (*types.User, error)
 }

pkg/types/models.go

@@ -5,6 +5,11 @@ import (
 
 	"go.mongodb.org/mongo-driver/bson/primitive"
 )
+type Role string
+const (
+    RoleUser  Role = "user"
+    RoleAdmin Role = "admin"
+)
 
 type User struct {
     ID        primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
@@ -13,6 +18,7 @@ type User struct {
     Password  string            `bson:"password" json:"password" validate:"required"`
     StudentId string            `bson:"studentId" json:"studentId" validate:"required"`
     CreatedAt time.Time         `bson:"createdAt" json:"createdAt"`
+    Role      Role              `bson:"role" json:"role" default:"user"`
 }
 
 type Contest struct {