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 {
      await (this.prisma as any).simUsageDaily.upsert({
        where: { account_date: { account, date: day } as any },
        update: { usageMb },
        create: { account, date: day, usageMb },
      });
    } catch (e: any) {
      this.logger.error("Failed to upsert daily usage", { account, error: e?.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));
    const rows = (await (this.prisma as any).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));
    const result = await (this.prisma as any).simUsageDaily.deleteMany({
      where: { date: { lt: firstOfMonth } },
    });
    return result.count;
  }
}