Client/Server/ToolProject/WebInfoManager/WebInfoManagerDlg.cpp

// WebInfoManagerDlg.cpp : 구현 파일
//

#include "stdafx.h"
#include "WebInfoManager.h"
#include "WebInfoManagerDlg.h"
#include ".\webinfomanagerdlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CWebInfoManagerDlg 대화 상자

const CHAR g_szSetupFileName[] = "./WebInfoManager.ini";

const CString CECListCtrlEx::GetToolTip(int, int, UINT nFlags, BOOL&)
{
	CString str = "";

//	if (nFlags == LVHT_ONITEMICON) VERIFY(str.LoadString(IDS_FLAG));

	return str;
}


CWebInfoManagerDlg::CWebInfoManagerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWebInfoManagerDlg::IDD, pParent)
{

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CWebInfoManagerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_INFO, m_InfoTitle);
	DDX_Control(pDX, IDC_LIST1, m_ServerListCtrl);
}

BEGIN_MESSAGE_MAP(CWebInfoManagerDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
	ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
END_MESSAGE_MAP()


// CWebInfoManagerDlg 메시지 처리기
bool ReadStringFromReg(const TCHAR *FileName_In, const TCHAR *Section_In, 
					   const TCHAR *KeyName_In, CString& Value_Out, int nMaxBuffer = 512)
{
	TCHAR* szBuffer = 
		reinterpret_cast<TCHAR*>(_alloca(nMaxBuffer * sizeof(TCHAR)));

	if (0 <= GetPrivateProfileString(Section_In, KeyName_In, "", szBuffer, nMaxBuffer, FileName_In))
	{
		Value_Out.SetString(szBuffer);
		return true;
	}

	return false;
}

BOOL CWebInfoManagerDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 이 대화 상자의 아이콘을 설정합니다. 응용 프로그램의 주 창이 대화 상자가 아닐 경우에는
	// 프레임워크가 이 작업을 자동으로 수행합니다.
	SetIcon(m_hIcon, TRUE);			// 큰 아이콘을 설정합니다.
	SetIcon(m_hIcon, FALSE);		// 작은 아이콘을 설정합니다.

	// TODO: 여기에 추가 초기화 작업을 추가합니다.
	ReadStringFromReg(g_szSetupFileName, "DB", "IP", m_IP);
	ReadStringFromReg(g_szSetupFileName, "DB", "NAME", m_NAME);
	ReadStringFromReg(g_szSetupFileName, "DB", "USER", m_USER);
	ReadStringFromReg(g_szSetupFileName, "DB", "PASSWORD", m_PASS);

	m_InitFlag = FALSE;

	if(!m_DBComp.Connect(m_IP, m_NAME, m_USER, m_PASS))
		AfxMessageBox("DB접근 실패");
	else 
		m_InitFlag = TRUE;


	CString title;
	if(m_InitFlag)
		ReadStringFromReg(g_szSetupFileName, "STRING", "MSG_O01", title);
	else
		ReadStringFromReg(g_szSetupFileName, "STRING", "MSG_O02", title);

	m_InfoTitle.SetWindowText(title);


	m_imglstFlags.Create(16, 16, ILC_MASK, 3, 0);

	m_imglstFlags.Add(theApp.LoadIcon(IDI_CLOSE));
	m_imglstFlags.Add(theApp.LoadIcon(IDI_CLOSE));
	m_imglstFlags.Add(theApp.LoadIcon(IDI_OPEN));


	m_ServerListCtrl.SetExtendedStyle(
		LVS_EX_FULLROWSELECT | LVS_EX_INFOTIP | 
		LVS_EX_LABELTIP | LVS_EX_CHECKBOXES);

	m_ServerListCtrl.SetImageList(&m_imglstFlags, LVSIL_SMALL);

	m_ServerListCtrl.InsertColumn(0, "ID", LVCFMT_LEFT,  200);
	m_ServerListCtrl.EnableColumnHiding(0, true);

	//	m_ServerListCtrl.InsertColumn(1, "Status", LVCFMT_LEFT, 100);

	UpdateServerList();

	return TRUE;  // 컨트롤에 대한 포커스를 설정하지 않을 경우 TRUE를 반환합니다.
}

void CWebInfoManagerDlg::UpdateServerList()
{
	///////////////////////////////////////////////////////////////////////////////////
	// 서버 상태 얻기.
	m_vetServer.clear();

	char szQuery[1024];
	strcpy(szQuery, "SELECT id, server, status FROM ServerStatus");

	if (!m_DBComp.ExecuteQuery(szQuery)) 
	{
		return;
	}

	SERVERSTATUS info[10];

	while(true)
	{
		int iCount = 0;
		if( !m_DBComp.GetData((void**)&info, sizeof(SERVERSTATUS), 10, &iCount) )
			break;

		for(int i = 0; i < iCount; ++i)
			m_vetServer.push_back(info[i]);

		if(iCount == 0)
			break;
	}

	m_ServerListCtrl.DeleteAllItems();

	CString str;
	int iCount = m_vetServer.size();
	for(int i = 0; i < iCount; ++i)
	{
		m_ServerListCtrl.InsertItem(
			LVIF_IMAGE | LVIF_PARAM | LVIF_TEXT, i, m_vetServer[i].name,
			INDEXTOSTATEIMAGEMASK(1), LVIS_STATEIMAGEMASK, m_vetServer[i].status+1, 0);
	}
}

// 대화 상자에 최소화 단추를 추가할 경우 아이콘을 그리려면 
// 아래 코드가 필요합니다. 문서/뷰 모델을 사용하는 MFC 응용 프로그램의 경우에는
// 프레임워크에서 이 작업을 자동으로 수행합니다.

void CWebInfoManagerDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 그리기를 위한 디바이스 컨텍스트

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 클라이언트 사각형에서 아이콘을 가운데에 맞춥니다.
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 아이콘을 그립니다.
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// 사용자가 최소화된 창을 끄는 동안에 커서가 표시되도록 시스템에서
//  이 함수를 호출합니다. 
HCURSOR CWebInfoManagerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CWebInfoManagerDlg::OnBnClickedButton1()
{
	// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
	int iCount = m_vetServer.size();

	char szQuery[1024];

	for(int i = 0; i < iCount; ++i)
	{
		if(m_ServerListCtrl.GetCheck(i))
		{
			// 쿼리 전송
			sprintf(szQuery, "UPDATE ServerStatus SET [status] = 1 WHERE id = %d", m_vetServer[i].id);
			m_DBComp.ExecuteQuery(szQuery);
		}
	}

	UpdateServerList();
}

void CWebInfoManagerDlg::OnBnClickedButton2()
{
	// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
	int iCount = m_vetServer.size();

	char szQuery[1024];

	for(int i = 0; i < iCount; ++i)
	{
		if(m_ServerListCtrl.GetCheck(i))
		{
			// 쿼리 전송
			sprintf(szQuery, "UPDATE ServerStatus SET [status] = 0 WHERE id = %d", m_vetServer[i].id);
			m_DBComp.ExecuteQuery(szQuery);
		}
	}

	UpdateServerList();
}