mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.3 KiB
Go
107 lines
3.3 KiB
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/synctv-org/synctv/internal/model"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
func CreateMovie(movie *model.Movie) error {
|
|
return db.Create(movie).Error
|
|
}
|
|
|
|
func GetAllMoviesByRoomID(roomID string) []*model.Movie {
|
|
movies := []*model.Movie{}
|
|
db.Where("room_id = ?", roomID).Order("position ASC").Find(&movies)
|
|
return movies
|
|
}
|
|
|
|
func DeleteMovieByID(roomID, id string) error {
|
|
err := db.Unscoped().Where("room_id = ? AND id = ?", roomID, id).Delete(&model.Movie{}).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("room or movie not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func LoadAndDeleteMovieByID(roomID, id string, columns ...clause.Column) (*model.Movie, error) {
|
|
movie := &model.Movie{}
|
|
err := db.Unscoped().Clauses(clause.Returning{Columns: columns}).Where("room_id = ? AND id = ?", roomID, id).Delete(movie).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return movie, errors.New("room or movie not found")
|
|
}
|
|
return movie, err
|
|
}
|
|
|
|
func DeleteMoviesByRoomID(roomID string) error {
|
|
err := db.Unscoped().Where("room_id = ?", roomID).Delete(&model.Movie{}).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("room not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func LoadAndDeleteMoviesByRoomID(roomID string, columns ...clause.Column) ([]*model.Movie, error) {
|
|
movies := []*model.Movie{}
|
|
err := db.Unscoped().Clauses(clause.Returning{Columns: columns}).Where("room_id = ?", roomID).Delete(&movies).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, errors.New("room not found")
|
|
}
|
|
return movies, err
|
|
}
|
|
|
|
func UpdateMovie(movie *model.Movie, columns ...clause.Column) error {
|
|
err := db.Model(movie).Clauses(clause.Returning{Columns: columns}).Where("room_id = ? AND id = ?", movie.RoomID, movie.ID).Updates(movie).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("room or movie not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func SaveMovie(movie *model.Movie, columns ...clause.Column) error {
|
|
err := db.Model(movie).Clauses(clause.Returning{Columns: columns}).Where("room_id = ? AND id = ?", movie.RoomID, movie.ID).Save(movie).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("room or movie not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func SwapMoviePositions(roomID, movie1ID, movie2ID string) (err error) {
|
|
tx := db.Begin()
|
|
defer func() {
|
|
if err != nil {
|
|
tx.Rollback()
|
|
} else {
|
|
tx.Commit()
|
|
}
|
|
}()
|
|
movie1 := &model.Movie{}
|
|
movie2 := &model.Movie{}
|
|
err = tx.Select("position").Where("room_id = ? AND id = ?", roomID, movie1ID).First(movie1).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
err = fmt.Errorf("movie with id %s not found", movie1ID)
|
|
}
|
|
return
|
|
}
|
|
err = tx.Select("position").Where("room_id = ? AND id = ?", roomID, movie2ID).First(movie2).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
err = fmt.Errorf("movie with id %s not found", movie2ID)
|
|
}
|
|
return
|
|
}
|
|
err = tx.Model(&model.Movie{}).Where("room_id = ? AND id = ?", roomID, movie1ID).Update("position", movie2.Position).Error
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = tx.Model(&model.Movie{}).Where("room_id = ? AND id = ?", roomID, movie2ID).Update("position", movie1.Position).Error
|
|
if err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|