diff --git a/service/utils/auth.go b/service/utils/auth.go index 29cbb3b..69d5116 100644 --- a/service/utils/auth.go +++ b/service/utils/auth.go @@ -53,6 +53,10 @@ func Auth(ctx context.Context, t yggdrasil.ValidateToken, client *ent.Client, pu ut, err := client.UserToken.Query().Where(usertoken.HasUserWith(user.ID(claims.UID))).First(ctx) if err != nil { + var ne *ent.NotFoundError + if !errors.As(err, &ne) { + return nil, fmt.Errorf("Auth: %w", errors.Join(err, ErrTokenInvalid)) + } return nil, fmt.Errorf("Auth: %w", err) } if strconv.FormatUint(ut.TokenID, 10) != claims.Tid { diff --git a/service/yggdrasil/texture.go b/service/yggdrasil/texture.go index e05a0a3..34445a9 100644 --- a/service/yggdrasil/texture.go +++ b/service/yggdrasil/texture.go @@ -99,7 +99,7 @@ func (y *Yggdrasil) PutTexture(ctx context.Context, token string, texturebyte [] } } if t == nil { - err = tx.Texture.Create().SetCreatedUser(u).SetTextureHash(hashstr).Exec(ctx) + t, err = tx.Texture.Create().SetCreatedUser(u).SetTextureHash(hashstr).Save(ctx) if err != nil { return err }