#ifndef ____H_EXTPACKMANAGERIMPL

#define ____H_EXTPACKMANAGERIMPL

#include "VirtualBoxBase.h"

#include <VBox/ExtPack/ExtPack.h>

#include <iprt/fs.h>

class ATL_NO_VTABLE ExtPackFile :

public VirtualBoxBase,

VBOX_SCRIPTABLE_IMPL(IExtPackFile)

{

public:

/** @name COM and internal init/term/mapping cruft.

VIRTUALBOXBASE_ADD_ERRORINFO_SUPPORT(ExtPackFile, IExtPackFile)

DECLARE_NOT_AGGREGATABLE(ExtPackFile)

DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(ExtPackFile)

COM_INTERFACE_ENTRY(ISupportErrorInfo)

COM_INTERFACE_ENTRY(IExtPackFile)

COM_INTERFACE_ENTRY(IExtPackBase)

COM_INTERFACE_ENTRY(IDispatch)

END_COM_MAP()

DECLARE_EMPTY_CTOR_DTOR(ExtPackFile)

HRESULT FinalConstruct();

void FinalRelease();

HRESULT initWithFile(const char *a_pszFile, class ExtPackManager *a_pExtPackMgr);

void uninit();

RTMEMEF_NEW_AND_DELETE_OPERATORS();

/** @} */

/** @name IExtPackBase interfaces

STDMETHOD(COMGETTER(Name))(BSTR *a_pbstrName);

STDMETHOD(COMGETTER(Description))(BSTR *a_pbstrDescription);

STDMETHOD(COMGETTER(Version))(BSTR *a_pbstrVersion);

STDMETHOD(COMGETTER(Revision))(ULONG *a_puRevision);

STDMETHOD(COMGETTER(VRDEModule))(BSTR *a_pbstrVrdeModule);

STDMETHOD(COMGETTER(PlugIns))(ComSafeArrayOut(IExtPackPlugIn *, a_paPlugIns));

STDMETHOD(COMGETTER(Usable))(BOOL *a_pfUsable);

STDMETHOD(COMGETTER(WhyUnusable))(BSTR *a_pbstrWhy);

STDMETHOD(COMGETTER(ShowLicense))(BOOL *a_pfShowIt);

STDMETHOD(COMGETTER(License))(BSTR *a_pbstrHtmlLicense);

STDMETHOD(QueryLicense)(IN_BSTR a_bstrPreferredLocale, IN_BSTR a_bstrPreferredLanguage,

IN_BSTR a_bstrFormat, BSTR *a_pbstrLicense);

/** @} */

/** @name IExtPackFile interfaces

STDMETHOD(COMGETTER(FilePath))(BSTR *a_pbstrPath);

STDMETHOD(Install)(void);

/** @} */

private:

/** @name Misc init helpers

HRESULT initFailed(const char *a_pszWhyFmt, ...);

/** @} */

private:

struct Data;

/** Pointer to the private instance. */

Data *m;

friend class ExtPackManager;

};

class ATL_NO_VTABLE ExtPack :

public VirtualBoxBase,

VBOX_SCRIPTABLE_IMPL(IExtPack)

{

public:

/** @name COM and internal init/term/mapping cruft.

VIRTUALBOXBASE_ADD_ERRORINFO_SUPPORT(ExtPack, IExtPack)

DECLARE_NOT_AGGREGATABLE(ExtPack)

DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(ExtPack)

COM_INTERFACE_ENTRY(ISupportErrorInfo)

COM_INTERFACE_ENTRY(IExtPack)

COM_INTERFACE_ENTRY(IExtPackBase)

COM_INTERFACE_ENTRY(IDispatch)

END_COM_MAP()

DECLARE_EMPTY_CTOR_DTOR(ExtPack)

HRESULT FinalConstruct();

void FinalRelease();

HRESULT initWithDir(VBOXEXTPACKCTX a_enmContext, const char *a_pszName, const char *a_pszDir);

void uninit();

RTMEMEF_NEW_AND_DELETE_OPERATORS();

/** @} */

/** @name IExtPackBase interfaces

STDMETHOD(COMGETTER(Name))(BSTR *a_pbstrName);

STDMETHOD(COMGETTER(Description))(BSTR *a_pbstrDescription);

STDMETHOD(COMGETTER(Version))(BSTR *a_pbstrVersion);

STDMETHOD(COMGETTER(Revision))(ULONG *a_puRevision);

STDMETHOD(COMGETTER(VRDEModule))(BSTR *a_pbstrVrdeModule);

STDMETHOD(COMGETTER(PlugIns))(ComSafeArrayOut(IExtPackPlugIn *, a_paPlugIns));

STDMETHOD(COMGETTER(Usable))(BOOL *a_pfUsable);

STDMETHOD(COMGETTER(WhyUnusable))(BSTR *a_pbstrWhy);

STDMETHOD(COMGETTER(ShowLicense))(BOOL *a_pfShowIt);

STDMETHOD(COMGETTER(License))(BSTR *a_pbstrHtmlLicense);

STDMETHOD(QueryLicense)(IN_BSTR a_bstrPreferredLocale, IN_BSTR a_bstrPreferredLanguage,

IN_BSTR a_bstrFormat, BSTR *a_pbstrLicense);

/** @} */

/** @name IExtPack interfaces

STDMETHOD(QueryObject)(IN_BSTR a_bstrObjectId, IUnknown **a_ppUnknown);

/** @} */

/** @name Internal interfaces used by ExtPackManager.

bool callInstalledHook(IVirtualBox *a_pVirtualBox, AutoWriteLock *a_pLock);

HRESULT callUninstallHookAndClose(IVirtualBox *a_pVirtualBox, bool a_fForcedRemoval);

bool callVirtualBoxReadyHook(IVirtualBox *a_pVirtualBox, AutoWriteLock *a_pLock);

bool callConsoleReadyHook(IConsole *a_pConsole, AutoWriteLock *a_pLock);

bool callVmCreatedHook(IVirtualBox *a_pVirtualBox, IMachine *a_pMachine, AutoWriteLock *a_pLock);

bool callVmConfigureVmmHook(IConsole *a_pConsole, PVM a_pVM, AutoWriteLock *a_pLock, int *a_pvrc);

bool callVmPowerOnHook(IConsole *a_pConsole, PVM a_pVM, AutoWriteLock *a_pLock, int *a_pvrc);

bool callVmPowerOffHook(IConsole *a_pConsole, PVM a_pVM, AutoWriteLock *a_pLock);

HRESULT checkVrde(void);

HRESULT getVrdpLibraryName(Utf8Str *a_pstrVrdeLibrary);

bool wantsToBeDefaultVrde(void) const;

HRESULT refresh(bool *pfCanDelete);

/** @} */

protected:

/** @name Internal helper methods.

void probeAndLoad(void);

bool findModule(const char *a_pszName, const char *a_pszExt, VBOXEXTPACKMODKIND a_enmKind,

Utf8Str *a_ppStrFound, bool *a_pfNative, PRTFSOBJINFO a_pObjInfo) const;

static bool objinfoIsEqual(PCRTFSOBJINFO pObjInfo1, PCRTFSOBJINFO pObjInfo2);

/** @} */

/** @name Extension Pack Helpers

static DECLCALLBACK(int) hlpFindModule(PCVBOXEXTPACKHLP pHlp, const char *pszName, const char *pszExt,

VBOXEXTPACKMODKIND enmKind, char *pszFound, size_t cbFound, bool *pfNative);

static DECLCALLBACK(int) hlpGetFilePath(PCVBOXEXTPACKHLP pHlp, const char *pszFilename, char *pszPath, size_t cbPath);

static DECLCALLBACK(VBOXEXTPACKCTX) hlpGetContext(PCVBOXEXTPACKHLP pHlp);

static DECLCALLBACK(int) hlpReservedN(PCVBOXEXTPACKHLP pHlp);

/** @} */

private:

struct Data;

/** Pointer to the private instance. */

Data *m;

friend class ExtPackManager;

};

class ATL_NO_VTABLE ExtPackManager :

public VirtualBoxBase,

VBOX_SCRIPTABLE_IMPL(IExtPackManager)

{

/** @name COM and internal init/term/mapping cruft.

VIRTUALBOXBASE_ADD_ERRORINFO_SUPPORT(ExtPackManager, IExtPackManager)

DECLARE_NOT_AGGREGATABLE(ExtPackManager)

DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(ExtPackManager)

COM_INTERFACE_ENTRY(ISupportErrorInfo)

COM_INTERFACE_ENTRY(IExtPackManager)

COM_INTERFACE_ENTRY(IDispatch)

END_COM_MAP()

DECLARE_EMPTY_CTOR_DTOR(ExtPackManager)

HRESULT FinalConstruct();

void FinalRelease();

HRESULT initExtPackManager(VirtualBox *a_pVirtualBox, VBOXEXTPACKCTX a_enmContext);

void uninit();

RTMEMEF_NEW_AND_DELETE_OPERATORS();

/** @} */

/** @name IExtPack interfaces

STDMETHOD(COMGETTER(InstalledExtPacks))(ComSafeArrayOut(IExtPack *, a_paExtPacks));

STDMETHOD(Find)(IN_BSTR a_bstrName, IExtPack **a_pExtPack);

STDMETHOD(OpenExtPackFile)(IN_BSTR a_bstrTarball, IExtPackFile **a_ppExtPackFile);

STDMETHOD(Uninstall)(IN_BSTR a_bstrName, BOOL a_fForcedRemoval);

STDMETHOD(Cleanup)(void);

STDMETHOD(QueryAllPlugInsForFrontend)(IN_BSTR a_bstrFrontend, ComSafeArrayOut(BSTR, a_pabstrPlugInModules));

/** @} */

/** @name Internal interfaces used by other Main classes.

HRESULT doInstall(ExtPackFile *a_pExtPackFile);

void callAllVirtualBoxReadyHooks(void);

void callAllConsoleReadyHooks(IConsole *a_pConsole);

void callAllVmCreatedHooks(IMachine *a_pMachine);

int callAllVmConfigureVmmHooks(IConsole *a_pConsole, PVM a_pVM);

int callAllVmPowerOnHooks(IConsole *a_pConsole, PVM a_pVM);

void callAllVmPowerOffHooks(IConsole *a_pConsole, PVM a_pVM);

HRESULT checkVrdeExtPack(Utf8Str const *a_pstrExtPack);

int getVrdeLibraryPathForExtPack(Utf8Str const *a_pstrExtPack, Utf8Str *a_pstrVrdeLibrary);

HRESULT getDefaultVrdeExtPack(Utf8Str *a_pstrExtPack);

bool isExtPackUsable(const char *a_pszExtPack);

/** @} */

private:

HRESULT runSetUidToRootHelper(const char *a_pszCommand, ...);

ExtPack *findExtPack(const char *a_pszName);

void removeExtPack(const char *a_pszName);

HRESULT refreshExtPack(const char *a_pszName, bool a_fUnsuableIsError, ExtPack **a_ppExtPack);

private:

struct Data;

/** Pointer to the private instance. */

Data *m;

};

#endif