import { MatrixClient, MatrixEvent } from 'matrix-js-sdk';
import { AccountDataEvent } from '../../types/matrix/accountData';

// Typed global account-data read/write helpers.
//
// matrix-js-sdk's `getAccountData`/`setAccountData` overloads only accept the
// SDK's built-in AccountDataEvents union and reject the fork's custom event
// names (the `io.lotus.*` / `in.cinny.*` values in the AccountDataEvent enum, as
// well as a few dynamic keys used by developer tools). We centralize the single
// `as any` cast here so every call site stays fully typed on its content shape.

export function getAccountData<T>(
  mx: MatrixClient,
  eventType: AccountDataEvent | string,
): T | undefined {
  const event = (mx as any).getAccountData(eventType) as MatrixEvent | undefined;
  return event?.getContent() as T | undefined;
}

export function setAccountData<T>(
  mx: MatrixClient,
  eventType: AccountDataEvent | string,
  content: T,
): Promise<void> {
  return (mx as any).setAccountData(eventType, content);
}