import {
  LiveList,
  LiveMap,
  LiveObject,
  Lson,
  LsonObject,
} from '@liveblocks/client';

type ExtractObjectShape<T extends LiveObject<LsonObject>> =
  T extends LiveObject<infer U> ? U : never;
type ExtractListShape<T extends LiveList<Lson>> = T extends LiveList<infer U>
  ? U
  : never;
type ExtractMapShape<T extends LiveMap> = T extends LiveMap<infer X, infer U>
  ? Record<X, U>
  : never;

type ApiJsonValues<T extends Record<string, any>> = {
  [K in keyof T]: ApiJsonData<T[K]>;
};

type ApiJsonObject<T extends LiveObject> = {
  liveblocksType: 'LiveObject';
  data: ApiJsonValues<ExtractObjectShape<T>>;
};
type ApiJsonList<T extends LiveList> = {
  liveblocksType: 'LiveList';
  data: ApiJsonData<ExtractListShape<T>>[];
};
type ApiJsonMap<T extends LiveMap> = {
  liveblocksType: 'LiveMap';
  data: ApiJsonData<ExtractMapShape<T>>;
};
type ApiJsonData<T> = T extends LiveObject
  ? ApiJsonObject<T>
  : T extends LiveList
  ? ApiJsonList<T>
  : T extends LiveMap
  ? ApiJsonMap<T>
  : T extends Record<string, any>
  ? ApiJsonValues<T>
  : T extends Array<any>
  ? Array<ApiJsonData<T[number]>>
  : T;

export type LiveblocksApiJson<
  SourceData extends LiveObject | Record<string, any>,
> = ApiJsonData<SourceData>;