From 9d283c12f6564e3311c80c5a1e100905e19152c4 Mon Sep 17 00:00:00 2001
From: Hiroshi Horie <548776+hiroshihorie@users.noreply.github.com>
Date: Tue, 11 Feb 2025 02:39:26 +0900
Subject: [PATCH] First impl

---
 .../Visualizer/AgentBarAudioVisualizer.swift  | 75 +++++++++++++++++++
 .../UI/Visualizer/Visualizer.swift            | 11 +--
 2 files changed, 79 insertions(+), 7 deletions(-)
 create mode 100644 Sources/LiveKitComponents/UI/Visualizer/AgentBarAudioVisualizer.swift

diff --git a/Sources/LiveKitComponents/UI/Visualizer/AgentBarAudioVisualizer.swift b/Sources/LiveKitComponents/UI/Visualizer/AgentBarAudioVisualizer.swift
new file mode 100644
index 0000000..86b776b
--- /dev/null
+++ b/Sources/LiveKitComponents/UI/Visualizer/AgentBarAudioVisualizer.swift
@@ -0,0 +1,75 @@
+/*
+ * Copyright 2025 LiveKit
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import LiveKit
+import SwiftUI
+
+// TODO: Move this to SDK
+extension Room {
+    var firstAgentParticipant: RemoteParticipant? {
+        remoteParticipants.values.first { $0.kind == .agent }
+    }
+}
+
+public struct AgentBarAudioVisualizer: View {
+    public let barColor: Color
+    public let barCount: Int
+    public let barCornerRadius: CGFloat
+    public let barSpacingFactor: CGFloat
+    public let barMinOpacity: Double
+    public let isCentered: Bool
+
+    @EnvironmentObject private var room: Room
+
+    public init(barColor: Color = .primary,
+                barCount: Int = 5,
+                barCornerRadius: CGFloat = 100,
+                barSpacingFactor: CGFloat = 0.015,
+                barMinOpacity: CGFloat = 0.35,
+                isCentered: Bool = true)
+    {
+        self.barColor = barColor
+        self.barCount = barCount
+        self.barCornerRadius = barCornerRadius
+        self.barSpacingFactor = barSpacingFactor
+        self.barMinOpacity = barMinOpacity
+        self.isCentered = isCentered
+    }
+
+    public var body: some View {
+        let firstAgentParticipant = room.firstAgentParticipant
+        let agentState = firstAgentParticipant?.agentState ?? .unknown
+        let firstAudioPublication = firstAgentParticipant?.firstAudioPublication as? RemoteTrackPublication
+        let track = firstAudioPublication?.track
+
+        let animation = agentState == .speaking
+            ? nil
+            : Animation.easeInOut(duration: 1)
+            .repeatForever(autoreverses: true)
+            .speed(1)
+
+        BarAudioVisualizer(audioTrack: track as? AudioTrack,
+                           barColor: barColor,
+                           barCount: barCount,
+                           barCornerRadius: barCornerRadius,
+                           barSpacingFactor: barSpacingFactor,
+                           barMinOpacity: barMinOpacity,
+                           isCentered: isCentered)
+            .id(track)
+            .opacity(agentState == .speaking ? 1 : 0.3)
+            .animation(animation, value: agentState)
+    }
+}
diff --git a/Sources/LiveKitComponents/UI/Visualizer/Visualizer.swift b/Sources/LiveKitComponents/UI/Visualizer/Visualizer.swift
index cdad109..9d0e55c 100644
--- a/Sources/LiveKitComponents/UI/Visualizer/Visualizer.swift
+++ b/Sources/LiveKitComponents/UI/Visualizer/Visualizer.swift
@@ -135,26 +135,23 @@ public class AudioProcessor: ObservableObject, AudioRenderer {
 /// BarAudioVisualizer(audioTrack: audioTrack, barColor: .blue, barCount: 10)
 /// ```
 public struct BarAudioVisualizer: View {
-    public let barCount: Int
     public let barColor: Color
+    public let barCount: Int
     public let barCornerRadius: CGFloat
     public let barSpacingFactor: CGFloat
     public let barMinOpacity: Double
     public let isCentered: Bool
 
-    public let audioTrack: AudioTrack
-
     @StateObject private var audioProcessor: AudioProcessor
 
-    public init(audioTrack: AudioTrack,
-                barColor: Color = .white,
-                barCount: Int = 7,
+    public init(audioTrack: AudioTrack?,
+                barColor: Color = .primary,
+                barCount: Int = 5,
                 barCornerRadius: CGFloat = 100,
                 barSpacingFactor: CGFloat = 0.015,
                 barMinOpacity: CGFloat = 0.35,
                 isCentered: Bool = true)
     {
-        self.audioTrack = audioTrack
         self.barColor = barColor
         self.barCount = barCount
         self.barCornerRadius = barCornerRadius