97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/fpmb/server/internal/database"
|
|
"github.com/fpmb/server/internal/models"
|
|
"github.com/gofiber/fiber/v2"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
func GetWhiteboard(c *fiber.Ctx) error {
|
|
userID, err := primitive.ObjectIDFromHex(c.Locals("user_id").(string))
|
|
if err != nil {
|
|
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid user"})
|
|
}
|
|
|
|
projectID, err := primitive.ObjectIDFromHex(c.Params("projectId"))
|
|
if err != nil {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid project ID"})
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
if _, err := getProjectRole(ctx, projectID, userID); err != nil {
|
|
return c.Status(fiber.StatusForbidden).JSON(fiber.Map{"error": "Access denied"})
|
|
}
|
|
|
|
var wb models.Whiteboard
|
|
err = database.GetCollection("whiteboards").FindOne(ctx, bson.M{"project_id": projectID}).Decode(&wb)
|
|
if err == mongo.ErrNoDocuments {
|
|
return c.JSON(fiber.Map{"id": nil, "project_id": projectID, "data": "", "updated_at": nil})
|
|
}
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Failed to fetch whiteboard"})
|
|
}
|
|
|
|
return c.JSON(wb)
|
|
}
|
|
|
|
func SaveWhiteboard(c *fiber.Ctx) error {
|
|
userID, err := primitive.ObjectIDFromHex(c.Locals("user_id").(string))
|
|
if err != nil {
|
|
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid user"})
|
|
}
|
|
|
|
projectID, err := primitive.ObjectIDFromHex(c.Params("projectId"))
|
|
if err != nil {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid project ID"})
|
|
}
|
|
|
|
var body struct {
|
|
Data string `json:"data"`
|
|
}
|
|
if err := c.BodyParser(&body); err != nil {
|
|
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request body"})
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
roleFlags, err := getProjectRole(ctx, projectID, userID)
|
|
if err != nil || !hasPermission(roleFlags, RoleEditor) {
|
|
return c.Status(fiber.StatusForbidden).JSON(fiber.Map{"error": "Insufficient permissions"})
|
|
}
|
|
|
|
now := time.Now()
|
|
col := database.GetCollection("whiteboards")
|
|
|
|
var existing models.Whiteboard
|
|
err = col.FindOne(ctx, bson.M{"project_id": projectID}).Decode(&existing)
|
|
|
|
if err == mongo.ErrNoDocuments {
|
|
wb := &models.Whiteboard{
|
|
ID: primitive.NewObjectID(),
|
|
ProjectID: projectID,
|
|
Data: body.Data,
|
|
CreatedBy: userID,
|
|
CreatedAt: now,
|
|
UpdatedAt: now,
|
|
}
|
|
col.InsertOne(ctx, wb)
|
|
return c.JSON(fiber.Map{"id": wb.ID, "project_id": projectID, "updated_at": now})
|
|
}
|
|
|
|
col.UpdateOne(ctx, bson.M{"project_id": projectID}, bson.M{"$set": bson.M{
|
|
"data": body.Data,
|
|
"updated_at": now,
|
|
}})
|
|
|
|
return c.JSON(fiber.Map{"id": existing.ID, "project_id": projectID, "updated_at": now})
|
|
}
|