/*
 * evolution-windows-sens.c
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with the program; if not, see <http://www.gnu.org/licenses/>
 *
 */

#ifdef __cplusplus
extern "C" {
#endif

#define INITGUID
#include <windows.h>
#include <eventsys.h>
#include <sensevts.h>
#include <rpc.h>

#include <shell/e-shell.h>
#include <e-util/e-extension.h>

#define NUM_ELEMENTS(x) (sizeof((x)) / sizeof((x)[0]))

/* 4E14FB9F-2E22-11D1-9964-00C04FBBB345 */
DEFINE_GUID(IID_IEventSystem, 0x4E14FB9F, 0x2E22, 0x11D1, 0x99, 0x64, 0x00, 0xC0, 0x4F, 0xBB, 0xB3, 0x45);

/* 4A6B0E15-2E38-11D1-9965-00C04FBBB345 */
DEFINE_GUID(IID_IEventSubscription, 0x4A6B0E15, 0x2E38, 0x11D1, 0x99, 0x65, 0x00, 0xC0, 0x4F, 0xBB, 0xB3, 0x45);

/* d597bab1-5b9f-11d1-8dd2-00aa004abd5e */
DEFINE_GUID(IID_ISensNetwork, 0xd597bab1, 0x5b9f, 0x11d1, 0x8d, 0xd2, 0x00, 0xaa, 0x00, 0x4a, 0xbd, 0x5e);

/* 4E14FBA2-2E22-11D1-9964-00C04FBBB345 */
DEFINE_GUID(CLSID_CEventSystem, 0x4E14FBA2, 0x2E22, 0x11D1, 0x99, 0x64, 0x00, 0xC0, 0x4F, 0xBB, 0xB3, 0x45);

/* 7542e960-79c7-11d1-88f9-0080c7d771bf */
DEFINE_GUID(CLSID_CEventSubscription, 0x7542e960, 0x79c7, 0x11d1, 0x88, 0xf9, 0x00, 0x80, 0xc7, 0xd7, 0x71, 0xbf);


/* Standard GObject macros */
#define E_TYPE_WINDOWS_SENS \
	(e_windows_sens_get_type ())
#define E_WINDOWS_SENS(obj) \
	(G_TYPE_CHECK_INSTANCE_CAST \
	((obj), E_TYPE_WINDOWS_SENS, EWindowsSENS))

typedef struct _EWindowsSENS EWindowsSENS;
typedef struct _EWindowsSENSClass EWindowsSENSClass;

struct _EWindowsSENS {
	EExtension parent;
};

struct _EWindowsSENSClass {
	EExtensionClass parent_class;
};

/* Module Entry Points */
void e_module_load (GTypeModule *type_module);
void e_module_unload (GTypeModule *type_module);

/* Forward Declarations */
GType e_windows_sens_get_type (void);

G_DEFINE_DYNAMIC_TYPE (EWindowsSENS, e_windows_sens, E_TYPE_EXTENSION)

static EShell *
windows_sens_get_shell (EWindowsSENS *extension)
{
	EExtensible *extensible;

	extensible = e_extension_get_extensible (E_EXTENSION (extension));

	return E_SHELL (extensible);
}

/* Object to receive the ISensNetwork events */

typedef struct ESensNetworkListener {
	ISensNetworkVtbl *lpVtbl;
	long ref;
	EWindowsSENS *ews_ptr;
} ESensNetworkListener;

static void e_sens_network_listener_init(ESensNetworkListener**,EWindowsSENS*);

/* Functions to implement ISensNetwork interface */

static HRESULT WINAPI e_sens_network_listener_queryinterface (ISensNetwork*,REFIID,void**);
static ULONG WINAPI e_sens_network_listener_addref (ISensNetwork*);
static ULONG WINAPI e_sens_network_listener_release (ISensNetwork*);
static HRESULT WINAPI e_sens_network_listener_gettypeinfocount (ISensNetwork*, UINT*);
static HRESULT WINAPI e_sens_network_listener_gettypeinfo (ISensNetwork*,UINT,LCID,ITypeInfo**);
static HRESULT WINAPI e_sens_network_listener_getidsofnames (ISensNetwork*,REFIID,LPOLESTR*,UINT,LCID, DISPID*);
static HRESULT WINAPI e_sens_network_listener_invoke (ISensNetwork*,DISPID,REFIID,LCID,WORD,DISPPARAMS*,VARIANT*,EXCEPINFO*,UINT*);
static HRESULT WINAPI e_sens_network_listener_connectionmade (ISensNetwork*,BSTR,ULONG,LPSENS_QOCINFO);
static HRESULT WINAPI e_sens_network_listener_connectionmadenoqocinfo (ISensNetwork*,BSTR,ULONG);
static HRESULT WINAPI e_sens_network_listener_connectionlost (ISensNetwork*,BSTR,ULONG);
static HRESULT WINAPI e_sens_network_listener_destinationreachable (ISensNetwork*,BSTR,BSTR,ULONG,LPSENS_QOCINFO);
static HRESULT WINAPI e_sens_network_listener_destinationreachablenoqocinfo (ISensNetwork*,BSTR,BSTR,ULONG);

/* Initializing the VTable of our ESensNetworkListener object */

static ISensNetworkVtbl ESensNetworkListenerVtbl = {
	e_sens_network_listener_queryinterface,
	e_sens_network_listener_addref,
	e_sens_network_listener_release,
	e_sens_network_listener_gettypeinfocount,
	e_sens_network_listener_gettypeinfo,
	e_sens_network_listener_getidsofnames,
	e_sens_network_listener_invoke,
	e_sens_network_listener_connectionmade,
	e_sens_network_listener_connectionmadenoqocinfo,
	e_sens_network_listener_connectionlost,
	e_sens_network_listener_destinationreachable,
	e_sens_network_listener_destinationreachablenoqocinfo
};


static HRESULT WINAPI
e_sens_network_listener_queryinterface (ISensNetwork *This,
                                        REFIID        iid,
                                        void        **ppv)
{
	if (IsEqualIID (iid, &IID_IUnknown) || IsEqualIID (iid, &IID_IDispatch) || IsEqualIID (iid, &IID_ISensNetwork)) {
		*ppv = This;
		((LPUNKNOWN)*ppv)->lpVtbl->AddRef((LPUNKNOWN)*ppv);
		return S_OK;
	}
	*ppv = NULL;
	return E_NOINTERFACE;
}

static ULONG WINAPI
e_sens_network_listener_addref (ISensNetwork *This)
{
	ESensNetworkListener *esnl_ptr=(ESensNetworkListener*)This;
	return InterlockedIncrement(&(esnl_ptr->ref));
}

static ULONG WINAPI
e_sens_network_listener_release (ISensNetwork *This)
{
	ESensNetworkListener *esnl_ptr=(ESensNetworkListener*)This;
	ULONG tmp = InterlockedDecrement(&(esnl_ptr->ref));
	return tmp;
}

static HRESULT WINAPI
e_sens_network_listener_gettypeinfocount (ISensNetwork *This,
                                          UINT         *pctinfo)
{
	return E_NOTIMPL;
}

static HRESULT WINAPI
e_sens_network_listener_gettypeinfo (ISensNetwork *This,
                                     UINT          iTInfo,
                                     LCID          lcid,
                                     ITypeInfo   **ppTInfo)
{
	return E_NOTIMPL;
}

static HRESULT WINAPI
e_sens_network_listener_getidsofnames (ISensNetwork *This,
                                       REFIID        riid,
                                       LPOLESTR     *rgszNames,
                                       UINT          cNames,
                                       LCID          lcid,
                                       DISPID       *rgDispId)
{
	return E_NOTIMPL;
}

static HRESULT WINAPI
e_sens_network_listener_invoke (ISensNetwork *This,
								DISPID        dispIdMember,
								REFIID        riid,
								LCID          lcid,
								WORD          wFlags,
								DISPPARAMS   *pDispParams,
								VARIANT      *pVarResult,
								EXCEPINFO    *pExcepInfo,
								UINT         *puArgErr)
{
	return E_NOTIMPL;
}

static HRESULT WINAPI
e_sens_network_listener_connectionmade (ISensNetwork  *This,
                                        BSTR           bstrConnection,
                                        ULONG          ulType,
                                        LPSENS_QOCINFO lpQOCInfo)
{
	if (ulType) {
		ESensNetworkListener *esnl_ptr=(ESensNetworkListener*)This;
		EShell *shell = windows_sens_get_shell (esnl_ptr->ews_ptr);
		/* Wait a second so that the connection stabilizes */
		g_usleep(G_USEC_PER_SEC);
		e_shell_set_network_available (shell, TRUE);
	}
	return S_OK;
}

static HRESULT WINAPI
e_sens_network_listener_connectionmadenoqocinfo (ISensNetwork *This, 
                                                 BSTR          bstrConnection,
                                                 ULONG         ulType)
{
	//Always followed by ConnectionMade
	return S_OK;
}

static HRESULT WINAPI
e_sens_network_listener_connectionlost (ISensNetwork *This,
                                        BSTR          bstrConnection,
                                        ULONG         ulType)
{
	if (ulType) {
		ESensNetworkListener *esnl_ptr=(ESensNetworkListener*)This;
		EShell *shell = windows_sens_get_shell (esnl_ptr->ews_ptr);
		e_shell_set_network_available (shell, FALSE);
	}
	return S_OK;
}

static HRESULT WINAPI
e_sens_network_listener_destinationreachable (ISensNetwork  *This,
                                              BSTR           bstrDestination,
                                              BSTR           bstrConnection,
                                              ULONG          ulType,
                                              LPSENS_QOCINFO lpQOCInfo)
{
	if (ulType) {
		ESensNetworkListener *esnl_ptr=(ESensNetworkListener*)This;
		EShell *shell = windows_sens_get_shell (esnl_ptr->ews_ptr);
		/* Wait a second so that the connection stabilizes */
		g_usleep(G_USEC_PER_SEC);
		e_shell_set_network_available (shell, TRUE);
	}
	return S_OK;
}

static HRESULT WINAPI
e_sens_network_listener_destinationreachablenoqocinfo (ISensNetwork *This,
                                                       BSTR          bstrDestination,
                                                       BSTR          bstrConnection,
                                                       ULONG         ulType)
{
	return S_OK;
}

static void
e_sens_network_listener_init(ESensNetworkListener **esnl_ptr,
                             EWindowsSENS          *ews)
{
	(*esnl_ptr) = g_new0(ESensNetworkListener,1);
	(*esnl_ptr)->lpVtbl = &ESensNetworkListenerVtbl;
	(*esnl_ptr)->ews_ptr = ews;
	(*esnl_ptr)->ref = 1;
}


static BSTR
_mb2wchar (const char* a)
{
	static WCHAR b[64];
	MultiByteToWideChar (0, 0, a, -1, b, 64);
	return b;
}

static const char* add_curly_braces_to_uuid (const char* string_uuid)
{
	static char curly_braced_uuid_string[64];
	int i;
	if (!string_uuid)
		return NULL;
	lstrcpy(curly_braced_uuid_string,"{");
	i = strlen(curly_braced_uuid_string);
	lstrcat(curly_braced_uuid_string+i,string_uuid);
	i = strlen(curly_braced_uuid_string);
	lstrcat(curly_braced_uuid_string+i,"}");
	return curly_braced_uuid_string;
}	
	
static void
windows_sens_constructed (GObject *object)
{
	HRESULT res;
	static IEventSystem *pEventSystem =0;
	static IEventSubscription* pEventSubscription = 0;
	static ESensNetworkListener *pESensNetworkListener = 0;
	static const char* eventclassid="{D5978620-5B9F-11D1-8DD2-00AA004ABD5E}";
	static const char* methods[]={
		"ConnectionMade",
		"ConnectionMadeNoQOCInfo",
		"ConnectionLost",
		"DestinationReachable",
		"DestinationReachableNoQOCInfo"
	};
	static const char* names[]={
		"EWS_ConnectionMade",
		"EWS_ConnectionMadeNoQOCInfo",
		"EWS_ConnectionLost",
		"EWS_DestinationReachable",
		"EWS_DestinationReachableNoQOCInfo"
	};
	unsigned char* subids[] = { 0, 0, 0, 0, 0 };

	EWindowsSENS *extension = (E_WINDOWS_SENS (object));
	e_sens_network_listener_init(&pESensNetworkListener, extension);

	CoInitialize(0);

	res=CoCreateInstance (&CLSID_CEventSystem, 0,CLSCTX_SERVER,&IID_IEventSystem,(LPVOID*)&pEventSystem);

	if (res == S_OK && pEventSystem) {

		unsigned i;

		for (i=0; i<NUM_ELEMENTS(methods); i++) {

			res=CoCreateInstance (&CLSID_CEventSubscription, 0, CLSCTX_SERVER, &IID_IEventSubscription, (LPVOID*)&pEventSubscription);

			if (res == S_OK && pEventSubscription) {
				UUID tmp_uuid;
				UuidCreate(&tmp_uuid);
				UuidToString(&tmp_uuid, &subids[i]);
				res=pEventSubscription->lpVtbl->put_SubscriptionID (pEventSubscription, _mb2wchar (add_curly_braces_to_uuid ((char*)subids[i])));
				if (res) {
					RpcStringFree (&subids[i]);
					break;
				}
				RpcStringFree (&subids[i]);
				res=pEventSubscription->lpVtbl->put_SubscriptionName (pEventSubscription, _mb2wchar (names[i]));
				if (res)
					break;
				res=pEventSubscription->lpVtbl->put_MethodName (pEventSubscription, _mb2wchar (methods[i]));
				if (res)
					break;
				res=pEventSubscription->lpVtbl->put_EventClassID (pEventSubscription, _mb2wchar (eventclassid));
				if (res)
					break;
				res=pEventSubscription->lpVtbl->put_SubscriberInterface (pEventSubscription, (IUnknown*)pESensNetworkListener);
				if (res)
					break;
				/* Make the subscription receive the event only if the owner of the subscription
				 * is logged on to the same computer as the publisher. This makes this module
				 * work with normal user account without administrative privileges.
				 */
				res=pEventSubscription->lpVtbl->put_PerUser (pEventSubscription, TRUE);
				if (res)
					break;

				res=pEventSystem->lpVtbl->Store (pEventSystem, (BSTR)PROGID_EventSubscription, (IUnknown*)pEventSubscription);
				if (res)
					break;
				pEventSubscription->lpVtbl->Release (pEventSubscription);
				pEventSubscription=0;
			}
		}
		if (pEventSubscription)
			pEventSubscription->lpVtbl->Release(pEventSubscription);
	}
	
	/* Do not try to get initial state when we are sure we will not get system events.
	 * Like that we don't get stuck with Disconnected status if we were disconnected
	 * on start.
	 */
	if (res == S_OK) {

		typedef BOOL (WINAPI* IsNetworkAlive_t) (LPDWORD);
		BOOL alive = TRUE;
		EShell *shell = windows_sens_get_shell (extension);

		IsNetworkAlive_t pIsNetworkAlive = NULL;

		HMODULE hDLL=LoadLibrary ("sensapi.dll");

		if ((pIsNetworkAlive=(IsNetworkAlive_t) GetProcAddress (hDLL, "IsNetworkAlive"))) {
			DWORD Network;
			alive=pIsNetworkAlive (&Network);
		}

		FreeLibrary(hDLL);

		e_shell_set_network_available (shell, alive);
	}
}

static void
e_windows_sens_class_init (EWindowsSENSClass *_class)
{
	GObjectClass *object_class;
	EExtensionClass *extension_class;

	object_class = G_OBJECT_CLASS (_class);
	object_class->constructed = windows_sens_constructed;

	extension_class = E_EXTENSION_CLASS (_class);
	extension_class->extensible_type = E_TYPE_SHELL;
}

static void
e_windows_sens_class_finalize (EWindowsSENSClass *_class)
{
}

static void
e_windows_sens_init (EWindowsSENS *extension)
{
}

G_MODULE_EXPORT void
e_module_load (GTypeModule *type_module)
{
	e_windows_sens_register_type (type_module);
}

G_MODULE_EXPORT void
e_module_unload (GTypeModule *type_module)
{
}

#ifdef __cplusplus
}
#endif