// AlarmDlg.cpp: 实现文件
//

#include "stdafx.h"
#include "Servo.h"
#include "afxdialogex.h"
#include "PageAlarm.h"
#include "AlarmManager.h"
#include "Common.h"
#include <iomanip>

#define PAGE_SIZE						100
#define PAGE_BACKGROUND_COLOR			RGB(252, 252, 255)

// CPageAlarm 对话框

IMPLEMENT_DYNAMIC(CPageAlarm, CDialogEx)

CPageAlarm::CPageAlarm(CWnd* pParent /*=nullptr*/)
	: CDialogEx(IDD_DIALOG_ALARM, pParent)
{
	m_crBkgnd = PAGE_BACKGROUND_COLOR;
	m_hbrBkgnd = nullptr;
	m_pObserver = nullptr;

	m_nSeverityLevel = 0;
	m_strDeviceName = "";
	m_strUnitName = "";
	m_strKeyword = "";
	m_nCurPage = 0;
	m_nTotalPages = 0;
	m_nDateTimeFlag = 0;

	memset(m_szTimeStart, 0, sizeof(m_szTimeStart));
	memset(m_szTimeEnd, 0, sizeof(m_szTimeEnd));
	m_szTimeStart[0] = '\0';
	m_szTimeEnd[0] = '\0';
}

CPageAlarm::~CPageAlarm()
{
}

void CPageAlarm::InitRxWindow()
{
	/* code */
	// 订阅数据
	IRxWindows* pRxWindows = RX_GetRxWindows();
	pRxWindows->enableLog(5);
	if (m_pObserver == NULL) {
		m_pObserver = pRxWindows->allocObserver([&](IAny* pAny) -> void {
			// onNext
			pAny->addRef();
			int code = pAny->getCode();

			if (RX_CODE_ALARM_SET == code) {
				UpdatePageData();
			}
			else if (RX_CODE_ALARM_CLEAR == code) {
				UpdatePageData();
			}

		pAny->release();
		}, [&]() -> void {
			// onComplete
		}, [&](IThrowable* pThrowable) -> void {
			// onErrorm
			pThrowable->printf();
		});

		theApp.m_model.getObservable()->observeOn(pRxWindows->mainThread())
			->subscribe(m_pObserver);
	}
}

void CPageAlarm::Resize()
{
	CWnd* pItem;
	CRect rcClient;
	GetClientRect(&rcClient);

	pItem = GetDlgItem(IDC_LIST_ALARM);
	pItem->MoveWindow(12, 58, rcClient.Width() - 24, rcClient.Height() - 64);
}

void CPageAlarm::LoadAlarms()
{
	// 刷新历史报警数据
	m_nCurPage = 1;
	UpdatePageData();
}

void CPageAlarm::UpdatePageData()
{
	// 根据过滤条件加载数据，提供描述和时间范围查询
	auto vecData = AlarmManager::getInstance().getFilteredAlarms("", "", m_strDeviceName, m_strUnitName, m_strKeyword, m_szTimeStart, m_szTimeEnd, m_nCurPage, PAGE_SIZE);

	// 填充数据到控件
	CListCtrl* pListCtrl = (CListCtrl*)GetDlgItem(IDC_LIST_ALARM);
	FillDataToListCtrl(pListCtrl, vecData);

	// 更新分页控件（页数和按钮）
	UpdatePageControls();
}

void CPageAlarm::UpdatePageControls()
{
	// 更新分页信息
	CString strPage;
	strPage.Format(_T("第 %d 页"), m_nCurPage);
	SetDlgItemText(IDC_LABEL_PAGE_NUMBER, strPage);

	// 启用/禁用翻页按钮
	GetDlgItem(IDC_BUTTON_PREV_PAGE)->EnableWindow(m_nCurPage > 1);
	GetDlgItem(IDC_BUTTON_NEXT_PAGE)->EnableWindow(m_nCurPage < m_nTotalPages);
}

void CPageAlarm::FillDataToListCtrl(CListCtrl* pListCtrl, const std::vector<AlarmData>& vecData)
{
	if (pListCtrl == nullptr || pListCtrl->m_hWnd == nullptr) {
		return;
	}

	// 清空当前CListCtrl中的所有项
	pListCtrl->DeleteAllItems();

	// 遍历数据并插入到CListCtrl中
	for (const auto& alarm : vecData) {
		InsertAlarmData(pListCtrl, alarm);
	}

	// 获取列数
	int nColCount = pListCtrl->GetHeaderCtrl()->GetItemCount();
	pListCtrl->SetColumnWidth(nColCount - 1, LVSCW_AUTOSIZE_USEHEADER);
}

static char* pszAlarmLevel[] = { "Warning", "Error" };
void CPageAlarm::InsertAlarmData(CListCtrl* pListCtrl, const AlarmData& alarmData)
{
	if (pListCtrl == nullptr || pListCtrl->m_hWnd == nullptr) {
		return;
	}

	int nRowCount = pListCtrl->GetItemCount();
	if (nRowCount >= PAGE_SIZE) {
		pListCtrl->DeleteItem(nRowCount - 1);
	}

	int nNewItem = pListCtrl->InsertItem(0, _T(""));

	// 设置每一列的数据
	CString str;
	str.Format(_T("%d"), alarmData.nId);
	pListCtrl->SetItemText(nNewItem, 1, str);											// 报警ID
	pListCtrl->SetItemText(nNewItem, 2, pszAlarmLevel[alarmData.nSeverityLevel % 2]);	// 等级
	pListCtrl->SetItemText(nNewItem, 3, alarmData.strDeviceName.c_str());               // 设备名称
	pListCtrl->SetItemText(nNewItem, 4, alarmData.strUnitName.c_str());                 // 单元名称
	pListCtrl->SetItemText(nNewItem, 5, alarmData.strStartTime.c_str());                // 开始时间
	pListCtrl->SetItemText(nNewItem, 6, alarmData.strEndTime.c_str());                  // 结束时间
	pListCtrl->SetItemText(nNewItem, 7, alarmData.strDescription.c_str());              // 描述
}

std::string CPageAlarm::getCurrentTimeString()
{
	auto now = std::chrono::system_clock::now();
	auto time_t_now = std::chrono::system_clock::to_time_t(now);

	std::tm tm_now = {};
	localtime_s(&tm_now, &time_t_now);

	std::stringstream ss;
	ss << std::put_time(&tm_now, "%Y-%m-%d %H:%M:%S");
	return ss.str();
}

void CPageAlarm::DoDataExchange(CDataExchange* pDX)
{
	DDX_Control(pDX, IDC_DATETIMEPICKER_START, m_dateTimeStart);
	DDX_Control(pDX, IDC_DATETIMEPICKER_END, m_dateTimeEnd);
	DDX_Control(pDX, IDC_LIST_ALARM, m_listCtrl);
	CDialogEx::DoDataExchange(pDX);
}


BEGIN_MESSAGE_MAP(CPageAlarm, CDialogEx)
	ON_WM_CTLCOLOR()
	ON_WM_DESTROY()
	ON_WM_CLOSE()
	ON_WM_SIZE()
	ON_CBN_SELCHANGE(IDC_COMBO_DATETIME, &CPageAlarm::OnCbnSelchangeComboDatetime)
	ON_BN_CLICKED(IDC_BUTTON_SEARCH, &CPageAlarm::OnBnClickedButtonSearch)
	ON_BN_CLICKED(IDC_BUTTON_EXPORT, &CPageAlarm::OnBnClickedButtonExport)
	ON_BN_CLICKED(IDC_BUTTON_PREV_PAGE, &CPageAlarm::OnBnClickedButtonPrevPage)
	ON_BN_CLICKED(IDC_BUTTON_NEXT_PAGE, &CPageAlarm::OnBnClickedButtonNextPage)
END_MESSAGE_MAP()


// CAlarmDlg 消息处理程序
BOOL CPageAlarm::OnInitDialog()
{
	CDialogEx::OnInitDialog();
	SetTimer(1, 3000, nullptr);

	// 下拉框控件
	CComboBox* pComboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DATETIME);
	pComboBox->AddString(_T("不限"));
	pComboBox->AddString(_T("今天"));
	pComboBox->AddString(_T("七天内"));
	pComboBox->AddString(_T("本月"));
	pComboBox->AddString(_T("今年"));
	pComboBox->AddString(_T("自定义"));
	pComboBox->SetCurSel(0);

	// 日期控件
	m_dateTimeStart.EnableWindow(FALSE);
	m_dateTimeEnd.EnableWindow(FALSE);


	// 读出列宽
	CString strIniFile, strItem;
	strIniFile.Format(_T("%s\\configuration.ini"), (LPTSTR)(LPCTSTR)theApp.m_strAppDir);
	int width[8] = { 0, 80, 180, 80, 80, 100, 80, 180};
	for (int i = 0; i < 8; i++) {
		strItem.Format(_T("Col_%d_Width"), i);
		width[i] = GetPrivateProfileInt("AlarmListCtrl", strItem, width[i], strIniFile);
	}


	// 报表控件
	CListCtrl* pListCtrl = (CListCtrl*)GetDlgItem(IDC_LIST_ALARM);
	DWORD dwStyle = pListCtrl->GetExtendedStyle();
	dwStyle |= LVS_EX_FULLROWSELECT;
	dwStyle |= LVS_EX_GRIDLINES;
	pListCtrl->SetExtendedStyle(dwStyle);

	HIMAGELIST imageList = ImageList_Create(24, 24, ILC_COLOR24, 1, 1);
	ListView_SetImageList(pListCtrl->GetSafeHwnd(), imageList, LVSIL_SMALL);
	pListCtrl->InsertColumn(0, _T(""), LVCFMT_RIGHT, width[0]);
	pListCtrl->InsertColumn(1, _T("警告ID"), LVCFMT_LEFT, width[1]);
	pListCtrl->InsertColumn(2, _T("报警等级"), LVCFMT_LEFT, width[2]);
	pListCtrl->InsertColumn(3, _T("设备名称"), LVCFMT_LEFT, width[3]);
	pListCtrl->InsertColumn(4, _T("单元名称"), LVCFMT_LEFT, width[4]);
	pListCtrl->InsertColumn(5, _T("发生时间"), LVCFMT_LEFT, width[5]);
	pListCtrl->InsertColumn(6, _T("解除时间"), LVCFMT_LEFT, width[6]);
	pListCtrl->InsertColumn(7, _T("描述"), LVCFMT_LEFT, width[7]);
	pListCtrl->SetColumnWidth(7, LVSCW_AUTOSIZE_USEHEADER);


	// 计算总页数
	int totalRecords = AlarmManager::getInstance().getTotalAlarmCount("", "", m_strDeviceName, m_strUnitName, m_strKeyword, m_szTimeStart, m_szTimeEnd);
	m_nTotalPages = (totalRecords + PAGE_SIZE - 1) / PAGE_SIZE;
	m_nCurPage = 1;

	Resize();
	LoadAlarms();

	return TRUE;  // return TRUE unless you set the focus to a control
	// 异常: OCX 属性页应返回 FALSE
}

HBRUSH CPageAlarm::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
{
	HBRUSH hbr = CDialogEx::OnCtlColor(pDC, pWnd, nCtlColor);

	if (nCtlColor == CTLCOLOR_STATIC) {
		pDC->SetBkColor(m_crBkgnd);
	}

	if (m_hbrBkgnd == nullptr) {
		m_hbrBkgnd = CreateSolidBrush(m_crBkgnd);
	}

	return m_hbrBkgnd;
}

void CPageAlarm::OnDestroy()
{
	CDialogEx::OnDestroy();


	// 保存列宽
	CString strIniFile, strItem, strTemp;
	strIniFile.Format(_T("%s\\configuration.ini"), (LPTSTR)(LPCTSTR)theApp.m_strAppDir);
	CHeaderCtrl* pHeader = m_listCtrl.GetHeaderCtrl();
	for (int i = 0; i < pHeader->GetItemCount(); i++) {
		RECT rect;
		pHeader->GetItemRect(i, &rect);
		strItem.Format(_T("Col_%d_Width"), i);
		strTemp.Format(_T("%d"), rect.right - rect.left);
		WritePrivateProfileString("AlarmListCtrl", strItem, strTemp, strIniFile);
	}


	if (m_hbrBkgnd != nullptr) {
		::DeleteObject(m_hbrBkgnd);
	}

	if (m_pObserver != NULL) {
		m_pObserver->unsubscribe();
		m_pObserver = NULL;
	}
}

void CPageAlarm::OnClose()
{
	ShowWindow(SW_HIDE);
	GetParent()->PostMessage(ID_MSG_ALARMDLG_HIDE, 0, 0);
}

void CPageAlarm::OnSize(UINT nType, int cx, int cy)
{
	CDialogEx::OnSize(nType, cx, cy);
	if (GetDlgItem(IDC_LIST_ALARM) == nullptr) return;
	Resize();
}

void CPageAlarm::OnCbnSelchangeComboDatetime()
{
	CComboBox* pComboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DATETIME);
	int nIndex = pComboBox->GetCurSel();
	int nCount = pComboBox->GetCount();
	m_dateTimeStart.EnableWindow(nIndex == nCount - 1);
	m_dateTimeEnd.EnableWindow(nIndex == nCount - 1);
}

void CPageAlarm::OnBnClickedButtonSearch()
{
	// 获取关键字
	CString cstrKeyword;
	GetDlgItemText(IDC_EDIT_KEYWORD, cstrKeyword);
	m_strKeyword = CT2A(cstrKeyword);

	// 获取日期
	CComboBox* pComboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DATETIME);
	m_nDateTimeFlag = pComboBox->GetCurSel();
	if (m_nDateTimeFlag == 0) {
		memset(m_szTimeStart, 0, sizeof(m_szTimeStart));
		memset(m_szTimeEnd, 0, sizeof(m_szTimeEnd));
		m_szTimeStart[0] = '\0';
		m_szTimeEnd[0] = '\0';
	}
	else {
		CTime time = CTime::GetCurrentTime();
		if (m_nDateTimeFlag == 1) {
			// 今天
			sprintf_s(m_szTimeStart, 64, "%d-%02d-%02d 00:00:00", time.GetYear(), time.GetMonth(), time.GetDay());
			sprintf_s(m_szTimeEnd, 64, "%d-%02d-%02d 23:59:59", time.GetYear(), time.GetMonth(), time.GetDay());
		}
		else if (m_nDateTimeFlag == 2) {
			// 7天内
			CTime time2 = time - CTimeSpan(7, 0, 0, 0);
			sprintf_s(m_szTimeStart, 64, "%d-%02d-%02d 00:00:00", time2.GetYear(), time2.GetMonth(), time2.GetDay());
			sprintf_s(m_szTimeEnd, 64, "%d-%02d-%02d 23:59:59", time.GetYear(), time.GetMonth(), time.GetDay());
		}
		else if (m_nDateTimeFlag == 3) {
			// 本月
			sprintf_s(m_szTimeStart, 64, "%d-%02d-01 00:00:00", time.GetYear(), time.GetMonth());
			sprintf_s(m_szTimeEnd, 64, "%d-%02d-%02d 23:59:59", time.GetYear(), time.GetMonth(), time.GetDay());
		}
		else if (m_nDateTimeFlag == 4) {
			// 今年
			sprintf_s(m_szTimeStart, 64, "%d-01-01 00:00:00", time.GetYear());
			sprintf_s(m_szTimeEnd, 64, "%d-12-31 23:59:59", time.GetYear());
		}
		else if (m_nDateTimeFlag == 5) {
			// 自定义
			SYSTEMTIME t1, t2;
			m_dateTimeStart.GetTime(&t1);
			m_dateTimeEnd.GetTime(&t2);

			sprintf_s(m_szTimeStart, 64, "%d-%02d-%02d %02d:%02d:%02d",
				t1.wYear, t1.wMonth, t1.wDay, t1.wHour, t1.wMinute, t1.wSecond);
			sprintf_s(m_szTimeEnd, 64, "%d-%02d-%02d %02d:%02d:%02d",
				t2.wYear, t2.wMonth, t2.wDay, t2.wHour, t2.wMinute, t2.wSecond);
		}
	}

	// 计算总页数
	int totalRecords = AlarmManager::getInstance().getTotalAlarmCount("", "", m_strDeviceName, m_strUnitName, m_strKeyword, m_szTimeStart, m_szTimeEnd);
	m_nTotalPages = (totalRecords + PAGE_SIZE - 1) / PAGE_SIZE;
	m_nCurPage = 1;

	UpdatePageData();  // 调用分页更新函数
}

void CPageAlarm::OnBnClickedButtonExport()
{
	CFileDialog fileDialog(FALSE, "csv", "", OFN_HIDEREADONLY, "csv.files(*.csv)|*.csv||");
	if (fileDialog.DoModal() != IDOK) {
		return;
	}

	CStdioFile file;
	if (!file.Open(fileDialog.GetPathName(), CFile::modeCreate | CFile::modeWrite)) {
		AfxMessageBox("创建文件失败！");
	}

	int nSubItemCount = 0;
	CString strSubItem, strHeader, strRow;
	char szItem[256];
	HDITEM hdItem[35];
	for (int i = 0; i < 35; i++) {
		hdItem[i].pszText = szItem;
		hdItem[i].cchTextMax = 256;
		hdItem[i].mask = HDI_TEXT | HDI_WIDTH;
	}

	// 先读表头
	CListCtrl* pListCtrl = (CListCtrl*)GetDlgItem(IDC_LIST_ALARM);
	CHeaderCtrl* pHeader = pListCtrl->GetHeaderCtrl();
	nSubItemCount = pHeader->GetItemCount();
	ASSERT(nSubItemCount <= 35);
	for (int i = 0; i < pHeader->GetItemCount(); i++) {
		pHeader->GetItem(i, &hdItem[i]);
		if (hdItem[i].cxy > 0) {
			if (!strHeader.IsEmpty()) {
				strHeader.Append(",");
			}
			strHeader.Append(CString(hdItem[i].pszText));
		}
	}
	strHeader.Append("\n");
	file.WriteString(strHeader);

	// 写表格内容
	int count = pListCtrl->GetItemCount();
	for (int i = 0; i < count; i++) {
		strRow.Empty();
		for (int j = 0; j < nSubItemCount; j++) {
			if (hdItem[j].cxy > 0) {
				if (!strRow.IsEmpty()) {
					strRow.Append(",");
				}
				CString strTemp = pListCtrl->GetItemText(i, j);
				strTemp.Replace("* ", "");
				strRow.Append(strTemp);
			}
		}
		strRow.Append("\n");
		file.WriteString(strRow);
	}

	file.Close();
}

void CPageAlarm::OnBnClickedButtonPrevPage()
{
	// 点击上一页
	m_nCurPage--;
	UpdatePageData();  // 调用分页更新函数
}

void CPageAlarm::OnBnClickedButtonNextPage()
{
	// 点击下一页
	m_nCurPage++;
	UpdatePageData();  // 调用分页更新函数
}