Skip to content

Commit eba8c3d

Browse files
committed
feature: Implemented zap logger
1 parent 97e335b commit eba8c3d

File tree

14 files changed

+79
-41
lines changed

14 files changed

+79
-41
lines changed

app/handlers/extension.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,12 @@ import (
77
"github.com/limanmys/render-engine/internal/sandbox"
88
"github.com/limanmys/render-engine/pkg/helpers"
99
"github.com/limanmys/render-engine/pkg/linux"
10+
"github.com/limanmys/render-engine/pkg/logger"
1011
)
1112

1213
func ExtensionRunner(c *fiber.Ctx) error {
1314
if len(c.FormValue("extension_id")) < 1 {
14-
return fiber.NewError(fiber.StatusNotFound, "extension not found")
15+
return logger.FiberError(fiber.StatusNotFound, "extension not found")
1516
}
1617

1718
extension, err := liman.GetExtension(&models.Extension{
@@ -23,7 +24,7 @@ func ExtensionRunner(c *fiber.Ctx) error {
2324
}
2425

2526
if extension.Status == "0" {
26-
return fiber.NewError(fiber.StatusServiceUnavailable, "extension is unavailable")
27+
return logger.FiberError(fiber.StatusServiceUnavailable, "extension is unavailable")
2728
}
2829

2930
credentials := &models.Credentials{}
@@ -38,7 +39,7 @@ func ExtensionRunner(c *fiber.Ctx) error {
3839
)
3940

4041
if err != nil || len(credentials.Username) < 1 {
41-
return fiber.NewError(fiber.StatusForbidden, "you need a key to use this extension")
42+
return logger.FiberError(fiber.StatusForbidden, "you need a key to use this extension")
4243
}
4344
}
4445

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package appLogger
1+
package app_logger
22

33
import (
44
"github.com/gofiber/fiber/v2"

app/middleware/auth/new.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55

66
"github.com/gofiber/fiber/v2"
77
"github.com/limanmys/render-engine/internal/liman"
8+
"github.com/limanmys/render-engine/pkg/logger"
89
)
910

1011
func New() fiber.Handler {
@@ -18,7 +19,7 @@ func authorization(c *fiber.Ctx) error {
1819
)
1920

2021
if err != nil {
21-
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
22+
return logger.FiberError(fiber.StatusUnauthorized, err.Error())
2223
}
2324

2425
c.Locals("user_id", user)
@@ -31,7 +32,7 @@ func authorization(c *fiber.Ctx) error {
3132
)
3233

3334
if err != nil {
34-
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
35+
return logger.FiberError(fiber.StatusUnauthorized, err.Error())
3536
}
3637

3738
c.Locals("user_id", user)
@@ -40,5 +41,5 @@ func authorization(c *fiber.Ctx) error {
4041

4142
// TODO: Log handlers
4243

43-
return fiber.NewError(fiber.StatusUnauthorized, "authorization token is missing")
44+
return logger.FiberError(fiber.StatusUnauthorized, "authorization token is missing")
4445
}

app/middleware/permission/new.go

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,22 +5,23 @@ import (
55
"github.com/limanmys/render-engine/app/models"
66
"github.com/limanmys/render-engine/internal/liman"
77
"github.com/limanmys/render-engine/pkg/helpers"
8+
"github.com/limanmys/render-engine/pkg/logger"
89
)
910

1011
func New() fiber.Handler {
1112
return permission
1213
}
1314

1415
func permission(c *fiber.Ctx) error {
15-
if helpers.Env("LIMAN_RESTRICTED", "false") == "true" {
16-
return c.Next()
17-
}
18-
1916
user, err := liman.GetUser(&models.User{
2017
ID: c.Locals("user_id").(string),
2118
})
2219
if err != nil {
23-
return fiber.NewError(fiber.StatusInternalServerError, "error while getting the user")
20+
return logger.FiberError(fiber.StatusInternalServerError, "error while getting the user")
21+
}
22+
23+
if helpers.Env("LIMAN_RESTRICTED", "false") == "true" {
24+
return c.Next()
2425
}
2526

2627
if user.Status == 1 {
@@ -29,12 +30,12 @@ func permission(c *fiber.Ctx) error {
2930

3031
perms, err := liman.GetObjectPermissions(user)
3132
if err != nil {
32-
return fiber.NewError(fiber.StatusInternalServerError, "error while getting the object permissions")
33+
return logger.FiberError(fiber.StatusInternalServerError, "error while getting the object permissions")
3334
}
3435

3536
if len(c.FormValue("server_id")) > 0 {
3637
if !helpers.Contains(perms, c.FormValue("server_id")) {
37-
return fiber.NewError(fiber.StatusForbidden, "you have no permission to do this")
38+
return logger.FiberError(fiber.StatusForbidden, "you have no permission to do this")
3839
}
3940
}
4041

@@ -50,7 +51,7 @@ func permission(c *fiber.Ctx) error {
5051
}
5152

5253
if !helpers.Contains(perms, extensionID) || len(extensionID) < 1 {
53-
return fiber.NewError(fiber.StatusForbidden, "you have no permission to do this")
54+
return logger.FiberError(fiber.StatusForbidden, "you have no permission to do this")
5455
}
5556

5657
c.Locals("extension_id", extensionID)

internal/liman/auth.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"github.com/gofiber/fiber/v2"
55
"github.com/limanmys/render-engine/app/models"
66
"github.com/limanmys/render-engine/internal/database"
7+
"github.com/limanmys/render-engine/pkg/logger"
78
)
89

910
func AuthWithToken(token string) (string, error) {
@@ -12,7 +13,7 @@ func AuthWithToken(token string) (string, error) {
1213
err := database.Connection().First(&tokenObj, "token = ?", token).Error
1314

1415
if err != nil || len(tokenObj.UserID) < 1 {
15-
return "", fiber.NewError(fiber.StatusUnauthorized, "authorization token is not valid")
16+
return "", logger.FiberError(fiber.StatusUnauthorized, "authorization token is not valid")
1617
}
1718

1819
return tokenObj.UserID, nil
@@ -24,7 +25,7 @@ func AuthWithAccessToken(token string) (string, error) {
2425
err := database.Connection().First(&tokenObj, "token = ?", token).Error
2526

2627
if err != nil || len(tokenObj.UserID) < 1 {
27-
return "", fiber.NewError(fiber.StatusUnauthorized, "authorization token is not valid")
28+
return "", logger.FiberError(fiber.StatusUnauthorized, "authorization token is not valid")
2829
}
2930

3031
return tokenObj.UserID, nil

internal/liman/credentials.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55
"github.com/gofiber/fiber/v2"
66
"github.com/limanmys/render-engine/app/models"
77
"github.com/limanmys/render-engine/internal/database"
8+
"github.com/limanmys/render-engine/pkg/logger"
89
)
910

1011
func GetCredentials(user *models.User, server *models.Server) (*models.Credentials, error) {
@@ -33,7 +34,7 @@ func GetCredentials(user *models.User, server *models.Server) (*models.Credentia
3334
credentials.Type = serverKey.Type
3435

3536
if len(credentials.Username) < 1 {
36-
return nil, fiber.NewError(fiber.StatusNotFound, "server not found")
37+
return nil, logger.FiberError(fiber.StatusNotFound, "server not found")
3738
}
3839

3940
return credentials, nil

internal/liman/extension.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package liman
22

33
import (
4+
"errors"
45
"fmt"
56
"io/ioutil"
67
"strings"
@@ -10,20 +11,21 @@ import (
1011
"github.com/limanmys/render-engine/app/models"
1112
"github.com/limanmys/render-engine/internal/constants"
1213
"github.com/limanmys/render-engine/internal/database"
14+
"github.com/limanmys/render-engine/pkg/logger"
1315
)
1416

1517
func GetExtension(extension *models.Extension) (*models.Extension, error) {
1618
result := database.Connection().First(&extension)
1719

1820
if result.Error != nil {
19-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found extension with this id")
21+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found extension with this id")
2022
}
2123

2224
if result.RowsAffected > 0 {
2325
return extension, nil
2426
}
2527

26-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found extension with this id")
28+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found extension with this id")
2729
}
2830

2931
func GetExtensionJSON(extension *models.Extension) (map[string]any, error) {
@@ -60,5 +62,5 @@ func GetLicence(extension *models.Extension) (*models.Licence, error) {
6062
return licence, nil
6163
}
6264

63-
return nil, fiber.NewError(fiber.StatusNotFound, "licence not found")
65+
return nil, errors.New("Licence not found")
6466
}

internal/liman/role_system.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"github.com/limanmys/render-engine/app/models"
88
"github.com/limanmys/render-engine/internal/database"
99
"github.com/limanmys/render-engine/pkg/helpers"
10+
"github.com/limanmys/render-engine/pkg/logger"
1011
)
1112

1213
func GetPermissions(user *models.User, extFilter string) ([]string, map[string]string, error) {
@@ -55,7 +56,7 @@ func getPermissionsFromMorph(morphID string, extFilter string) ([]string, map[st
5556

5657
err := database.Connection().Find(&permission, "morph_id = ?", morphID).Error
5758
if err != nil {
58-
return nil, nil, fiber.NewError(fiber.StatusInternalServerError, "error while fetching the permissions")
59+
return nil, nil, logger.FiberError(fiber.StatusInternalServerError, "error while fetching the permissions")
5960
}
6061

6162
funcPerms := []string{}
@@ -101,7 +102,7 @@ func getRoleMaps(user *models.User) ([]string, error) {
101102

102103
err := database.Connection().Find(&roles, "user_id = ?", user.ID).Error
103104
if err != nil {
104-
return nil, fiber.NewError(fiber.StatusInternalServerError, "error while fetching the roles")
105+
return nil, logger.FiberError(fiber.StatusInternalServerError, "error while fetching the roles")
105106
}
106107

107108
roleID := []string{user.ID}

internal/liman/server.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,19 @@ import (
44
"github.com/gofiber/fiber/v2"
55
"github.com/limanmys/render-engine/app/models"
66
"github.com/limanmys/render-engine/internal/database"
7+
"github.com/limanmys/render-engine/pkg/logger"
78
)
89

910
func GetServer(server *models.Server) (*models.Server, error) {
1011
result := database.Connection().First(&server)
1112

1213
if result.Error != nil {
13-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found server with this id")
14+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found server with this id")
1415
}
1516

1617
if result.RowsAffected > 0 {
1718
return server, nil
1819
}
1920

20-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found server with this id")
21+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found server with this id")
2122
}

internal/liman/user.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,19 @@ import (
44
"github.com/gofiber/fiber/v2"
55
"github.com/limanmys/render-engine/app/models"
66
"github.com/limanmys/render-engine/internal/database"
7+
"github.com/limanmys/render-engine/pkg/logger"
78
)
89

910
func GetUser(user *models.User) (*models.User, error) {
1011
result := database.Connection().First(&user)
1112

1213
if result.Error != nil {
13-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found user with this id")
14+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found user with this id")
1415
}
1516

1617
if result.RowsAffected > 0 {
1718
return user, nil
1819
}
1920

20-
return nil, fiber.NewError(fiber.StatusNotFound, "cannot found user with this id")
21+
return nil, logger.FiberError(fiber.StatusNotFound, "cannot found user with this id")
2122
}

0 commit comments

Comments
 (0)