mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 22:28:36 +00:00
96 lines
2.3 KiB
C++
96 lines
2.3 KiB
C++
/*
|
|
* 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;
|
|
tmp.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;
|
|
}
|
|
};
|
|
|