From 8312d08653fe53767b76e26367074bc8e14f5eea Mon Sep 17 00:00:00 2001
From: chenhang <averych@126.com>
Date: Fri, 10 Jul 2026 11:15:59 +0800
Subject: [PATCH] fix(chat): apply unlocked voice audio url

---
 src/app/chat/components/message-content.tsx   |  5 ++--
 src/app/chat/components/voice-bubble.tsx      |  9 +++----
 .../__tests__/unlock_private_response.test.ts |  4 ++++
 .../chat/response/unlock_private_response.ts  |  1 +
 .../repositories/chat_local_message_store.ts  | 24 ++++++++++++++++---
 src/data/repositories/chat_repository.ts      |  6 ++---
 .../interfaces/ichat_repository.ts            |  8 ++++++-
 .../schemas/chat/unlock_private_response.ts   |  1 +
 .../__tests__/chat-machine.helpers.test.ts    | 11 ++++-----
 .../chat-machine.transitions.test.ts          | 16 +++++++++----
 src/stores/chat/chat-unlock-flow.ts           |  6 ++++-
 src/stores/chat/chat-unlock-helpers.ts        | 10 ++++++++
 12 files changed, 76 insertions(+), 25 deletions(-)

diff --git a/src/app/chat/components/message-content.tsx b/src/app/chat/components/message-content.tsx
index ea0e007b..54deabef 100644
--- a/src/app/chat/components/message-content.tsx
+++ b/src/app/chat/components/message-content.tsx
@@ -47,6 +47,7 @@ export function MessageContent({
     lockedPrivate === true ||
     (locked === true && lockReason === "private_message");
   const isLockedVoiceMessage = locked === true && lockReason === "voice_message";
+  const shouldRenderVoiceMessage = hasAudio || isLockedVoiceMessage;
   const handleUnlockPrivateMessage =
     messageId && onUnlockPrivateMessage
       ? () => onUnlockPrivateMessage(messageId)
@@ -79,7 +80,7 @@ export function MessageContent({
               onOpenImage={onOpenImage}
             />
           )}
-          {hasAudio && audioUrl ? (
+          {shouldRenderVoiceMessage ? (
             <VoiceBubble
               messageId={messageId}
               audioUrl={audioUrl}
@@ -90,7 +91,7 @@ export function MessageContent({
               onUnlock={handleUnlockVoiceMessage}
             />
           ) : null}
-          {hasText && !hasAudio ? (
+          {hasText && !shouldRenderVoiceMessage ? (
             <TextBubble content={content} isFromAI={isFromAI} />
           ) : null}
         </>
diff --git a/src/app/chat/components/voice-bubble.tsx b/src/app/chat/components/voice-bubble.tsx
index fb0792a7..b126e9cc 100644
--- a/src/app/chat/components/voice-bubble.tsx
+++ b/src/app/chat/components/voice-bubble.tsx
@@ -8,7 +8,7 @@ import styles from "./voice-bubble.module.css";
 
 export interface VoiceBubbleProps {
   messageId?: string;
-  audioUrl: string;
+  audioUrl?: string | null;
   isFromAI: boolean;
   locked?: boolean;
   hint?: string | null;
@@ -28,6 +28,7 @@ export function VoiceBubble({
   const audioRef = useRef<HTMLAudioElement>(null);
   const [isPlaying, setIsPlaying] = useState(false);
   const [errorMediaUrl, setErrorMediaUrl] = useState<string | null>(null);
+  const hasAudio = audioUrl != null && audioUrl.length > 0;
   const { mediaUrl, isUsingCachedMedia, reportMediaError } =
     useCachedChatMediaUrl({
       messageId,
@@ -66,7 +67,7 @@ export function VoiceBubble({
   const hasError = errorMediaUrl === mediaUrl;
 
   const handleToggle = () => {
-    if (locked) return;
+    if (locked || !hasAudio) return;
 
     const audio = audioRef.current;
     if (!audio || hasError) return;
@@ -92,7 +93,7 @@ export function VoiceBubble({
         type="button"
         className={styles.playButton}
         onClick={handleToggle}
-        disabled={locked || hasError}
+        disabled={locked || hasError || !hasAudio}
         aria-label={isPlaying ? "Pause voice message" : "Play voice message"}
       >
         {locked ? (
@@ -132,7 +133,7 @@ export function VoiceBubble({
           </>
         ) : null}
       </div>
-      {!locked ? (
+      {!locked && hasAudio ? (
         <audio ref={audioRef} src={mediaUrl} preload="metadata" />
       ) : null}
     </div>
diff --git a/src/data/dto/chat/__tests__/unlock_private_response.test.ts b/src/data/dto/chat/__tests__/unlock_private_response.test.ts
index 0c8e9178..b4719909 100644
--- a/src/data/dto/chat/__tests__/unlock_private_response.test.ts
+++ b/src/data/dto/chat/__tests__/unlock_private_response.test.ts
@@ -7,6 +7,7 @@ describe("UnlockPrivateResponse", () => {
     const response = UnlockPrivateResponse.from({
       unlocked: true,
       content: "完整消息内容",
+      audioUrl: "https://example.com/unlocked-voice.mp3",
       reason: "ok",
       creditBalance: 9180,
       creditsCharged: 10,
@@ -24,6 +25,7 @@ describe("UnlockPrivateResponse", () => {
 
     expect(response.unlocked).toBe(true);
     expect(response.content).toBe("完整消息内容");
+    expect(response.audioUrl).toBe("https://example.com/unlocked-voice.mp3");
     expect(response.reason).toBe("ok");
     expect(response.creditBalance).toBe(9180);
     expect(response.creditsCharged).toBe(10);
@@ -66,6 +68,7 @@ describe("UnlockPrivateResponse", () => {
     const response = UnlockPrivateResponse.from({
       unlocked: false,
       content: null,
+      audioUrl: null,
       reason: "insufficient_balance",
       lockDetail: {
         locked: true,
@@ -78,6 +81,7 @@ describe("UnlockPrivateResponse", () => {
     });
 
     expect(response.content).toBe("");
+    expect(response.audioUrl).toBe("");
   });
 
   it("defaults nullable credit fields to zero", () => {
diff --git a/src/data/dto/chat/response/unlock_private_response.ts b/src/data/dto/chat/response/unlock_private_response.ts
index 085cc9d2..001da173 100644
--- a/src/data/dto/chat/response/unlock_private_response.ts
+++ b/src/data/dto/chat/response/unlock_private_response.ts
@@ -12,6 +12,7 @@ import type { ChatLockDetailData } from "@/data/schemas/chat";
 export class UnlockPrivateResponse {
   declare readonly unlocked: boolean;
   declare readonly content: string;
+  declare readonly audioUrl: string;
   declare readonly reason: UnlockPrivateReason;
   declare readonly creditBalance: number;
   declare readonly creditsCharged: number;
diff --git a/src/data/repositories/chat_local_message_store.ts b/src/data/repositories/chat_local_message_store.ts
index 15cd7bb0..66fa9404 100644
--- a/src/data/repositories/chat_local_message_store.ts
+++ b/src/data/repositories/chat_local_message_store.ts
@@ -1,7 +1,7 @@
 "use client";
 
-import type { ChatLockDetailData } from "@/data/schemas/chat";
 import { ChatMessage } from "@/data/dto/chat";
+import type { UnlockedPrivateMessageLocalPatch } from "@/data/repositories/interfaces";
 import { LocalChatStorage, LocalMessage } from "@/data/storage/chat";
 import { Result } from "@/utils";
 
@@ -10,7 +10,7 @@ export class ChatLocalMessageStore {
 
   async markMessageUnlocked(
     messageId: string,
-    lockDetail?: ChatLockDetailData,
+    patch: UnlockedPrivateMessageLocalPatch = {},
   ): Promise<Result<void>> {
     return Result.wrap(async () => {
       const localResult = await this.getMessages();
@@ -24,7 +24,9 @@ export class ChatLocalMessageStore {
         changed = true;
         return ChatMessage.from({
           ...message.toJson(),
-          lockDetail: lockDetail ?? {
+          content: normalizeUnlockedContent(patch.content, message.content),
+          audioUrl: normalizeUnlockedAudioUrl(patch.audioUrl, message.audioUrl),
+          lockDetail: patch.lockDetail ?? {
             ...message.lockDetail,
             locked: false,
             showContent: true,
@@ -105,6 +107,22 @@ export class ChatLocalMessageStore {
   }
 }
 
+function normalizeUnlockedContent(
+  nextContent: string | undefined,
+  currentContent: string,
+): string {
+  if (nextContent == null || nextContent.length === 0) return currentContent;
+  return nextContent;
+}
+
+function normalizeUnlockedAudioUrl(
+  nextAudioUrl: string | null | undefined,
+  currentAudioUrl: string | null,
+): string | null {
+  if (nextAudioUrl == null || nextAudioUrl.length === 0) return currentAudioUrl;
+  return nextAudioUrl;
+}
+
 function shouldMarkMessageUnlocked(
   message: ChatMessage,
   messageId: string,
diff --git a/src/data/repositories/chat_repository.ts b/src/data/repositories/chat_repository.ts
index 81a583d7..7e933302 100644
--- a/src/data/repositories/chat_repository.ts
+++ b/src/data/repositories/chat_repository.ts
@@ -7,7 +7,6 @@ import type {
   UnlockHistoryResponse,
   UnlockPrivateResponse,
 } from "@/data/dto/chat";
-import type { ChatLockDetailData } from "@/data/schemas/chat";
 import { chatApi } from "@/data/services/api";
 import {
   LocalChatMediaStorage,
@@ -18,6 +17,7 @@ import type {
   CacheRemoteChatMediaInput,
   ChatMediaLookupInput,
   IChatRepository,
+  UnlockedPrivateMessageLocalPatch,
 } from "@/data/repositories/interfaces";
 import type { Result } from "@/utils";
 
@@ -66,11 +66,11 @@ export class ChatRepository implements IChatRepository {
   /** 把本地缓存中的单条锁定消息标记为已解锁。 */
   async markPrivateMessageUnlockedInLocal(
     messageId: string,
-    lockDetail?: ChatLockDetailData,
+    patch?: UnlockedPrivateMessageLocalPatch,
   ): Promise<Result<void>> {
     return this.localMessages.markMessageUnlocked(
       messageId,
-      lockDetail,
+      patch,
     );
   }
 
diff --git a/src/data/repositories/interfaces/ichat_repository.ts b/src/data/repositories/interfaces/ichat_repository.ts
index 70918458..e7d4af80 100644
--- a/src/data/repositories/interfaces/ichat_repository.ts
+++ b/src/data/repositories/interfaces/ichat_repository.ts
@@ -32,6 +32,12 @@ export interface CacheRemoteChatMediaInput extends ChatMediaLookupInput {
   remoteUrl: string;
 }
 
+export interface UnlockedPrivateMessageLocalPatch {
+  lockDetail?: ChatLockDetailData;
+  audioUrl?: string | null;
+  content?: string;
+}
+
 export interface IChatRepository {
   /** 发送一条消息。 */
   sendMessage(
@@ -51,7 +57,7 @@ export interface IChatRepository {
   /** 把本地缓存中的单条锁定消息标记为已解锁。 */
   markPrivateMessageUnlockedInLocal(
     messageId: string,
-    lockDetail?: ChatLockDetailData,
+    patch?: UnlockedPrivateMessageLocalPatch,
   ): Promise<Result<void>>;
 
   /** 把一条消息写入本地存储。 */
diff --git a/src/data/schemas/chat/unlock_private_response.ts b/src/data/schemas/chat/unlock_private_response.ts
index 5aa954b6..da71b214 100644
--- a/src/data/schemas/chat/unlock_private_response.ts
+++ b/src/data/schemas/chat/unlock_private_response.ts
@@ -11,6 +11,7 @@ export const UnlockPrivateReasonSchema = stringOrEmpty;
 export const UnlockPrivateResponseSchema = z.object({
   unlocked: booleanOrFalse,
   content: stringOrEmpty,
+  audioUrl: stringOrEmpty,
   reason: UnlockPrivateReasonSchema,
   creditBalance: numberOrZero,
   creditsCharged: numberOrZero,
diff --git a/src/stores/chat/__tests__/chat-machine.helpers.test.ts b/src/stores/chat/__tests__/chat-machine.helpers.test.ts
index 164715a1..9837c759 100644
--- a/src/stores/chat/__tests__/chat-machine.helpers.test.ts
+++ b/src/stores/chat/__tests__/chat-machine.helpers.test.ts
@@ -66,8 +66,7 @@ describe("sendResponseToUiMessage", () => {
   it("maps locked voice messages without treating them as private text", () => {
     const message = sendResponseToUiMessage(
       makeResponse({
-        audioUrl:
-          "https://proapi.banlv-ai.com/audio/5198b93c-2915-406a.mp3",
+        audioUrl: "",
         lockDetail: {
           locked: true,
           showContent: false,
@@ -80,9 +79,7 @@ describe("sendResponseToUiMessage", () => {
     );
 
     expect(message.content).toBe("AI reply");
-    expect(message.audioUrl).toBe(
-      "https://proapi.banlv-ai.com/audio/5198b93c-2915-406a.mp3",
-    );
+    expect(message.audioUrl).toBeUndefined();
     expect(message.locked).toBe(true);
     expect(message.lockReason).toBe("voice_message");
     expect(message.lockedPrivate).toBeUndefined();
@@ -243,7 +240,7 @@ describe("localMessagesToUi", () => {
         type: "voice",
         content: "hidden voice transcript",
         createdAt: "2026-06-25T12:00:00.000Z",
-        audioUrl: "https://example.com/voice.mp3",
+        audioUrl: null,
         image: { type: null, url: null },
         lockDetail: {
           locked: true,
@@ -257,7 +254,7 @@ describe("localMessagesToUi", () => {
     ]);
 
     expect(message.content).toBe("hidden voice transcript");
-    expect(message.audioUrl).toBe("https://example.com/voice.mp3");
+    expect(message.audioUrl).toBeUndefined();
     expect(message.locked).toBe(true);
     expect(message.lockReason).toBe("voice_message");
     expect(message.lockedPrivate).toBeUndefined();
diff --git a/src/stores/chat/__tests__/chat-machine.transitions.test.ts b/src/stores/chat/__tests__/chat-machine.transitions.test.ts
index bfb88462..55337aeb 100644
--- a/src/stores/chat/__tests__/chat-machine.transitions.test.ts
+++ b/src/stores/chat/__tests__/chat-machine.transitions.test.ts
@@ -59,6 +59,7 @@ function makeUnlockPrivateResponse(
   return UnlockPrivateResponse.from({
     unlocked: true,
     content: "unlocked content",
+    audioUrl: "",
     reason: "ok",
     creditBalance: 90,
     creditsCharged: 10,
@@ -740,7 +741,7 @@ describe("chatMachine transitions", () => {
     actor.stop();
   });
 
-  it("keeps voice message content unchanged after unlock succeeds", async () => {
+  it("applies the real voice audio url after unlock succeeds", async () => {
     const actor = createActor(
       createTestChatMachine({
         historyMessages: [
@@ -749,7 +750,6 @@ describe("chatMachine transitions", () => {
             content: "Original voice transcript.",
             isFromAI: true,
             date: "2026-06-29",
-            audioUrl: "https://example.com/voice.mp3",
             locked: true,
             lockReason: "voice_message",
             privateMessageHint: "A voice message is waiting.",
@@ -759,6 +759,7 @@ describe("chatMachine transitions", () => {
           messageId: "msg-voice-locked",
           response: makeUnlockPrivateResponse({
             content: "This response content must be ignored.",
+            audioUrl: "https://example.com/unlocked-voice.mp3",
           }),
         },
       }),
@@ -783,7 +784,7 @@ describe("chatMachine transitions", () => {
       {
         id: "msg-voice-locked",
         content: "Original voice transcript.",
-        audioUrl: "https://example.com/voice.mp3",
+        audioUrl: "https://example.com/unlocked-voice.mp3",
         locked: false,
         lockReason: null,
         privateMessageHint: null,
@@ -802,7 +803,6 @@ describe("chatMachine transitions", () => {
             content: "",
             isFromAI: true,
             date: "2026-06-29",
-            audioUrl: "https://example.com/voice.mp3",
             locked: true,
             lockReason: "voice_message",
             privateMessageHint: "A voice message is waiting.",
@@ -855,6 +855,14 @@ describe("chatMachine transitions", () => {
       requiredCredits: 10,
       shortfallCredits: 7,
     });
+    expect(actor.getSnapshot().context.messages).toMatchObject([
+      {
+        id: "msg-voice-locked",
+        locked: true,
+        lockReason: "voice_message",
+      },
+    ]);
+    expect(actor.getSnapshot().context.messages[0]?.audioUrl).toBeUndefined();
     expect(actor.getSnapshot().context.isUnlockingMessage).toBe(false);
 
     actor.stop();
diff --git a/src/stores/chat/chat-unlock-flow.ts b/src/stores/chat/chat-unlock-flow.ts
index c50a55be..2a9bba0c 100644
--- a/src/stores/chat/chat-unlock-flow.ts
+++ b/src/stores/chat/chat-unlock-flow.ts
@@ -64,7 +64,11 @@ export const unlockMessageActor = fromPromise<
   if (unlockResult.data.unlocked) {
     const markResult = await chatRepo.markPrivateMessageUnlockedInLocal(
       input.messageId,
-      unlockResult.data.lockDetail,
+      {
+        content: unlockResult.data.content,
+        audioUrl: unlockResult.data.audioUrl,
+        lockDetail: unlockResult.data.lockDetail,
+      },
     );
     if (Result.isErr(markResult)) {
       log.warn("[chat-machine] mark unlocked local message failed", {
diff --git a/src/stores/chat/chat-unlock-helpers.ts b/src/stores/chat/chat-unlock-helpers.ts
index 6b8208bf..e1aa5db2 100644
--- a/src/stores/chat/chat-unlock-helpers.ts
+++ b/src/stores/chat/chat-unlock-helpers.ts
@@ -20,6 +20,7 @@ export function applySingleUnlockOutput(
 
     return {
       ...message,
+      audioUrl: getUnlockedAudioUrl(message, output.response),
       locked: output.response.lockDetail.locked,
       lockReason: output.response.lockDetail.reason,
       imagePaywalled: message.imageUrl
@@ -32,6 +33,15 @@ export function applySingleUnlockOutput(
   });
 }
 
+function getUnlockedAudioUrl(
+  message: UiMessage,
+  response: UnlockPrivateResponse,
+): string | undefined {
+  if (message.lockReason !== "voice_message") return message.audioUrl;
+  if (response.audioUrl.length === 0) return message.audioUrl;
+  return response.audioUrl;
+}
+
 function shouldApplySingleUnlock(message: UiMessage, messageId: string): boolean {
   if (message.id !== messageId) return false;
   if (!message.isFromAI) return false;