#ifndef ___VBoxCredential_h

#define ___VBoxCredential_h

#include <windows.h>

#include <strsafe.h>

#include <shlguid.h>

#include "helpers.h"

#include "dll.h"

#include "resource.h"

class VBoxCredProv;

class VBoxCredential : public ICredentialProviderCredential

{

public:

VBoxCredential(VBoxCredProv *pProvider);

virtual ~VBoxCredential();

public:

// IUnknown

STDMETHOD_(ULONG, AddRef)()

{

return m_cRef++;

}

STDMETHOD_(ULONG, Release)()

{

LONG cRef = m_cRef--;

if (!cRef)

{

delete this;

}

return cRef;

}

STDMETHOD (QueryInterface)(REFIID riid, void** ppv)

{

HRESULT hr;

if (ppv != NULL)

{

if (IID_IUnknown == riid ||

IID_ICredentialProviderCredential == riid)

{

*ppv = static_cast<IUnknown*>(this);

reinterpret_cast<IUnknown*>(*ppv)->AddRef();

hr = S_OK;

}

else

{

*ppv = NULL;

hr = E_NOINTERFACE;

}

}

else

{

hr = E_INVALIDARG;

}

return hr;

}

public:

// ICredentialProviderCredential

IFACEMETHODIMP Advise(ICredentialProviderCredentialEvents* pcpce);

IFACEMETHODIMP UnAdvise();

IFACEMETHODIMP SetSelected(BOOL* pbAutoLogon);

IFACEMETHODIMP SetDeselected();

IFACEMETHODIMP GetFieldState(DWORD dwFieldID,

CREDENTIAL_PROVIDER_FIELD_STATE* pcpfs,

CREDENTIAL_PROVIDER_FIELD_INTERACTIVE_STATE* pcpfis);

IFACEMETHODIMP GetStringValue(DWORD dwFieldID, PWSTR* ppwsz);

IFACEMETHODIMP GetBitmapValue(DWORD dwFieldID, HBITMAP* phbmp);

IFACEMETHODIMP GetCheckboxValue(DWORD dwFieldID, BOOL* pbChecked, PWSTR* ppwszLabel);

IFACEMETHODIMP GetComboBoxValueCount(DWORD dwFieldID, DWORD* pcItems, DWORD* pdwSelectedItem);

IFACEMETHODIMP GetComboBoxValueAt(DWORD dwFieldID, DWORD dwItem, PWSTR* ppwszItem);

IFACEMETHODIMP GetSubmitButtonValue(DWORD dwFieldID, DWORD* pdwAdjacentTo);

IFACEMETHODIMP SetStringValue(DWORD dwFieldID, PCWSTR pcwzString);

IFACEMETHODIMP SetCheckboxValue(DWORD dwFieldID, BOOL bChecked);

IFACEMETHODIMP SetComboBoxSelectedValue(DWORD dwFieldID, DWORD dwSelectedItem);

IFACEMETHODIMP CommandLinkClicked(DWORD dwFieldID);

IFACEMETHODIMP GetSerialization(CREDENTIAL_PROVIDER_GET_SERIALIZATION_RESPONSE *pcpGetSerializationResponse,

CREDENTIAL_PROVIDER_CREDENTIAL_SERIALIZATION *pcpCredentialSerialization,

PWSTR *ppwszOptionalStatusText,

CREDENTIAL_PROVIDER_STATUS_ICON *pcpsiOptionalStatusIcon);

IFACEMETHODIMP ReportResult(NTSTATUS ntsStatus,

NTSTATUS ntsSubstatus,

PWSTR* ppwszOptionalStatusText,

CREDENTIAL_PROVIDER_STATUS_ICON* pcpsiOptionalStatusIcon);

public:

void WipeString(const PWSTR pwszString);

void Reset();

HRESULT Initialize(CREDENTIAL_PROVIDER_USAGE_SCENARIO cpus,

const CREDENTIAL_PROVIDER_FIELD_DESCRIPTOR* rgcpfd,

const FIELD_STATE_PAIR* rgfsp);

int Update(const char *pszUser,

const char *pszPw,

const char *pszDomain);

BOOL TranslateAccountName(PWSTR pwszDisplayName, PWSTR *ppwszAccoutName);

BOOL ExtractAccoutData(PWSTR pwszAccountData, PWSTR *ppwszAccoutName, PWSTR *ppwszDomain);

private:

/** @todo Merge all arrays which depend on SFI_NUM_FIELDS below

/** Pointer to parent. */

VBoxCredProv *m_pProvider;

/** Internal reference count. */

LONG m_cRef;

/** The usage scenario for which we were enumerated. */

CREDENTIAL_PROVIDER_USAGE_SCENARIO m_cpUS;

/** Holding type and name of each field in the tile. */

CREDENTIAL_PROVIDER_FIELD_DESCRIPTOR m_rgCredProvFieldDescriptors[SFI_NUM_FIELDS];

/** Holding state of each field in the tile. */

FIELD_STATE_PAIR m_rgFieldStatePairs[SFI_NUM_FIELDS];

/** Holding string value of each field. This is different from the name of

PWSTR m_rgFieldStrings[SFI_NUM_FIELDS];

ICredentialProviderCredentialEvents *m_pCredProvCredentialEvents;

};

#endif /* ___VBoxCredential_h */