stash/pkg/studio/export.go

package studio

import (
	"context"
	"fmt"

	"github.com/stashapp/stash/pkg/logger"
	"github.com/stashapp/stash/pkg/models"
	"github.com/stashapp/stash/pkg/models/json"
	"github.com/stashapp/stash/pkg/models/jsonschema"
	"github.com/stashapp/stash/pkg/utils"
)

type FinderImageStashIDGetter interface {
	models.StudioGetter
	models.AliasLoader
	models.URLLoader
	models.StashIDLoader
	GetImage(ctx context.Context, studioID int) ([]byte, error)
	models.CustomFieldsReader
}

// ToJSON converts a Studio object into its JSON equivalent.
func ToJSON(ctx context.Context, reader FinderImageStashIDGetter, studio *models.Studio) (*jsonschema.Studio, error) {
	newStudioJSON := jsonschema.Studio{
		Name:          studio.Name,
		Details:       studio.Details,
		Favorite:      studio.Favorite,
		IgnoreAutoTag: studio.IgnoreAutoTag,
		CreatedAt:     json.JSONTime{Time: studio.CreatedAt},
		UpdatedAt:     json.JSONTime{Time: studio.UpdatedAt},
	}

	if studio.ParentID != nil {
		parent, err := reader.Find(ctx, *studio.ParentID)
		if err != nil {
			return nil, fmt.Errorf("error getting parent studio: %v", err)
		}

		if parent != nil {
			newStudioJSON.ParentStudio = parent.Name
		}
	}

	if studio.Rating != nil {
		newStudioJSON.Rating = *studio.Rating
	}

	if err := studio.LoadAliases(ctx, reader); err != nil {
		return nil, fmt.Errorf("loading studio aliases: %w", err)
	}
	newStudioJSON.Aliases = studio.Aliases.List()

	if err := studio.LoadURLs(ctx, reader); err != nil {
		return nil, fmt.Errorf("loading studio URLs: %w", err)
	}
	newStudioJSON.URLs = studio.URLs.List()

	if err := studio.LoadStashIDs(ctx, reader); err != nil {
		return nil, fmt.Errorf("loading studio stash ids: %w", err)
	}
	newStudioJSON.StashIDs = studio.StashIDs.List()

	var err error
	newStudioJSON.CustomFields, err = reader.GetCustomFields(ctx, studio.ID)
	if err != nil {
		return nil, fmt.Errorf("getting studio custom fields: %v", err)
	}

	image, err := reader.GetImage(ctx, studio.ID)
	if err != nil {
		logger.Errorf("Error getting studio image: %v", err)
	}

	if len(image) > 0 {
		newStudioJSON.Image = utils.GetBase64StringFromData(image)
	}

	return &newStudioJSON, nil
}