import type { PageResult } from '@vben/request';
import { requestClient } from '#/api/request';

export type EnergyType = 'ELECTRICITY' | 'STEAM' | 'WATER';
export interface EnergyRecord {
  id?: number;
  recordDate: string;
  energyType: EnergyType;
  usageArea: string;
  productCategory: string;
  usagePurpose: string;
  quantity: number;
  unitPrice: number;
  amount?: number;
  recorder: string;
  remark?: string;
  createTime?: string;
}
export interface EnergyFilters {
  startDate?: string;
  endDate?: string;
  energyType?: EnergyType;
  usageArea?: string;
  productCategory?: string;
}
export interface EnergySummary {
  energyType: EnergyType;
  quantity: number;
  amount: number;
  recordCount: number;
}
export interface EnergyBucket extends EnergySummary { label: string }
export interface EnergyAnalysis {
  totalAmount: number;
  recordCount: number;
  summary: EnergySummary[];
  trend: EnergyBucket[];
  areas: EnergyBucket[];
}
export function getEnergyRecordPage(params: EnergyFilters & { pageNo: number; pageSize: number }) {
  return requestClient.get<PageResult<EnergyRecord>>('/energy/record/page', { params });
}
export function getEnergyRecord(id: number) {
  return requestClient.get<EnergyRecord>('/energy/record/get', { params: { id } });
}
export function createEnergyRecord(data: EnergyRecord) {
  return requestClient.post<number>('/energy/record/create', data);
}
export function updateEnergyRecord(data: EnergyRecord) {
  return requestClient.put('/energy/record/update', data);
}
export function deleteEnergyRecord(id: number) {
  return requestClient.delete('/energy/record/delete', { params: { id } });
}
export function getEnergyAnalysis(params: EnergyFilters & { startDate: string; endDate: string; groupBy: 'DAY' | 'MONTH' }) {
  return requestClient.get<EnergyAnalysis>('/energy/record/analysis', { params });
}

export function exportEnergyRecords(params: EnergyFilters) {
  return requestClient.download('/energy/record/export-excel', { params });
}
export function exportEnergyAnalysis(params: EnergyFilters & { startDate: string; endDate: string; groupBy: 'DAY' | 'MONTH' }) {
  return requestClient.download('/energy/record/analysis-export', { params });
}
export type EnergyAlertMetric = 'AMOUNT' | 'QUANTITY' | 'MONTH_AMOUNT' | 'MONTH_QUANTITY';
export interface EnergyAlertRule {
  id?: number;
  name: string;
  energyType: EnergyType;
  usageArea?: string;
  productCategory?: string;
  metric: EnergyAlertMetric;
  threshold: number;
  enabled: boolean;
  recipientIds: number[];
}
export interface EnergyAlertLog {
  id: number; ruleName: string; recordDate: string; energyType: EnergyType;
  scope: string; metric: EnergyAlertMetric; periodType: 'DAY' | 'MONTH'; actualValue: number; threshold: number;
  recipientName: string; status: 'FAILED' | 'SENT'; errorMessage: string; createTime: string;
}
export async function getEnergyAlertRules(params: { pageNo: number; pageSize: number }) {
  const result = await requestClient.get<PageResult<Omit<EnergyAlertRule, 'recipientIds'> & { recipientIds: string }>>('/energy/alert/page', { params });
  return { ...result, list: result.list.map(row => ({ ...row, recipientIds: row.recipientIds.split(',').filter(Boolean).map(Number) })) };
}
export function getEnergyAlertLogs(params: { pageNo: number; pageSize: number; date?: string; status?: string }) {
  return requestClient.get<PageResult<EnergyAlertLog>>('/energy/alert/logs', { params });
}
export function getEnergyAlertRecipients() {
  return requestClient.get<{ id: number; nickname: string }[]>('/energy/alert/recipients');
}
export function saveEnergyAlertRule(data: EnergyAlertRule) {
  return data.id ? requestClient.put('/energy/alert/update', data) : requestClient.post('/energy/alert/create', data);
}
export function deleteEnergyAlertRule(id: number) { return requestClient.delete('/energy/alert/delete', { params: { id } }); }
export function checkEnergyAlerts(date: string) { return requestClient.post<{ sent: number; failed: number }>('/energy/alert/check', undefined, { params: { date } }); }