import { useSDKModel } from './use-models';
import { ApiKeys, PingData } from '../types';
import { useAsync, UseAsyncResult } from './use-async';
import { RawModel } from '../models';

export type UseRawModelType = {
  model: RawModel;
  ping: () => UseAsyncResult<PingData>;
};

export const useRaw = (): UseRawModelType => {
  const model = <RawModel>(<unknown>useSDKModel(ApiKeys.RAW));
  return {
    model,
    // @ts-ignore
    ping: usePing(model),
  };
};

export const usePing = (model: RawModel) => (): UseAsyncResult<PingData> => {
  return useAsync<PingData>(model.ping.bind(model));
};