#include "pch.h"
#include "SublayersView.h"
#include "StringHelper.h"
#include <SortHelper.h>
#include "resource.h"
#include <WFPEnumerators.h>
#include "WFPHelper.h"

CSublayersView::CSublayersView(IMainFrame* frame, WFPEngine& engine) : CGenericListViewBase(frame), m_Engine(engine) {
}

LRESULT CSublayersView::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"Name", 0, 300, ColumnType::Name);
	cm->AddColumn(L"Layer Key", 0, 300, ColumnType::Key, ColumnFlags::Visible | ColumnFlags::Numeric);
	cm->AddColumn(L"Flags", 0, 120, ColumnType::Flags);
	cm->AddColumn(L"Weight", LVCFMT_RIGHT, 80, ColumnType::Weight, ColumnFlags::Visible | ColumnFlags::Numeric);
	cm->AddColumn(L"Provider", 0, 200, ColumnType::Provider);
	cm->AddColumn(L"Provider Data", LVCFMT_RIGHT, 90, ColumnType::ProviderData);
	cm->AddColumn(L"Description", 0, 300, ColumnType::Desc);

	CImageList images;
	images.Create(16, 16, ILC_COLOR32 | ILC_MASK, 1, 1);
	images.AddIcon(AtlLoadIconImage(IDI_SUBLAYER, 0, 16, 16));
	m_List.SetImageList(images, LVSIL_SMALL);

	Refresh();

	return 0;
}

LRESULT CSublayersView::OnRefresh(WORD, WORD, HWND, BOOL&) {
	Refresh();
	return 0;
}

LRESULT CSublayersView::OnDeleteSubLayer(WORD, WORD, HWND, BOOL&) {
	int selected = m_List.GetSelectedCount();
	ATLASSERT(selected > 0);
	if (AtlMessageBox(m_hWnd, L"Delete selected sublayer(s)?", IDS_TITLE, MB_ICONWARNING | MB_YESNO | MB_DEFBUTTON2) == IDNO)
		return 0;

	int deleted = 0;
	for (auto index : SelectedItemsView(m_List)) {
		auto const& sl = m_Layers[index];
		if (m_Engine.DeleteSubLayer(sl.Data->subLayerKey))
			deleted++;
	}
	if (deleted > 0) {
		m_List.SelectAllItems(false);
		Refresh();
	}
	if (deleted < selected)
		AtlMessageBox(m_hWnd, std::format(L"Deleted {}/{} sublayers.", deleted, selected).c_str(), IDS_TITLE, MB_ICONWARNING);

	return 0;
}

LRESULT CSublayersView::OnProperties(WORD, WORD, HWND, BOOL&) {
	auto row = m_List.GetNextItem(-1, LVNI_SELECTED);
	if (row >= 0) {
		auto& sl = m_Layers[row];
		WFPHelper::ShowSublayerProperties(m_Engine, sl.Data);
	}
	return 0;
}

void CSublayersView::Refresh() {
	m_Layers = WFPSubLayerEnumerator(m_Engine.Handle()).Next<SubLayerInfo>(512);
	m_List.SetItemCountEx((int)m_Layers.size(), LVSICF_NOSCROLL);
	Frame()->SetStatusText(6, std::format(L"{} Sublayers", m_Layers.size()).c_str());
}

void CSublayersView::UpdateUI() const {
	auto& ui = Frame()->UI();
	auto selected = m_List.GetSelectedCount();
	ui.UIEnable(ID_EDIT_PROPERTIES, selected == 1);
	ui.UIEnable(ID_EDIT_DELETE, selected > 0);
	ui.UIEnable(ID_EDIT_COPY, selected > 0);
	ui.UIEnable(ID_EDIT_FINDNEXT, Frame()->GetFindDialog() != nullptr);
}

CString CSublayersView::GetColumnText(HWND, int row, int col) {
	auto& sl = m_Layers[row];
	auto info = sl.Data;
	switch (GetColumnManager(m_List)->GetColumnTag<ColumnType>(col)) {
		case ColumnType::Key: return StringHelper::GuidToString(info->subLayerKey);
		case ColumnType::Name: return sl.Name();
		case ColumnType::Desc: return sl.Desc();
		case ColumnType::ProviderData: return info->providerData.size == 0 ? L"" : std::format(L"{} Bytes", info->providerData.size).c_str();
		case ColumnType::Flags:
			if (info->flags == 0)
				return L"0";
			return std::format(L"0x{:X} ({})", info->flags,
				(PCWSTR)StringHelper::WFPSubLayerFlagsToString(info->flags)).c_str();

		case ColumnType::Weight: return std::to_wstring(info->weight).c_str();
		case ColumnType::Provider:
			if (sl.ProviderName.IsEmpty()) {
				if (info->providerKey) {
					auto provider = m_Engine.GetProviderByKey(*info->providerKey);
					if (provider)
						sl.ProviderName = StringHelper::ParseMUIString(provider->displayData.name);
					else
						sl.ProviderName = StringHelper::GuidToString(*info->providerKey);
				}
			}
			return sl.ProviderName;

	}
	return CString();
}

void CSublayersView::DoSort(SortInfo const* si) {
	auto col = GetColumnManager(m_List)->GetColumnTag<ColumnType>(si->SortColumn);
	auto asc = si->SortAscending;

	auto compare = [&](auto& sl1, auto& sl2) {
		auto l1 = sl1.Data, l2 = sl2.Data;
		switch (col) {
			case ColumnType::Key: return SortHelper::Sort(StringHelper::GuidToString(l1->subLayerKey), StringHelper::GuidToString(l2->subLayerKey), asc);
			case ColumnType::Name: return SortHelper::Sort(sl1.Name(), sl2.Name(), asc);
			case ColumnType::Desc: return SortHelper::Sort(sl1.Desc(), sl2.Desc(), asc);
			case ColumnType::Flags: return SortHelper::Sort(l1->flags, l2->flags, asc);
			case ColumnType::Provider: return SortHelper::Sort(sl1.ProviderName, sl2.ProviderName, asc);
			case ColumnType::Weight: return SortHelper::Sort(l1->weight, l2->weight, asc);
			case ColumnType::ProviderData: return SortHelper::Sort(l1->providerData.size, l2->providerData.size, asc);
		}
		return false;
	};
	std::ranges::sort(m_Layers, compare);
}

int CSublayersView::GetSaveColumnRange(HWND, int&) const {
	return 1;
}

int CSublayersView::GetRowImage(HWND, int row, int col) const {
	return 0;
}

CString CSublayersView::GetDefaultSaveFile() const {
	return L"sublayers.csv";
}

bool CSublayersView::OnDoubleClickList(HWND, int row, int col, POINT const& pt) {
	if (row >= 0) {
		auto& sl = m_Layers[row];
		WFPHelper::ShowSublayerProperties(m_Engine, sl.Data);
		return true;
	}
	return false;
}

bool CSublayersView::OnRightClickList(HWND, int row, int col, POINT const& pt) const {
	if (row < 0)
		return false;

	CMenu menu;
	menu.LoadMenu(IDR_CONTEXT);

	return Frame()->TrackPopupMenu(menu.GetSubMenu(2), 0, pt.x, pt.y);
}

LRESULT CSublayersView::OnActivate(UINT, WPARAM activate, LPARAM, BOOL&) {
	if (activate)
		UpdateUI();
	return 0;
}

CString const& CSublayersView::SubLayerInfo::Name() const {
	if (m_Name.IsEmpty())
		m_Name = StringHelper::ParseMUIString(Data->displayData.name);
	return m_Name;
}

CString const& CSublayersView::SubLayerInfo::Desc() const {
	if (m_Desc.IsEmpty())
		m_Desc = StringHelper::ParseMUIString(Data->displayData.description);
	return m_Desc;
}

void CSublayersView::OnStateChanged(HWND, int from, int to, UINT oldState, UINT newState) {
	if ((newState & LVIS_SELECTED) || (oldState & LVIS_SELECTED))
		UpdateUI();
}