stash/api/resolver_mutation_performer.go

package api

import (
	"context"
	"database/sql"
	"github.com/stashapp/stash/database"
	"github.com/stashapp/stash/models"
	"github.com/stashapp/stash/utils"
	"strconv"
	"time"
)

func (r *mutationResolver) PerformerCreate(ctx context.Context, input models.PerformerCreateInput) (*models.Performer, error) {
	// Process the base 64 encoded image string
	checksum, imageData, err := utils.ProcessBase64Image(input.Image)
	if err != nil {
		return nil, err
	}

	// Populate a new performer from the input
	currentTime := time.Now()
	newPerformer := models.Performer{
		Image: imageData,
		Checksum: checksum,
		CreatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
		UpdatedAt: models.SQLiteTimestamp{ Timestamp: currentTime },
	}
	if input.Name != nil {
		newPerformer.Name = sql.NullString{ String: *input.Name, Valid: true }
	}
	if input.URL != nil {
		newPerformer.Url = sql.NullString{ String: *input.URL, Valid: true }
	}
	if input.Birthdate != nil {
		newPerformer.Birthdate = sql.NullString{ String: *input.Birthdate, Valid: true }
	}
	if input.Ethnicity != nil {
		newPerformer.Ethnicity = sql.NullString{ String: *input.Ethnicity, Valid: true }
	}
	if input.Country != nil {
		newPerformer.Country = sql.NullString{ String: *input.Country, Valid: true }
	}
	if input.EyeColor != nil {
		newPerformer.EyeColor = sql.NullString{ String: *input.EyeColor, Valid: true }
	}
	if input.Height != nil {
		newPerformer.Height = sql.NullString{ String: *input.Height, Valid: true }
	}
	if input.Measurements != nil {
		newPerformer.Measurements = sql.NullString{ String: *input.Measurements, Valid: true }
	}
	if input.FakeTits != nil {
		newPerformer.FakeTits = sql.NullString{ String: *input.FakeTits, Valid: true }
	}
	if input.CareerLength != nil {
		newPerformer.CareerLength = sql.NullString{ String: *input.CareerLength, Valid: true }
	}
	if input.Tattoos != nil {
		newPerformer.Tattoos = sql.NullString{ String: *input.Tattoos, Valid: true }
	}
	if input.Piercings != nil {
		newPerformer.Piercings = sql.NullString{ String: *input.Piercings, Valid: true }
	}
	if input.Aliases != nil {
		newPerformer.Aliases = sql.NullString{ String: *input.Aliases, Valid: true }
	}
	if input.Twitter != nil {
		newPerformer.Twitter = sql.NullString{ String: *input.Twitter, Valid: true }
	}
	if input.Instagram != nil {
		newPerformer.Instagram = sql.NullString{ String: *input.Instagram, Valid: true }
	}
	if input.Favorite != nil {
		newPerformer.Favorite = sql.NullBool{ Bool: *input.Favorite, Valid: true }
	}

	// Start the transaction and save the performer
	tx := database.DB.MustBeginTx(ctx, nil)
	qb := models.NewPerformerQueryBuilder()
	performer, err := qb.Create(newPerformer, tx)
	if err != nil {
		_ = tx.Rollback()
		return nil, err
	}

	// Commit
	if err := tx.Commit(); err != nil {
		return nil, err
	}

	return performer, nil
}

func (r *mutationResolver) PerformerUpdate(ctx context.Context, input models.PerformerUpdateInput) (*models.Performer, error) {
	// Populate performer from the input
	performerID, _ := strconv.Atoi(input.ID)
	updatedPerformer := models.Performer{
		ID: performerID,
		UpdatedAt: models.SQLiteTimestamp{ Timestamp: time.Now() },
	}
	if input.Image != nil {
		checksum, imageData, err := utils.ProcessBase64Image(*input.Image)
		if err != nil {
			return nil, err
		}
		updatedPerformer.Image = imageData
		updatedPerformer.Checksum = checksum
	}
	if input.Name != nil {
		updatedPerformer.Name = sql.NullString{ String: *input.Name, Valid: true }
	}
	if input.URL != nil {
		updatedPerformer.Url = sql.NullString{ String: *input.URL, Valid: true }
	}
	if input.Birthdate != nil {
		updatedPerformer.Birthdate = sql.NullString{ String: *input.Birthdate, Valid: true }
	}
	if input.Ethnicity != nil {
		updatedPerformer.Ethnicity = sql.NullString{ String: *input.Ethnicity, Valid: true }
	}
	if input.Country != nil {
		updatedPerformer.Country = sql.NullString{ String: *input.Country, Valid: true }
	}
	if input.EyeColor != nil {
		updatedPerformer.EyeColor = sql.NullString{ String: *input.EyeColor, Valid: true }
	}
	if input.Height != nil {
		updatedPerformer.Height = sql.NullString{ String: *input.Height, Valid: true }
	}
	if input.Measurements != nil {
		updatedPerformer.Measurements = sql.NullString{ String: *input.Measurements, Valid: true }
	}
	if input.FakeTits != nil {
		updatedPerformer.FakeTits = sql.NullString{ String: *input.FakeTits, Valid: true }
	}
	if input.CareerLength != nil {
		updatedPerformer.CareerLength = sql.NullString{ String: *input.CareerLength, Valid: true }
	}
	if input.Tattoos != nil {
		updatedPerformer.Tattoos = sql.NullString{ String: *input.Tattoos, Valid: true }
	}
	if input.Piercings != nil {
		updatedPerformer.Piercings = sql.NullString{ String: *input.Piercings, Valid: true }
	}
	if input.Aliases != nil {
		updatedPerformer.Aliases = sql.NullString{ String: *input.Aliases, Valid: true }
	}
	if input.Twitter != nil {
		updatedPerformer.Twitter = sql.NullString{ String: *input.Twitter, Valid: true }
	}
	if input.Instagram != nil {
		updatedPerformer.Instagram = sql.NullString{ String: *input.Instagram, Valid: true }
	}
	if input.Favorite != nil {
		updatedPerformer.Favorite = sql.NullBool{ Bool: *input.Favorite, Valid: true }
	}

	// Start the transaction and save the performer
	tx := database.DB.MustBeginTx(ctx, nil)
	qb := models.NewPerformerQueryBuilder()
	performer, err := qb.Update(updatedPerformer, tx)
	if err != nil {
		_ = tx.Rollback()
		return nil, err
	}

	// Commit
	if err := tx.Commit(); err != nil {
		return nil, err
	}

	return performer, nil
}