package api

import (
	"context"
	"github.com/stashapp/stash/pkg/models"
	"strconv"
)

func (r *queryResolver) FindStudio(ctx context.Context, id string) (*models.Studio, error) {
	qb := models.NewStudioQueryBuilder()
	idInt, _ := strconv.Atoi(id)
	return qb.Find(idInt, nil)
}

func (r *queryResolver) FindStudios(ctx context.Context, filter *models.FindFilterType) (models.FindStudiosResultType, error) {
	qb := models.NewStudioQueryBuilder()
	studios, total := qb.Query(filter)
	return models.FindStudiosResultType{
		Count:   total,
		Studios: studios,
	}, nil
}

func (r *queryResolver) AllStudios(ctx context.Context) ([]models.Studio, error) {
	qb := models.NewStudioQueryBuilder()
	return qb.All()
}