#include <windows.h>
#include <nsis/pluginapi.h>
#include <wuapi.h>
#include "main.h"

static const LPWSTR MicrosoftUpdateServiceID = L"7971f918-a847-4430-9279-4a52d1efe18d";

PLUGIN_METHOD(EnableMicrosoftUpdate) {
	PLUGIN_INIT();

	IUpdateServiceManager2 *serviceManager;
	IUpdateServiceRegistration *registration;

	HRESULT hr = CoCreateInstance(&CLSID_UpdateServiceManager, NULL, CLSCTX_INPROC_SERVER, &IID_IUpdateServiceManager2, (void **)&serviceManager);
	if (!SUCCEEDED(hr)) {
		goto end;
	}

	BSTR serviceID = SysAllocString(MicrosoftUpdateServiceID);
	BSTR serviceCab = SysAllocString(L"");
	hr = IUpdateServiceManager2_AddService2(serviceManager, serviceID, asfAllowPendingRegistration | asfAllowOnlineRegistration | asfRegisterServiceWithAU, serviceCab, &registration);
	SysFreeString(serviceID);
	SysFreeString(serviceCab);

end:
	if (registration) {
		IUpdateServiceManager2_Release(registration);
	}

	if (serviceManager) {
		IUpdateServiceManager2_Release(serviceManager);
	}

	pushint(hr);
}