siadrive/SiaDrive/SiaDriveDlg.cpp

// SiaDriveDlg.cpp : implementation file
//

#include "stdafx.h"
#include "SiaDriveApp.h"
#include "SiaDriveDlg.h"
#include "afxdialogex.h"
#include <stdbool.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

/* Home Test
whipped wizard hefty avatar gnome knuckle uptight jury ruling evenings urchins visited dating pioneer offend trying oyster setup maximum library mayor yearbook keep voyage excess bifocals kettle gutter aces
*/

// CAboutDlg dialog used for App About

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// Dialog Data
#ifdef AFX_DESIGN_TIME
	enum { IDD = IDD_ABOUTBOX };
#endif

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

// Implementation
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CSiaDriveDlg dialog

BEGIN_DHTML_EVENT_MAP(CSiaDriveDlg)
	DHTML_EVENT_ONCLICK(_T("ButtonOK"), OnButtonOK)
	DHTML_EVENT_ONCLICK(_T("ButtonCancel"), OnButtonCancel)
	DHTML_EVENT_ONCLICK(_T("CreateWalletButton"), OnButtonCreateWallet)
	DHTML_EVENT_ONCLICK(_T("ConfirmSeedButton"), OnButtonConfirmSeed)
END_DHTML_EVENT_MAP()


CSiaDriveDlg::CSiaDriveDlg(CWnd* pParent /*=NULL*/)
	: CDHtmlDialog(IDD_SIADRIVE_DIALOG, IDR_HTML_SIADRIVE_DIALOG, pParent),
	_siaApi({L"localhost", 9980, L"1.1.0"})
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSiaDriveDlg::DoDataExchange(CDataExchange* pDX)
{
	CDHtmlDialog::DoDataExchange(pDX);
	DDX_DHtml_ElementInnerText(pDX, _T("ServerVersion"), _serverVersion);
	DDX_DHtml_ElementInnerText(pDX, _T("ClientVersion"), _clientVersion);
	DDX_DHtml_ElementInnerText(pDX, _T("WalletBalanceTotal"), _walletBalanceTotal);
	DDX_DHtml_ElementInnerText(pDX, _T("WalletBalanceConfirmed"), _walletBalanceConfirmed);
	DDX_DHtml_ElementInnerText(pDX, _T("WalletBalanceUnconfirmed"), _walletBalanceUnconfirmed);
	DDX_DHtml_ElementInnerText(pDX, _T("WalletCreatedSeed"), _walletCreatedSeed);
}

BEGIN_MESSAGE_MAP(CSiaDriveDlg, CDHtmlDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_TIMER()
END_MESSAGE_MAP()


// CSiaDriveDlg message handlers

BOOL CSiaDriveDlg::OnInitDialog()
{
	CDHtmlDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CSiaDriveDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDHtmlDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CSiaDriveDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

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

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDHtmlDialog::OnPaint();
	}
}

// The system calls this function to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CSiaDriveDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

HRESULT CSiaDriveDlg::OnButtonOK(IHTMLElement* /*pElement*/)
{
	OnOK();
	return S_OK;
}

HRESULT CSiaDriveDlg::OnButtonCancel(IHTMLElement* /*pElement*/)
{
	OnCancel();
	return S_OK;
}

HRESULT CSiaDriveDlg::OnButtonConfirmSeed(IHTMLElement* /*pElement*/)
{
	_seedCreation = false;
	_walletCreatedSeed = L"";
	UpdateData(FALSE);
	this->Navigate(this->m_strCurrentUrl);

	return S_OK;
}

BOOL CSiaDriveDlg::PreTranslateMessage(MSG* pMsg)
{
	//TODO: Implement copy/paste context menu
	if ((pMsg->message == WM_RBUTTONDOWN) || (pMsg->message == WM_RBUTTONDBLCLK)) 
	{
		return TRUE;
	}

	return CDHtmlDialog::PreTranslateMessage(pMsg);
}

bool IsRefreshKeyMessage(const MSG *message)
{
	return message
		&& (message->message == WM_KEYUP)
		&& (message->wParam == VK_F5);
}

HRESULT CSiaDriveDlg::TranslateAccelerator(MSG *message, const GUID * guid, DWORD dw)
{
	return IsRefreshKeyMessage(message) ? E_FAIL : CDHtmlDialog::TranslateAccelerator(message, guid, dw);
}

void CSiaDriveDlg::OnDocumentComplete(LPDISPATCH, LPCTSTR)
{
	KillTimer(IDT_UPDATE);

	_serverVersion = L"x.x.x";
	_clientVersion = L"1.0.0";
	_walletBalanceConfirmed = L"";
	_walletBalanceUnconfirmed = L"";
	_walletBalanceTotal = L"";

	if ((_connected = UpdateUi(false)))
	{
		switch (_siaConfig.GetUI_Main_TabIndex())
		{
		case WALLET_TAB:
		{
			DisplayWalletTab();
		}
		break;
		}
	}
	else
	{
		UpdateData(FALSE);
	}

	SetTimer(IDT_UPDATE, 2000, nullptr);
}

void CSiaDriveDlg::OnTimer(UINT_PTR nIDEvent)
{
	switch (nIDEvent)
	{
	case IDT_UPDATE:
		{
			UpdateUi();
		}
		break;
	}
}

bool CSiaDriveDlg::DisplaySiaInfo()
{
	_serverVersion = _siaApi.GetServerVersion().c_str();
	if (_serverVersion)
	{
		SiaCurrency confirmed;
		if (API_SUCCESS(SiaApiError, _siaApi.GetWallet()->GetConfirmedBalance(confirmed)))
		{
			SiaCurrency unconfirmed;
			if (API_SUCCESS(SiaApiError, _siaApi.GetWallet()->GetUnonfirmedBalance(unconfirmed)))
			{
				SiaCurrency total = confirmed + unconfirmed;

				_walletBalanceConfirmed = SiaCurrencyToString(confirmed).c_str();
				_walletBalanceUnconfirmed = SiaCurrencyToString(unconfirmed).c_str();
				_walletBalanceTotal = SiaCurrencyToString(total).c_str();

				return true;
			}
		}
	}

	_serverVersion = L"x.x.x";
	_walletBalanceConfirmed = _walletBalanceUnconfirmed = _walletBalanceTotal = L"";

	return false;
}

bool CSiaDriveDlg::UpdateUi(const bool& refresh)
{
	bool ret = DisplaySiaInfo();
	if (ret)
	{
		if (_connected)
		{
			UpdateData(FALSE);
		}
		else if (!_seedCreation)
		{
			KillTimer(IDT_UPDATE);
			this->Navigate(this->m_strCurrentUrl);
		}
	}
	else if (refresh && !_seedCreation)
	{
		_connected = false;
		KillTimer(IDT_UPDATE);
		this->Navigate(this->m_strCurrentUrl);
	}

	return ret;
}

HRESULT CSiaDriveDlg::GetDomElementById(const CString& id, CComPtr<IHTMLDOMNode>& node)
{
	CComPtr<IHTMLElement> elem;
	return GetDomElementAndNodeById(id, node, elem);
}

HRESULT CSiaDriveDlg::GetDomElementAndNodeById(const CString& id, CComPtr<IHTMLDOMNode>& node, CComPtr<IHTMLElement>& elem)
{
	HRESULT hr;
	if (SUCCEEDED((hr = GetElement(id, &elem))))
	{
		hr = elem->QueryInterface(IID_IHTMLDOMNode, reinterpret_cast<void**>(&node));
	}

	return hr;
}

void CSiaDriveDlg::DisplayCreateWallet()
{
	CComPtr<IHTMLDOMNode> mainNode;
	if (SUCCEEDED(GetDomElementById(L"main_window", mainNode)))
	{
		CComPtr<IHTMLElement> div;
		CComPtr<IHTMLDOMNode> walletCreateNode;
		if (SUCCEEDED(GetDomElementAndNodeById(L"create_wallet", walletCreateNode, div)))
		{
			CComPtr<IHTMLDOMNode> parent;
			if (SUCCEEDED(walletCreateNode->get_parentNode(&parent)))
			{
				CComPtr<IHTMLDOMNode> removedNode;
				if (SUCCEEDED(parent->removeChild(walletCreateNode, &removedNode)))
				{
					CComPtr<IHTMLDOMNode> appendedNode;
					if (SUCCEEDED(mainNode->appendChild(removedNode, &appendedNode)))
					{
						CComPtr<IHTMLStyle> style;
						if (SUCCEEDED(div->get_style(&style)))
						{
							style->put_display(L"block");
						}
					}
				}
			}
		}
	}
}

void CSiaDriveDlg::RemoveElementById(const String& id)
{
	CComPtr<IHTMLDOMNode> element;
	if (SUCCEEDED(GetDomElementById(id.c_str(), element)))
	{
		CComPtr<IHTMLDOMNode> parent;
		if (SUCCEEDED(element->get_parentNode(&parent)))
		{
			CComPtr<IHTMLDOMNode> removed;
			parent->removeChild(element, &removed);
		}
	}
}

void CSiaDriveDlg::ClearCreateElements()
{
	RemoveElementById(L"create_wallet");
	RemoveElementById(L"disp_wallet_seed");
}

void CSiaDriveDlg::DisplayWalletTab()
{
	if (_siaApi.GetWallet()->GetCreated())
	{
		ClearCreateElements();
		if (_siaApi.GetWallet()->GetLocked())
		{
			DisplayUnlockWallet();
		}
		else
		{
		
		}
	}
	else
	{
		DisplayCreateWallet();
	}
}

void CSiaDriveDlg::DisplaySeedCreated(const String& seed)
{
	CComPtr<IHTMLDOMNode> mainNode;
	if (SUCCEEDED(GetDomElementById(L"main_window", mainNode)))
	{
		CComPtr<IHTMLElement> div;
		CComPtr<IHTMLDOMNode> dispWalletSeedNode;
		if (SUCCEEDED(GetDomElementAndNodeById(L"disp_wallet_seed", dispWalletSeedNode, div)))
		{
			CComPtr<IHTMLDOMNode> parent;
			if (SUCCEEDED(dispWalletSeedNode->get_parentNode(&parent)))
			{
				CComPtr<IHTMLDOMNode> removedNode;
				if (SUCCEEDED(parent->removeChild(dispWalletSeedNode, &removedNode)))
				{
					CComPtr<IHTMLDOMNode> child;
					if (SUCCEEDED(mainNode->get_firstChild(&child)))
					{
						CComPtr<IHTMLDOMNode> removed;
						if (SUCCEEDED(mainNode->removeChild(child, &removed)))
						{
							CComPtr<IHTMLDOMNode> appendedNode;
							if (SUCCEEDED(mainNode->appendChild(removedNode, &appendedNode)))
							{
								_walletCreatedSeed = seed.c_str();
								UpdateData(FALSE);

								CComPtr<IHTMLStyle> style;
								if (SUCCEEDED(div->get_style(&style)))
								{
									style->put_display(L"block");
								}
							}
						}
					}
				}
			}
		}
	}
}

void CSiaDriveDlg::DisplayUnlockWallet()
{
	
}

HRESULT CSiaDriveDlg::OnButtonCreateWallet(IHTMLElement* pElement)
{
	if (!_seedCreation)
	{
		_seedCreation = true;
		KillTimer(IDT_UPDATE);

		String seed;
		if (API_SUCCESS(SiaApiError, _siaApi.GetWallet()->Create(SiaSeedLanguage::English, seed)))
		{
			DisplaySeedCreated(seed);
		}
	}

	return S_OK;
}