Thomas Peetz
128 lines
3.9 KiB
Go
128 lines
3.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"kontor-api-echo/internal/helpers"
|
|
"kontor-api-echo/internal/models"
|
|
"log/slog"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
type MediaMusicRepository struct {
|
|
db *bun.DB
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) Import(dtos models.MediaMusicDTOList) {
|
|
slog.Info("MediaMusicRepository#Import")
|
|
for index, dto := range dtos {
|
|
slog.Info("importing", "index", index, "video", dto)
|
|
model := dto.ConvertDTOToModel()
|
|
existing_item := models.MediaMusic{ID: dto.ID}
|
|
result_existing := repository.db.First(&existing_item)
|
|
if result_existing.Error != nil {
|
|
slog.Info("check existing id: not found - inserting")
|
|
result := repository.db.Create(&model)
|
|
if result.Error != nil {
|
|
slog.Info("error during import", "error", result.Error.Error())
|
|
}
|
|
} else {
|
|
slog.Info("check existing id: found - updating")
|
|
result := repository.db.Save(&model)
|
|
if result.Error != nil {
|
|
slog.Info("error during updating", "error", result.Error.Error())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func NewMediaMusicRepository(db *bun.DB) *MediaMusicRepository {
|
|
repo := &MediaMusicRepository{db: db}
|
|
return repo
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) FindAll(context context.Context) ([]models.MediaMusic, error) {
|
|
var musics []models.MediaMusic
|
|
|
|
result := repository.db.Find(&musics)
|
|
slog.Info("MediaMusicRepository#FindAll", "count", result.RowsAffected)
|
|
|
|
return musics, nil
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) GetById(context context.Context, id string) (*models.MediaMusic, error) {
|
|
var music = models.MediaMusic{ID: id}
|
|
slog.Info("MediaMusicRepository#GetById", "id", id)
|
|
result := repository.db.First(&music)
|
|
if result.Error != nil {
|
|
slog.Info("MediaMusicRepository#GetById", "error", result.Error.Error())
|
|
}
|
|
return &music, result.Error
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) Add(context context.Context, url string) (*models.MediaMusic, error) {
|
|
slog.Info("MediaMusicRepository#Add", "url", url)
|
|
webLink := helpers.StripUrl(url, "v")
|
|
var music models.MediaMusic
|
|
search_result := repository.db.Where("url = ?", webLink).First(&music)
|
|
if search_result.RowsAffected > 0 {
|
|
slog.Info("search for existing url", "result", search_result, "music", music)
|
|
return &music, nil
|
|
}
|
|
if errors.Is(search_result.Error, sql.ErrNoRows) {
|
|
slog.Info("MediaMusicRepository#Add URL not found", "url", webLink)
|
|
music = models.MediaMusic{
|
|
ID: uuid.NewString(),
|
|
CreatedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
DeletedAt: nil,
|
|
Version: 0,
|
|
WebLink: webLink,
|
|
ShouldDownload: false,
|
|
Review: true,
|
|
Title: "",
|
|
FileName: "",
|
|
}
|
|
add_result := repository.db.Create(&music)
|
|
if add_result.Error != nil {
|
|
slog.Info("add new MediaMusic", "error", add_result.Error)
|
|
return nil, add_result.Error
|
|
}
|
|
slog.Info("add new MediaMusic", "music", music)
|
|
return &music, nil
|
|
}
|
|
if search_result.Error != nil {
|
|
return nil, search_result.Error
|
|
}
|
|
return nil, fmt.Errorf("MediaMusicRepository#Add unknown error")
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) Update(context context.Context, update models.MediaMusicDTO) (*models.MediaMusic, error) {
|
|
slog.Info("MediaMusicRepository#Update", "music", update)
|
|
var music = models.MediaMusic{ID: update.ID}
|
|
slog.Info("MediaMusicRepository#Update", "id", update.ID)
|
|
result := repository.db.First(&music)
|
|
if result.Error != nil {
|
|
slog.Info("MediaMusicRepository#Update", "error", result.Error.Error())
|
|
return nil, result.Error
|
|
}
|
|
music.Update(update)
|
|
repository.db.Save(&music)
|
|
return &music, nil
|
|
}
|
|
|
|
func (repository *MediaMusicRepository) Delete(context context.Context, id string) error {
|
|
var actor = models.MediaMusic{ID: id}
|
|
slog.Info("MediaMusicRepository#Delete", "id", id)
|
|
result := repository.db.Delete(&actor)
|
|
if result.Error != nil {
|
|
slog.Info("MediaMusicRepository#Delete", "error", result.Error.Error())
|
|
}
|
|
return result.Error
|
|
}
|