Exile 1.8 - The Password Manager

// EditElementDialog.cpp : implementation file
//

#include "stdafx.h"
#include "exile.h"
#include "EditElementDialog.h"
#include "ChangeIconDialog.h"
#include "PasswordGeneratorDialog.h"
#include "Security.h"
#include "predefined.h"
#include "messagebox.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CEditElementDialog dialog


CEditElementDialog::CEditElementDialog(CWnd* pParent /*=NULL*/)
	: CDialog(CEditElementDialog::IDD, pParent)
{
	//{{AFX_DATA_INIT(CEditElementDialog)
	m_strAddress = _T("");
	m_strElementName = _T("");
	m_strLogin = _T("");
	m_strNotes = _T("");
	m_strPassword = _T("");
	//}}AFX_DATA_INIT
	m_cRating = 5;
}

CEditElementDialog::~CEditElementDialog()
{
	CleanString(m_strAddress);
	CleanString(m_strElementName);
	CleanString(m_strLogin);
	CleanString(m_strNotes);
	CleanString(m_strPassword);
}

void CEditElementDialog::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CEditElementDialog)
	DDX_Control(pDX, IDC_RATING, m_ctrlRating);
	DDX_Control(pDX, IDC_HOTKEY, m_ctrlHotKey);
	DDX_Control(pDX, IDC_ELEMENTICON, m_ctrlElementIcon);
	DDX_Text(pDX, IDC_ADDRESS, m_strAddress);
	DDX_CBString(pDX, IDC_ELEMENTNAME, m_strElementName);
	DDX_Text(pDX, IDC_LOGIN, m_strLogin);
	DDX_Text(pDX, IDC_NOTES, m_strNotes);
	DDX_Text(pDX, IDC_PASSWORD, m_strPassword);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CEditElementDialog, CDialog)
	//{{AFX_MSG_MAP(CEditElementDialog)
	ON_CBN_SELCHANGE(IDC_ELEMENTNAME, OnSelchangeElementname)
	ON_BN_CLICKED(IDC_CHANGEICON, OnChangeicon)
	ON_BN_CLICKED(IDC_PASSWORD_TITLE, OnPasswordTitle)
	ON_COMMAND(ID_PASSWORDCONTEXT_GENERATEPASSWORD, OnPasswordcontextGeneratepassword)
	ON_COMMAND(ID_PASSWORDCONTEXT_HIDEPASSWORD, OnPasswordcontextHidepassword)
	ON_COMMAND(ID_PASSWORDCONTEXT_SHOWPASSWORD, OnPasswordcontextShowpassword)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CEditElementDialog message handlers

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

	CComboBox *pCombo = (CComboBox *)GetDlgItem(IDC_ELEMENTNAME);

	for(int n = 0; n < sizeof(g_aElements) / sizeof(g_aElements[0]); ++n)
	{
		pCombo->AddString(g_aElements[n].szElementName);
	} // for

	// Ratings
	for(n = 0; n < sizeof(g_aRatings) / sizeof(g_aRatings[0]); ++n)
	{
		m_ctrlRating.InsertString(n, g_aRatings[n].szName);
		m_ctrlRating.SetItemData(n, (DWORD)g_aRatings[n].cRating);
	} // for

	m_ctrlElementIcon.SetIcon(m_pimgList->ExtractIcon(m_lIconID));
	m_ctrlRating.SetCurSel(10 - m_cRating);

	m_bPassword = TRUE;

	((CEdit *)GetDlgItem(IDC_PASSWORD))->SetPasswordChar(m_bPassword ? _T('\0') : _T('*'));

	m_ctrlPasswordLink.SubclassDlgItem(IDC_PASSWORD_TITLE, this);
	m_ctrlHotKey.SetHotKey(m_wVKCode, m_wModifiers);
		
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CEditElementDialog::OnSelchangeElementname() 
{
	CComboBox *pCombo = (CComboBox *)GetDlgItem(IDC_ELEMENTNAME);

	if(CB_ERR != pCombo->GetCurSel())
	{
		TCHAR szName[nNameLength];
		pCombo->GetLBText(pCombo->GetCurSel(), szName);

		// Look in predefined categories
		for(int n = 0; n < sizeof(g_aElements) / sizeof(g_aElements[0]); ++n)
		{
#ifdef _UNICODE
			if(0 == wcscmp(szName, g_aElements[n].szElementName))
			{
				m_strAddress = g_aElements[n].szAddress;
				m_strNotes = g_aElements[n].szNotes;
				m_lIconID = g_aElements[n].lIconID;
				m_ctrlElementIcon.SetIcon(m_pimgList->ExtractIcon(m_lIconID));

				UpdateData(FALSE);
				break;
			} // if
#else
			if(0 == strcmp(szName, g_aElements[n].szElementName))
			{
				m_strAddress = g_aElements[n].szAddress;
				m_strNotes = g_aElements[n].szNotes;
				m_lIconID = g_aElements[n].lIconID;
				m_ctrlElementIcon.SetIcon(m_pimgList->ExtractIcon(m_lIconID));
				
				UpdateData(FALSE);
				break;
			} // if
#endif // _UNICODE
		} // for
	} // if
}

void CEditElementDialog::OnChangeicon() 
{
	CChangeIconDialog dlgChange;
	dlgChange.SetImageList(m_pimgList);

	if(IDOK == dlgChange.DoModal()) 
	{
		m_lIconID = dlgChange.GetIconID();
		m_ctrlElementIcon.SetIcon(m_pimgList->ExtractIcon(m_lIconID));
	} // if	
}

void CEditElementDialog::OnPasswordTitle() 
{
	CRect rc;
	GetDlgItem(IDC_PASSWORD_TITLE)->GetWindowRect(rc);

	CMenu mCtx;
	mCtx.LoadMenu(IDR_PASSWORD_CONTEXT);

	CMenu *pmCtx = mCtx.GetSubMenu(0);

	pmCtx->EnableMenuItem(ID_PASSWORDCONTEXT_HIDEPASSWORD, MF_BYCOMMAND | 
		(m_bPassword ? MF_ENABLED : MF_DISABLED | MF_GRAYED));
	pmCtx->EnableMenuItem(ID_PASSWORDCONTEXT_SHOWPASSWORD, MF_BYCOMMAND | 
		(!m_bPassword ? MF_ENABLED : MF_DISABLED | MF_GRAYED));

	pmCtx->TrackPopupMenu(TPM_LEFTALIGN, rc.left, rc.bottom, this);
}

void CEditElementDialog::OnPasswordcontextGeneratepassword() 
{
	CPasswordGeneratorDialog dlgPassword;

	//
	// Settings
	//
	dlgPassword.m_bDecimalDigits = m_bDecimalDigits;
	dlgPassword.m_bLowercase = m_bLowercase;
	dlgPassword.m_bPunctuation = m_bPunctuation;
	dlgPassword.m_bSpecial = m_bSpecial;
	dlgPassword.m_bUppercase = m_bUppercase;
	dlgPassword.m_nPasswordLength = m_nPasswordLength;

	if(IDOK == dlgPassword.DoModal()) 
	{
		UpdateData();
		m_strPassword = dlgPassword.m_strPassword;
		UpdateData(FALSE);
		CleanString(dlgPassword.m_strPassword);

		//
		// Settings back
		//
		m_bDecimalDigits = dlgPassword.m_bDecimalDigits;
		m_bLowercase = dlgPassword.m_bLowercase;
		m_bPunctuation = dlgPassword.m_bPunctuation;
		m_bSpecial = dlgPassword.m_bSpecial;
		m_bUppercase = dlgPassword.m_bUppercase;
		m_nPasswordLength = dlgPassword.m_nPasswordLength;
	} // if
}

void CEditElementDialog::OnPasswordcontextHidepassword() 
{
	m_bPassword = FALSE;

	((CEdit *)GetDlgItem(IDC_PASSWORD))->SetPasswordChar(_T('*'));
	GetDlgItem(IDC_PASSWORD)->Invalidate();
}

void CEditElementDialog::OnPasswordcontextShowpassword() 
{
	m_bPassword = TRUE;
	
	((CEdit *)GetDlgItem(IDC_PASSWORD))->SetPasswordChar(_T(0));
	GetDlgItem(IDC_PASSWORD)->Invalidate();
}

void CEditElementDialog::OnOK() 
{
	UpdateData();

	if(m_strElementName.IsEmpty()) 
	{
		MessageBoxEx(*this, IDS_ENTERELEMENTNAME, IDS_TITLE, MB_OK | MB_ICONINFORMATION);
		return;
	} // if

	if(CB_ERR == m_ctrlRating.GetCurSel())
	{
		MessageBoxEx(*this, IDS_SELECTRATING, IDS_TITLE);
		return;
	} // if

	m_cRating = (BYTE)m_ctrlRating.GetItemData(m_ctrlRating.GetCurSel());

	m_ctrlHotKey.GetHotKey(m_wVKCode, m_wModifiers);

	CDialog::OnOK();
}