songs list and serializable list query

lib/database/dao/library_dao.dart

@@ -2,37 +2,78 @@ import 'package:drift/drift.dart';
 
 import '../../sources/models.dart' as models;
 import '../database.dart';
+import '../query.dart';
 
 part 'library_dao.g.dart';
 
 typedef AristListItem = ({models.Artist artist, int? albumCount});
 
+extension on SortDirection {
+  OrderingMode toMode() => switch (this) {
+    SortDirection.asc => OrderingMode.asc,
+    SortDirection.desc => OrderingMode.desc,
+  };
+}
+
 @DriftAccessor(include: {'../tables.drift'})
 class LibraryDao extends DatabaseAccessor<SubtracksDatabase>
     with _$LibraryDaoMixin {
   LibraryDao(super.db);
 
-  Future<List<models.Album>> listAlbums({
-    required int sourceId,
-    required int limit,
-    required int offset,
-  }) {
+  Future<List<models.Album>> listAlbums(AlbumsQuery q) {
     final query = albums.select()
-      ..where(
-        (f) => f.sourceId.equals(sourceId),
-      )
-      ..limit(limit, offset: offset);
+      ..where((albums) {
+        var filter = albums.sourceId.equals(q.sourceId);
+        for (final queryFilter in q.filter) {
+          filter &= switch (queryFilter) {
+            AlbumsFilterArtistId(:final artistId) => albums.artistId.equals(
+              artistId,
+            ),
+            AlbumsFilterYearEquals(:final year) => albums.year.equals(year),
+            _ => CustomExpression(''),
+          };
+        }
+
+        return filter;
+      })
+      ..orderBy(
+        q.sort
+            .map(
+              (sort) =>
+                  (albums) => OrderingTerm(
+                    expression: switch (sort.by) {
+                      AlbumsColumn.name => albums.name,
+                      AlbumsColumn.created => albums.created,
+                      AlbumsColumn.year => albums.year,
+                      AlbumsColumn.starred => albums.starred,
+                    },
+                    mode: sort.dir.toMode(),
+                  ),
+            )
+            .toList(),
+      );
+
+    _limitQuery(query: query, limit: q.limit, offset: q.offset);
 
     return query.get();
   }
 
-  Future<List<AristListItem>> listArtists({
-    required int sourceId,
-    required int limit,
-    required int offset,
-  }) async {
+  Future<List<AristListItem>> listArtists(ArtistsQuery q) {
     final albumCount = albums.id.count();
 
+    var filter =
+        artists.sourceId.equals(q.sourceId) &
+        albums.sourceId.equals(q.sourceId);
+
+    for (final queryFilter in q.filter) {
+      filter &= switch (queryFilter) {
+        ArtistsFilterStarred(:final starred) =>
+          starred ? artists.starred.isNotNull() : artists.starred.isNull(),
+        ArtistsFilterNameSearch() => CustomExpression(''),
+        _ => CustomExpression(''),
+      };
+    }
+
     final query =
         artists.select().join([
             leftOuterJoin(
@@ -41,22 +82,78 @@ class LibraryDao extends DatabaseAccessor<SubtracksDatabase>
             ),
           ])
           ..addColumns([albumCount])
-          ..where(
-            artists.sourceId.equals(sourceId) &
-                albums.sourceId.equals(sourceId),
-          )
+          ..where(filter)
           ..groupBy([artists.sourceId, artists.id])
-          ..orderBy([OrderingTerm.asc(artists.name)])
-          ..limit(limit, offset: offset);
+          ..orderBy(
+            q.sort
+                .map(
+                  (sort) => OrderingTerm(
+                    expression: switch (sort.by) {
+                      ArtistsColumn.name => artists.name,
+                      ArtistsColumn.starred => artists.starred,
+                      ArtistsColumn.albumCount => albumCount,
+                    },
+                    mode: sort.dir.toMode(),
+                  ),
+                )
+                .toList(),
+          );
 
-    return (await query.get())
+    _limitQuery(query: query, limit: q.limit, offset: q.offset);
+
+    return query
         .map(
           (row) => (
             artist: row.readTable(artists),
             albumCount: row.read(albumCount) ?? 0,
           ),
         )
-        .toList();
+        .get();
+  }
+
+  Future<List<models.Song>> listSongs(SongsQuery q) {
+    var joinPlaylistSongs = false;
+    var filter = songs.sourceId.equals(q.sourceId);
+
+    for (final queryFilter in q.filter) {
+      switch (queryFilter) {
+        case SongsFilterAlbumId(:final albumId):
+          filter &= songs.albumId.equals(albumId);
+        case SongsFilterPlaylistId(:final playlistId):
+          joinPlaylistSongs = true;
+          filter &= playlistSongs.playlistId.equals(playlistId);
+      }
+    }
+
+    final query =
+        songs.select().join([
+            if (joinPlaylistSongs)
+              leftOuterJoin(
+                playlistSongs,
+                playlistSongs.sourceId.equals(q.sourceId) &
+                    playlistSongs.songId.equalsExp(songs.id),
+              ),
+          ])
+          ..where(filter)
+          ..orderBy(
+            q.sort
+                .map(
+                  (sort) => OrderingTerm(
+                    expression: switch (sort.by) {
+                      SongsColumn.title => songs.title,
+                      SongsColumn.starred => songs.starred,
+                      SongsColumn.disc => songs.disc,
+                      SongsColumn.track => songs.track,
+                    },
+                    mode: sort.dir.toMode(),
+                  ),
+                )
+                .toList(),
+          );
+
+    _limitQuery(query: query, limit: q.limit, offset: q.offset);
+
+    return query.map((row) => (row.readTable(songs))).get();
   }
 
   Selectable<models.Album> getAlbum(int sourceId, String id) {
@@ -64,4 +161,14 @@ class LibraryDao extends DatabaseAccessor<SubtracksDatabase>
       (f) => f.sourceId.equals(sourceId) & f.id.equals(id),
     );
   }
+
+  void _limitQuery({
+    required LimitContainerMixin query,
+    required int? limit,
+    required int? offset,
+  }) {
+    if (limit != null) {
+      query.limit(limit, offset: offset);
+    }
+  }
 }