mirror of
https://github.com/navidrome/navidrome.git
synced 2025-04-04 13:07:36 +03:00
Move artwork id encoding to public package
This commit is contained in:
parent
13ba08157a
commit
e40da183bb
7 changed files with 97 additions and 62 deletions
|
@ -7,9 +7,7 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/lestrrat-go/jwx/v2/jwt"
|
|
||||||
"github.com/navidrome/navidrome/core"
|
"github.com/navidrome/navidrome/core"
|
||||||
"github.com/navidrome/navidrome/core/auth"
|
|
||||||
"github.com/navidrome/navidrome/core/ffmpeg"
|
"github.com/navidrome/navidrome/core/ffmpeg"
|
||||||
"github.com/navidrome/navidrome/log"
|
"github.com/navidrome/navidrome/log"
|
||||||
"github.com/navidrome/navidrome/model"
|
"github.com/navidrome/navidrome/model"
|
||||||
|
@ -111,31 +109,3 @@ func (a *artwork) getArtworkReader(ctx context.Context, artID model.ArtworkID, s
|
||||||
}
|
}
|
||||||
return artReader, err
|
return artReader, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func EncodeArtworkID(artID model.ArtworkID) string {
|
|
||||||
token, _ := auth.CreatePublicToken(map[string]any{"id": artID.String()})
|
|
||||||
return token
|
|
||||||
}
|
|
||||||
|
|
||||||
func DecodeArtworkID(tokenString string) (model.ArtworkID, error) {
|
|
||||||
token, err := auth.TokenAuth.Decode(tokenString)
|
|
||||||
if err != nil {
|
|
||||||
return model.ArtworkID{}, err
|
|
||||||
}
|
|
||||||
if token == nil {
|
|
||||||
return model.ArtworkID{}, errors.New("unauthorized")
|
|
||||||
}
|
|
||||||
err = jwt.Validate(token, jwt.WithRequiredClaim("id"))
|
|
||||||
if err != nil {
|
|
||||||
return model.ArtworkID{}, err
|
|
||||||
}
|
|
||||||
claims, err := token.AsMap(context.Background())
|
|
||||||
if err != nil {
|
|
||||||
return model.ArtworkID{}, err
|
|
||||||
}
|
|
||||||
id, ok := claims["id"].(string)
|
|
||||||
if !ok {
|
|
||||||
return model.ArtworkID{}, errors.New("invalid id type")
|
|
||||||
}
|
|
||||||
return model.ParseArtworkID(id)
|
|
||||||
}
|
|
||||||
|
|
|
@ -4,12 +4,10 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"io"
|
"io"
|
||||||
|
|
||||||
"github.com/go-chi/jwtauth/v5"
|
|
||||||
"github.com/navidrome/navidrome/conf"
|
"github.com/navidrome/navidrome/conf"
|
||||||
"github.com/navidrome/navidrome/conf/configtest"
|
"github.com/navidrome/navidrome/conf/configtest"
|
||||||
"github.com/navidrome/navidrome/consts"
|
"github.com/navidrome/navidrome/consts"
|
||||||
"github.com/navidrome/navidrome/core/artwork"
|
"github.com/navidrome/navidrome/core/artwork"
|
||||||
"github.com/navidrome/navidrome/core/auth"
|
|
||||||
"github.com/navidrome/navidrome/model"
|
"github.com/navidrome/navidrome/model"
|
||||||
"github.com/navidrome/navidrome/resources"
|
"github.com/navidrome/navidrome/resources"
|
||||||
"github.com/navidrome/navidrome/tests"
|
"github.com/navidrome/navidrome/tests"
|
||||||
|
@ -46,31 +44,4 @@ var _ = Describe("Artwork", func() {
|
||||||
Expect(result).To(Equal(phBytes))
|
Expect(result).To(Equal(phBytes))
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Context("Public ID Encoding", func() {
|
|
||||||
BeforeEach(func() {
|
|
||||||
auth.TokenAuth = jwtauth.New("HS256", []byte("super secret"), nil)
|
|
||||||
})
|
|
||||||
It("returns a reversible string representation", func() {
|
|
||||||
id := model.NewArtworkID(model.KindArtistArtwork, "1234")
|
|
||||||
encoded := artwork.EncodeArtworkID(id)
|
|
||||||
decoded, err := artwork.DecodeArtworkID(encoded)
|
|
||||||
Expect(err).ToNot(HaveOccurred())
|
|
||||||
Expect(decoded).To(Equal(id))
|
|
||||||
})
|
|
||||||
It("fails to decode an invalid token", func() {
|
|
||||||
_, err := artwork.DecodeArtworkID("xx-123")
|
|
||||||
Expect(err).To(MatchError("invalid JWT"))
|
|
||||||
})
|
|
||||||
It("fails to decode an invalid id", func() {
|
|
||||||
encoded := artwork.EncodeArtworkID(model.ArtworkID{})
|
|
||||||
_, err := artwork.DecodeArtworkID(encoded)
|
|
||||||
Expect(err).To(MatchError("invalid artwork id"))
|
|
||||||
})
|
|
||||||
It("fails to decode a token without an id", func() {
|
|
||||||
token, _ := auth.CreatePublicToken(map[string]any{})
|
|
||||||
_, err := artwork.DecodeArtworkID(token)
|
|
||||||
Expect(err).To(HaveOccurred())
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
})
|
||||||
|
|
38
server/public/encode_artwork_id.go
Normal file
38
server/public/encode_artwork_id.go
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
package public
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"github.com/lestrrat-go/jwx/v2/jwt"
|
||||||
|
"github.com/navidrome/navidrome/core/auth"
|
||||||
|
"github.com/navidrome/navidrome/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func EncodeArtworkID(artID model.ArtworkID) string {
|
||||||
|
token, _ := auth.CreatePublicToken(map[string]any{"id": artID.String()})
|
||||||
|
return token
|
||||||
|
}
|
||||||
|
|
||||||
|
func DecodeArtworkID(tokenString string) (model.ArtworkID, error) {
|
||||||
|
token, err := auth.TokenAuth.Decode(tokenString)
|
||||||
|
if err != nil {
|
||||||
|
return model.ArtworkID{}, err
|
||||||
|
}
|
||||||
|
if token == nil {
|
||||||
|
return model.ArtworkID{}, errors.New("unauthorized")
|
||||||
|
}
|
||||||
|
err = jwt.Validate(token, jwt.WithRequiredClaim("id"))
|
||||||
|
if err != nil {
|
||||||
|
return model.ArtworkID{}, err
|
||||||
|
}
|
||||||
|
claims, err := token.AsMap(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return model.ArtworkID{}, err
|
||||||
|
}
|
||||||
|
id, ok := claims["id"].(string)
|
||||||
|
if !ok {
|
||||||
|
return model.ArtworkID{}, errors.New("invalid id type")
|
||||||
|
}
|
||||||
|
return model.ParseArtworkID(id)
|
||||||
|
}
|
39
server/public/encode_artwork_id_test.go
Normal file
39
server/public/encode_artwork_id_test.go
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
package public_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/go-chi/jwtauth/v5"
|
||||||
|
"github.com/navidrome/navidrome/core/auth"
|
||||||
|
"github.com/navidrome/navidrome/model"
|
||||||
|
"github.com/navidrome/navidrome/server/public"
|
||||||
|
. "github.com/onsi/ginkgo/v2"
|
||||||
|
. "github.com/onsi/gomega"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ = Describe("EncodeArtworkID", func() {
|
||||||
|
Context("Public ID Encoding", func() {
|
||||||
|
BeforeEach(func() {
|
||||||
|
auth.TokenAuth = jwtauth.New("HS256", []byte("super secret"), nil)
|
||||||
|
})
|
||||||
|
It("returns a reversible string representation", func() {
|
||||||
|
id := model.NewArtworkID(model.KindArtistArtwork, "1234")
|
||||||
|
encoded := public.EncodeArtworkID(id)
|
||||||
|
decoded, err := public.DecodeArtworkID(encoded)
|
||||||
|
Expect(err).ToNot(HaveOccurred())
|
||||||
|
Expect(decoded).To(Equal(id))
|
||||||
|
})
|
||||||
|
It("fails to decode an invalid token", func() {
|
||||||
|
_, err := public.DecodeArtworkID("xx-123")
|
||||||
|
Expect(err).To(MatchError("invalid JWT"))
|
||||||
|
})
|
||||||
|
It("fails to decode an invalid id", func() {
|
||||||
|
encoded := public.EncodeArtworkID(model.ArtworkID{})
|
||||||
|
_, err := public.DecodeArtworkID(encoded)
|
||||||
|
Expect(err).To(MatchError("invalid artwork id"))
|
||||||
|
})
|
||||||
|
It("fails to decode a token without an id", func() {
|
||||||
|
token, _ := auth.CreatePublicToken(map[string]any{})
|
||||||
|
_, err := public.DecodeArtworkID(token)
|
||||||
|
Expect(err).To(HaveOccurred())
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
|
@ -46,7 +46,7 @@ func (p *Router) handleImages(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
artId, err := artwork.DecodeArtworkID(id)
|
artId, err := DecodeArtworkID(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
return
|
return
|
||||||
|
|
17
server/public/public_suite_test.go
Normal file
17
server/public/public_suite_test.go
Normal file
|
@ -0,0 +1,17 @@
|
||||||
|
package public
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/navidrome/navidrome/log"
|
||||||
|
"github.com/navidrome/navidrome/tests"
|
||||||
|
. "github.com/onsi/ginkgo/v2"
|
||||||
|
. "github.com/onsi/gomega"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestPublicEndpoints(t *testing.T) {
|
||||||
|
tests.Init(t, false)
|
||||||
|
log.SetLevel(log.LevelFatal)
|
||||||
|
RegisterFailHandler(Fail)
|
||||||
|
RunSpecs(t, "Public Endpoints Suite")
|
||||||
|
}
|
|
@ -11,10 +11,10 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/navidrome/navidrome/consts"
|
"github.com/navidrome/navidrome/consts"
|
||||||
"github.com/navidrome/navidrome/core/artwork"
|
|
||||||
"github.com/navidrome/navidrome/model"
|
"github.com/navidrome/navidrome/model"
|
||||||
"github.com/navidrome/navidrome/model/request"
|
"github.com/navidrome/navidrome/model/request"
|
||||||
"github.com/navidrome/navidrome/server"
|
"github.com/navidrome/navidrome/server"
|
||||||
|
"github.com/navidrome/navidrome/server/public"
|
||||||
"github.com/navidrome/navidrome/server/subsonic/responses"
|
"github.com/navidrome/navidrome/server/subsonic/responses"
|
||||||
"github.com/navidrome/navidrome/utils"
|
"github.com/navidrome/navidrome/utils"
|
||||||
)
|
)
|
||||||
|
@ -116,7 +116,7 @@ func toArtistID3(r *http.Request, a model.Artist) responses.ArtistID3 {
|
||||||
}
|
}
|
||||||
|
|
||||||
func publicImageURL(r *http.Request, artID model.ArtworkID, size int) string {
|
func publicImageURL(r *http.Request, artID model.ArtworkID, size int) string {
|
||||||
link := artwork.EncodeArtworkID(artID)
|
link := public.EncodeArtworkID(artID)
|
||||||
path := filepath.Join(consts.URLPathPublicImages, link)
|
path := filepath.Join(consts.URLPathPublicImages, link)
|
||||||
params := url.Values{}
|
params := url.Values{}
|
||||||
if size > 0 {
|
if size > 0 {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue