Skip to content

Commit

Permalink
i needed it
Browse files Browse the repository at this point in the history
  • Loading branch information
@Nansess
Nansess committed May 1, 2024
1 parent 38885d5 commit 3862b56
Showing 1 changed file with 80 additions and 115 deletions.
195 changes: 80 additions & 115 deletions main/src/main/java/com/github/topi314/lavasrc/tidal/TidalSourceManager.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,8 @@
import com.sedmelluq.discord.lavaplayer.tools.io.HttpInterface;
import com.sedmelluq.discord.lavaplayer.tools.io.HttpInterfaceManager;
import com.sedmelluq.discord.lavaplayer.track.*;

import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.HttpClientBuilder;
import org.slf4j.Logger;
Expand All @@ -31,9 +33,7 @@
import java.util.function.Function;
import java.util.regex.Pattern;

public class TidalSourceManager
extends MirroringAudioSourceManager
implements HttpConfigurable {
public class TidalSourceManager extends MirroringAudioSourceManager implements HttpConfigurable {

public static final Pattern URL_PATTERN = Pattern.compile(
"https?://(?:(?:listen|www)\\.)?tidal\\.com/(?:browse/)?(?<type>album|track|playlist)/(?<id>[a-zA-Z0-9\\-]+)(?:\\?.*)?");
Expand All @@ -45,32 +45,19 @@ public class TidalSourceManager
private static final String USER_AGENT = "TIDAL/3704 CFNetwork/1220.1 Darwin/20.3.0";
private static final String TIDAL_TOKEN = "i4ZDjcyhed7Mu47q";

private static final Logger log = LoggerFactory.getLogger(
TidalSourceManager.class);
private static final Logger log = LoggerFactory.getLogger(TidalSourceManager.class);

private final HttpInterfaceManager httpInterfaceManager = HttpClientTools.createDefaultThreadLocalManager();
private int searchLimit = 6;
private final String countryCode;

public TidalSourceManager(
String[] providers,
String countryCode,
Function<Void, AudioPlayerManager> audioPlayerManager) {
this(
countryCode,
audioPlayerManager,
new DefaultMirroringAudioTrackResolver(providers));
public TidalSourceManager(String[] providers, String countryCode, Function<Void, AudioPlayerManager> audioPlayerManager) {
this(countryCode, audioPlayerManager, new DefaultMirroringAudioTrackResolver(providers));
}

public TidalSourceManager(
String countryCode,
Function<Void, AudioPlayerManager> audioPlayerManager,
MirroringAudioTrackResolver mirroringAudioTrackResolver) {
public TidalSourceManager(String countryCode, Function<Void, AudioPlayerManager> audioPlayerManager, MirroringAudioTrackResolver mirroringAudioTrackResolver) {
super(audioPlayerManager, mirroringAudioTrackResolver);
if (countryCode == null || countryCode.isEmpty()) {
countryCode = "US";
}
this.countryCode = countryCode;
this.countryCode = (countryCode == null || countryCode.isEmpty()) ? "US" : countryCode;
}

public void setSearchLimit(int searchLimit) {
Expand All @@ -83,24 +70,13 @@ public String getSourceName() {
}

@Override
public AudioTrack decodeTrack(AudioTrackInfo trackInfo, DataInput input)
throws IOException {
public AudioTrack decodeTrack(AudioTrackInfo trackInfo, DataInput input) throws IOException {
var extendedAudioTrackInfo = super.decodeTrack(input);
return new TidalAudioTrack(
trackInfo,
extendedAudioTrackInfo.albumName,
extendedAudioTrackInfo.albumUrl,
extendedAudioTrackInfo.artistUrl,
extendedAudioTrackInfo.artistArtworkUrl,
extendedAudioTrackInfo.previewUrl,
extendedAudioTrackInfo.isPreview,
this);
return new TidalAudioTrack(trackInfo, extendedAudioTrackInfo.albumName, extendedAudioTrackInfo.albumUrl, extendedAudioTrackInfo.artistUrl, extendedAudioTrackInfo.artistArtworkUrl, extendedAudioTrackInfo.previewUrl, extendedAudioTrackInfo.isPreview, this);
}

@Override
public AudioItem loadItem(
AudioPlayerManager manager,
AudioReference reference) {
public AudioItem loadItem(AudioPlayerManager manager, AudioReference reference) {
try {
var identifier = reference.identifier;
var matcher = URL_PATTERN.matcher(identifier);
Expand Down Expand Up @@ -136,38 +112,67 @@ private JsonBrowser getApiResponse(String apiUrl) throws IOException {
var request = new HttpGet(apiUrl);
request.setHeader("user-agent", USER_AGENT);
request.setHeader("x-tidal-token", TIDAL_TOKEN);
return LavaSrcTools.fetchResponseAsJson(
httpInterfaceManager.getInterface(),
request);
return LavaSrcTools.fetchResponseAsJson(httpInterfaceManager.getInterface(), request);
}

private List<AudioTrack> parseItems(JsonBrowser itemsJson, Function<JsonBrowser, AudioTrack> parser) {
private List<AudioTrack> parseTracks(JsonBrowser json) {
var tracks = new ArrayList<AudioTrack>();
for (var item : itemsJson.values()) {
var parsedItem = parser.apply(item);
if (parsedItem != null) {
tracks.add(parsedItem);
for (var audio : json.values()) {
var parsedTrack = parseTrack(audio);
if (parsedTrack != null) {
tracks.add(parsedTrack);
}
}
return tracks;
}

private List<AudioTrack> parseTracks(JsonBrowser json) {
return parseItems(json.get("items"), this::parseTrack);
private AudioItem getSearchWithRetry(String query, int maxRetries) throws IOException {
for (int retry = 0; retry <= maxRetries; retry++) {
try {
String apiUrl = PUBLIC_API_BASE +
"search?query=" +
URLEncoder.encode(query, StandardCharsets.UTF_8) +
"&offset=0&limit=" +
searchLimit +
"&countryCode=" +
countryCode;
var json = getApiResponse(apiUrl);

if (json.get("tracks").get("items").isNull()) {
return AudioReference.NO_TRACK; // devo bullied me :(
}

var tracks = parseTracks(json.get("tracks").get("items"));

if (tracks.isEmpty()) {
return AudioReference.NO_TRACK;
}

return new BasicAudioPlaylist("Tidal Music Search: " + query, tracks, null, true);
} catch (SocketTimeoutException e) {
if (retry == maxRetries) {
return AudioReference.NO_TRACK;
}
}
}
return AudioReference.NO_TRACK;
}

private List<AudioTrack> parseTrackItem(JsonBrowser json) {
return parseItems(json.get("items"), this::parseItem);
private AudioItem getSearch(String query) throws IOException {
int maxRetries = 2;
return getSearchWithRetry(query, maxRetries);
}

private AudioTrack parseItem(JsonBrowser audio) {
private AudioTrack parseTrack(JsonBrowser audio) {
return newMethod(audio);
}

private AudioTrack newMethod(JsonBrowser audio) {
var id = audio.get("id").text();
var rawDuration = audio.get("duration").text();

if (rawDuration == null) {
log.warn(
"Skipping track with null duration. Audio JSON: {}",
audio);
log.warn("Skipping track with null duration. Audio JSON: {}", audio);
return null;
}

Expand All @@ -193,73 +198,17 @@ private AudioTrack parseItem(JsonBrowser audio) {
formattedCoverIdentifier +
"/1280x1280.jpg";
return new TidalAudioTrack(
new AudioTrackInfo(
title,
artistName.toString(),
duration,
id,
false,
originalUrl,
artworkUrl,
isrc),
new AudioTrackInfo(title, artistName.toString(), duration, id, false, originalUrl, artworkUrl, isrc),
this);
} catch (NumberFormatException e) {
log.error(
"Error parsing duration for track. Audio JSON: {}",
audio,
e);
log.error("Error parsing duration for track. Audio JSON: {}", audio, e);
return null;
}
}

private AudioItem getSearchWithRetry(String query, int maxRetries)
throws IOException {
for (int retry = 0; retry <= maxRetries; retry++) {
try {
String apiUrl = PUBLIC_API_BASE +
"search?query=" +
URLEncoder.encode(query, StandardCharsets.UTF_8) +
"&offset=0&limit=" +
searchLimit +
"&countryCode=" +
countryCode;
var json = getApiResponse(apiUrl);

if (json.get("tracks").get("items").isNull()) {
return AudioReference.NO_TRACK; // devo bullied me :(
}

var tracks = parseTracks(json.get("tracks").get("items"));

if (tracks.isEmpty()) {
return AudioReference.NO_TRACK;
}

return new BasicAudioPlaylist(
"Tidal Music Search: " + query,
tracks,
null,
true);
} catch (SocketTimeoutException e) {
if (retry == maxRetries) {
return AudioReference.NO_TRACK;
}
}
}
return AudioReference.NO_TRACK;
}

private AudioItem getSearch(String query) throws IOException {
int maxRetries = 2;
return getSearchWithRetry(query, maxRetries);
}

private AudioItem getAlbumOrPlaylist(
String itemId,
String type,
int maxPageItems) throws IOException {
private AudioItem getAlbumOrPlaylist(String itemId, String type, int maxPageItems) throws IOException {
try {
// Fetch tracks directly inside of albums
// Fetch tracks directly for albums
String apiUrl = PUBLIC_API_BASE +
type +
"s/" +
Expand All @@ -284,10 +233,10 @@ private AudioItem getAlbumOrPlaylist(
String itemInfoUrl = "";

if (type.equalsIgnoreCase("playlist")) {
// Fetch playlist
// Fetch playlist information
itemInfoUrl = PUBLIC_API_BASE + "playlists/" + itemId + "?countryCode=" + countryCode;
} else if (type.equalsIgnoreCase("album")) {
// Fetch album
// Fetch album information
itemInfoUrl = PUBLIC_API_BASE + "albums/" + itemId + "?countryCode=" + countryCode;
}

Expand Down Expand Up @@ -332,13 +281,29 @@ public AudioItem getTrack(String trackId) throws IOException {
}
}

private List<AudioTrack> parseTrackItem(JsonBrowser json) {
var tracks = new ArrayList<AudioTrack>();
var items = json.get("items");

for (var audio : items.values()) {
var parsedTrack = parseItem(audio);
if (parsedTrack != null) {
tracks.add(parsedTrack);
}
}
return tracks;
}

private AudioTrack parseItem(JsonBrowser audio) {
return newMethod(audio);
}

@Override
public void encodeTrack(AudioTrack track, DataOutput output) {
}

@Override
public void configureRequests(
Function<RequestConfig, RequestConfig> configurator) {
public void configureRequests(Function<RequestConfig, RequestConfig> configurator) {
httpInterfaceManager.configureRequests(configurator);
}

Expand Down

0 comments on commit 3862b56

Please sign in to comment.