umami/src/app/api/users/[userId]/usage/route.ts

import { z } from 'zod';
import { json, unauthorized, badRequest } from 'lib/response';
import { getAllUserWebsitesIncludingTeamOwner } from 'queries/prisma/website';
import { getEventUsage } from 'queries/analytics/events/getEventUsage';
import { getEventDataUsage } from 'queries/analytics/events/getEventDataUsage';
import { checkAuth } from 'lib/auth';
import { checkRequest } from 'lib/request';

export async function GET(request: Request, { params }: { params: Promise<{ userId: string }> }) {
  const schema = z.object({
    startAt: z.coerce.number().int(),
    endAt: z.coerce.number().int(),
  });

  const { query, error } = await checkRequest(request, schema);

  if (error) {
    return badRequest(error);
  }

  const auth = await checkAuth(request);

  if (!auth || !auth.user.isAdmin) {
    return unauthorized();
  }

  const { userId } = await params;
  const { startAt, endAt } = query;

  const startDate = new Date(+startAt);
  const endDate = new Date(+endAt);

  const websites = await getAllUserWebsitesIncludingTeamOwner(userId);

  const websiteIds = websites.map(a => a.id);

  const websiteEventUsage = await getEventUsage(websiteIds, startDate, endDate);
  const eventDataUsage = await getEventDataUsage(websiteIds, startDate, endDate);

  const websiteUsage = websites.map(a => ({
    websiteId: a.id,
    websiteName: a.name,
    websiteEventUsage: websiteEventUsage.find(b => a.id === b.websiteId)?.count || 0,
    eventDataUsage: eventDataUsage.find(b => a.id === b.websiteId)?.count || 0,
    deletedAt: a.deletedAt,
  }));

  const usage = websiteUsage.reduce(
    (acc, cv) => {
      acc.websiteEventUsage += cv.websiteEventUsage;
      acc.eventDataUsage += cv.eventDataUsage;

      return acc;
    },
    { websiteEventUsage: 0, eventDataUsage: 0 },
  );

  const filteredWebsiteUsage = websiteUsage.filter(
    a => !a.deletedAt && (a.websiteEventUsage > 0 || a.eventDataUsage > 0),
  );

  return json({
    ...usage,
    websites: filteredWebsiteUsage,
  });
}