aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--pkg/ext/middleware.go10
-rw-r--r--pkg/ext/responses.go2
-rw-r--r--pkg/list/list.go1
-rw-r--r--pkg/service/auth.go15
-rw-r--r--pkg/view/album.go3
-rw-r--r--pkg/view/auth.go4
-rw-r--r--pkg/view/filesystem.go2
-rw-r--r--pkg/view/media.go2
-rw-r--r--pkg/worker/scanner/album_scanner.go1
9 files changed, 21 insertions, 19 deletions
diff --git a/pkg/ext/middleware.go b/pkg/ext/middleware.go
index fe2d185..2cf70cf 100644
--- a/pkg/ext/middleware.go
+++ b/pkg/ext/middleware.go
@@ -56,7 +56,7 @@ func NewAuthMiddleware(key []byte, log *logrus.Entry) *AuthMiddleware {
func (a *AuthMiddleware) LoggedIn(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
- path := string(r.URL.Path)
+ path := r.URL.Path
if path == "/login" || path == "/initial" {
next(w, r)
return
@@ -82,7 +82,7 @@ func (a *AuthMiddleware) LoggedIn(next http.HandlerFunc) http.HandlerFunc {
http.Redirect(w, r, redirectLogin, http.StatusTemporaryRedirect)
return
}
- r = r.WithContext(context.WithValue(r.Context(), "token", token))
+ r = r.WithContext(context.WithValue(r.Context(), service.TokenKey, token))
a.entry.
WithField("userID", token.UserID).
WithField("username", token.Username).
@@ -91,8 +91,8 @@ func (a *AuthMiddleware) LoggedIn(next http.HandlerFunc) http.HandlerFunc {
}
}
-func GetTokenFromCtx(w http.ResponseWriter, r *http.Request) *service.Token {
- tokenValue := r.Context().Value("token")
+func GetTokenFromCtx(r *http.Request) *service.Token {
+ tokenValue := r.Context().Value(service.TokenKey)
if token, ok := tokenValue.(*service.Token); ok {
return token
}
@@ -112,7 +112,7 @@ func NewInitialSetupMiddleware(userRepository repository.UserRepository) *Initia
func (i *InitialSetupMiddleware) Check(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
// if user has been set to context it is logged in already
- token := GetTokenFromCtx(w, r)
+ token := GetTokenFromCtx(r)
if token == nil {
next(w, r)
return
diff --git a/pkg/ext/responses.go b/pkg/ext/responses.go
index ba58dd5..34e5f27 100644
--- a/pkg/ext/responses.go
+++ b/pkg/ext/responses.go
@@ -7,7 +7,7 @@ import (
"git.sr.ht/~gabrielgio/img/templates"
)
-func NotFound(w http.ResponseWriter, r *http.Request) {
+func NotFound(w http.ResponseWriter) {
templates.WritePageTemplate(w, &templates.ErrorPage{
Err: "Not Found",
})
diff --git a/pkg/list/list.go b/pkg/list/list.go
index b6b6b89..482e5bf 100644
--- a/pkg/list/list.go
+++ b/pkg/list/list.go
@@ -26,7 +26,6 @@ func Distribuite[T any](slice []T, size int) [][]T {
chuncks[x] = append(chuncks[x], slice[end])
}
-
}
return chuncks
diff --git a/pkg/service/auth.go b/pkg/service/auth.go
index f27cf88..30e574a 100644
--- a/pkg/service/auth.go
+++ b/pkg/service/auth.go
@@ -105,7 +105,7 @@ func (u *AuthController) Upsert(
) error {
if id != nil {
if err := u.userRepository.Update(ctx, *id, &repository.UpdateUser{
- Username: string(username),
+ Username: username,
Name: name,
IsAdmin: isAdmin,
Path: path,
@@ -140,10 +140,15 @@ func (u *AuthController) Upsert(
return err
}
-type Token struct {
- UserID uint
- Username string
-}
+type (
+ AuthKey string
+ Token struct {
+ UserID uint
+ Username string
+ }
+)
+
+const TokenKey AuthKey = "token"
func ReadToken(data []byte, key []byte) (*Token, error) {
block, err := aes.NewCipher(key)
diff --git a/pkg/view/album.go b/pkg/view/album.go
index 9e70b4b..b19e381 100644
--- a/pkg/view/album.go
+++ b/pkg/view/album.go
@@ -30,7 +30,7 @@ func NewAlbumView(
func (self *AlbumView) Index(w http.ResponseWriter, r *http.Request) error {
p := getPagination(r)
- token := ext.GetTokenFromCtx(w, r)
+ token := ext.GetTokenFromCtx(r)
// TODO: optmize call, GetPathFromUserID may no be necessary
userPath, err := self.userRepository.GetPathFromUserID(r.Context(), token.UserID)
@@ -67,7 +67,6 @@ func (self *AlbumView) Index(w http.ResponseWriter, r *http.Request) error {
if err != nil {
return err
}
-
}
medias, err := self.mediaRepository.List(r.Context(), p)
diff --git a/pkg/view/auth.go b/pkg/view/auth.go
index 2a4b95e..1b87235 100644
--- a/pkg/view/auth.go
+++ b/pkg/view/auth.go
@@ -19,7 +19,7 @@ func NewAuthView(userController *service.AuthController) *AuthView {
}
}
-func (v *AuthView) LoginView(w http.ResponseWriter, r *http.Request) error {
+func (v *AuthView) LoginView(w http.ResponseWriter, _ *http.Request) error {
templates.WritePageTemplate(w, &templates.LoginPage{})
return nil
}
@@ -72,7 +72,7 @@ func Index(w http.ResponseWriter, r *http.Request) {
http.Redirect(w, r, "/login", http.StatusTemporaryRedirect)
}
-func (v *AuthView) InitialRegisterView(w http.ResponseWriter, r *http.Request) error {
+func (v *AuthView) InitialRegisterView(w http.ResponseWriter, _ *http.Request) error {
templates.WritePageTemplate(w, &templates.RegisterPage{})
return nil
}
diff --git a/pkg/view/filesystem.go b/pkg/view/filesystem.go
index d49ad4f..24f0ce6 100644
--- a/pkg/view/filesystem.go
+++ b/pkg/view/filesystem.go
@@ -34,7 +34,7 @@ func NewFileSystemView(
func (self *FileSystemView) Index(w http.ResponseWriter, r *http.Request) error {
var (
pathValue = r.FormValue("path")
- token = ext.GetTokenFromCtx(w, r)
+ token = ext.GetTokenFromCtx(r)
)
page, err := self.fsService.GetPage(r.Context(), token.UserID, pathValue)
diff --git a/pkg/view/media.go b/pkg/view/media.go
index f490ce1..3041998 100644
--- a/pkg/view/media.go
+++ b/pkg/view/media.go
@@ -71,7 +71,7 @@ func NewMediaView(
func (self *MediaView) Index(w http.ResponseWriter, r *http.Request) error {
p := getPagination(r)
- token := ext.GetTokenFromCtx(w, r)
+ token := ext.GetTokenFromCtx(r)
userPath, err := self.userRepository.GetPathFromUserID(r.Context(), token.UserID)
if err != nil {
diff --git a/pkg/worker/scanner/album_scanner.go b/pkg/worker/scanner/album_scanner.go
index 04af9bc..6413cb1 100644
--- a/pkg/worker/scanner/album_scanner.go
+++ b/pkg/worker/scanner/album_scanner.go
@@ -85,7 +85,6 @@ func (e *AlbumScanner) GetAndCreateNestedAlbuns(ctx context.Context, paths []str
Name: filepath.Base(paths[0]),
Path: paths[0],
})
-
}
func FanInwards(paths []string) []string {