import { badRequest } from "../errors";
import { billingAccountFor } from "../metering/access";
import { periodOf } from "../metering/record";
import { usageSummary } from "../metering/summary";
import type { ApiContext } from "./http";

const PERIOD = /^\d{4}-(0[1-9]|1[0-2])$/;

function periodParam(ctx: ApiContext): string {
  const period = ctx.url.searchParams.get("period") ?? periodOf();
  if (!PERIOD.test(period)) throw badRequest("period must be YYYY-MM");
  return period;
}

export async function getUserUsage(ctx: ApiContext): Promise<Response> {
  const account = await billingAccountFor(ctx.viewer, null);
  return Response.json(await usageSummary(account.accountId, periodParam(ctx)));
}

export async function getOrgUsage(ctx: ApiContext, org: string): Promise<Response> {
  let name: string;
  try {
    name = decodeURIComponent(org);
  } catch {
    throw badRequest("Invalid organization name");
  }
  const account = await billingAccountFor(ctx.viewer, name);
  return Response.json(await usageSummary(account.accountId, periodParam(ctx)));
}