stash/pkg/models/model_studio.go

package models

import (
	"database/sql"
	"time"

	"github.com/stashapp/stash/pkg/hash/md5"
)

type Studio struct {
	ID        int             `db:"id" json:"id"`
	Checksum  string          `db:"checksum" json:"checksum"`
	Name      sql.NullString  `db:"name" json:"name"`
	URL       sql.NullString  `db:"url" json:"url"`
	ParentID  sql.NullInt64   `db:"parent_id,omitempty" json:"parent_id"`
	CreatedAt SQLiteTimestamp `db:"created_at" json:"created_at"`
	UpdatedAt SQLiteTimestamp `db:"updated_at" json:"updated_at"`
	// Rating expressed in 1-100 scale
	Rating        sql.NullInt64  `db:"rating" json:"rating"`
	Details       sql.NullString `db:"details" json:"details"`
	IgnoreAutoTag bool           `db:"ignore_auto_tag" json:"ignore_auto_tag"`
	// TODO - this is only here because of database code in the models package
	ImageBlob sql.NullString `db:"image_blob" json:"-"`
}

type StudioPartial struct {
	ID        int              `db:"id" json:"id"`
	Checksum  *string          `db:"checksum" json:"checksum"`
	Name      *sql.NullString  `db:"name" json:"name"`
	URL       *sql.NullString  `db:"url" json:"url"`
	ParentID  *sql.NullInt64   `db:"parent_id,omitempty" json:"parent_id"`
	CreatedAt *SQLiteTimestamp `db:"created_at" json:"created_at"`
	UpdatedAt *SQLiteTimestamp `db:"updated_at" json:"updated_at"`
	// Rating expressed in 1-100 scale
	Rating        *sql.NullInt64  `db:"rating" json:"rating"`
	Details       *sql.NullString `db:"details" json:"details"`
	IgnoreAutoTag *bool           `db:"ignore_auto_tag" json:"ignore_auto_tag"`
}

func NewStudio(name string) *Studio {
	currentTime := time.Now()
	return &Studio{
		Checksum:  md5.FromString(name),
		Name:      sql.NullString{String: name, Valid: true},
		CreatedAt: SQLiteTimestamp{Timestamp: currentTime},
		UpdatedAt: SQLiteTimestamp{Timestamp: currentTime},
	}
}

type Studios []*Studio

func (s *Studios) Append(o interface{}) {
	*s = append(*s, o.(*Studio))
}

func (s *Studios) New() interface{} {
	return &Studio{}
}