package urlbuilders

import (
	"strconv"

	"github.com/stashapp/stash/pkg/models"
)

type StudioURLBuilder struct {
	BaseURL   string
	StudioID  string
	UpdatedAt string
}

func NewStudioURLBuilder(baseURL string, studio *models.Studio) StudioURLBuilder {
	return StudioURLBuilder{
		BaseURL:   baseURL,
		StudioID:  strconv.Itoa(studio.ID),
		UpdatedAt: strconv.FormatInt(studio.UpdatedAt.Unix(), 10),
	}
}

func (b StudioURLBuilder) GetStudioImageURL(hasImage bool) string {
	url := b.BaseURL + "/studio/" + b.StudioID + "/image?t=" + b.UpdatedAt
	if !hasImage {
		url += "&default=true"
	}
	return url
}