folders for playlists

Author: James-Cherished

app/src/main/java/org/schabi/newpipe/NewPipeDatabase.kt

@@ -17,6 +17,7 @@ import org.schabi.newpipe.database.Migrations.MIGRATION_5_6
 import org.schabi.newpipe.database.Migrations.MIGRATION_6_7
 import org.schabi.newpipe.database.Migrations.MIGRATION_7_8
 import org.schabi.newpipe.database.Migrations.MIGRATION_8_9
+import org.schabi.newpipe.database.Migrations.MIGRATION_9_10
 import kotlin.concurrent.Volatile
 
 object NewPipeDatabase {
@@ -27,8 +28,8 @@ object NewPipeDatabase {
     private fun getDatabase(context: Context): AppDatabase {
         return databaseBuilder(
             context.applicationContext,
-            AppDatabase::class.java,
-            AppDatabase.Companion.DATABASE_NAME
+             AppDatabase::class.java,
+             AppDatabase.Companion.DATABASE_NAME
         ).addMigrations(
             MIGRATION_1_2,
             MIGRATION_2_3,
@@ -37,7 +38,8 @@ object NewPipeDatabase {
             MIGRATION_5_6,
             MIGRATION_6_7,
             MIGRATION_7_8,
-            MIGRATION_8_9
+            MIGRATION_8_9,
+            MIGRATION_9_10
         ).build()
     }
 

app/src/main/java/org/schabi/newpipe/database/AppDatabase.kt

@@ -34,7 +34,7 @@ import org.schabi.newpipe.database.subscription.SubscriptionEntity
 
 @TypeConverters(Converters::class)
 @Database(
-    version = Migrations.DB_VER_9,
+    version = Migrations.DB_VER_10,
     entities = [
         SubscriptionEntity::class,
         SearchHistoryEntry::class,
@@ -44,6 +44,7 @@ import org.schabi.newpipe.database.subscription.SubscriptionEntity
         PlaylistEntity::class,
         PlaylistStreamEntity::class,
         PlaylistRemoteEntity::class,
+        org.schabi.newpipe.database.playlist.model.PlaylistFolderEntity::class,
         FeedEntity::class,
         FeedGroupEntity::class,
         FeedGroupSubscriptionEntity::class,
@@ -56,6 +57,7 @@ abstract class AppDatabase : RoomDatabase() {
     abstract fun playlistDAO(): PlaylistDAO
     abstract fun playlistRemoteDAO(): PlaylistRemoteDAO
     abstract fun playlistStreamDAO(): PlaylistStreamDAO
+    abstract fun playlistFolderDAO(): org.schabi.newpipe.database.playlist.dao.PlaylistFolderDAO
     abstract fun searchHistoryDAO(): SearchHistoryDAO
     abstract fun streamDAO(): StreamDAO
     abstract fun streamHistoryDAO(): StreamHistoryDAO

app/src/main/java/org/schabi/newpipe/database/Migrations.kt

@@ -30,6 +30,7 @@ object Migrations {
     const val DB_VER_7 = 7
     const val DB_VER_8 = 8
     const val DB_VER_9 = 9
+    const val DB_VER_10 = 10
 
     private val TAG = Migrations::class.java.getName()
     private val isDebug = MainActivity.DEBUG
@@ -365,4 +366,33 @@ object Migrations {
             }
         }
     }
+
+    val MIGRATION_9_10 = object : Migration(DB_VER_9, DB_VER_10) {
+        override fun migrate(db: SupportSQLiteDatabase) {
+            // Add folder table and folder_id column to playlists
+            db.execSQL(
+                "CREATE TABLE IF NOT EXISTS `playlist_folders` (" +
+                    "`uid` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, " +
+                    "`name` TEXT NOT NULL, " +
+                    "`sort_order` INTEGER NOT NULL DEFAULT 0)"
+            )
+
+            // Add nullable folder_id column to playlists
+            db.execSQL(
+                "ALTER TABLE `playlists` ADD COLUMN `folder_id` INTEGER"
+            )
+        }
+    }
+
+    val ALL_MIGRATIONS = arrayOf(
+        MIGRATION_1_2,
+        MIGRATION_2_3,
+        MIGRATION_3_4,
+        MIGRATION_4_5,
+        MIGRATION_5_6,
+        MIGRATION_6_7,
+        MIGRATION_7_8,
+        MIGRATION_8_9,
+        MIGRATION_9_10
+    )
 }

app/src/main/java/org/schabi/newpipe/database/playlist/PlaylistMetadataEntry.kt

@@ -29,6 +29,9 @@ open class PlaylistMetadataEntry(
     @ColumnInfo(name = PlaylistEntity.PLAYLIST_THUMBNAIL_STREAM_ID)
     open val thumbnailStreamId: Long?,
 
+    @ColumnInfo(name = PlaylistEntity.PLAYLIST_FOLDER_ID)
+    open val folderId: Long?,
+
     @ColumnInfo(name = PLAYLIST_STREAM_COUNT)
     open val streamCount: Long
 ) : PlaylistLocalItem {

app/src/main/java/org/schabi/newpipe/database/playlist/dao/PlaylistFolderDAO.kt

@@ -0,0 +1,27 @@
+package org.schabi.newpipe.database.playlist.dao
+
+import androidx.room.Dao
+import androidx.room.Query
+import androidx.room.Update
+import androidx.room.Insert
+import io.reactivex.rxjava3.core.Flowable
+import io.reactivex.rxjava3.core.Maybe
+import org.schabi.newpipe.database.playlist.model.PlaylistFolderEntity
+
+@Dao
+interface PlaylistFolderDAO {
+    @Query("SELECT * FROM playlist_folders ORDER BY sort_order ASC")
+    fun getAll(): Flowable<List<PlaylistFolderEntity>>
+
+    @Insert
+    fun insert(folder: PlaylistFolderEntity): Long
+
+    @Update
+    fun update(folder: PlaylistFolderEntity): Int
+
+    @Query("DELETE FROM playlist_folders WHERE uid = :folderId")
+    fun delete(folderId: Long): Int
+
+    @Query("SELECT * FROM playlist_folders WHERE uid = :folderId LIMIT 1")
+    fun get(folderId: Long): Maybe<PlaylistFolderEntity>
+}

app/src/main/java/org/schabi/newpipe/database/playlist/dao/PlaylistStreamDAO.kt

@@ -71,9 +71,9 @@ interface PlaylistStreamDAO : BasicDAO<PlaylistStreamEntity> {
     @Transaction
     @Query(
         """
-        SELECT uid, name, is_thumbnail_permanent, thumbnail_stream_id, display_index,
+        SELECT uid, name,
         (SELECT thumbnail_url FROM streams WHERE streams.uid = thumbnail_stream_id) AS thumbnail_url,
-
+        display_index, is_thumbnail_permanent, thumbnail_stream_id, folder_id,
         COALESCE(COUNT(playlist_id), 0) AS streamCount FROM playlists
 
         LEFT JOIN playlist_stream_join
@@ -106,8 +106,9 @@ interface PlaylistStreamDAO : BasicDAO<PlaylistStreamEntity> {
     @Transaction
     @Query(
         """
-        SELECT playlists.uid, name, is_thumbnail_permanent, thumbnail_stream_id, display_index,
+        SELECT playlists.uid, name,
         (SELECT thumbnail_url FROM streams WHERE streams.uid = thumbnail_stream_id) AS thumbnail_url,
+        display_index, is_thumbnail_permanent, thumbnail_stream_id, folder_id,
 
         COALESCE(COUNT(playlist_id), 0) AS streamCount,
         COALESCE(SUM(url = :streamUrl), 0) AS timesStreamIsContained FROM playlists

app/src/main/java/org/schabi/newpipe/database/playlist/model/PlaylistEntity.kt

@@ -26,9 +26,13 @@ data class PlaylistEntity @JvmOverloads constructor(
 
     @ColumnInfo(name = PLAYLIST_THUMBNAIL_STREAM_ID)
     var thumbnailStreamId: Long,
-
+    
+    @ColumnInfo(name = PLAYLIST_FOLDER_ID)
     @ColumnInfo(name = PLAYLIST_DISPLAY_INDEX)
-    var displayIndex: Long
+    var displayIndex: Long,
+
+    @ColumnInfo(name = PLAYLIST_FOLDER_ID)
+    var folderId: Long? = null,
 ) {
 
     @Ignore
@@ -46,6 +50,7 @@ data class PlaylistEntity @JvmOverloads constructor(
         const val PLAYLIST_TABLE = "playlists"
         const val PLAYLIST_ID = "uid"
         const val PLAYLIST_NAME = "name"
+        const val PLAYLIST_FOLDER_ID = "folder_id"
         const val PLAYLIST_THUMBNAIL_URL = "thumbnail_url"
         const val PLAYLIST_DISPLAY_INDEX = "display_index"
         const val PLAYLIST_THUMBNAIL_PERMANENT = "is_thumbnail_permanent"

app/src/main/java/org/schabi/newpipe/database/playlist/model/PlaylistFolderEntity.kt

@@ -0,0 +1,30 @@
+/*
+ * SPDX-FileCopyrightText: 2025 NewPipe contributors
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+package org.schabi.newpipe.database.playlist.model
+
+import androidx.room.ColumnInfo
+import androidx.room.Entity
+import androidx.room.PrimaryKey
+
+@Entity(tableName = PlaylistFolderEntity.FOLDER_TABLE)
+data class PlaylistFolderEntity(
+    @PrimaryKey(autoGenerate = true)
+    @ColumnInfo(name = FOLDER_ID)
+    var uid: Long = 0,
+
+    @ColumnInfo(name = FOLDER_NAME)
+    var name: String,
+
+    @ColumnInfo(name = FOLDER_SORT_ORDER)
+    var sortOrder: Long = 0
+){
+    companion object {
+        const val FOLDER_TABLE = "playlist_folders"
+        const val FOLDER_ID = "uid"
+        const val FOLDER_NAME = "name"
+        const val FOLDER_SORT_ORDER = "sort_order"
+    }
+}