mirror of
https://github.com/zodiacon/WFPExplorer
synced 2026-06-08 18:39:31 +00:00
Pavel Yosifovich
107 lines
4.6 KiB
C++
107 lines
4.6 KiB
C++
#include "pch.h"
|
|
#include "ProviderContextView.h"
|
|
#include "StringHelper.h"
|
|
#include <SortHelper.h>
|
|
#include "resource.h"
|
|
#include <WFPEnumerators.h>
|
|
|
|
CProviderContextView::CProviderContextView(IMainFrame* frame, WFPEngine& engine) : CGenericListViewBase(frame), m_Engine(engine) {
|
|
}
|
|
|
|
CString CProviderContextView::GetProviderName(GUID const* key) const {
|
|
CString name;
|
|
if (key) {
|
|
if (auto provider = m_Engine.GetProviderByKey(*key); provider)
|
|
name = StringHelper::ParseMUIString(provider->displayData.name);
|
|
if(name.IsEmpty() || name[0] == L'@')
|
|
name =StringHelper::GuidToString(*key);
|
|
}
|
|
return name;
|
|
}
|
|
|
|
LRESULT CProviderContextView::OnCreate(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/) {
|
|
m_hWndClient = m_List.Create(m_hWnd, rcDefault, nullptr,
|
|
WS_CHILD | WS_VISIBLE | WS_CLIPCHILDREN | LVS_OWNERDATA | LVS_REPORT | LVS_SHOWSELALWAYS);
|
|
m_List.SetExtendedListViewStyle(LVS_EX_DOUBLEBUFFER | LVS_EX_FULLROWSELECT | LVS_EX_INFOTIP);
|
|
|
|
auto cm = GetColumnManager(m_List);
|
|
cm->AddColumn(L"Context Key", 0, 300, ColumnType::Key, ColumnFlags::Visible | ColumnFlags::Numeric);
|
|
cm->AddColumn(L"Type", 0, 150, ColumnType::Type);
|
|
cm->AddColumn(L"Name", 0, 220, ColumnType::Name);
|
|
cm->AddColumn(L"Flags", 0, 100, ColumnType::Flags);
|
|
cm->AddColumn(L"Data Size", LVCFMT_RIGHT, 90, ColumnType::DataSize);
|
|
cm->AddColumn(L"Context ID", 0, 150, ColumnType::Id, ColumnFlags::Visible | ColumnFlags::Numeric);
|
|
cm->AddColumn(L"Provider", 0, 240, ColumnType::Provider);
|
|
cm->AddColumn(L"Description", 0, 300, ColumnType::Desc);
|
|
|
|
CImageList images;
|
|
images.Create(16, 16, ILC_COLOR32 | ILC_MASK, 2, 2);
|
|
images.AddIcon(AtlLoadIconImage(IDI_CONTEXT, 0, 16, 16));
|
|
//images.AddIcon(AtlLoadIconImage(IDI_PROVIDER_PERSISTENT, 0, 16, 16));
|
|
m_List.SetImageList(images, LVSIL_SMALL);
|
|
|
|
Refresh();
|
|
|
|
return 0;
|
|
}
|
|
|
|
LRESULT CProviderContextView::OnRefresh(WORD, WORD, HWND, BOOL&) {
|
|
Refresh();
|
|
return 0;
|
|
}
|
|
|
|
void CProviderContextView::Refresh() {
|
|
m_Contexts = WFPProviderContextEnumerator(m_Engine.Handle()).Next(256);
|
|
m_List.SetItemCountEx((int)m_Contexts.size(), LVSICF_NOSCROLL);
|
|
}
|
|
|
|
CString CProviderContextView::GetColumnText(HWND, int row, int col) {
|
|
auto info = m_Contexts[row];
|
|
switch (GetColumnManager(m_List)->GetColumnTag<ColumnType>(col)) {
|
|
case ColumnType::Key: return StringHelper::GuidToString(info->providerContextKey);
|
|
case ColumnType::Name: return StringHelper::ParseMUIString(info->displayData.name);
|
|
case ColumnType::Type: return std::format(L"{} ({})", StringHelper::WFPProviderContextTypeToString(info->type), (UINT32)info->type).c_str();
|
|
case ColumnType::Provider: return GetProviderName(info->providerKey);
|
|
case ColumnType::Desc: return StringHelper::ParseMUIString(info->displayData.description);
|
|
case ColumnType::Id: return std::format(L"0x{:X}", info->providerContextId).c_str();
|
|
case ColumnType::DataSize: return info->providerData.size ? std::format(L"{} bytes", info->providerData.size).c_str() : L"";
|
|
case ColumnType::Flags:
|
|
if (info->flags == 0)
|
|
return L"0";
|
|
return std::format(L"0x{:X} ({})", info->flags, (PCWSTR)StringHelper::WFPProviderContextFlagsToString(info->flags)).c_str();
|
|
}
|
|
return CString();
|
|
}
|
|
|
|
void CProviderContextView::DoSort(SortInfo const* si) {
|
|
auto col = GetColumnManager(m_List)->GetColumnTag<ColumnType>(si->SortColumn);
|
|
auto asc = si->SortAscending;
|
|
|
|
auto compare = [&](auto p1, auto p2) {
|
|
switch (col) {
|
|
case ColumnType::Key: return SortHelper::Sort(StringHelper::GuidToString(p1->providerContextKey), StringHelper::GuidToString(p2->providerContextKey), asc);
|
|
case ColumnType::Name: return SortHelper::Sort(StringHelper::ParseMUIString(p1->displayData.name), StringHelper::ParseMUIString(p2->displayData.name), asc);
|
|
case ColumnType::Desc: return SortHelper::Sort(StringHelper::ParseMUIString(p1->displayData.description), StringHelper::ParseMUIString(p2->displayData.description), asc);
|
|
case ColumnType::Flags: return SortHelper::Sort(p1->flags, p2->flags, asc);
|
|
case ColumnType::Type: return SortHelper::Sort(p1->type, p2->type, asc);
|
|
case ColumnType::Id: return SortHelper::Sort(p1->providerContextId, p2->providerContextId, asc);
|
|
case ColumnType::DataSize: return SortHelper::Sort(p1->providerData.size, p2->providerData.size, asc);
|
|
case ColumnType::Provider: return SortHelper::Sort(GetProviderName(p1->providerKey), GetProviderName(p2->providerKey), asc);
|
|
}
|
|
return false;
|
|
};
|
|
std::ranges::sort(m_Contexts, compare);
|
|
}
|
|
|
|
int CProviderContextView::GetSaveColumnRange(HWND, int&) const {
|
|
return 1;
|
|
}
|
|
|
|
int CProviderContextView::GetRowImage(HWND, int row, int col) const {
|
|
return 0;
|
|
}
|
|
|
|
CString CProviderContextView::GetDefaultSaveFile() const {
|
|
return L"providercontexts.csv";
|
|
}
|