#include "stdafx.h"

ToastNotification::ToastNotification(_In_ wchar_t* text, _In_ wchar_t* caption, _In_ wchar_t* imagePath)
	: _text(text), _caption(caption), _imagePath(imagePath)
{
}

ToastNotification::~ToastNotification()
{
}

HRESULT ToastNotification::Initialize()
{
	HRESULT hr = Windows::Foundation::GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager).Get(), &notificationManager);
	if (SUCCEEDED(hr))
	{
		hr = notificationManager->CreateToastNotifierWithId(StringReferenceWrapper(::AppUserModelID).Get(), &notifier);
		if (SUCCEEDED(hr))
		{
			hr = Create(&notification);
		}
	}
	return hr;
}

HRESULT ToastNotification::CreateXml(_Outptr_ ABI::Windows::Data::Xml::Dom::IXmlDocument** xml)
{
	Microsoft::WRL::ComPtr<ABI::Windows::Data::Xml::Dom::IXmlDocumentIO> xmlDocument;
	HRESULT hr = Windows::Foundation::ActivateInstance(StringReferenceWrapper(RuntimeClass_Windows_Data_Xml_Dom_XmlDocument).Get(), &xmlDocument);
	if (FAILED(hr))
		return hr;

	HXML xmlToast = xmlCreateNode(L"toast", NULL, 0);
	HXML xmlAudioNode = xmlAddChild(xmlToast, L"audio", NULL);
	xmlAddAttr(xmlAudioNode, L"silent", L"true");
	HXML xmlVisualNode = xmlAddChild(xmlToast, L"visual", NULL);
	HXML xmlBindingNode = xmlAddChild(xmlVisualNode, L"binding", NULL);
	if (_imagePath)
	{
		xmlAddAttr(xmlBindingNode, L"template", L"ToastImageAndText02");
		HXML xmlImageNode = xmlAddChild(xmlBindingNode, L"image", NULL);
		xmlAddAttr(xmlImageNode, L"id", L"1");
		xmlAddAttr(xmlImageNode, L"src", CMStringW(FORMAT, L"file:///%s", _imagePath));
	}
	else
	{
		xmlAddAttr(xmlBindingNode, L"template", L"ToastText02");
	}
	HXML xmlTitleNode = xmlAddChild(xmlBindingNode, L"text", _caption != NULL ? _caption : L"Miranda NG");
	xmlAddAttr(xmlTitleNode, L"id", L"1");
	if (_text)
	{
		HXML xmlTextNode = xmlAddChild(xmlBindingNode, L"text", _text);
		xmlAddAttr(xmlTextNode, L"id", L"2");
	}

	TCHAR *xtmp = xmlToString(xmlToast, NULL);
	size_t xlen = mir_tstrlen(xtmp);

	hr = xmlDocument->LoadXml(StringReferenceWrapper(xtmp, xlen).Get());
	if (FAILED(hr))
		return hr;

	return xmlDocument.CopyTo(xml);
}

HRESULT ToastNotification::Create(_Outptr_ ABI::Windows::UI::Notifications::IToastNotification** _notification)
{
	Microsoft::WRL::ComPtr<ABI::Windows::Data::Xml::Dom::IXmlDocument> xml;
	HRESULT hr = CreateXml(&xml);
	if (FAILED(hr))
		return hr;

	Microsoft::WRL::ComPtr<ABI::Windows::UI::Notifications::IToastNotificationFactory> factory;
	hr = Windows::Foundation::GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotification).Get(), &factory);
	if (FAILED(hr))
		return hr;
	return hr = factory->CreateToastNotification(xml.Get(), _notification);
}

HRESULT ToastNotification::Show()
{
	return Show(new ToastEventHandler(nullptr));
}

HRESULT ToastNotification::Show(_In_ ToastEventHandler* handler)
{
	EventRegistrationToken activatedToken, dismissedToken, failedToken;
	Microsoft::WRL::ComPtr<ToastEventHandler> eventHandler(handler);

	notification->add_Activated(eventHandler.Get(), &activatedToken);
	notification->add_Dismissed(eventHandler.Get(), &dismissedToken);
	notification->add_Failed(eventHandler.Get(), &failedToken);

	return notifier->Show(notification.Get());
}

HRESULT ToastNotification::Hide()
{
	return notifier->Hide(notification.Get());
}