From 167f04a4dfab2b1d7834c7348185ce2b2ccb0dc8 Mon Sep 17 00:00:00 2001
From: Jay-716 <13422525511@163.com>
Date: Thu, 25 Jan 2024 22:51:07 +0800
Subject: [PATCH] pulseaudio: reconnect context when pulseaudio server restarts

When pulseaudio server restarts, the context is not reconnect
automatically. So the pulseaudio module will stop updating.
---
 src/util/audio_backend.cpp | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/util/audio_backend.cpp b/src/util/audio_backend.cpp
index 7eef1448..b7319d58 100644
--- a/src/util/audio_backend.cpp
+++ b/src/util/audio_backend.cpp
@@ -80,7 +80,22 @@ void AudioBackend::contextStateCb(pa_context *c, void *data) {
                            nullptr, nullptr);
       break;
     case PA_CONTEXT_FAILED:
-      backend->mainloop_api_->quit(backend->mainloop_api_, 1);
+      // When pulseaudio server restarts, the connection is "failed". Try to reconnect.
+      // pa_threaded_mainloop_lock is already acquired in callback threads.
+      // So there is no need to lock it again.
+      if (backend->context_ != nullptr) {
+        pa_context_disconnect(backend->context_);
+      }
+      backend->context_ = pa_context_new(backend->mainloop_api_, "waybar");
+      if (backend->context_ == nullptr) {
+        throw std::runtime_error("pa_context_new() failed.");
+      }
+      pa_context_set_state_callback(backend->context_, contextStateCb, data);
+      if (pa_context_connect(backend->context_, nullptr, PA_CONTEXT_NOFAIL, nullptr) < 0) {
+        auto err =
+            fmt::format("pa_context_connect() failed: {}", pa_strerror(pa_context_errno(backend->context_)));
+        throw std::runtime_error(err);
+      }
       break;
     case PA_CONTEXT_CONNECTING:
     case PA_CONTEXT_AUTHORIZING: