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/v4" "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} 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 NewMediaLoFiRepository(db *bun.DB) *MediaLoFiRepository { repo := &MediaLoFiRepository{db: db} return repo } func (repository *MediaLoFiRepository) FindAll(context context.Context) ([]models.MediaLoFi, error) { var models []models.MediaLoFi err := repository.db.NewSelect().Model(&models).Scan(context) if err != nil { slog.Info(err.Error()) return nil, echo.ErrInternalServerError } slog.Info("MailRepository#FindAll", "count", len(models)) return models, nil } func (repository *MediaLoFiRepository) GetById(context context.Context, id string) (*models.MediaLoFi, error) { var model = models.MediaLoFi{ID: id} slog.Info("MediaLoFiRepository#GetById", "id", id) err := repository.db.NewSelect().Model(&model).Where("id = ?", id).Scan(context) if err != nil { slog.Info("AssignmentRepository#GetById", "error", err.Error()) return nil, err } return &model, err } func (repository *MediaLoFiRepository) Add(context context.Context, url string) (*models.MediaLoFi, error) { slog.Info("MediaLoFiRepository#Add", "url", url) webLink := helpers.StripUrl(url, "v") var model models.MediaLoFi err := repository.db.NewSelect().Model(&model).Where("url = ?", url).Scan(context) if errors.Is(err, 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_error := repository.db.NewInsert().Model(&model).Exec(context) if add_error != nil { slog.Info("add new MediaLofi", "error", add_error) return nil, add_error } slog.Info("add new MediaLofi", "lofi", lofi) return &lofi, nil } else { slog.Info("search for existing url", "lofi", model) return &model, nil } } func (repository *MediaLoFiRepository) Update(context context.Context, update models.MediaLoFiDTO) (*models.MediaLoFi, error) { slog.Info("MediaLoFiRepository#Update", "lofi", update) var model = models.MediaLoFi{ID: update.ID} model.Update(update) slog.Info("MediaLoFiRepository#Update", "id", update.ID) _, err := repository.db.NewUpdate().Model(&model). Set("updated_at = ? ", time.Now). Where("id = ?", update.ID). Exec(context) if err != nil { slog.Info("MediaLoFiRepository#Update", "error", err.Error()) } return &model, nil } func (repository *MediaLoFiRepository) Delete(context context.Context, id string) error { slog.Info("MediaLoFiRepository#Delete", "id", id) _, err := repository.db.NewDelete().Model((*models.MediaLoFi)(nil)).Where("id = ?", id).Exec(context) if err != nil { slog.Info("MediaLoFiRepository#Delete", "error", err.Error()) } return err }