package repository import ( "context" "database/sql" "errors" "kontor-api-echo/internal/helpers" "kontor-api-echo/internal/models" "log/slog" "time" "github.com/google/uuid" "github.com/labstack/echo/v5" "github.com/uptrace/bun" ) type MediaVideoRepository struct { db *bun.DB } func (repository *MediaVideoRepository) Import(dtos models.MediaVideoDTOList) { slog.Info("MediaVideoRepository#Import") for index, dto := range dtos { slog.Info("importing", "index", index, "video", dto) model := dto.ConvertDTOToModel() existing_item := models.MediaVideo{ID: dto.ID} err := repository.db.NewSelect().Model(&existing_item).Where("id = ?", dto.ID).Scan(context.Background()) if err != nil { slog.Info("check existing id: not found - inserting") _, err := repository.db.NewInsert().Model(&model).Exec(context.Background()) if err != nil { slog.Info("error during import", "error", err.Error()) } } else { slog.Info("check existing id: found - updating") _, err := repository.db.NewUpdate().Model(&model). Set("updated_at = ? ", time.Now). Where("id = ?", model.ID). Exec(context.Background()) if err != nil { slog.Info("error during updating", "error", err.Error()) } } } } func NewMediaVideoRepository(db *bun.DB) *MediaVideoRepository { repo := &MediaVideoRepository{db: db} return repo } func (repository *MediaVideoRepository) FindAll(context context.Context) ([]models.MediaVideo, error) { var videos []models.MediaVideo err := repository.db.NewSelect().Model(&videos).Scan(context) if err != nil { slog.Info(err.Error()) return nil, echo.ErrInternalServerError } slog.Info("MediaVideoRepository#FindAll", "count", len(videos)) return videos, nil } func (repository *MediaVideoRepository) GetById(context context.Context, id string) (*models.MediaVideo, error) { var video models.MediaVideo slog.Info("MediaVideoRepository#GetById", "id", id) err := repository.db.NewSelect().Model(&video).Where("id = ?", id).Scan(context) if err != nil { return nil, err } return &video, nil } func (repository *MediaVideoRepository) Add(context context.Context, url string) (*models.MediaVideo, error) { slog.Info("MediaVideoRepository#Add", "url", url) webLink := helpers.StripUrl(url, "v") var model models.MediaVideo err := repository.db.NewSelect().Model(&model).Where("url = ?", url).Scan(context) if errors.Is(err, sql.ErrNoRows) { slog.Info("MediaMusicRepository#Add URL not found", "url", webLink) video := models.MediaVideo{ ID: uuid.NewString(), CreatedAt: time.Now(), UpdatedAt: time.Now(), DeletedAt: nil, Version: 0, WebLink: webLink, ShouldDownload: false, Review: true, Title: "", FileName: "", } _, add_error := repository.db.NewInsert().Model(&model).Exec(context) if add_error != nil { slog.Info("add new MediaMusic", "error", add_error) return nil, add_error } slog.Info("add new MediaMusic", "lofi", video) return &video, nil } else { slog.Info("search for existing url", "lofi", model) return &model, nil } } func (repository *MediaVideoRepository) Update(context context.Context, update models.MediaVideoDTO) (*models.MediaVideo, error) { slog.Info("MediaVideoRepository#Update", "video", update) var video models.MediaVideo slog.Info("MediaVideoRepository#Update", "id", update.ID) err := repository.db.NewSelect().Model(&video).Where("id = ?", update.ID).Scan(context) if err != nil { return nil, err } _, err = repository.db.NewUpdate().Model(&video). Set("updated_at = ? ", time.Now). Where("id = ?", update.ID). Exec(context) if err != nil { slog.Info("error during updating", "error", err.Error()) } return &video, nil } func (repository *MediaVideoRepository) Delete(context context.Context, id string) error { slog.Info("MediaVideoRepository#Delete", "id", id) _, err := repository.db.NewDelete().Model((*models.MediaVideo)(nil)).Where("id = ?", id).Exec(context) if err != nil { slog.Info("MediaVideoRepository#Delete", "error", err.Error()) } return err }