Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| ac729da784 |
@@ -1,14 +0,0 @@
|
||||
const LEGACY_CHARACTER_ID_ALIASES: Readonly<Record<string, string>> =
|
||||
Object.freeze({
|
||||
character_elio: "elio",
|
||||
character_maya: "maya-tan",
|
||||
character_nayeli: "nayeli-cervantes",
|
||||
});
|
||||
|
||||
/**
|
||||
* Converts character IDs persisted by pre-catalog releases to the current
|
||||
* backend IDs. Unknown and already-current IDs pass through unchanged.
|
||||
*/
|
||||
export function normalizeLegacyCharacterId(characterId: string): string {
|
||||
return LEGACY_CHARACTER_ID_ALIASES[characterId] ?? characterId;
|
||||
}
|
||||
@@ -1,241 +0,0 @@
|
||||
import "fake-indexeddb/auto";
|
||||
|
||||
import Dexie from "dexie";
|
||||
import { afterEach, describe, expect, it } from "vitest";
|
||||
|
||||
import {
|
||||
buildChatConversationKey,
|
||||
buildChatMediaCacheKey,
|
||||
} from "@/lib/chat/chat_cache_keys";
|
||||
|
||||
import {
|
||||
LocalChatDB,
|
||||
type LocalChatMediaRow,
|
||||
type LocalMessageRow,
|
||||
} from "../local_chat_db";
|
||||
|
||||
const DATABASE_SCHEMA = {
|
||||
messages: "++dbId, sessionId",
|
||||
media:
|
||||
"cacheKey, ownerKey, messageId, kind, remoteUrl, updatedAt, lastAccessedAt",
|
||||
};
|
||||
|
||||
const openDatabases: Dexie[] = [];
|
||||
const databaseNames = new Set<string>();
|
||||
|
||||
afterEach(async () => {
|
||||
for (const database of openDatabases.splice(0)) database.close();
|
||||
await Promise.all([...databaseNames].map((name) => Dexie.delete(name)));
|
||||
databaseNames.clear();
|
||||
});
|
||||
|
||||
describe("LocalChatDB v5 migration", () => {
|
||||
it("moves owner-only and old character messages to current conversations", async () => {
|
||||
const legacy = await createLegacyDatabase(3);
|
||||
await legacy.table<LocalMessageRow, number>("messages").bulkAdd([
|
||||
messageRow("owner-only", "user:account-1"),
|
||||
messageRow(
|
||||
"old-elio",
|
||||
buildChatConversationKey("user:account-1", "character_elio"),
|
||||
),
|
||||
messageRow(
|
||||
"old-maya",
|
||||
buildChatConversationKey("user:account-1", "character_maya"),
|
||||
),
|
||||
messageRow(
|
||||
"old-nayeli",
|
||||
buildChatConversationKey("device:guest-1", "character_nayeli"),
|
||||
),
|
||||
messageRow(
|
||||
"current-elio",
|
||||
buildChatConversationKey("user:account-2", "elio"),
|
||||
),
|
||||
messageRow(
|
||||
"unknown-character",
|
||||
buildChatConversationKey("user:account-2", "character_aria"),
|
||||
),
|
||||
]);
|
||||
legacy.close();
|
||||
|
||||
const upgraded = track(new LocalChatDB(legacy.name));
|
||||
await upgraded.open();
|
||||
|
||||
const rows = await upgraded.messages.orderBy("dbId").toArray();
|
||||
expect(rows.map(({ id, sessionId }) => ({ id, sessionId }))).toEqual([
|
||||
{
|
||||
id: "owner-only",
|
||||
sessionId: buildChatConversationKey("user:account-1", "elio"),
|
||||
},
|
||||
{
|
||||
id: "old-elio",
|
||||
sessionId: buildChatConversationKey("user:account-1", "elio"),
|
||||
},
|
||||
{
|
||||
id: "old-maya",
|
||||
sessionId: buildChatConversationKey("user:account-1", "maya-tan"),
|
||||
},
|
||||
{
|
||||
id: "old-nayeli",
|
||||
sessionId: buildChatConversationKey(
|
||||
"device:guest-1",
|
||||
"nayeli-cervantes",
|
||||
),
|
||||
},
|
||||
{
|
||||
id: "current-elio",
|
||||
sessionId: buildChatConversationKey("user:account-2", "elio"),
|
||||
},
|
||||
{
|
||||
id: "unknown-character",
|
||||
sessionId: buildChatConversationKey(
|
||||
"user:account-2",
|
||||
"character_aria",
|
||||
),
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
it("rekeys media while preserving current data and the newest legacy row", async () => {
|
||||
const legacy = await createLegacyDatabase(4);
|
||||
const owner = "user:account-1";
|
||||
const currentElioOwner = buildChatConversationKey(owner, "elio");
|
||||
const currentElioKey = buildChatMediaCacheKey({
|
||||
ownerKey: currentElioOwner,
|
||||
messageId: "current-wins",
|
||||
kind: "image",
|
||||
});
|
||||
await legacy.table<LocalChatMediaRow, string>("media").bulkAdd([
|
||||
mediaRow({
|
||||
ownerKey: buildChatConversationKey(owner, "character_elio"),
|
||||
messageId: "current-wins",
|
||||
kind: "image",
|
||||
remoteUrl: "https://example.com/legacy-newer.jpg",
|
||||
updatedAt: 20,
|
||||
}),
|
||||
mediaRow({
|
||||
ownerKey: currentElioOwner,
|
||||
messageId: "current-wins",
|
||||
kind: "image",
|
||||
remoteUrl: "https://example.com/current.jpg",
|
||||
updatedAt: 10,
|
||||
}),
|
||||
mediaRow({
|
||||
ownerKey: owner,
|
||||
messageId: "newest-legacy-wins",
|
||||
kind: "audio",
|
||||
remoteUrl: "https://example.com/older.mp3",
|
||||
updatedAt: 5,
|
||||
}),
|
||||
mediaRow({
|
||||
ownerKey: buildChatConversationKey(owner, "character_elio"),
|
||||
messageId: "newest-legacy-wins",
|
||||
kind: "audio",
|
||||
remoteUrl: "https://example.com/newer.mp3",
|
||||
updatedAt: 15,
|
||||
}),
|
||||
mediaRow({
|
||||
ownerKey: buildChatConversationKey(owner, "character_maya"),
|
||||
messageId: "maya-media",
|
||||
kind: "image",
|
||||
remoteUrl: "https://example.com/maya.jpg",
|
||||
updatedAt: 8,
|
||||
}),
|
||||
]);
|
||||
legacy.close();
|
||||
|
||||
const upgraded = track(new LocalChatDB(legacy.name));
|
||||
await upgraded.open();
|
||||
|
||||
expect(await upgraded.media.get(currentElioKey)).toMatchObject({
|
||||
ownerKey: currentElioOwner,
|
||||
remoteUrl: "https://example.com/current.jpg",
|
||||
});
|
||||
|
||||
const legacyCollisionKey = buildChatMediaCacheKey({
|
||||
ownerKey: currentElioOwner,
|
||||
messageId: "newest-legacy-wins",
|
||||
kind: "audio",
|
||||
});
|
||||
expect(await upgraded.media.get(legacyCollisionKey)).toMatchObject({
|
||||
ownerKey: currentElioOwner,
|
||||
remoteUrl: "https://example.com/newer.mp3",
|
||||
});
|
||||
|
||||
const mayaOwner = buildChatConversationKey(owner, "maya-tan");
|
||||
expect(
|
||||
await upgraded.media.get(
|
||||
buildChatMediaCacheKey({
|
||||
ownerKey: mayaOwner,
|
||||
messageId: "maya-media",
|
||||
kind: "image",
|
||||
}),
|
||||
),
|
||||
).toMatchObject({ ownerKey: mayaOwner });
|
||||
expect(await upgraded.media.count()).toBe(3);
|
||||
});
|
||||
|
||||
it("upgrades an existing v4 database and remains idempotent", async () => {
|
||||
const legacy = await createLegacyDatabase(4);
|
||||
await legacy.table<LocalMessageRow, number>("messages").add(
|
||||
messageRow(
|
||||
"old-elio",
|
||||
buildChatConversationKey("device:guest-1", "character_elio"),
|
||||
),
|
||||
);
|
||||
legacy.close();
|
||||
|
||||
const upgraded = track(new LocalChatDB(legacy.name));
|
||||
await upgraded.open();
|
||||
const firstRows = await upgraded.messages.toArray();
|
||||
upgraded.close();
|
||||
|
||||
const reopened = track(new LocalChatDB(legacy.name));
|
||||
await reopened.open();
|
||||
expect(await reopened.messages.toArray()).toEqual(firstRows);
|
||||
expect(firstRows[0]?.sessionId).toBe(
|
||||
buildChatConversationKey("device:guest-1", "elio"),
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
async function createLegacyDatabase(version: 3 | 4): Promise<Dexie> {
|
||||
const name = `cozsweet-chat-v${version}-${crypto.randomUUID()}`;
|
||||
databaseNames.add(name);
|
||||
const database = track(new Dexie(name));
|
||||
database.version(version).stores(DATABASE_SCHEMA);
|
||||
await database.open();
|
||||
return database;
|
||||
}
|
||||
|
||||
function track<T extends Dexie>(database: T): T {
|
||||
openDatabases.push(database);
|
||||
return database;
|
||||
}
|
||||
|
||||
function messageRow(id: string, sessionId: string): LocalMessageRow {
|
||||
return {
|
||||
id,
|
||||
role: "assistant",
|
||||
type: "text",
|
||||
content: id,
|
||||
createdAt: "2026-07-17T00:00:00.000Z",
|
||||
sessionId,
|
||||
};
|
||||
}
|
||||
|
||||
function mediaRow(
|
||||
input: Pick<
|
||||
LocalChatMediaRow,
|
||||
"ownerKey" | "messageId" | "kind" | "remoteUrl" | "updatedAt"
|
||||
>,
|
||||
): LocalChatMediaRow {
|
||||
return {
|
||||
...input,
|
||||
cacheKey: buildChatMediaCacheKey(input),
|
||||
bytes: new Uint8Array([1, 2, 3]).buffer,
|
||||
mimeType: input.kind === "image" ? "image/jpeg" : "audio/mpeg",
|
||||
byteSize: 3,
|
||||
createdAt: input.updatedAt,
|
||||
lastAccessedAt: input.updatedAt,
|
||||
};
|
||||
}
|
||||
@@ -9,16 +9,12 @@
|
||||
* 构造时 `dbName` 可注入,便于测试时每个用例用独立 DB 互不污染。
|
||||
*/
|
||||
|
||||
import Dexie, { type Table, type Transaction } from "dexie";
|
||||
import Dexie, { type Table } from "dexie";
|
||||
import type { ChatMediaKind } from "@/data/schemas/chat";
|
||||
import type {
|
||||
ChatImageData,
|
||||
ChatLockDetailData,
|
||||
} from "@/data/schemas/chat";
|
||||
import {
|
||||
buildChatMediaCacheKey,
|
||||
migrateLegacyChatConversationKey,
|
||||
} from "@/lib/chat/chat_cache_keys";
|
||||
|
||||
export interface LocalMessageRow {
|
||||
/** Dexie 自增主键(数据库内部使用,不暴露给 LocalMessage 类)。 */
|
||||
@@ -80,71 +76,5 @@ export class LocalChatDB extends Dexie {
|
||||
messages: "++dbId, sessionId",
|
||||
media: "cacheKey, ownerKey, messageId, kind, remoteUrl, updatedAt, lastAccessedAt",
|
||||
});
|
||||
this.version(5)
|
||||
.stores({
|
||||
messages: "++dbId, sessionId",
|
||||
media: "cacheKey, ownerKey, messageId, kind, remoteUrl, updatedAt, lastAccessedAt",
|
||||
})
|
||||
.upgrade(migrateLegacyChatCache);
|
||||
}
|
||||
}
|
||||
|
||||
async function migrateLegacyChatCache(
|
||||
transaction: Transaction,
|
||||
): Promise<void> {
|
||||
const messages = transaction.table<LocalMessageRow, number>("messages");
|
||||
await messages.toCollection().modify((message) => {
|
||||
message.sessionId = migrateLegacyChatConversationKey(message.sessionId);
|
||||
});
|
||||
|
||||
const media = transaction.table<LocalChatMediaRow, string>("media");
|
||||
const rows = await media.toArray();
|
||||
const canonicalCacheKeys = new Set(
|
||||
rows.flatMap((row) => {
|
||||
const ownerKey = migrateLegacyChatConversationKey(row.ownerKey);
|
||||
const cacheKey = buildChatMediaCacheKey({
|
||||
ownerKey,
|
||||
messageId: row.messageId,
|
||||
kind: row.kind,
|
||||
});
|
||||
return ownerKey === row.ownerKey && cacheKey === row.cacheKey
|
||||
? [cacheKey]
|
||||
: [];
|
||||
}),
|
||||
);
|
||||
|
||||
for (const row of rows) {
|
||||
const ownerKey = migrateLegacyChatConversationKey(row.ownerKey);
|
||||
const cacheKey = buildChatMediaCacheKey({
|
||||
ownerKey,
|
||||
messageId: row.messageId,
|
||||
kind: row.kind,
|
||||
});
|
||||
if (ownerKey === row.ownerKey && cacheKey === row.cacheKey) continue;
|
||||
|
||||
const existing = await media.get(cacheKey);
|
||||
await media.delete(row.cacheKey);
|
||||
if (existing && canonicalCacheKeys.has(cacheKey)) continue;
|
||||
if (existing && !isNewerMediaRow(row, existing)) continue;
|
||||
|
||||
await media.put({
|
||||
...row,
|
||||
cacheKey,
|
||||
ownerKey,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
function isNewerMediaRow(
|
||||
candidate: LocalChatMediaRow,
|
||||
existing: LocalChatMediaRow,
|
||||
): boolean {
|
||||
return (
|
||||
candidate.updatedAt > existing.updatedAt ||
|
||||
(candidate.updatedAt === existing.updatedAt &&
|
||||
candidate.lastAccessedAt > existing.lastAccessedAt) ||
|
||||
(candidate.updatedAt === existing.updatedAt &&
|
||||
candidate.lastAccessedAt === existing.lastAccessedAt &&
|
||||
candidate.createdAt > existing.createdAt)
|
||||
);
|
||||
}
|
||||
|
||||
@@ -2,7 +2,6 @@ import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { createStorage } from "unstorage";
|
||||
import memoryDriver from "unstorage/drivers/memory";
|
||||
|
||||
import { StorageKeys } from "@/data/storage/storage_keys";
|
||||
import { SessionAsyncUtil } from "@/utils/session-storage";
|
||||
|
||||
import { NavigationStorage } from "../navigation_storage";
|
||||
@@ -166,51 +165,4 @@ describe("NavigationStorage", () => {
|
||||
).resolves.toMatchObject({ characterId: CHARACTER_ID });
|
||||
});
|
||||
|
||||
it("normalizes old character ids in pending navigation sessions", async () => {
|
||||
const createdAt = Date.now();
|
||||
const legacyPromotion = {
|
||||
characterId: "character_elio",
|
||||
promotionType: "image" as const,
|
||||
lockType: "image_paywall" as const,
|
||||
clientLockId: "promotion_legacy",
|
||||
createdAt,
|
||||
};
|
||||
await SessionAsyncUtil.setJson(
|
||||
StorageKeys.pendingChatPromotion,
|
||||
legacyPromotion,
|
||||
);
|
||||
await SessionAsyncUtil.setJson(StorageKeys.pendingChatUnlock, {
|
||||
reason: "single_message_unlock",
|
||||
characterId: "character_elio",
|
||||
displayMessageId: "promotion:promotion_legacy",
|
||||
kind: "image",
|
||||
lockType: "image_paywall",
|
||||
clientLockId: "promotion_legacy",
|
||||
promotion: legacyPromotion,
|
||||
returnUrl: "/characters/elio/chat",
|
||||
stage: "payment",
|
||||
createdAt,
|
||||
});
|
||||
await SessionAsyncUtil.setJson(StorageKeys.pendingChatImageReturn, {
|
||||
reason: "image_paywall",
|
||||
characterId: "character_elio",
|
||||
messageId: "message_legacy",
|
||||
returnUrl: "/characters/elio/chat?image=message_legacy",
|
||||
createdAt,
|
||||
});
|
||||
|
||||
await expect(
|
||||
NavigationStorage.consumePendingChatPromotion(CHARACTER_ID),
|
||||
).resolves.toMatchObject({ characterId: CHARACTER_ID });
|
||||
await expect(
|
||||
NavigationStorage.peekPendingChatUnlock(CHARACTER_ID),
|
||||
).resolves.toMatchObject({
|
||||
characterId: CHARACTER_ID,
|
||||
promotion: { characterId: CHARACTER_ID },
|
||||
});
|
||||
await expect(
|
||||
NavigationStorage.consumePendingChatImageReturn(CHARACTER_ID),
|
||||
).resolves.toMatchObject({ characterId: CHARACTER_ID });
|
||||
});
|
||||
|
||||
});
|
||||
|
||||
@@ -5,7 +5,6 @@ import { z } from "zod";
|
||||
import { StorageKeys } from "@/data/storage/storage_keys";
|
||||
import { ChatLockTypeSchema } from "@/data/schemas/chat";
|
||||
import { getCharacterById } from "@/data/constants/character";
|
||||
import { normalizeLegacyCharacterId } from "@/data/constants/character_id_aliases";
|
||||
import { Result } from "@/utils/result";
|
||||
import { SessionAsyncUtil } from "@/utils/session-storage";
|
||||
|
||||
@@ -139,9 +138,7 @@ export class NavigationStorage {
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatUnlock);
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== normalizeLegacyCharacterId(characterId)) {
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== characterId) return null;
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatUnlock);
|
||||
return value;
|
||||
}
|
||||
@@ -161,9 +158,7 @@ export class NavigationStorage {
|
||||
if (!value) {
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatUnlock);
|
||||
}
|
||||
return value?.characterId === normalizeLegacyCharacterId(characterId)
|
||||
? value
|
||||
: null;
|
||||
return value?.characterId === characterId ? value : null;
|
||||
}
|
||||
|
||||
static async hasPendingChatUnlock(characterId: string): Promise<boolean> {
|
||||
@@ -211,9 +206,7 @@ export class NavigationStorage {
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatPromotion);
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== normalizeLegacyCharacterId(characterId)) {
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== characterId) return null;
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatPromotion);
|
||||
return value;
|
||||
}
|
||||
@@ -257,9 +250,7 @@ export class NavigationStorage {
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatImageReturn);
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== normalizeLegacyCharacterId(characterId)) {
|
||||
return null;
|
||||
}
|
||||
if (value.characterId !== characterId) return null;
|
||||
await SessionAsyncUtil.remove(StorageKeys.pendingChatImageReturn);
|
||||
return value;
|
||||
}
|
||||
@@ -272,15 +263,13 @@ export class NavigationStorage {
|
||||
value: PendingChatUnlock | null,
|
||||
): PendingChatUnlock | null {
|
||||
if (!value || Date.now() - value.createdAt > MAX_AGE_MS) return null;
|
||||
const characterId = normalizeLegacyCharacterId(value.characterId);
|
||||
if (!getCharacterById(characterId)) return null;
|
||||
if (!getCharacterById(value.characterId)) return null;
|
||||
const promotion = value.promotion
|
||||
? NavigationStorage.parsePendingChatPromotion(value.promotion)
|
||||
: undefined;
|
||||
if (value.promotion && !promotion) return null;
|
||||
return PendingChatUnlockSchema.parse({
|
||||
...value,
|
||||
characterId,
|
||||
...(promotion ? { promotion } : {}),
|
||||
});
|
||||
}
|
||||
@@ -289,20 +278,14 @@ export class NavigationStorage {
|
||||
value: PendingChatImageReturn | null,
|
||||
): PendingChatImageReturn | null {
|
||||
if (!value || Date.now() - value.createdAt > MAX_AGE_MS) return null;
|
||||
const characterId = normalizeLegacyCharacterId(value.characterId);
|
||||
return getCharacterById(characterId)
|
||||
? PendingChatImageReturnSchema.parse({ ...value, characterId })
|
||||
: null;
|
||||
return getCharacterById(value.characterId) ? value : null;
|
||||
}
|
||||
|
||||
private static parsePendingChatPromotion(
|
||||
value: PendingChatPromotion | null,
|
||||
): PendingChatPromotion | null {
|
||||
if (!value || Date.now() - value.createdAt > MAX_AGE_MS) return null;
|
||||
const characterId = normalizeLegacyCharacterId(value.characterId);
|
||||
return getCharacterById(characterId)
|
||||
? PendingChatPromotionSchema.parse({ ...value, characterId })
|
||||
: null;
|
||||
return getCharacterById(value.characterId) ? value : null;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1,9 +1,6 @@
|
||||
import type { ChatMediaKind } from "@/data/schemas/chat";
|
||||
import { normalizeLegacyCharacterId } from "@/data/constants/character_id_aliases";
|
||||
|
||||
const CHARACTER_NAMESPACE = "::character:";
|
||||
const LEGACY_OWNER_KEY_PATTERN = /^(?:user|device):.+$/u;
|
||||
const LEGACY_DEFAULT_CHARACTER_ID = "elio";
|
||||
|
||||
export function buildChatConversationKey(
|
||||
ownerKey: string,
|
||||
@@ -22,37 +19,3 @@ export function buildChatMediaCacheKey(input: {
|
||||
}): string {
|
||||
return `${input.ownerKey}:${input.messageId}:${input.kind}`;
|
||||
}
|
||||
|
||||
/**
|
||||
* Migrates both pre-character owner keys and keys containing the first
|
||||
* frontend-only character IDs. This is intentionally idempotent so it can be
|
||||
* used by database upgrades and compatibility reads.
|
||||
*/
|
||||
export function migrateLegacyChatConversationKey(ownerKey: string): string {
|
||||
const namespaceIndex = ownerKey.lastIndexOf(CHARACTER_NAMESPACE);
|
||||
if (namespaceIndex < 0) {
|
||||
return LEGACY_OWNER_KEY_PATTERN.test(ownerKey)
|
||||
? buildChatConversationKey(ownerKey, LEGACY_DEFAULT_CHARACTER_ID)
|
||||
: ownerKey;
|
||||
}
|
||||
|
||||
const encodedCharacterId = ownerKey.slice(
|
||||
namespaceIndex + CHARACTER_NAMESPACE.length,
|
||||
);
|
||||
if (encodedCharacterId.length === 0) return ownerKey;
|
||||
|
||||
let characterId: string;
|
||||
try {
|
||||
characterId = decodeURIComponent(encodedCharacterId);
|
||||
} catch {
|
||||
return ownerKey;
|
||||
}
|
||||
|
||||
const normalizedCharacterId = normalizeLegacyCharacterId(characterId);
|
||||
if (normalizedCharacterId === characterId) return ownerKey;
|
||||
|
||||
return buildChatConversationKey(
|
||||
ownerKey.slice(0, namespaceIndex),
|
||||
normalizedCharacterId,
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user