#ifndef ____H_EXTPACKMANAGERIMPL

#define ____H_EXTPACKMANAGERIMPL

#include "VirtualBoxBase.h"

#include <VBox/ExtPack/ExtPack.h>

#include <iprt/fs.h>

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(IDispatch)

END_COM_MAP()

DECLARE_EMPTY_CTOR_DTOR(ExtPack)

HRESULT FinalConstruct();

void FinalRelease();

HRESULT init(const char *a_pszName, const char *a_pszParentDir);

void uninit();

/** @} */

/** @name IExtPack interfaces

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

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

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

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

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

/** @} */

/** @name Internal interfaces used by ExtPackManager.

void *getCallbackTable();

HRESULT refresh(bool *pfCanDelete);

/** @} */

protected:

/** @name Internal helper methods.

void probeAndLoad(void);

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

Utf8Str *a_pStrFound, 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,

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

/** @} */

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 init();

void uninit();

/** @} */

/** @name IExtPack interfaces

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

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

STDMETHOD(Install)(IN_BSTR a_bstrTarball, BSTR *a_pbstrName);

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

/** @} */

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

int callAllConfigHooks(IConsole *a_pConsole, PVM a_pVM);

int callAllNewMachineHooks(IMachine *a_pMachine);

/** @} */

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