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.
72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"github.com/synctv-org/synctv/internal/model"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
func CreateMovie(movie *model.Movie) error {
|
|
return db.Create(movie).Error
|
|
}
|
|
|
|
func GetAllMoviesByRoomID(roomID uint) ([]*model.Movie, error) {
|
|
movies := []*model.Movie{}
|
|
err := db.Where("room_id = ?", roomID).Order("position ASC").Find(&movies).Error
|
|
return movies, err
|
|
}
|
|
|
|
func DeleteMovieByID(roomID, id uint) error {
|
|
return db.Unscoped().Where("room_id = ? AND id = ?", roomID, id).Delete(&model.Movie{}).Error
|
|
}
|
|
|
|
// TODO: delete error
|
|
func LoadAndDeleteMovieByID(roomID, id uint, 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
|
|
return movie, err
|
|
}
|
|
|
|
func DeleteMoviesByRoomID(roomID uint) error {
|
|
return db.Unscoped().Where("room_id = ?", roomID).Delete(&model.Movie{}).Error
|
|
}
|
|
|
|
func LoadAndDeleteMoviesByRoomID(roomID uint, columns ...clause.Column) ([]*model.Movie, error) {
|
|
movies := []*model.Movie{}
|
|
err := db.Unscoped().Clauses(clause.Returning{Columns: columns}).Where("room_id = ?", roomID).Delete(&movies).Error
|
|
return movies, err
|
|
}
|
|
|
|
func UpdateMovie(movie *model.Movie, columns ...clause.Column) error {
|
|
return db.Model(movie).Clauses(clause.Returning{Columns: columns}).Where("room_id = ? AND id = ?", movie.RoomID, movie.ID).Updates(movie).Error
|
|
}
|
|
|
|
func SwapMoviePositions(roomID uint, movie1ID uint, movie2ID uint) (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 {
|
|
return
|
|
}
|
|
err = tx.Select("position").Where("room_id = ? AND id = ?", roomID, movie2ID).First(movie2).Error
|
|
if err != nil {
|
|
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
|
|
}
|