@lvce-editor/chat-storage-worker 1.0.0 → 1.2.0

package/dist/chatStorageWorkerMain.js

@@ -888,6 +888,631 @@ const create = async ({
   return rpc;
 };
 
+const toError = error => {
+  if (error instanceof Error) {
+    return error;
+  }
+  return new Error('IndexedDB request failed');
+};
+
+const requestToPromise = async createRequest => {
+  const request = createRequest();
+  const {
+    promise,
+    reject,
+    resolve
+  } = Promise.withResolvers();
+  request.addEventListener('success', () => {
+    resolve(request.result);
+  });
+  request.addEventListener('error', () => {
+    reject(toError(request.error));
+  });
+  return promise;
+};
+
+const transactionToPromise = async createTransaction => {
+  const transaction = createTransaction();
+  const {
+    promise,
+    reject,
+    resolve
+  } = Promise.withResolvers();
+  transaction.addEventListener('complete', () => {
+    resolve();
+  });
+  transaction.addEventListener('error', () => {
+    reject(toError(transaction.error));
+  });
+  transaction.addEventListener('abort', () => {
+    reject(toError(transaction.error));
+  });
+  return promise;
+};
+
+const toChatViewEvent = event => {
+  const {
+    eventId,
+    ...chatViewEvent
+  } = event;
+  return chatViewEvent;
+};
+const now$1 = () => {
+  return new Date().toISOString();
+};
+const isSameMessage$1 = (a, b) => {
+  return a.id === b.id && a.inProgress === b.inProgress && a.role === b.role && a.text === b.text && a.time === b.time && JSON.stringify(a.toolCalls || []) === JSON.stringify(b.toolCalls || []);
+};
+const canAppendMessages$1 = (previousMessages, nextMessages) => {
+  if (nextMessages.length < previousMessages.length) {
+    return false;
+  }
+  return previousMessages.every((message, index) => isSameMessage$1(message, nextMessages[index]));
+};
+const canUpdateMessages$1 = (previousMessages, nextMessages) => {
+  if (previousMessages.length !== nextMessages.length) {
+    return false;
+  }
+  for (let i = 0; i < previousMessages.length; i += 1) {
+    const previous = previousMessages[i];
+    const next = nextMessages[i];
+    if (previous.id !== next.id || previous.role !== next.role) {
+      return false;
+    }
+  }
+  return true;
+};
+const getMutationEvents$1 = (previous, next) => {
+  const timestamp = now$1();
+  const events = [];
+  if (!previous) {
+    events.push({
+      sessionId: next.id,
+      timestamp,
+      title: next.title,
+      type: 'chat-session-created'
+    });
+    for (const message of next.messages) {
+      events.push({
+        message,
+        sessionId: next.id,
+        timestamp,
+        type: 'chat-message-added'
+      });
+    }
+    return events;
+  }
+  if (previous.title !== next.title) {
+    events.push({
+      sessionId: next.id,
+      timestamp,
+      title: next.title,
+      type: 'chat-session-title-updated'
+    });
+  }
+  if (canAppendMessages$1(previous.messages, next.messages)) {
+    for (let i = previous.messages.length; i < next.messages.length; i += 1) {
+      events.push({
+        message: next.messages[i],
+        sessionId: next.id,
+        timestamp,
+        type: 'chat-message-added'
+      });
+    }
+    return events;
+  }
+  if (canUpdateMessages$1(previous.messages, next.messages)) {
+    for (let i = 0; i < previous.messages.length; i += 1) {
+      const previousMessage = previous.messages[i];
+      const nextMessage = next.messages[i];
+      if (!isSameMessage$1(previousMessage, nextMessage)) {
+        events.push({
+          inProgress: nextMessage.inProgress,
+          messageId: nextMessage.id,
+          sessionId: next.id,
+          text: nextMessage.text,
+          time: nextMessage.time,
+          timestamp,
+          toolCalls: nextMessage.toolCalls,
+          type: 'chat-message-updated'
+        });
+      }
+    }
+    return events;
+  }
+  events.push({
+    messages: [...next.messages],
+    sessionId: next.id,
+    timestamp,
+    type: 'chat-session-messages-replaced'
+  });
+  return events;
+};
+const replaySession$1 = (id, summary, events) => {
+  let deleted = false;
+  let title = summary?.title || '';
+  let messages = summary?.messages ? [...summary.messages] : [];
+  for (const event of events) {
+    if (event.sessionId !== id) {
+      continue;
+    }
+    if (event.type === 'chat-session-created') {
+      const {
+        title: eventTitle
+      } = event;
+      deleted = false;
+      title = eventTitle;
+      continue;
+    }
+    if (event.type === 'chat-session-deleted') {
+      deleted = true;
+      continue;
+    }
+    if (event.type === 'chat-session-title-updated') {
+      const {
+        title: eventTitle
+      } = event;
+      title = eventTitle;
+      continue;
+    }
+    if (event.type === 'chat-message-added') {
+      messages = [...messages, event.message];
+      continue;
+    }
+    if (event.type === 'chat-message-updated') {
+      messages = messages.map(message => {
+        if (message.id !== event.messageId) {
+          return message;
+        }
+        return {
+          ...message,
+          ...(event.inProgress === undefined ? {} : {
+            inProgress: event.inProgress
+          }),
+          text: event.text,
+          time: event.time,
+          ...(event.toolCalls === undefined ? {} : {
+            toolCalls: event.toolCalls
+          })
+        };
+      });
+      continue;
+    }
+    if (event.type === 'chat-session-messages-replaced') {
+      messages = [...event.messages];
+    }
+  }
+  if (deleted || !title) {
+    return undefined;
+  }
+  return {
+    id,
+    messages,
+    title
+  };
+};
+class IndexedDbChatSessionStorage {
+  constructor(options = {}) {
+    this.state = {
+      databaseName: options.databaseName || 'lvce-chat-view-sessions',
+      databasePromise: undefined,
+      databaseVersion: options.databaseVersion || 2,
+      eventStoreName: options.eventStoreName || 'chat-view-events',
+      storeName: options.storeName || 'chat-sessions'
+    };
+  }
+  openDatabase = async () => {
+    if (this.state.databasePromise) {
+      return this.state.databasePromise;
+    }
+    const request = indexedDB.open(this.state.databaseName, this.state.databaseVersion);
+    request.addEventListener('upgradeneeded', () => {
+      const database = request.result;
+      if (!database.objectStoreNames.contains(this.state.storeName)) {
+        database.createObjectStore(this.state.storeName, {
+          keyPath: 'id'
+        });
+      }
+      if (database.objectStoreNames.contains(this.state.eventStoreName)) {
+        const {
+          transaction
+        } = request;
+        if (!transaction) {
+          return;
+        }
+        const eventStore = transaction.objectStore(this.state.eventStoreName);
+        if (!eventStore.indexNames.contains('sessionId')) {
+          eventStore.createIndex('sessionId', 'sessionId', {
+            unique: false
+          });
+        }
+      } else {
+        const eventStore = database.createObjectStore(this.state.eventStoreName, {
+          autoIncrement: true,
+          keyPath: 'eventId'
+        });
+        eventStore.createIndex('sessionId', 'sessionId', {
+          unique: false
+        });
+      }
+    });
+    const databasePromise = requestToPromise(() => request);
+    this.state.databasePromise = databasePromise;
+    return databasePromise;
+  };
+  listSummaries = async () => {
+    const database = await this.openDatabase();
+    const transaction = database.transaction(this.state.storeName, 'readonly');
+    const store = transaction.objectStore(this.state.storeName);
+    const summaries = await requestToPromise(() => store.getAll());
+    return summaries;
+  };
+  getSummary = async id => {
+    const database = await this.openDatabase();
+    const transaction = database.transaction(this.state.storeName, 'readonly');
+    const store = transaction.objectStore(this.state.storeName);
+    const summary = await requestToPromise(() => store.get(id));
+    return summary;
+  };
+  getEventsBySessionId = async sessionId => {
+    const database = await this.openDatabase();
+    const transaction = database.transaction(this.state.eventStoreName, 'readonly');
+    const store = transaction.objectStore(this.state.eventStoreName);
+    const index = store.index('sessionId');
+    const events = await requestToPromise(() => index.getAll(IDBKeyRange.only(sessionId)));
+    return events.map(toChatViewEvent);
+  };
+  listEventsInternal = async () => {
+    const database = await this.openDatabase();
+    const transaction = database.transaction(this.state.eventStoreName, 'readonly');
+    const store = transaction.objectStore(this.state.eventStoreName);
+    const events = await requestToPromise(() => store.getAll());
+    return events.map(toChatViewEvent);
+  };
+  appendEvents = async events => {
+    if (events.length === 0) {
+      return;
+    }
+    const database = await this.openDatabase();
+    const transaction = database.transaction([this.state.storeName, this.state.eventStoreName], 'readwrite');
+    const summaryStore = transaction.objectStore(this.state.storeName);
+    const eventStore = transaction.objectStore(this.state.eventStoreName);
+    for (const event of events) {
+      eventStore.add(event);
+      if (event.type === 'chat-session-created' || event.type === 'chat-session-title-updated') {
+        summaryStore.put({
+          id: event.sessionId,
+          title: event.title
+        });
+      }
+      if (event.type === 'chat-session-deleted') {
+        summaryStore.delete(event.sessionId);
+      }
+    }
+    await transactionToPromise(() => transaction);
+  };
+  async appendEvent(event) {
+    await this.appendEvents([event]);
+  }
+  async clear() {
+    const database = await this.openDatabase();
+    const transaction = database.transaction([this.state.storeName, this.state.eventStoreName], 'readwrite');
+    transaction.objectStore(this.state.storeName).clear();
+    transaction.objectStore(this.state.eventStoreName).clear();
+    await transactionToPromise(() => transaction);
+  }
+  async deleteSession(id) {
+    await this.appendEvent({
+      sessionId: id,
+      timestamp: now$1(),
+      type: 'chat-session-deleted'
+    });
+  }
+  async getEvents(sessionId) {
+    if (sessionId) {
+      return this.getEventsBySessionId(sessionId);
+    }
+    return this.listEventsInternal();
+  }
+  async getSession(id) {
+    const [summary, events] = await Promise.all([this.getSummary(id), this.getEventsBySessionId(id)]);
+    return replaySession$1(id, summary, events);
+  }
+  async listSessions() {
+    const summaries = await this.listSummaries();
+    const sessions = [];
+    for (const summary of summaries) {
+      const events = await this.getEventsBySessionId(summary.id);
+      const session = replaySession$1(summary.id, summary, events);
+      if (!session) {
+        continue;
+      }
+      sessions.push(session);
+    }
+    return sessions;
+  }
+  async setSession(session) {
+    const previous = await this.getSession(session.id);
+    const events = getMutationEvents$1(previous, session);
+    await this.appendEvents(events);
+    if (events.length === 0) {
+      const database = await this.openDatabase();
+      const transaction = database.transaction(this.state.storeName, 'readwrite');
+      const summaryStore = transaction.objectStore(this.state.storeName);
+      summaryStore.put({
+        id: session.id,
+        title: session.title
+      });
+      await transactionToPromise(() => transaction);
+    }
+  }
+}
+
+const now = () => {
+  return new Date().toISOString();
+};
+const isSameMessage = (a, b) => {
+  return a.id === b.id && a.inProgress === b.inProgress && a.role === b.role && a.text === b.text && a.time === b.time && JSON.stringify(a.toolCalls || []) === JSON.stringify(b.toolCalls || []);
+};
+const canAppendMessages = (previousMessages, nextMessages) => {
+  if (nextMessages.length < previousMessages.length) {
+    return false;
+  }
+  return previousMessages.every((message, index) => isSameMessage(message, nextMessages[index]));
+};
+const canUpdateMessages = (previousMessages, nextMessages) => {
+  if (previousMessages.length !== nextMessages.length) {
+    return false;
+  }
+  for (let i = 0; i < previousMessages.length; i += 1) {
+    const previous = previousMessages[i];
+    const next = nextMessages[i];
+    if (previous.id !== next.id || previous.role !== next.role) {
+      return false;
+    }
+  }
+  return true;
+};
+const getMutationEvents = (previous, next) => {
+  const timestamp = now();
+  const events = [];
+  if (!previous) {
+    events.push({
+      sessionId: next.id,
+      timestamp,
+      title: next.title,
+      type: 'chat-session-created'
+    });
+    for (const message of next.messages) {
+      events.push({
+        message,
+        sessionId: next.id,
+        timestamp,
+        type: 'chat-message-added'
+      });
+    }
+    return events;
+  }
+  if (previous.title !== next.title) {
+    events.push({
+      sessionId: next.id,
+      timestamp,
+      title: next.title,
+      type: 'chat-session-title-updated'
+    });
+  }
+  if (canAppendMessages(previous.messages, next.messages)) {
+    for (let i = previous.messages.length; i < next.messages.length; i += 1) {
+      events.push({
+        message: next.messages[i],
+        sessionId: next.id,
+        timestamp,
+        type: 'chat-message-added'
+      });
+    }
+    return events;
+  }
+  if (canUpdateMessages(previous.messages, next.messages)) {
+    for (let i = 0; i < previous.messages.length; i += 1) {
+      const previousMessage = previous.messages[i];
+      const nextMessage = next.messages[i];
+      if (!isSameMessage(previousMessage, nextMessage)) {
+        events.push({
+          inProgress: nextMessage.inProgress,
+          messageId: nextMessage.id,
+          sessionId: next.id,
+          text: nextMessage.text,
+          time: nextMessage.time,
+          timestamp,
+          toolCalls: nextMessage.toolCalls,
+          type: 'chat-message-updated'
+        });
+      }
+    }
+    return events;
+  }
+  events.push({
+    messages: [...next.messages],
+    sessionId: next.id,
+    timestamp,
+    type: 'chat-session-messages-replaced'
+  });
+  return events;
+};
+const replaySession = (id, title, events) => {
+  let deleted = false;
+  let currentTitle = title || '';
+  let messages = [];
+  for (const event of events) {
+    if (event.sessionId !== id) {
+      continue;
+    }
+    if (event.type === 'chat-session-created') {
+      deleted = false;
+      currentTitle = event.title;
+      continue;
+    }
+    if (event.type === 'chat-session-deleted') {
+      deleted = true;
+      continue;
+    }
+    if (event.type === 'chat-session-title-updated') {
+      currentTitle = event.title;
+      continue;
+    }
+    if (event.type === 'chat-message-added') {
+      messages = [...messages, event.message];
+      continue;
+    }
+    if (event.type === 'chat-message-updated') {
+      messages = messages.map(message => {
+        if (message.id !== event.messageId) {
+          return message;
+        }
+        return {
+          ...message,
+          ...(event.inProgress === undefined ? {} : {
+            inProgress: event.inProgress
+          }),
+          text: event.text,
+          time: event.time,
+          ...(event.toolCalls === undefined ? {} : {
+            toolCalls: event.toolCalls
+          })
+        };
+      });
+      continue;
+    }
+    if (event.type === 'chat-session-messages-replaced') {
+      messages = [...event.messages];
+    }
+  }
+  if (deleted || !currentTitle) {
+    return undefined;
+  }
+  return {
+    id,
+    messages,
+    title: currentTitle
+  };
+};
+class InMemoryChatSessionStorage {
+  events = [];
+  summaries = new Map();
+  async appendEvent(event) {
+    this.events.push(event);
+    if (event.type === 'chat-session-created' || event.type === 'chat-session-title-updated') {
+      this.summaries.set(event.sessionId, event.title);
+      return;
+    }
+    if (event.type === 'chat-session-deleted') {
+      this.summaries.delete(event.sessionId);
+    }
+  }
+  async clear() {
+    this.events.length = 0;
+    this.summaries.clear();
+  }
+  async deleteSession(id) {
+    await this.appendEvent({
+      sessionId: id,
+      timestamp: now(),
+      type: 'chat-session-deleted'
+    });
+  }
+  async getEvents(sessionId) {
+    if (!sessionId) {
+      return [...this.events];
+    }
+    return this.events.filter(event => event.sessionId === sessionId);
+  }
+  async getSession(id) {
+    return replaySession(id, this.summaries.get(id), this.events);
+  }
+  async listSessions() {
+    const ids = new Set();
+    for (const id of this.summaries.keys()) {
+      ids.add(id);
+    }
+    for (const event of this.events) {
+      ids.add(event.sessionId);
+    }
+    const sessions = [];
+    for (const id of ids) {
+      const session = replaySession(id, this.summaries.get(id), this.events);
+      if (!session) {
+        continue;
+      }
+      sessions.push(session);
+    }
+    return sessions;
+  }
+  async setSession(session) {
+    const previous = await this.getSession(session.id);
+    const events = getMutationEvents(previous, session);
+    for (const event of events) {
+      await this.appendEvent(event);
+    }
+    this.summaries.set(session.id, session.title);
+  }
+}
+
+const createDefaultStorage = () => {
+  if (typeof indexedDB === 'undefined') {
+    return new InMemoryChatSessionStorage();
+  }
+  return new IndexedDbChatSessionStorage();
+};
+let chatSessionStorage = createDefaultStorage();
+const setSession = async session => {
+  await chatSessionStorage.setSession(session);
+};
+const listChatSessions = async () => {
+  const sessions = await chatSessionStorage.listSessions();
+  return sessions.map(session => {
+    const summary = {
+      id: session.id,
+      messages: [],
+      title: session.title
+    };
+    if (!session.projectId) {
+      return summary;
+    }
+    return {
+      ...summary,
+      projectId: session.projectId
+    };
+  });
+};
+const getChatSession = async id => {
+  const session = await chatSessionStorage.getSession(id);
+  if (!session) {
+    return undefined;
+  }
+  const resultBase = {
+    id: session.id,
+    messages: [...session.messages],
+    title: session.title
+  };
+  const result = session.projectId ? {
+    ...resultBase,
+    projectId: session.projectId
+  } : resultBase;
+  return result;
+};
+const deleteChatSession = async id => {
+  await chatSessionStorage.deleteSession(id);
+};
+const clearChatSessions = async () => {
+  await chatSessionStorage.clear();
+};
+const appendChatViewEvent = async event => {
+  await chatSessionStorage.appendEvent(event);
+};
+const getChatViewEvents = async sessionId => {
+  return chatSessionStorage.getEvents(sessionId);
+};
+
 const handleMessagePort = async port => {
   await create$1({
     commandMap: commandMap,
@@ -896,6 +1521,13 @@ const handleMessagePort = async port => {
 };
 
 const commandMap = {
+  'ChatStorage.appendEvent': appendChatViewEvent,
+  'ChatStorage.clear': clearChatSessions,
+  'ChatStorage.deleteSession': deleteChatSession,
+  'ChatStorage.getEvents': getChatViewEvents,
+  'ChatStorage.getSession': getChatSession,
+  'ChatStorage.listSessions': listChatSessions,
+  'ChatStorage.setSession': setSession,
   'HandleMessagePort.handleMessagePort': handleMessagePort
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lvce-editor/chat-storage-worker",
-  "version": "1.0.0",
+  "version": "1.2.0",
   "description": "Chat Storage Worker",
   "repository": {
     "type": "git",