reactos/dll/shellext/zipfldr/CZipEnumerator.hpp

/*
 * PROJECT:     ReactOS Zip Shell Extension
 * LICENSE:     GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
 * PURPOSE:     CZipEnumerator
 * COPYRIGHT:   Copyright 2017 Mark Jansen (mark.jansen@reactos.org)
 */

struct CZipEnumerator
{
private:
    CComPtr<IZip> m_Zip;
    bool m_First;
    CAtlList<CStringA> m_Returned;
public:
    CZipEnumerator()
        :m_First(true)
    {
    }

    bool initialize(IZip* zip)
    {
        m_Zip = zip;
        return reset();
    }

    bool reset()
    {
        unzFile uf = m_Zip->getZip();
        m_First = true;
        if (unzGoToFirstFile(uf) != UNZ_OK)
            return false;
        m_Returned.RemoveAll();
        return true;
    }

    bool next_unique(const char* prefix, CStringA& name, bool& folder, unz_file_info64& info)
    {
        size_t len = strlen(prefix);
        CStringA tmp;
        while (next(tmp, info))
        {
            if (!_strnicmp(tmp, prefix, len))
            {
                int pos = tmp.Find('/', len);
                if (pos < 0)
                {
                    name = tmp.Mid(len);
                    folder = false;
                }
                else
                {
                    name = tmp.Mid(len, pos - len);
                    folder = true;
                }
                tmp = name.MakeLower();

                POSITION it = m_Returned.Find(tmp);
                if (!name.IsEmpty() && !it)
                {
                    m_Returned.AddTail(tmp);
                    return true;
                }
            }
        }
        return false;
    }

    bool next(CStringA& name, unz_file_info64& info)
    {
        int err;

        unzFile uf = m_Zip->getZip();
        if (!m_First)
        {
            err = unzGoToNextFile(uf);
            if (err == UNZ_END_OF_LIST_OF_FILE)
            {
                return false;
            }
        }
        m_First = false;

        err = unzGetCurrentFileInfo64(uf, &info, NULL, 0, NULL, 0, NULL, 0);
        if (err == UNZ_OK)
        {
            PSTR buf = name.GetBuffer(info.size_filename);
            err = unzGetCurrentFileInfo64(uf, NULL, buf, name.GetAllocLength(), NULL, 0, NULL, 0);
            name.ReleaseBuffer(info.size_filename);
            name.Replace('\\', '/');
        }
        return err == UNZ_OK;
    }
};