import { fetchGraphData } from './fetch-graph';
import {
  PumActionHistoryEntities,
  transform,
} from './query/pumpum/pum-action-history';
import { PrexApiService } from '../api';
import {
  PUM_TIMELINE_QUERY,
  PUM_ACTION_HISTORY_QUERY,
} from './query/pumpum/pum-action-history';
import { Address } from 'viem';
import { PagingOptions } from '../types';

export async function queryPumTimeline(
  apiService: PrexApiService,
  pageOptions: PagingOptions
) {
  const body = await fetchGraphData(
    apiService,
    'pumTimeline',
    PUM_TIMELINE_QUERY,
    {
      limit: pageOptions.limit,
      offset: pageOptions.offset,
    }
  );

  const pumActionHistory = body as PumActionHistoryEntities;

  return transform(pumActionHistory);
}

export async function queryPumActionHistory(
  apiService: PrexApiService,
  user: Address,
  pageOptions: PagingOptions
) {
  const body = await fetchGraphData(
    apiService,
    'pumActionHistory',
    PUM_ACTION_HISTORY_QUERY,
    {
      limit: pageOptions.limit,
      offset: pageOptions.offset,
      user: user.toLowerCase(),
    }
  );

  const pumActionHistory = body as PumActionHistoryEntities;

  return transform(pumActionHistory);
}