/*++

Copyright (C) Microsoft Corporation, 1995 - 1999
All rights reserved.

Module Name:

    folder.hxx

Abstract:

    Holds TServerNotify definitions.

Author:

    Albert Ting (AlbertT)  30-Oct-1995

Revision History:

--*/

#ifndef _FOLDER_HXX
#define _FOLDER_HXX

class TFolder;

/********************************************************************

    Class summary:

    TFolder: This objects represents the print folder (local or
        remote) that a user opens in the UI.  It uses TConnectionNotify
        to watch for new changes, and then creates TServers to hold
        the server/printer data.

        Destruction refcounted.

    TConnectionNotify: if the Folder is local, then TFolder creates
        and owns this.  This object is responsible for telling
        TFolder (via ConnectionNotifyClient) that the registry has
        changed and TFolder should walk through its TServers, either
        adding new ones or deleting old ones.  (Compare EnumPrinters
        with INFO_4 vs TFolder's linked list.)

        Destruction sync'd with TFolder.

    VDataSource: watch set of printers. This class is responsible for
        keeping and updating data about a server/printer.  It also
        sends out notifications to the shell when something changes.
        It uses TPrinter to get the data.  TDSServer and VDSConnection
        derive from this class.

        Destruction refcounted.

    TDSServer: watch a server with >=0 printers.
        The single data source will return many printers.  It
        will internally enumerate masq printers, but it won't
        return them via Get/EnumPrinters.

        Inherits from VDataSource.

    VDSConnection: watch a connection with 1 printer.  This is
        needed is true connects don't appear on the server, and
        masq printers appear but are inaccurate.

        Inherits from VDataSource.

    TDSCTrue: watch a VDSConnection (true connect)

        Inherits from VDSConnection since we gather information
        from a separate handle.

    TDSCMasq: watch a VDSConnection (masq connect).

        Inherits from VDSConnection since we gather information
        from a separate handle.

    TNotifyHandler:

    Utility class. Holder of IFolderNotify* handlers allowing to be 
    chained in a linked list.

********************************************************************/

/********************************************************************

    TConnectionNotify

    Watch the registry looking for new printer addition.  When
    a new connection has been created, add the the TServer (actually
    a printer) to pFolder.

    Client constructs with a registry path off of HKEY_CURRENT_USER
    and passes in a callback.  When the registry node changes, the
    callback is triggered.

********************************************************************/

class TConnectionNotify : public MNotifyWork
{
    SIGNATURE( 'regn' )
    SAFE_NEW

public:
    TConnectionNotify(
        TFolder* pFolder
        );
    ~TConnectionNotify(
        VOID
        );

    BOOL
    bValid(
        VOID
        ) const;

private:
    TFolder            *_pFolder;
    CAutoHandleHKEY     _shKeyConnections;
    CAutoHandleNT       _shEvent;
    BOOL                _bRegistered;

    BOOL
    bSetupNotify(
        VOID
        );

    /********************************************************************

        Virtual definition MNotifyWork.

    ********************************************************************/

    HANDLE
    hEvent(
        VOID
        ) const;

    VOID
    vProcessNotifyWork(
        TNotify* pNotify
        );
};

inline
BOOL
TConnectionNotify::
bValid(
    VOID
    ) const
{
    return _shEvent != NULL && MNotifyWork::bValid();
}

/********************************************************************

    VDataSource

    VDataSource uses the name constructor class factory idiom.
    Instead of using the normal ctr, client should use pNew(),
    which instantiates a derieved, concrete class of VDataSource.

********************************************************************/

class VDataSource : public MPrinterClient {

    SIGNATURE( 'daso' )
    SAFE_NEW

public:

    enum CONNECT_TYPE {
        kServer,
        kTrue,
        kMasq
    };

    VAR( TString, strDataSource );
    VAR( CONNECT_TYPE, ConnectType );
    DLINK( VDataSource, DataSource );

    /********************************************************************

        Separate constructor/destructor since we always build
        a derived class of VDataSource (based on ConnectType).

    ********************************************************************/

    static
    VDataSource*
    pNew(
        TFolder* pFolder,
        LPCTSTR pszServer,
        CONNECT_TYPE ConnectType
        );

    VOID
    vDelete(
        VOID
        );

    BOOL
    bValid(
        VOID
        ) const;

    HANDLE
    hItemFindByName(
        LPCTSTR pszPrinter
        ) const;

    BOOL
    bSkipItem(
        HANDLE hItem
        ) const;

    /********************************************************************

        Overrides that derived classes can define.

    ********************************************************************/

    virtual
    BOOL
    bRefresh(
        VOID
        ) = 0;

    virtual
    BOOL
    bReopen(
        VOID
        ) = 0;

    virtual
    BOOL
    bGetPrinter(
        LPCTSTR pszPrinter,
        PFOLDER_PRINTER_DATA pData,
        DWORD cbData,
        PDWORD pcbNeeded
        ) const;

    virtual
    BOOL
    bAdministrator(
        VOID
        ) const = 0;

    virtual
    COUNTB
    cbSinglePrinterData(
        HANDLE hItem
        ) const;

    virtual
    VOID
    vPackSinglePrinterData(
        HANDLE hItem,
        PBYTE& pBegin,
        PBYTE& pEnd
        ) const;

    virtual
    COUNTB
    cbAllPrinterData(
        VOID
        ) const;

    virtual
    COUNT
    cPackAllPrinterData(
        PBYTE& pBegin,
        PBYTE& pEnd
        ) const;

protected:

    TFolder* _pFolder;
    COUNT _cIgnoreNotifications;
    LONG  _lItems;

    VDataSource(
        TFolder* pFolder,
        LPCTSTR pszDataSource,
        CONNECT_TYPE ConnectType
        );

    ~VDataSource(
        VOID
        );

    virtual
    VOID
    vContainerChanged(
        CONTAINER_CHANGE ContainerChange,
        INFO Info
        );

    virtual
    BOOL
    bGetPrintLib(
        TRefLock<TPrintLib> &refLock
        ) const;

    /********************************************************************

        Data retrieval.

    ********************************************************************/

    virtual
    LPCTSTR
    pszGetPrinterName(
        HANDLE hItem
        ) const = 0;

    virtual
    LPCTSTR
    pszGetCommentString(
        HANDLE hItem
        ) const;

    virtual
    LPCTSTR
    pszGetLocationString(
        HANDLE hItem
        ) const;

    virtual
    LPCTSTR
    pszGetModelString(
        HANDLE hItem
        ) const;

    virtual
    LPCTSTR
    VDataSource::
    pszGetPortString(
        HANDLE hItem
        ) const;

private:

    /********************************************************************

        Change callbacks and support.

    ********************************************************************/

    virtual
    VOID
    vReloadItems(
        VOID
        ) = 0;

    virtual
    VOID
    vRefreshComplete(
        VOID
        ) const = 0;

    virtual
    FOLDER_NOTIFY_TYPE
    uItemCreate(
        LPCTSTR pszPrinter,
        BOOL bNotify
        ) = 0;

    /********************************************************************

        MPrinterClient / MDataClient virtual definitions.

    ********************************************************************/

    VOID
    vItemChanged(
        ITEM_CHANGE ItemChange,
        HITEM hItem,
        INFO Info,
        INFO InfoNew
        );

    VDataNotify*
    pNewNotify(
        MDataClient* pDataClient
        ) const;

    VDataRefresh*
    pNewRefresh(
        MDataClient* pDataClient
        ) const;

    VOID
    vRefZeroed(
        VOID
        );
};

inline
BOOL
VDataSource::
bValid(
    VOID
    ) const
{
    return _pPrinter != NULL && MPrinterClient::bValid();
}

/********************************************************************

    TDSServer

    Watch a server for printers.  We can also watch a particular
    printer (\\server\share) and it will act like a server with
    just one printer.

********************************************************************/

class TDSServer : public VDataSource {
friend
VDataSource*
VDataSource::pNew(
    TFolder* pFolder,
    LPCTSTR pszServer,
    CONNECT_TYPE ConnectType
    );

    SIGNATURE( 'dssv' )

private:

    BOOL _bDiscardRefresh;

    TDSServer(
        TFolder* pFolder,
        LPCTSTR pszDataSource
        );

    BOOL
    bRefresh(
        VOID
        );

    BOOL
    bReopen(
        VOID
        );

    /********************************************************************

        Data retrieval.

    ********************************************************************/

    LPCTSTR
    pszGetPrinterName(
        HANDLE hItem
        ) const;

    BOOL
    bAdministrator(
        VOID
        ) const;

    BOOL
    bGetPrinter(
        LPCTSTR pszPrinter,
        PFOLDER_PRINTER_DATA pData,
        DWORD cbData,
        PDWORD pcbNeeded
        ) const;

    /********************************************************************

        Change callbacks and support.

    ********************************************************************/

    VOID
    vContainerChanged(
        CONTAINER_CHANGE ContainerChange,
        INFO Info
        );

    VOID
    vReloadItems(
        VOID
        );

    VOID
    vRefreshComplete(
        VOID
        ) const;

    FOLDER_NOTIFY_TYPE
    uItemCreate(
        LPCTSTR pszPrinter,
        BOOL bNotify
        );
};


/********************************************************************

    VDSConnection

    Watch a connection.

********************************************************************/

class VDSConnection : public VDataSource {

    SIGNATURE( 'dscn' )

public:

    static
    BOOL
    bStaticInitShutdown(
        BOOL bShutdown = FALSE
        );

protected:

    VDSConnection(
        TFolder* pFolder,
        LPCTSTR pszDataSource,
        CONNECT_TYPE ConnectType
        );

private:

    friend
    VDataSource*
    VDataSource::pNew(
        TFolder* pFolder,
        LPCTSTR pszServer,
        CONNECT_TYPE ConnectType
        );

    CONNECT_STATUS _ConnectStatus;

    static TString *gpstrConnectStatusOpen;
    static TString *gpstrConnectStatusOpenError;
    static TString *gpstrConnectStatusAccessDenied;
    static TString *gpstrConnectStatusInvalidPrinterName;

    BOOL
    bRefresh(
        VOID
        );

    BOOL
    bReopen(
        VOID
        );

    /********************************************************************

        Data retrieval.

    ********************************************************************/

    LPCTSTR
    pszGetPrinterName(
        HANDLE hItem
        ) const;

    LPCTSTR
    pszGetCommentString(
        HANDLE hItem
        ) const;

    LPCTSTR
    pszGetStatusString(
        HANDLE hItem
        ) const;

    BOOL
    bAdministrator(
        VOID
        ) const;

    BOOL
    bGetPrinter(
        LPCTSTR pszPrinter,
        PFOLDER_PRINTER_DATA pData,
        DWORD cbData,
        PDWORD pcbNeeded
        ) const;

    COUNTB
    cbSinglePrinterData(
        HANDLE hItem
        ) const;

    VOID
    vPackSinglePrinterData(
        HANDLE hItem,
        PBYTE& pBegin,
        PBYTE& pEnd
        ) const;

    COUNTB
    cbAllPrinterData(
        VOID
        ) const;

    COUNT
    cPackAllPrinterData(
        PBYTE& pBegin,
        PBYTE& pEnd
        ) const;

    BOOL
    bUseFakePrinterData(
        VOID
        ) const;

    /********************************************************************

        Change callbacks and support.

    ********************************************************************/

    VOID
    vContainerChanged(
        CONTAINER_CHANGE ContainerChange,
        INFO Info
        );

    VOID
    vReloadItems(
        VOID
        );

    VOID
    vRefreshComplete(
        VOID
        ) const;

    FOLDER_NOTIFY_TYPE
    uItemCreate(
        LPCTSTR pszPrinter,
        BOOL bNotify
        );

    VOID
    vUpdateConnectStatus(
        CONNECT_STATUS ConnectStatusNew
        );
};


/********************************************************************

    TDSCTrue: watch a true connection.

********************************************************************/

class TDSCTrue : public VDSConnection {
friend
VDataSource*
VDataSource::pNew(
    TFolder* pFolder,
    LPCTSTR pszServer,
    CONNECT_TYPE ConnectType
    );

private:

    TDSCTrue(
        TFolder *pFolder,
        LPCTSTR pszDataSource
        );

    ~TDSCTrue(
        VOID
        );

    BOOL
    bValid(
        VOID
        ) const;
};

inline
BOOL
TDSCTrue::
bValid(
    VOID
    ) const
{
    return VDSConnection::bValid();
}

/********************************************************************

    TDSCMasq: watch a masq or 'false' connection.

********************************************************************/

class TDSCMasq : public VDSConnection {
friend
VDataSource*
VDataSource::pNew(
    TFolder* pFolder,
    LPCTSTR pszServer,
    CONNECT_TYPE ConnectType
    );

private:

    TDSCMasq(
        TFolder *pFolder,
        LPCTSTR pszDataSource
        );

    ~TDSCMasq(
        VOID
        );

    BOOL
    bValid(
        VOID
        ) const;
};


inline
BOOL
TDSCMasq::
bValid(
    VOID
    ) const
{
    return VDSConnection::bValid();
}

/********************************************************************

    TFolder

    Watch a server.  If the server is local (NULL) then we have
    to watch printer connections also.  Too bad the spooler wasn't
    architected to do this for us... maybe the next one will.

********************************************************************/

class TFolder : public MRefCom
{
    SIGNATURE( 'fold' )
    SAFE_NEW

    class TNotifyHandler {

        SIGNATURE( 'tnhd' )
        SAFE_NEW

        //
        // To prevent accidental copying of this object
        //
        TNotifyHandler( const TNotifyHandler& );
        TNotifyHandler& operator =( const TNotifyHandler& );

    public:

        TNotifyHandler(
            IFolderNotify *pClientNotify
            );

        ~TNotifyHandler(
            VOID
            );

        VAR( BOOL, bValid );
        VAR( IFolderNotify*, pClientNotify );
        DLINK( TNotifyHandler, Link );
    };

public:

    VAR( BOOL, bValid );
    VAR( TConnectionNotify*, pConnectionNotify );
    DLINK_BASE( VDataSource, DataSource, DataSource );
    VAR( CCSLock, CritSec );
    VAR( TRefLock<TPrintLib>, pPrintLib );

    DLINK_BASE( TNotifyHandler, Handlers, Link );
    VAR( TString, strLocalDataSource );
    DLINK( TFolder, Link );

    TFolder(
        IN LPCTSTR           pszDataSource
        );

    ~TFolder(
        VOID
        );

    HRESULT
    pLookupNotifyHandler(
        IN  IFolderNotify    *pClientNotify,
        OUT TNotifyHandler  **ppHandler
        );

    HRESULT
    RegisterNotifyHandler(
        IN IFolderNotify    *pClientNotify
        );

    HRESULT
    UnregisterNotifyHandler(
        IN IFolderNotify    *pClientNotify
        );

    BOOL
    bNotifyAllClients( 
        IN FOLDER_NOTIFY_TYPE NotifyType, 
        IN LPCWSTR pszName, 
        IN LPCWSTR pszNewName 
        );

    BOOL
    bLocal(
        VOID
        ) const;

    //
    // Called when the system needs to refresh a window because
    // of a large change.
    //
    VOID
    vRefreshUI(
        VOID
        );

    VOID
    vCleanup(
        VOID
        );

    VOID
    vAddDataSource(
        LPCTSTR pszPrinter,
        VDataSource::CONNECT_TYPE ConnectType,
        BOOL bNotify
        );

    VOID
    vAddMasqDataSource(
        LPCTSTR pszPrinter,
        BOOL bNotify
        );

    VOID
    vDeleteMasqDataSource(
        LPCTSTR pszPrinter
        );

    VOID
    vRevalidateMasqPrinters(
        VOID
        );

    static
    VOID
    vDefaultPrinterChanged(
        VOID
        );

    static
    VOID
    vCheckDeleteDefault(
        LPCTSTR pszDeleted
        );

    /********************************************************************

        Connection notify override.

        We receive change notification callbacks from this virtual
        function definition.

    ********************************************************************/

    VOID
    vConnectionNotifyChange(
        BOOL bNotify
        );

private:

    VDataSource*
    pFindDataSource(
        LPCTSTR pszPrinter,
        VDataSource::CONNECT_TYPE ConnectType
        ) const;

    /********************************************************************

        MRefCom override.

    ********************************************************************/

    VOID
    vRefZeroed(
        VOID
        );
};

/********************************************************************

    class TFolderList

 ********************************************************************/

class TFolderList {

    SIGNATURE( 'fldl' )
    SAFE_NEW

    //
    // To prevent accidental copying of this object
    //
    TFolderList( const TFolderList& );
    TFolderList& operator =( const TFolderList& );

public:

    TFolderList( );
    ~TFolderList( );

    static HRESULT
    RegisterDataSource( 
        IN  LPCTSTR         pszDataSource, 
        IN  IFolderNotify   *pClientNotify,
        OUT LPHANDLE        phFolder,
        OUT  PBOOL          pbAdministrator OPTIONAL
        );

    static HRESULT
    UnregisterDataSource( 
        IN  LPCTSTR         pszDataSource, 
        IN  IFolderNotify   *pClientNotify,
        OUT LPHANDLE        phFolder
        );

    static BOOL
    bValidFolderObject( 
        IN  const TFolder *pFolder
        );

    BOOL
    bValid(
        VOID
        ) const;

    //
    // This global lock serializes the calls to the external PrintUI cache  
    // private APIs.
    //
    static CCSLock    *gpFolderLock;           

private:

    static HRESULT
    pLookupFolder( 
        IN  LPCTSTR     pszDataSource,
        OUT TFolder   **ppFolder
        );

    DLINK_BASE( TFolder, Folders, Link );       // List of the registered folders
    static TFolderList *gpFolders;              // Global TFolder cashe for the library
};


#endif // ndef _FOLDER_HXX