cinny/src/client/initMatrix.ts

import { createClient, MatrixClient, IndexedDBStore, IndexedDBCryptoStore } from 'matrix-js-sdk';

import { cryptoCallbacks } from './secretStorageKeys';
import { clearNavToActivePathStore } from '../app/state/navToActivePath';
import { getFallbackSession, removeFallbackSession, Session } from '../app/state/sessions';
import { LotusOidcTokenRefresher } from './oidcTokenRefresher';
import { revokeOidcTokens } from './oidcLogout';
import { pushSessionToSW } from '../sw-session';
import { deleteSearchCacheDatabase } from '../app/utils/searchCache';

// Thrown when the local IndexedDB has a higher schema version than this SDK expects.
// This happens after a downgrade (e.g. matrix-js-sdk was briefly upgraded and then reverted).
export const IDB_VERSION_CONFLICT = 'IDB_VERSION_CONFLICT';

export const initClient = async (session: Session): Promise<MatrixClient> => {
  const indexedDBStore = new IndexedDBStore({
    indexedDB: globalThis.indexedDB,
    localStorage: globalThis.localStorage,
    dbName: 'web-sync-store',
  });

  const legacyCryptoStore = new IndexedDBCryptoStore(globalThis.indexedDB, 'crypto-store');

  // OIDC/next-gen-auth sessions carry a refresh token; wire automatic refresh
  // (the client calls this reactively on a 401) and persist rotated tokens.
  const oidcRefresher =
    session.refreshToken && session.oidc
      ? new LotusOidcTokenRefresher(session.oidc, session.deviceId, session.userId, session.baseUrl)
      : undefined;

  const mx = createClient({
    baseUrl: session.baseUrl,
    accessToken: session.accessToken,
    refreshToken: session.refreshToken,
    userId: session.userId,
    store: indexedDBStore,
    cryptoStore: legacyCryptoStore,
    deviceId: session.deviceId,
    timelineSupport: true,
    cryptoCallbacks: cryptoCallbacks as any,
    // SAS (emoji) + QR-code verification (show/scan/reciprocate).
    verificationMethods: ['m.sas.v1', 'm.qr_code.show.v1', 'm.qr_code.scan.v1', 'm.reciprocate.v1'],
    tokenRefreshFunction: oidcRefresher
      ? (refreshToken) => oidcRefresher.doRefreshAccessToken(refreshToken)
      : undefined,
  });

  try {
    await indexedDBStore.startup();
  } catch (e) {
    // IDB VersionError = local DB was written by a newer SDK version (schema downgrade).
    if (e instanceof DOMException && e.name === 'VersionError') {
      throw new Error(IDB_VERSION_CONFLICT);
    }
    throw e;
  }
  await mx.initRustCrypto();

  mx.setMaxListeners(150);
  mx.matrixRTC.setMaxListeners(150);

  return mx;
};

export const startClient = async (mx: MatrixClient) => {
  await mx.startClient({
    lazyLoadMembers: true,
    // P3-8: partition m.thread relations into Thread objects/timelines. Thread
    // replies leave the main timeline (roots stay + get a summary chip); the
    // thread panel renders them. markAsRead sends UNTHREADED receipts
    // (utils/notifications.ts) so room badges keep clearing.
    threadSupport: true,
  });
};

export const clearCacheAndReload = async (mx: MatrixClient) => {
  mx.stopClient();
  clearNavToActivePathStore(mx.getSafeUserId());
  await mx.store.deleteAllData();
  window.location.reload();
};

export const logoutClient = async (mx: MatrixClient) => {
  pushSessionToSW();
  mx.stopClient();
  // For OIDC sessions, revoke the tokens at the issuer too (best-effort).
  const session = getFallbackSession();
  if (session?.oidc) {
    await revokeOidcTokens(session);
  }
  try {
    await mx.logout();
  } catch {
    // ignore if failed to logout
  }
  await mx.clearStores();
  // The opt-in local search index stores decrypted plaintext — always wipe it
  // on logout. (clearLoginData below nukes all IDB databases, covering it too.)
  await deleteSearchCacheDatabase();
  // Remove only the session credential keys, preserving user preferences and
  // unsent drafts (N98). The factory-reset path is clearLoginData() below.
  removeFallbackSession();
  window.location.reload();
};

export const clearLoginData = async () => {
  const dbs = await window.indexedDB.databases();

  dbs.forEach((idbInfo) => {
    const { name } = idbInfo;
    if (name) {
      window.indexedDB.deleteDatabase(name);
    }
  });

  // Unregister service workers so stale caches don't interfere after a reset
  if ('serviceWorker' in navigator) {
    const regs = await navigator.serviceWorker.getRegistrations();
    await Promise.all(regs.map((r) => r.unregister()));
    const cacheNames = await caches.keys();
    await Promise.all(cacheNames.map((c) => caches.delete(c)));
  }

  window.localStorage.clear();
  window.location.reload();
};