128 lines
3.8 KiB
Go
128 lines
3.8 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 MediaLoFiRepository struct {
|
|
db *bun.DB
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) Import(dtos models.MediaLoFiDTOList) {
|
|
slog.Info("MediaLoFiRepository#Import")
|
|
for index, dto := range dtos {
|
|
slog.Info("importing", "index", index, "video", dto)
|
|
model := dto.ConvertDTOToModel()
|
|
existing_item := models.MediaLoFi{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 NewMediaLoFiRepository(db *bun.DB) *MediaLoFiRepository {
|
|
repo := &MediaLoFiRepository{db: db}
|
|
return repo
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) FindAll(context context.Context) ([]models.MediaLoFi, error) {
|
|
var lofis []models.MediaLoFi
|
|
|
|
result := repository.db.Find(&lofis)
|
|
slog.Info("MediaLoFiRepository#FindAll", "count", result.RowsAffected)
|
|
|
|
return lofis, nil
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) GetById(context context.Context, id string) (*models.MediaLoFi, error) {
|
|
var lofi = models.MediaLoFi{ID: id}
|
|
slog.Info("MediaLoFiRepository#GetById", "id", id)
|
|
result := repository.db.First(&lofi)
|
|
if result.Error != nil {
|
|
slog.Info("MediaLoFiRepository#GetById", "error", result.Error.Error())
|
|
}
|
|
return &lofi, result.Error
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) Add(context context.Context, url string) (*models.MediaLoFi, error) {
|
|
slog.Info("MediaLoFiRepository#Add", "url", url)
|
|
webLink := helpers.StripUrl(url, "v")
|
|
var lofi models.MediaLoFi
|
|
search_result := repository.db.Where("url = ?", webLink).First(&lofi)
|
|
if search_result.RowsAffected > 0 {
|
|
slog.Info("search for existing url", "result", search_result, "lofi", lofi)
|
|
return &lofi, nil
|
|
}
|
|
if errors.Is(search_result.Error, sql.ErrNoRows) {
|
|
slog.Info("MediaLoFiRepository#Add URL not found", "url", webLink)
|
|
lofi = models.MediaLoFi{
|
|
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(&lofi)
|
|
if add_result.Error != nil {
|
|
slog.Info("add new MediaLofi", "error", add_result.Error)
|
|
return nil, add_result.Error
|
|
}
|
|
slog.Info("add new MediaLofi", "lofi", lofi)
|
|
return &lofi, nil
|
|
}
|
|
if search_result.Error != nil {
|
|
return nil, search_result.Error
|
|
}
|
|
return nil, fmt.Errorf("MediaLoFiRepository#Add unknown error")
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) Update(context context.Context, update models.MediaLoFiDTO) (*models.MediaLoFi, error) {
|
|
slog.Info("MediaLoFiRepository#Update", "lofi", update)
|
|
var lofi = models.MediaLoFi{ID: update.ID}
|
|
slog.Info("MediaLoFiRepository#Update", "id", update.ID)
|
|
result := repository.db.First(&lofi)
|
|
if result.Error != nil {
|
|
slog.Info("MediaLoFiRepository#Update", "error", result.Error.Error())
|
|
return nil, result.Error
|
|
}
|
|
lofi.Update(update)
|
|
repository.db.Save(&lofi)
|
|
return &lofi, nil
|
|
}
|
|
|
|
func (repository *MediaLoFiRepository) Delete(context context.Context, id string) error {
|
|
var actor = models.MediaLoFi{ID: id}
|
|
slog.Info("MediaLoFiRepository#Delete", "id", id)
|
|
result := repository.db.Delete(&actor)
|
|
if result.Error != nil {
|
|
slog.Info("MediaLoFiRepository#Delete", "error", result.Error.Error())
|
|
}
|
|
return result.Error
|
|
}
|