#include "pch.h"
#pragma hdrstop

#include "ownerlst.h"


COwnerListEntry::COwnerListEntry(
    IDiskQuotaUser *pOwner
    ) : m_pOwner(pOwner)
{ 
    if (m_pOwner) 
    {
        TCHAR szDisplayName[MAX_PATH] = { TEXT('\0') };
        TCHAR szLogonName[MAX_PATH]   = { TEXT('\0') };
        m_pOwner->AddRef();
        if (SUCCEEDED(m_pOwner->GetName(NULL, 0, 
                                        szLogonName, ARRAYSIZE(szLogonName),
                                        szDisplayName, ARRAYSIZE(szDisplayName))))
        {
            if (TEXT('\0') != szDisplayName[0])
            {
                m_strOwnerName.Format(TEXT("%1 (%2)"), szDisplayName, szLogonName);
            }
            else if (TEXT('\0') != szLogonName[0])
            {
                m_strOwnerName = szLogonName;
            }
            else
            {
                //
                // If the account SID has NOT been resolved to a name, display
                // the SID as a string.  We do the same thing in the Quota Entries
                // view.
                //
                BYTE Sid[MAX_SID_LEN];
                DWORD cchSidStr = MAX_PATH;
                if (SUCCEEDED(m_pOwner->GetSid(Sid, ARRAYSIZE(Sid))))
                {
                    SidToString(Sid, m_strOwnerName.GetBuffer(cchSidStr), &cchSidStr);
                    m_strOwnerName.ReleaseBuffer();
                }
            }
        }
    }    
}


int
COwnerListEntry::AddFile(
    LPCTSTR pszFile,
    bool bDirectory
    )
{
    m_rgFiles.Append(COwnerListFile(pszFile, bDirectory));
    return m_rgFiles.Count() - 1;
}


int
COwnerListEntry::FileCount(
    bool bIncludeDeleted       // optional.  Default = false.
    )
{
    int cFiles = m_rgFiles.Count();
    int n = 0;
    for (int i = 0; i < cFiles; i++)
    {
        if (bIncludeDeleted || !m_rgFiles[i].IsEmpty())
            n++;
    }
    return n;
}


#if DBG
void
COwnerListEntry::Dump(
    void
    ) const
{
    DBGERROR((TEXT("\tm_pOwner........: 0x%08X"), m_pOwner));
    DBGERROR((TEXT("\tm_strOwnerName..: \"%s\""), m_strOwnerName.Cstr()));
    int cFiles = m_rgFiles.Count();
    DBGERROR((TEXT("\tcFiles..........: %d"), cFiles));
    for (int i = 0; i < cFiles; i++)
    {
        DBGERROR((TEXT("\tFile[%3d].......: \"%s\""), i, m_rgFiles[i].Cstr()));
    }
}
#endif // DBG


COwnerList::~COwnerList(
    void
    )
{
    Clear();
}


void
COwnerList::Clear(
    void
    )
{
    int cOwners = m_rgpOwners.Count();
    for (int i = 0; i < cOwners; i++)
    {
        delete m_rgpOwners[i];
        m_rgpOwners[i] = NULL;
    }
    m_rgpOwners.Clear();
}


int
COwnerList::AddOwner(
    IDiskQuotaUser *pOwner
    )
{
    m_rgpOwners.Append(new COwnerListEntry(pOwner));
    return m_rgpOwners.Count() - 1;
}


IDiskQuotaUser *
COwnerList::GetOwner(
    int iOwner
    ) const
{
    //
    // Caller must call Release() when done with returned iface pointer.
    //
    return m_rgpOwners[iOwner]->GetOwner();
}


int 
COwnerList::FileCount(
    int iOwner,
    bool bIncludeDeleted    // optional.  Default = false
    ) const
{
    int cFiles = 0;
    int iFirst = iOwner;
    int iLast  = iOwner;
    if (-1 == iOwner)
    {
        //
        // Count ALL files.
        //
        iFirst = 0;
        iLast = m_rgpOwners.Count() - 1;
    }
    for (int i = iFirst; i <= iLast; i++)
    {
        cFiles += m_rgpOwners[i]->FileCount(bIncludeDeleted);
    }
    return cFiles;
}

#if DBG
void
COwnerList::Dump(
    void
    ) const
{
    int cEntries = m_rgpOwners.Count();
    DBGERROR((TEXT("COwnerList::Dump - %d entries"), cEntries));
    for (int i = 0; i < cEntries; i++)
    {
        DBGERROR((TEXT("COwnerListEntry[%d] ========================="), i));
        m_rgpOwners[i]->Dump();
    }
}
#endif // DBG