aboutsummaryrefslogtreecommitdiff
path: root/Src/Plugins/Library/ml_webdev/forceUrl.cpp
blob: 90b55742ccf322937c5624d0657190a0776e9011 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#include "main.h"
#include "./forceUrl.h"

typedef struct __FORCEURLDATA
{
	UINT	serviceId;
	LPWSTR	url;
} FORCEURLDATA;

#define FORCEURLPROP		L"MLWEBDEV_FORCEURL"

void ForceUrl_Remove()
{
	HWND hLibrary = Plugin_GetLibrary();
	if (NULL == hLibrary) return;

	FORCEURLDATA *data = (FORCEURLDATA*)GetProp(hLibrary, FORCEURLPROP);
	RemoveProp(hLibrary, FORCEURLPROP);
	if (NULL != data)
	{
		Plugin_FreeString(data->url);
		free(data);
	}
}

HRESULT ForceUrl_Set(UINT serviceId, LPCWSTR pszUrl)
{
	if (NULL == pszUrl) return E_INVALIDARG;

	HWND hLibrary = Plugin_GetLibrary();
	if (NULL == hLibrary) return E_FAIL;

	FORCEURLDATA *data = (FORCEURLDATA*)GetProp(hLibrary, FORCEURLPROP);

	if (NULL != data)
	{
		Plugin_FreeString(data->url);
		if (data->serviceId != serviceId)
		{
			free(data);
			data = NULL;
		}
	}

	if (NULL == data)
	{
		data = (FORCEURLDATA*)malloc(sizeof(FORCEURLDATA));
		if (NULL == data) return E_OUTOFMEMORY;
		data->serviceId = serviceId;
	}
	
	data->url = Plugin_CopyString(pszUrl);
	if (NULL == data->url || FALSE == SetProp(hLibrary, FORCEURLPROP, data))
	{
		ForceUrl_Remove();
		return E_FAIL;
	}

	return S_OK;
}

HRESULT ForceUrl_Get(UINT serviceId, const wchar_t **ppszUrl)
{
	if (NULL == ppszUrl) return E_POINTER;
	*ppszUrl = NULL;

	HWND hLibrary = Plugin_GetLibrary();
	if (NULL == hLibrary) return E_FAIL;

	FORCEURLDATA *data = (FORCEURLDATA*)GetProp(hLibrary, FORCEURLPROP);

	if (NULL == data || data->serviceId != serviceId)
		return E_NOINTERFACE;

    *ppszUrl = data->url;
	return S_OK;
}