#pragma once

DEFINE_PROPERTYKEY(PKEY_AppUserModel_ID, 0x9F4C2855, 0x9F79, 0x4B39, 0xA8, 0xD0, 0xE1, 0xD4, 0x2D, 0xE1, 0xD5, 0xF3, 5);
HRESULT TryCreateShortcut();

extern wchar_t wszTempDir[];

class ToasterImage
{
	HBITMAP _hBitmap;
	ptrW tszId;

public:
	__inline explicit ToasterImage(HICON hIcon) : _hBitmap(NULL), tszId(CMStringW(FORMAT, L"%p", hIcon).Detach())
	{
		ICONINFO icon = { 0 };
		if (GetIconInfo(hIcon, &icon))
		{
			_hBitmap = icon.hbmColor;
			DeleteObject(icon.hbmMask);
		}
	}

	__inline explicit ToasterImage(HBITMAP bmp) : _hBitmap(bmp), tszId(CMStringW(FORMAT, L"%p", bmp).Detach())
	{  
	}

	__inline explicit ToasterImage(const char *szProto) : _hBitmap(NULL), tszId(mir_a2u(szProto))
	{
		ICONINFO icon = { 0 };
		if (GetIconInfo(Skin_LoadProtoIcon(szProto, ID_STATUS_ONLINE, 1), &icon))
		{
			_hBitmap = icon.hbmColor;
			DeleteObject(icon.hbmMask);
		}
	}

	__inline ~ToasterImage()
	{
		DeleteObject(_hBitmap);
	}

	wchar_t* Save()
	{
		wchar_t wszSavePath[MAX_PATH];
		mir_snwprintf(wszSavePath, L"%s\\MirandaToaster.%s.png", wszTempDir, tszId);

		if (!(GetFileAttributes(wszSavePath) < 0xFFFFFFF))
		{
			IMGSRVC_INFO isi = { sizeof(isi) };
			isi.wszName = mir_wstrdup(wszSavePath);
			isi.hbm = _hBitmap;
			isi.dwMask = IMGI_HBITMAP;
			isi.fif = FREE_IMAGE_FORMAT::FIF_PNG;
			__try
			{
				CallService(MS_IMG_SAVE, (WPARAM)&isi, IMGL_WCHAR);
			}
			__except (EXCEPTION_EXECUTE_HANDLER)
			{
				return nullptr;
			}
		}
		return mir_wstrdup(wszSavePath);
	}

	__inline operator wchar_t*()
	{	return Save();
	}
};