mirror of https://github.com/stashapp/stash.git
107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/stashapp/stash/pkg/api/urlbuilders"
|
|
"github.com/stashapp/stash/pkg/image"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
)
|
|
|
|
func (r *imageResolver) Title(ctx context.Context, obj *models.Image) (*string, error) {
|
|
ret := image.GetTitle(obj)
|
|
return &ret, nil
|
|
}
|
|
|
|
func (r *imageResolver) Rating(ctx context.Context, obj *models.Image) (*int, error) {
|
|
if obj.Rating.Valid {
|
|
rating := int(obj.Rating.Int64)
|
|
return &rating, nil
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *imageResolver) File(ctx context.Context, obj *models.Image) (*models.ImageFileType, error) {
|
|
width := int(obj.Width.Int64)
|
|
height := int(obj.Height.Int64)
|
|
size := int(obj.Size.Int64)
|
|
return &models.ImageFileType{
|
|
Size: &size,
|
|
Width: &width,
|
|
Height: &height,
|
|
}, nil
|
|
}
|
|
|
|
func (r *imageResolver) Paths(ctx context.Context, obj *models.Image) (*models.ImagePathsType, error) {
|
|
baseURL, _ := ctx.Value(BaseURLCtxKey).(string)
|
|
builder := urlbuilders.NewImageURLBuilder(baseURL, obj)
|
|
thumbnailPath := builder.GetThumbnailURL()
|
|
imagePath := builder.GetImageURL()
|
|
return &models.ImagePathsType{
|
|
Image: &imagePath,
|
|
Thumbnail: &thumbnailPath,
|
|
}, nil
|
|
}
|
|
|
|
func (r *imageResolver) Galleries(ctx context.Context, obj *models.Image) (ret []*models.Gallery, err error) {
|
|
if err := r.withReadTxn(ctx, func(repo models.ReaderRepository) error {
|
|
var err error
|
|
ret, err = repo.Gallery().FindByImageID(obj.ID)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (r *imageResolver) Studio(ctx context.Context, obj *models.Image) (ret *models.Studio, err error) {
|
|
if !obj.StudioID.Valid {
|
|
return nil, nil
|
|
}
|
|
|
|
if err := r.withReadTxn(ctx, func(repo models.ReaderRepository) error {
|
|
ret, err = repo.Studio().Find(int(obj.StudioID.Int64))
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (r *imageResolver) Tags(ctx context.Context, obj *models.Image) (ret []*models.Tag, err error) {
|
|
if err := r.withReadTxn(ctx, func(repo models.ReaderRepository) error {
|
|
ret, err = repo.Tag().FindByImageID(obj.ID)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (r *imageResolver) Performers(ctx context.Context, obj *models.Image) (ret []*models.Performer, err error) {
|
|
if err := r.withReadTxn(ctx, func(repo models.ReaderRepository) error {
|
|
ret, err = repo.Performer().FindByImageID(obj.ID)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (r *imageResolver) CreatedAt(ctx context.Context, obj *models.Image) (*time.Time, error) {
|
|
return &obj.CreatedAt.Timestamp, nil
|
|
}
|
|
|
|
func (r *imageResolver) UpdatedAt(ctx context.Context, obj *models.Image) (*time.Time, error) {
|
|
return &obj.UpdatedAt.Timestamp, nil
|
|
}
|
|
|
|
func (r *imageResolver) FileModTime(ctx context.Context, obj *models.Image) (*time.Time, error) {
|
|
return &obj.FileModTime.Timestamp, nil
|
|
}
|