logto/packages/console/src/hooks/use-api-resources-usage.ts

import { isManagementApi } from '@logto/schemas';
import { useContext, useMemo } from 'react';
import useSWR from 'swr';

import { type ApiResource } from '@/consts';
import { isCloud } from '@/consts/env';
import { TenantsContext } from '@/contexts/TenantsProvider';
import { hasReachedQuotaLimit, hasSurpassedQuotaLimit } from '@/utils/quota';

import useSubscriptionPlan from './use-subscription-plan';

const useApiResourcesUsage = () => {
  const { currentTenantId } = useContext(TenantsContext);
  const { data: currentPlan } = useSubscriptionPlan(currentTenantId);
  /**
   * Note: we only need to fetch all resources when the user is in cloud environment.
   * The oss version doesn't have the quota limit.
   */
  const { data: allResources } = useSWR<ApiResource[]>(isCloud && 'api/resources');

  const resourceCount = useMemo(
    () => allResources?.filter(({ indicator }) => !isManagementApi(indicator)).length ?? 0,
    [allResources]
  );

  const hasReachedLimit = useMemo(
    () =>
      Boolean(
        currentPlan &&
          hasReachedQuotaLimit({
            quotaKey: 'resourcesLimit',
            plan: currentPlan,
            usage: resourceCount,
          })
      ),
    [currentPlan, resourceCount]
  );

  const hasSurpassedLimit = useMemo(
    () =>
      Boolean(
        currentPlan &&
          hasSurpassedQuotaLimit({
            quotaKey: 'resourcesLimit',
            plan: currentPlan,
            usage: resourceCount,
          })
      ),
    [currentPlan, resourceCount]
  );

  return {
    hasReachedLimit,
    hasSurpassedLimit,
  };
};

export default useApiResourcesUsage;