From 44ccc885ee59c53e6501aebbb698ff81023ea9c5 Mon Sep 17 00:00:00 2001
From: Reeverflow <serenereeverflow@gmail.com>
Date: Sat, 21 Feb 2026 19:51:51 +0100
Subject: [PATCH] Added Playlist One to Many and Video Many to One relation to
 the DB.

---
 .../video_downloader/dto/SaveNewPlaylistRequest.java     | 2 ++
 .../com/example/video_downloader/entity/Playlist.java    | 9 +++++++++
 .../java/com/example/video_downloader/entity/Video.java  | 5 +++++
 .../video_downloader/repositories/VideoRepository.java   | 2 +-
 .../video_downloader/services/PlaylistService.java       | 3 +++
 .../video_downloader/services/ProcessService.java        | 2 --
 .../example/video_downloader/services/YTDLPService.java  | 5 ++++-
 7 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/example/video_downloader/dto/SaveNewPlaylistRequest.java b/src/main/java/com/example/video_downloader/dto/SaveNewPlaylistRequest.java
index a81293b..3d2ae2e 100644
--- a/src/main/java/com/example/video_downloader/dto/SaveNewPlaylistRequest.java
+++ b/src/main/java/com/example/video_downloader/dto/SaveNewPlaylistRequest.java
@@ -4,5 +4,7 @@ import lombok.Data;
 
 @Data
 public class SaveNewPlaylistRequest {
+    private String playlist_name;
     private String url;
+
 }
diff --git a/src/main/java/com/example/video_downloader/entity/Playlist.java b/src/main/java/com/example/video_downloader/entity/Playlist.java
index d39ab4a..feab01a 100644
--- a/src/main/java/com/example/video_downloader/entity/Playlist.java
+++ b/src/main/java/com/example/video_downloader/entity/Playlist.java
@@ -4,16 +4,25 @@ import jakarta.persistence.*;
 import lombok.Data;
 import lombok.ToString;
 
+import java.util.ArrayList;
+import java.util.List;
+
 @Data
 @ToString
 @Entity
 @Table(name = "playlists")
+
 public class Playlist {
     @Id
     @GeneratedValue(strategy = GenerationType.SEQUENCE)
     private Long id;
 
+    private String playlist_name;
+
     private String url;
     private StatusEnum status;
 
+    @OneToMany(mappedBy = "playlist", cascade = CascadeType.ALL, orphanRemoval = true)
+    private List<Video> videos = new ArrayList<>();
+
 }
diff --git a/src/main/java/com/example/video_downloader/entity/Video.java b/src/main/java/com/example/video_downloader/entity/Video.java
index 2c8c495..c898124 100644
--- a/src/main/java/com/example/video_downloader/entity/Video.java
+++ b/src/main/java/com/example/video_downloader/entity/Video.java
@@ -21,4 +21,9 @@ public class Video {
     private StatusEnum status;
 
     private String fullPath;
+
+    @ManyToOne
+    @JoinColumn(name = "playlist_id")
+    private Playlist playlist;
+
 }
diff --git a/src/main/java/com/example/video_downloader/repositories/VideoRepository.java b/src/main/java/com/example/video_downloader/repositories/VideoRepository.java
index 84405bb..976010f 100644
--- a/src/main/java/com/example/video_downloader/repositories/VideoRepository.java
+++ b/src/main/java/com/example/video_downloader/repositories/VideoRepository.java
@@ -9,5 +9,5 @@ import java.util.List;
 
 @Repository
 public interface VideoRepository extends JpaRepository<Video, Long> {
-    List<Video> findUrlByStatus(StatusEnum status);
+    List<Video> findByStatus(StatusEnum status);
 }
diff --git a/src/main/java/com/example/video_downloader/services/PlaylistService.java b/src/main/java/com/example/video_downloader/services/PlaylistService.java
index 44f7e99..6923df8 100644
--- a/src/main/java/com/example/video_downloader/services/PlaylistService.java
+++ b/src/main/java/com/example/video_downloader/services/PlaylistService.java
@@ -5,9 +5,11 @@ import com.example.video_downloader.entity.Playlist;
 import com.example.video_downloader.entity.StatusEnum;
 import com.example.video_downloader.repositories.PlaylistRepository;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
 
 import java.util.List;
 
+@Service
 public class PlaylistService {
 
     @Autowired
@@ -17,6 +19,7 @@ public class PlaylistService {
         Playlist playlist = new Playlist();
 
         playlist.setStatus(StatusEnum.NEW);
+        playlist.setPlaylistName(request.getPlaylistName());
         playlist.setUrl(request.getUrl());
         playlistRepository.save(playlist);
         return playlist;
diff --git a/src/main/java/com/example/video_downloader/services/ProcessService.java b/src/main/java/com/example/video_downloader/services/ProcessService.java
index 00f16a1..f1261a6 100644
--- a/src/main/java/com/example/video_downloader/services/ProcessService.java
+++ b/src/main/java/com/example/video_downloader/services/ProcessService.java
@@ -13,8 +13,6 @@ public class ProcessService {
         builder.redirectError(ProcessBuilder.Redirect.INHERIT);
         Process process = builder.start();
         int exitCode = process.waitFor();
-
-
     }
 
 
diff --git a/src/main/java/com/example/video_downloader/services/YTDLPService.java b/src/main/java/com/example/video_downloader/services/YTDLPService.java
index 8066c37..c6debee 100644
--- a/src/main/java/com/example/video_downloader/services/YTDLPService.java
+++ b/src/main/java/com/example/video_downloader/services/YTDLPService.java
@@ -18,12 +18,15 @@ public class YTDLPService {
     @Autowired
     private VideoRepository videoRepository;
 
+    @Autowired
+    private PlaylistService playlistService;
+
     @Autowired
     private ProcessService processService;
 
     @PostConstruct
     public void validateVideos() throws HttpClientErrorException, IOException, InterruptedException {
-        List<Video> response = videoRepository.findUrlByStatus(StatusEnum.NEW);
+        List<Video> response = videoRepository.findByStatus(StatusEnum.NEW);
         for(Video video : response) {
             if(video.getStatus().equals(StatusEnum.NEW)){
                 downloadVideo(video.getUrl(), video.getId());