kitty/spotube
1
0
Fork 0
mirror of https://github.com/KRTirtho/spotube.git synced 2025-09-13 07:55:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: better track matching on youtube API

Browse Source
This commit is contained in:
Kingkor Roy Tirtho 2023-08-18 14:06:03 +06:00
parent 02fa54ae89
commit 904a0d3e15
1 changed files with 58 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
lib/models/spotube_track.dart
View File

@ -4,10 +4,16 @@ import 'package:spotify/spotify.dart';
import 'package:spotube/extensions/album_simple.dart';
import 'package:spotube/extensions/artist_simple.dart';
import 'package:spotube/models/matched_track.dart';
import 'package:spotube/provider/user_preferences_provider.dart';
import 'package:spotube/services/youtube/youtube.dart';
import 'package:spotube/utils/service_utils.dart';
import 'package:collection/collection.dart';
final officialMusicRegex = RegExp(
r"official\s(video|audio|music\svideo)",
caseSensitive: false,
);
class SpotubeTrack extends Track {
final YoutubeVideoInfo ytTrack;
final String ytUri;
@ -65,19 +71,59 @@ class SpotubeTrack extends Track {
final List<YoutubeVideoInfo> siblings =
await client.search("$title - ${artists.join(", ")}").then(
(res) {
final siblings = res
final isYoutubeApi =
client.preferences.youtubeApiType == YoutubeApiType.youtube;
final siblings = isYoutubeApi ||
client.preferences.searchMode == SearchMode.youtube
? res
.sorted((a, b) => b.views.compareTo(a.views))
.where((item) {
.map((sibling) {
int score = 0;
for (final artist in artists) {
final isSameChannelArtist =
sibling.channelName.toLowerCase() ==
artist.toLowerCase();
final channelContainsArtist = sibling.channelName
.toLowerCase()
.contains(artist.toLowerCase());
if (isSameChannelArtist || channelContainsArtist) {
score += 1;
}
final titleContainsArtist = sibling.title
.toLowerCase()
.contains(artist.toLowerCase());
if (titleContainsArtist) {
score += 1;
}
}
if (sibling.title
.toLowerCase()
.contains(track.name!.toLowerCase())) {
score += 2;
}
if (officialMusicRegex
.hasMatch(sibling.title.toLowerCase())) {
score += 1;
}
return (sibling: sibling, score: score);
})
.sorted((a, b) => b.score.compareTo(a.score))
.map((e) => e.sibling)
: res.sorted((a, b) => b.views.compareTo(a.views)).where((item) {
return artists.any(
(artist) =>
client.preferences.searchMode == SearchMode.youtube ||
artist.toLowerCase() == item.channelName.toLowerCase(),
);
})
.take(10)
.toList();
});
return siblings;
return siblings.take(10).toList();
},
);