Skip to content

Commit 24cce55

Browse files
authored
fix: add validate profile image file type (#16)
1 parent 4840705 commit 24cce55

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

platform/server/controller/user.go

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ import (
44
"time"
55

66
"github.com/codern-org/codern/domain"
7+
errs "github.com/codern-org/codern/domain/error"
8+
"github.com/codern-org/codern/internal/validator"
79
"github.com/codern-org/codern/platform/server/middleware"
810
"github.com/codern-org/codern/platform/server/payload"
911
"github.com/codern-org/codern/platform/server/response"
@@ -34,6 +36,14 @@ func (c *UserController) Update(ctx *fiber.Ctx) error {
3436

3537
user := middleware.GetUserFromCtx(ctx)
3638

39+
fileMimeType, err := validator.GetMimeType(pl.Profile)
40+
if err != nil {
41+
return err
42+
}
43+
if fileMimeType != "image/png" && fileMimeType != "image/jpeg" && fileMimeType != "image/jpg" && fileMimeType != "image/gif" {
44+
return errs.New(errs.ErrBodyParser, "unsupported file type")
45+
}
46+
3747
if err := c.userUsecase.Update(
3848
user.Id,
3949
&domain.UpdateUser{

0 commit comments

Comments
 (0)