Assist_Design/apps/bff/src/subscriptions/sim-usage-store.service.ts

import { Injectable, Inject } from "@nestjs/common";
import { PrismaService } from "../common/prisma/prisma.service";
import { Logger } from "nestjs-pino";

@Injectable()
export class SimUsageStoreService {
  constructor(
    private readonly prisma: PrismaService,
    @Inject(Logger) private readonly logger: Logger
  ) {}

  private normalizeDate(date?: Date): Date {
    const d = date ? new Date(date) : new Date();
    // strip time to YYYY-MM-DD
    const iso = d.toISOString().split("T")[0];
    return new Date(iso + "T00:00:00.000Z");
  }

  async upsertToday(account: string, usageMb: number, date?: Date): Promise<void> {
    const day = this.normalizeDate(date);
    try {
      // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
      await this.prisma.simUsageDaily.upsert({
        // @ts-expect-error composite unique input type depends on Prisma schema
        where: { account_date: { account, date: day } as unknown },
        update: { usageMb },
        create: { account, date: day, usageMb },
      });
    } catch (e: unknown) {
      const message = e instanceof Error ? e.message : String(e);
      this.logger.error("Failed to upsert daily usage", { account, error: message });
    }
  }

  async getLastNDays(
    account: string,
    days = 30
  ): Promise<Array<{ date: string; usageMb: number }>> {
    const end = this.normalizeDate();
    const start = new Date(end);
    start.setUTCDate(end.getUTCDate() - (days - 1));
    // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
    const rows = (await this.prisma.simUsageDaily.findMany({
      where: { account, date: { gte: start, lte: end } },
      orderBy: { date: "desc" },
    })) as Array<{ date: Date; usageMb: number }>;
    return rows.map(r => ({ date: r.date.toISOString().split("T")[0], usageMb: r.usageMb }));
  }

  async cleanupPreviousMonths(): Promise<number> {
    const now = new Date();
    const firstOfMonth = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1));
    // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
    const result = (await this.prisma.simUsageDaily.deleteMany({
      where: { date: { lt: firstOfMonth } },
    })) as unknown as { count: number };
    return result.count;
  }
}