From 7640609b410a1babc2973bc33ed8894729b3b6a1 Mon Sep 17 00:00:00 2001
From: Max Kellermann <max@musicpd.org>
Date: Wed, 24 Jan 2018 15:18:19 +0100
Subject: [PATCH] input/tidal: retry if session is expired

---
 src/input/plugins/TidalInputPlugin.cxx | 33 ++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/input/plugins/TidalInputPlugin.cxx b/src/input/plugins/TidalInputPlugin.cxx
index 9de31fda7..547b501c6 100644
--- a/src/input/plugins/TidalInputPlugin.cxx
+++ b/src/input/plugins/TidalInputPlugin.cxx
@@ -21,6 +21,7 @@
 #include "TidalInputPlugin.hxx"
 #include "TidalSessionManager.hxx"
 #include "TidalTrackRequest.hxx"
+#include "TidalError.hxx"
 #include "CurlInputPlugin.hxx"
 #include "PluginUnavailable.hxx"
 #include "input/ProxyInputStream.hxx"
@@ -29,6 +30,7 @@
 #include "config/Block.hxx"
 #include "thread/Mutex.hxx"
 #include "util/Domain.hxx"
+#include "util/Exception.hxx"
 #include "util/StringCompare.hxx"
 #include "Log.hxx"
 
@@ -48,6 +50,12 @@ class TidalInputStream final
 
 	std::exception_ptr error;
 
+	/**
+	 * Retry to login if TidalError::IsInvalidSession() returns
+	 * true?
+	 */
+	bool retry_login = true;
+
 public:
 	TidalInputStream(const char *_uri, const char *_track_id,
 			 Mutex &_mutex, Cond &_cond) noexcept
@@ -119,11 +127,36 @@ TidalInputStream::OnTidalTrackSuccess(std::string url) noexcept
 	}
 }
 
+gcc_pure
+static bool
+IsInvalidSession(std::exception_ptr e) noexcept
+{
+	try {
+		std::rethrow_exception(e);
+	} catch (const TidalError &te) {
+		return te.IsInvalidSession();
+	} catch (...) {
+		return false;
+	}
+}
+
 void
 TidalInputStream::OnTidalTrackError(std::exception_ptr e) noexcept
 {
 	const std::lock_guard<Mutex> protect(mutex);
 
+	if (retry_login && IsInvalidSession(e)) {
+		/* the session has expired - obtain a new session id
+		   by logging in again */
+
+		FormatInfo(tidal_domain, "Session expired ('%s'), retrying to log in",
+			   GetFullMessage(e).c_str());
+
+		retry_login = false;
+		tidal_session->AddLoginHandler(*this);
+		return;
+	}
+
 	Failed(e);
 }