Merge branch 'liuyang' into clh

LAPTOP-SNT8I5JK\Boounion

2024-12-02 e5d448c95b18b46068453fa95527f1de1f7f0d45

 SourceCode/Bond/BondEq/UserManagerDlg.cpp

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,498 @@
// UserManagerDlg.cpp: å®žçŽ°æ–‡ä»¶
//

#include "stdafx.h"
#include "BondEq.h"
#include "afxdialogex.h"
#include "UserManagerDlg.h"
#include "UserManager.h"
#include "NewCellTypes/GridCellCombo.h"
#include "NewCellTypes/GridCellNumeric.h"


// CUserManagerDlg å¯¹è¯æ¡†

IMPLEMENT_DYNAMIC(CUserManagerDlg, CDialogEx)

CUserManagerDlg::CUserManagerDlg(CWnd* pParent /*=nullptr*/)
   : CDialogEx(IDD_DIALOG_USER_MANAGER, pParent)
{

}

CUserManagerDlg::~CUserManagerDlg()
{
}

void CUserManagerDlg::DoDataExchange(CDataExchange* pDX)
{
   DDX_Control(pDX, IDC_CUSTOM_USER, m_gridUserManager);
   CDialogEx::DoDataExchange(pDX);
}

void CUserManagerDlg::InitUserManager()
{
   if (m_gridUserManager.GetSafeHwnd() == NULL)
      return;

   int nRows = 1;
   int nCols = 7;

   int nFixRows = 1;
   int nFixCols = 0;
   int nRowIdx = 0;
   int nColIdx = 0;

   m_gridUserManager.DeleteAllItems();
   m_gridUserManager.SetVirtualMode(FALSE);
   m_gridUserManager.GetDefaultCell(TRUE, FALSE)->SetBackClr(g_nGridFixCellColor);
   m_gridUserManager.GetDefaultCell(FALSE, TRUE)->SetBackClr(g_nGridFixCellColor);
   m_gridUserManager.GetDefaultCell(FALSE, FALSE)->SetBackClr(g_nGridCellColor);
   m_gridUserManager.SetFixedTextColor(g_nGridFixFontColor);

   m_gridUserManager.SetRowCount(nRows);
   m_gridUserManager.SetColumnCount(nCols);
   m_gridUserManager.SetFixedRowCount(nFixRows);
   m_gridUserManager.SetFixedColumnCount(nFixCols);

   // Col
   m_gridUserManager.SetColumnWidth(nColIdx, 90);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("No."));
   m_gridUserManager.SetColumnWidth(nColIdx, 100);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("用户名"));
   m_gridUserManager.SetColumnWidth(nColIdx, 70);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("密码"));
   m_gridUserManager.SetColumnWidth(nColIdx, 70);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("权限"));
   m_gridUserManager.SetColumnWidth(nColIdx, 70);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("会话超时（分钟）"));
   m_gridUserManager.SetColumnWidth(nColIdx, 70);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("会话过期（小时）"));
   m_gridUserManager.SetColumnWidth(nColIdx, 70);
   m_gridUserManager.SetItemText(nRowIdx, nColIdx++, _T("最后一次登录时间"));

   m_gridUserManager.SetFixedRowSelection(FALSE);
   m_gridUserManager.SetFixedColumnSelection(FALSE);
   m_gridUserManager.EnableSelection(TRUE);
   m_gridUserManager.SetEditable(TRUE);
   m_gridUserManager.SetRowResize(FALSE);
   m_gridUserManager.SetColumnResize(FALSE);
   m_gridUserManager.ExpandColumnsToFit(TRUE);

   FillUserManager();
}

void CUserManagerDlg::FillUserManager()
{
   UserManager& userManager = UserManager::getInstance();
   if (!userManager.isLoggedIn()) {
      AfxMessageBox(_T("未登录"));
      return;
   }

   if (userManager.getCurrentUserRole() != UserRole::SuperAdmin) {
      AfxMessageBox(_T("非管理员用户"));
      return;
   }

   std::vector<std::vector<std::string>> usersData = userManager.getUsers();
   if (!usersData.empty()) {
      m_gridUserManager.SetRowCount(usersData.size() + 1);

      for (size_t i = 0; i < usersData.size(); i++) {
         int nRowIdx = i + 1;
         int nColIdx = 0;

         m_gridUserManager.SetItemText(nRowIdx, nColIdx++, std::to_string(i + 1).c_str());
         for (size_t j = 0; j < usersData[i].size(); j++) {
            if (usersData[i][1].empty()) {
               continue;
            }
            m_gridUserManager.SetItemText(nRowIdx, nColIdx++, usersData[i][j].c_str());
         }
      }

      m_gridUserManager.Invalidate();
      m_gridUserManager.UpdateWindow();
   }

   CStringArray permissions;
   permissions.Add(_T("管理员"));
   permissions.Add(_T("工程师"));
   permissions.Add(_T("操作员"));

   int nCols = m_gridUserManager.GetColumnCount();
   for (int i = 1; i < m_gridUserManager.GetRowCount(); ++i) {
      m_gridUserManager.SetItemState(i, 0, GVIS_READONLY); // ç¬¬ä¸€åˆ—只读
      m_gridUserManager.SetItemState(i, nCols - 1, GVIS_READONLY); // æœ€åŽä¸€åˆ—只读

      // ç¬¬å››åˆ—设置权限列为下拉框
      if (m_gridUserManager.SetCellType(i, 3, RUNTIME_CLASS(CGridCellCombo))) {
         CGridCellCombo* pCell = static_cast<CGridCellCombo*>(m_gridUserManager.GetCell(i, 3));
         pCell->SetOptions(permissions);
         pCell->SetStyle(CBS_DROPDOWNLIST);

         CString cstrRole = m_gridUserManager.GetItemText(i, 3);
         int nRole = _ttoi(cstrRole);
         if (nRole < 0 || nRole > 2) {
            CString cstrMessage;
            cstrMessage.Format(_T("用户 [%s]，权限异常！将设置成操作员！"), m_gridUserManager.GetItemText(i, 1));
            AfxMessageBox(cstrMessage);
            nRole = 2;
         }

         m_gridUserManager.SetItemText(i, 3, permissions.GetAt(nRole));
      }

      // ç¬¬äº”、六列（会话超时）设置为Numeric
      m_gridUserManager.SetCellType(i, 4, RUNTIME_CLASS(CGridCellNumeric));
      m_gridUserManager.SetCellType(i, 5, RUNTIME_CLASS(CGridCellNumeric));
   }
}

void CUserManagerDlg::AddRow(CGridCtrl* pGridCtrl)
{
   if (!pGridCtrl) return;

   if (pGridCtrl->GetRowCount() <= 0 || pGridCtrl->GetColumnCount() <= 0) {
      AfxMessageBox(_T("网格控件未正确初始化，请检查初始化逻辑！"));
      return;
   }

   int nRowCount = pGridCtrl->GetRowCount();
   pGridCtrl->SetRowCount(nRowCount + 1);
   int newRowIndex = nRowCount;

   CString strText;
   strText.Format(_T("%d"), newRowIndex);
   pGridCtrl->SetItemText(newRowIndex, 0, strText);
   strText.Format(_T("User%d"), newRowIndex);
   pGridCtrl->SetItemText(newRowIndex, 1, strText);
   pGridCtrl->SetItemText(newRowIndex, 2, _T("123456"));
   pGridCtrl->SetItemText(newRowIndex, 3, _T("操作员"));
   pGridCtrl->SetItemText(newRowIndex, 4, _T("30"));
   pGridCtrl->SetItemText(newRowIndex, 5, _T("72"));
   pGridCtrl->SetItemText(newRowIndex, 6, _T("2024-01-01 00:00:00"));

   int nCols = m_gridUserManager.GetColumnCount();
   m_gridUserManager.SetItemState(newRowIndex, 0, GVIS_READONLY); // ç¬¬ä¸€åˆ—
   m_gridUserManager.SetItemState(newRowIndex, nCols - 1, GVIS_READONLY); // æœ€åŽä¸€åˆ—

   // ç¬¬å››åˆ—设置（权限列）为下拉框
   CStringArray permissions;
   permissions.Add(_T("管理员"));
   permissions.Add(_T("工程师"));
   permissions.Add(_T("操作员"));

   if (m_gridUserManager.SetCellType(newRowIndex, 3, RUNTIME_CLASS(CGridCellCombo))) {
      CGridCellCombo* pCell = static_cast<CGridCellCombo*>(m_gridUserManager.GetCell(newRowIndex, 3));
      pCell->SetOptions(permissions);
      pCell->SetStyle(CBS_DROPDOWNLIST);

      m_gridUserManager.SetItemText(newRowIndex, 3, permissions.GetAt(2));
   }

   // ç¬¬äº”、六列（会话超时）设置为Numeric
   m_gridUserManager.SetCellType(newRowIndex, 4, RUNTIME_CLASS(CGridCellNumeric));
   m_gridUserManager.SetCellType(newRowIndex, 5, RUNTIME_CLASS(CGridCellNumeric));

   pGridCtrl->Invalidate();
   pGridCtrl->UpdateWindow();
}

void CUserManagerDlg::DeleteSelectedRow(CGridCtrl* pGridCtrl)
{
   if (!pGridCtrl) return;
   CCellRange selectedRange = pGridCtrl->GetSelectedCellRange();
   if (selectedRange.IsValid()) {
      CString currentUser = UserManager::getInstance().getCurrentUser().c_str();
      std::vector<int> rowsToDelete;

      for (int row = selectedRange.GetMinRow(); row <= selectedRange.GetMaxRow(); ++row) {
         BOOL isRowSelected = FALSE;
         for (int col = selectedRange.GetMinCol(); col <= selectedRange.GetMaxCol(); ++col) {
            if (pGridCtrl->IsCellSelected(row, col)) {
               isRowSelected = TRUE;
               break;
            }
         }

         if (!isRowSelected) {
            continue;
         }

         CString selectedUser = pGridCtrl->GetItemText(row, 1);
         if (selectedUser == currentUser) {
            CString message;
            message.Format(_T("用户 [%s] æ˜¯å½“前登录用户，不能删除！"), currentUser);
            AfxMessageBox(message, MB_OK | MB_ICONEXCLAMATION);
            return;
         }

         rowsToDelete.push_back(row);
      }

      if (rowsToDelete.empty()) {
         AfxMessageBox(_T("请先选择要删除的行！"), MB_OK | MB_ICONINFORMATION);
         return;
      }

      CString message;
      if (rowsToDelete.size() == 1) {
         CString selectedUser = pGridCtrl->GetItemText(rowsToDelete[0], 1);
         message.Format(_T("确定要删除选中用户 [%s] å—？"), selectedUser);
      }
      else {
         message.Format(_T("确定要删除选中的 %d ä¸ªç”¨æˆ·å—？"), rowsToDelete.size());
      }

      if (AfxMessageBox(message, MB_YESNO | MB_ICONQUESTION) == IDYES) {
         for (auto it = rowsToDelete.rbegin(); it != rowsToDelete.rend(); ++it) {
            pGridCtrl->DeleteRow(*it);
         }

         pGridCtrl->Invalidate();
         pGridCtrl->UpdateWindow();
      }
   }
   else {
      AfxMessageBox(_T("请先选择要删除的用户！"), MB_OK | MB_ICONINFORMATION);
   }
}

bool CUserManagerDlg::IsUsernameDuplicate(const CString& username)
{
   for (int row = 1; row < m_gridUserManager.GetRowCount(); ++row) {
      if (m_gridUserManager.GetItemText(row, 1) == username) {
         return true;
      }
   }
   return false;
}

void CUserManagerDlg::AdjustControls(int nWidth, int nHeight)
{
   CWnd* pWnd = GetWindow(GW_CHILD);
   while (pWnd) {
      UINT nCtrlID = pWnd->GetDlgCtrlID();

      CRect ctrlRect;
      pWnd->GetWindowRect(&ctrlRect);
      ScreenToClient(&ctrlRect);

      // è®¡ç®—控件的新位置和大小，按比例调整
      int newX = (int)(ctrlRect.left * (nWidth / (float)m_nInitialWidth));
      int newY = (int)(ctrlRect.top * (nHeight / (float)m_nInitialHeight));
      int newWidth = (int)(ctrlRect.Width() * (nWidth / (float)m_nInitialWidth));
      int newHeight = (int)(ctrlRect.Height() * (nHeight / (float)m_nInitialHeight));

      TCHAR szClassName[256];
      GetClassName(pWnd->m_hWnd, szClassName, sizeof(szClassName));


      if (_tcsicmp(szClassName, _T("MFCGridCtrl")) == 0) {
         CGridCtrl* pGridCtrl = (CGridCtrl*)pWnd;
         pGridCtrl->SetDefCellHeight(newHeight / 20);
         pGridCtrl->ExpandColumnsToFit(TRUE);
         pGridCtrl->Invalidate();
      }

      pWnd->MoveWindow(newX, newY, newWidth, newHeight);
      AdjustControlFont(pWnd, newWidth, newHeight);

      // èŽ·å–下一个控件
      pWnd = pWnd->GetNextWindow();
   }
}

void CUserManagerDlg::AdjustControlFont(CWnd* pWnd, int nWidth, int nHeight)
{
   TCHAR szClassName[256];
   GetClassName(pWnd->m_hWnd, szClassName, sizeof(szClassName));

   if (_tcsicmp(szClassName, _T("Static")) == 0) {
      CStatic* pStatic = (CStatic*)pWnd;
      pStatic->ModifyStyle(0, SS_CENTER | SS_CENTERIMAGE);
      return;
   }

   if (_tcsicmp(szClassName, _T("MFCGridCtrl")) == 0) {
      return;
   }

   int fontSize = nHeight - 10;
   CFont* pCurrentFont = pWnd->GetFont();
   LOGFONT logFont;
   pCurrentFont->GetLogFont(&logFont);
   logFont.lfHeight = -fontSize;

   CFont newFont;
   newFont.CreateFontIndirect(&logFont);

   pWnd->SetFont(&newFont);
   pWnd->Invalidate();
}

BEGIN_MESSAGE_MAP(CUserManagerDlg, CDialogEx)
   ON_WM_SIZE()
   ON_BN_CLICKED(IDC_BUTTON_ADD, &CUserManagerDlg::OnBnClickedButtonAdd)
   ON_BN_CLICKED(IDC_BUTTON_DEL, &CUserManagerDlg::OnBnClickedButtonDel)
   ON_BN_CLICKED(IDOK, &CUserManagerDlg::OnBnClickedOk)
   ON_BN_CLICKED(IDC_BUTTON_INSERT, &CUserManagerDlg::OnBnClickedButtonInsert)
END_MESSAGE_MAP()


// CUserManagerDlg æ¶ˆæ¯å¤„理程序


BOOL CUserManagerDlg::OnInitDialog()
{
   CDialogEx::OnInitDialog();

   // TODO:  åœ¨æ­¤æ·»åŠ é¢å¤–的初始化
   CRect screenRect, dlgRect, clientRect;
   SetWindowText(_T("用户管理"));
   SystemParametersInfo(SPI_GETWORKAREA, 0, &screenRect, 0);

   GetClientRect(&clientRect);
   m_nInitialWidth = clientRect.Width();
   m_nInitialHeight = clientRect.Height();

   GetWindowRect(&dlgRect);
   int dlgWidth = dlgRect.Width() * 3;
   int dlgHeight = dlgRect.Height() * 3;

   if (dlgWidth > screenRect.Width()) {
      dlgWidth = screenRect.Width();
   }
   if (dlgHeight > screenRect.Height()) {
      dlgHeight = screenRect.Height();
   }

   int centerX = screenRect.left + (screenRect.Width() - dlgWidth) / 2;
   int centerY = screenRect.top + (screenRect.Height() - dlgHeight) / 2;
   MoveWindow(centerX, centerY, dlgWidth, dlgHeight);

   // åˆå§‹åŒ–用户管理表格
   InitUserManager();

   return TRUE;  // return TRUE unless you set the focus to a control
   // å¼‚常: OCX å±žæ€§é¡µåº”返回 FALSE
}

void CUserManagerDlg::OnSize(UINT nType, int cx, int cy)
{
   CDialogEx::OnSize(nType, cx, cy);

   // TODO: åœ¨æ­¤å¤„添加消息处理程序代码
   CRect rect;
   GetClientRect(&rect);

   // éåŽ†å¯¹è¯æ¡†ä¸­çš„所有控件
   AdjustControls(rect.Width(), rect.Height());
}

void CUserManagerDlg::OnBnClickedButtonAdd()
{
   // TODO: åœ¨æ­¤æ·»åŠ æŽ§ä»¶é€šçŸ¥å¤„理程序代码
   AddRow(&m_gridUserManager);
}

void CUserManagerDlg::OnBnClickedButtonInsert()
{
   // TODO: åœ¨æ­¤æ·»åŠ æŽ§ä»¶é€šçŸ¥å¤„理程序代码
   if (!m_gridUserManager.GetSafeHwnd()) return;

   CCellRange selectedRange = m_gridUserManager.GetSelectedCellRange();
   if (!selectedRange.IsValid()) {
      AfxMessageBox(_T("请先选择要插入的位置！"), MB_OK | MB_ICONINFORMATION);
      return;
   }

   int minRow = selectedRange.GetMinRow();
   int maxRow = selectedRange.GetMaxRow();
   if (minRow < 1) {
      AfxMessageBox(_T("请选择有效的行！"), MB_OK | MB_ICONEXCLAMATION);
      return;
   }

   for (int row = maxRow; row >= minRow; --row) {
      std::vector<CString> rowData;
      for (int col = 0; col < m_gridUserManager.GetColumnCount(); ++col) {
         rowData.push_back(m_gridUserManager.GetItemText(row, col));
      }

      m_gridUserManager.InsertRow(_T("新用户"), row);
      CString newUsername = rowData[1];
      int suffix = 1;
      while (IsUsernameDuplicate(newUsername)) {
         newUsername.Format(_T("%s_%d"), rowData[1], suffix++);
      }
      rowData[1] = newUsername;

      for (int col = 0; col < m_gridUserManager.GetColumnCount(); ++col) {
         m_gridUserManager.SetItemText(row, col, rowData[col]);
      }

      CStringArray permissions;
      permissions.Add(_T("管理员"));
      permissions.Add(_T("工程师"));
      permissions.Add(_T("操作员"));

      if (m_gridUserManager.SetCellType(row, 3, RUNTIME_CLASS(CGridCellCombo))) {
         CGridCellCombo* pCell = static_cast<CGridCellCombo*>(m_gridUserManager.GetCell(row, 3));
         pCell->SetOptions(permissions);
         pCell->SetStyle(CBS_DROPDOWNLIST);

         m_gridUserManager.SetItemText(row, 3, rowData[3]);
      }

      m_gridUserManager.SetCellType(row, 4, RUNTIME_CLASS(CGridCellNumeric));
      m_gridUserManager.SetCellType(row, 5, RUNTIME_CLASS(CGridCellNumeric));
   }

   for (int row = 1; row < m_gridUserManager.GetRowCount(); ++row) {
      CString strIndex;
      strIndex.Format(_T("%d"), row);
      m_gridUserManager.SetItemText(row, 0, strIndex);
   }

   m_gridUserManager.Invalidate();
   m_gridUserManager.UpdateWindow();
}

void CUserManagerDlg::OnBnClickedButtonDel()
{
   // TODO: åœ¨æ­¤æ·»åŠ æŽ§ä»¶é€šçŸ¥å¤„理程序代码
   DeleteSelectedRow(&m_gridUserManager);
}

void CUserManagerDlg::OnBnClickedOk()
{
   // TODO: åœ¨æ­¤æ·»åŠ æŽ§ä»¶é€šçŸ¥å¤„理程序代码
   std::vector<std::vector<std::string>> vecData;
   for (int i = 1; i < m_gridUserManager.GetRowCount(); ++i) {
      std::vector<std::string> rowData;
      for (int j = 1; j < m_gridUserManager.GetColumnCount(); ++j) {
         CString cellText = m_gridUserManager.GetItemText(i, j);
         std::string cellString = CT2A(cellText.GetString());
         if (j == 3) {
            if (cellText == _T("管理员"))
               cellString = "0";
            else if (cellText == _T("工程师"))
               cellString = "1";
            else if (cellText == _T("操作员"))
               cellString = "2";
            else
               cellString = "2";
         }

         rowData.push_back(cellString);
      }

      vecData.push_back(rowData);
   }

   UserManager::getInstance().setUsers(vecData);
   CDialogEx::OnOK();
}