path: root/MyRuLib/FbBookPanel.cpp

#include "FbBookPanel.h"
#include "FbConst.h"
#include "FbManager.h"
#include "FbFrameHtml.h"
#include "InfoCash.h"
#include "FbBookMenu.h"
#include "MyRuLibApp.h"
#include "FbDownloader.h"

BEGIN_EVENT_TABLE(FbBookPanel, wxSplitterWindow)
	EVT_MENU(ID_BOOKINFO_UPDATE, FbBookPanel::OnInfoUpdate)
	EVT_TREE_SEL_CHANGED(ID_BOOKS_LISTCTRL, FbBookPanel::OnBooksListViewSelected)
	EVT_TREE_ITEM_ACTIVATED(ID_BOOKS_LISTCTRL, FbBookPanel::OnBooksListActivated)
	EVT_TREE_ITEM_COLLAPSING(ID_BOOKS_LISTCTRL, FbBookPanel::OnBooksListCollapsing)
	EVT_TREE_KEY_DOWN(ID_BOOKS_LISTCTRL, FbBookPanel::OnBooksListKeyDown)
	EVT_TREE_STATE_IMAGE_CLICK(ID_BOOKS_LISTCTRL, FbBookPanel::OnImageClick)
	EVT_TREE_ITEM_MENU(ID_BOOKS_LISTCTRL, FbBookPanel::OnContextMenu)
	EVT_MENU(wxID_SELECTALL, FbBookPanel::OnSelectAll)
	EVT_MENU(ID_UNSELECTALL, FbBookPanel::OnUnselectAll)
	EVT_MENU(ID_OPEN_BOOK, FbBookPanel::OnOpenBook)
	EVT_MENU(ID_DOWNLOAD_BOOK, FbBookPanel::OnDownloadBook)
	EVT_MENU(ID_SYSTEM_DOWNLOAD, FbBookPanel::OnSystemDownload)
	EVT_MENU(ID_DELETE_DOWNLOAD, FbBookPanel::OnDeleteDownload)
	EVT_MENU(ID_FAVORITES_ADD, FbBookPanel::OnFavoritesAdd)
	EVT_MENU(ID_EDIT_COMMENTS, FbBookPanel::OnEditComments)
	EVT_MENU(ID_RATING_5, FbBookPanel::OnChangeRating)
	EVT_MENU(ID_RATING_4, FbBookPanel::OnChangeRating)
	EVT_MENU(ID_RATING_3, FbBookPanel::OnChangeRating)
	EVT_MENU(ID_RATING_2, FbBookPanel::OnChangeRating)
	EVT_MENU(ID_RATING_1, FbBookPanel::OnChangeRating)
	EVT_MENU(ID_RATING_0, FbBookPanel::OnChangeRating)
END_EVENT_TABLE()

FbBookPanel::FbBookPanel()
	:wxSplitterWindow(), m_BookInfo(NULL), m_folder(fbNO_FOLDER), m_selected(0)
{
}

bool FbBookPanel::Create(wxWindow *parent, const wxSize& size, long style, int keyType, int keyMode)
{
	bool res = wxSplitterWindow::Create(parent, wxID_ANY, wxDefaultPosition, size, wxSP_NOBORDER, wxT("bookspanel"));
	if (res) {
		SetMinimumPaneSize(50);
		SetSashGravity(0.5);
		m_BookList = new FbBookList(this, ID_BOOKS_LISTCTRL, style);
		CreateBookInfo( (bool) FbParams::GetValue(keyType) );
		CreateColumns( (bool) FbParams::GetValue(keyMode) ? FB2_MODE_TREE : FB2_MODE_LIST );
	}
	return res;
}

void FbBookPanel::CreateBookInfo(bool bVertical)
{
	if (m_BookInfo) Unsplit(m_BookInfo);

	m_BookInfo = new wxHtmlWindow(this, ID_BOOKS_INFO_PANEL, wxDefaultPosition, wxDefaultSize, wxSUNKEN_BORDER);
	int fontsizes[] = {6, 8, 9, 10, 12, 16, 18};
	m_BookInfo->SetFonts(wxT("Tahoma"), wxT("Tahoma"), fontsizes);

	if (bVertical)
		SplitVertically(m_BookList, m_BookInfo, GetSize().GetWidth()/2);
	else
		SplitHorizontally(m_BookList, m_BookInfo, GetSize().GetHeight()/2);

	BookTreeItemData * book = GetSelectedBook();
	if (!book) {
		m_BookInfo->SetPage(wxEmptyString);
	} else {
		InfoCash::LoadIcon(book->file_type);
		InfoCash::UpdateInfo(this, book->GetId(), bVertical);
	}
}

BookTreeItemData * FbBookPanel::GetSelectedBook()
{
	wxTreeItemId selected = m_BookList->GetSelection();
	if (selected.IsOk()) {
		return (BookTreeItemData*) m_BookList->GetItemData(selected);
	} else
		return NULL;
}

void FbBookPanel::OnBooksListViewSelected(wxTreeEvent & event)
{
	m_BookInfo->SetPage(wxEmptyString);

	wxTreeItemId selected = event.GetItem();
	if (selected.IsOk()) {
		BookTreeItemData * data = (BookTreeItemData*) m_BookList->GetItemData(selected);
		if (data) {
			InfoCash::LoadIcon(data->file_type);
			InfoCash::UpdateInfo(this, data->GetId(), GetSplitMode() == wxSPLIT_VERTICAL);
		}
	}
}

void FbBookPanel::OnBooksListCollapsing(wxTreeEvent & event)
{
	event.Veto();
}

void FbBookPanel::OnImageClick(wxTreeEvent &event)
{
	wxTreeItemId item = event.GetItem();
	if (item.IsOk()) {
		int image = m_BookList->GetItemImage(item);
		image = ( image == 1 ? 0 : 1);
		if (m_BookList->GetItemBold(item)) {
			m_BookList->SetItemImage(item, image);
			wxTreeItemIdValue cookie;
			item = m_BookList->GetFirstChild(item, cookie);
			while (item.IsOk()) {
				wxTreeItemIdValue cookie;
				wxTreeItemId subitem = m_BookList->GetFirstChild(item, cookie);
				while (subitem.IsOk()) {
					m_BookList->SetItemImage(subitem, image);
					subitem = m_BookList->GetNextSibling(subitem);
				}
				m_BookList->SetItemImage(item, image);
				item = m_BookList->GetNextSibling (item);
			}
		} else {
			m_BookList->SetItemImage(item, image);
			wxTreeItemId parent = m_BookList->GetItemParent(item);
			wxTreeItemIdValue cookie;
			item = m_BookList->GetFirstChild(parent, cookie);
			while (item.IsOk()) {
				if (image != m_BookList->GetItemImage(item)) {
					image = 2;
					break;
				}
				item = m_BookList->GetNextSibling (item);
			}
			m_BookList->SetItemImage(parent, image);
		}
	}
	event.Veto();
}

void FbBookPanel::OnBooksListActivated(wxTreeEvent & event)
{
	wxTreeItemId selected = event.GetItem();
	if (selected.IsOk()) {
		BookTreeItemData * data = (BookTreeItemData*)m_BookList->GetItemData(selected);
		if (data) FbManager::OpenBook(data->GetId(), data->file_type);
	}
}

void FbBookPanel::OnBooksListKeyDown(wxTreeEvent & event)
{
	if (event.GetKeyCode() == 0x20) {
		wxArrayTreeItemIds selections;
		size_t count = m_BookList->GetSelections(selections);
		int image = 0;
		for (size_t i=0; i<count; ++i) {
			wxTreeItemId selected = selections[i];
			if (i==0)
				image = (m_BookList->GetItemImage(selected) + 1) % 2;
			m_BookList->SetItemImage(selected, image);
		}
		event.Veto();
	} else
		event.Skip();
}

void FbBookPanel::OnInfoUpdate(wxCommandEvent& event)
{
	wxTreeItemId selected = m_BookList->GetSelection();
	if (selected.IsOk()) {
		BookTreeItemData * data= (BookTreeItemData*)m_BookList->GetItemData(selected);
		if (data && (data->GetId() == event.GetInt())) {
			wxString html = event.GetString();
			m_BookInfo->SetPage(html);
		}
	}
}

void FbBookPanel::OnSubmenu(wxCommandEvent& event)
{
	wxPostEvent(m_BookList, event);
}

void FbBookPanel::OnContextMenu(wxTreeEvent& event)
{
	wxPoint point = event.GetPoint();
	// If from keyboard
	if (point.x == -1 && point.y == -1) {
		wxSize size = GetSize();
		point.x = size.x / 3;
		point.y = size.y / 3;
	}
	ShowContextMenu(point, event.GetItem());
}

void FbBookPanel::ShowContextMenu(const wxPoint& pos, wxTreeItemId item)
{
	int id = 0;
	if (item.IsOk()) {
		BookTreeItemData * data = (BookTreeItemData*)m_BookList->GetItemData(item);
		if (data) id = data->GetId();
	}
	FbBookMenu menu(id, m_folder);
	menu.ConnectFolders(this, wxCommandEventHandler(FbBookPanel::OnFolderAdd));
	menu.ConnectAuthors(this, wxCommandEventHandler(FbBookPanel::OnOpenAuthor));
	PopupMenu(&menu, pos.x, pos.y);
}

void FbBookPanel::OnSelectAll(wxCommandEvent& event)
{
	m_BookList->SelectAll();
}

void FbBookPanel::OnUnselectAll(wxCommandEvent& event)
{
	m_BookList->SelectAll(0);
}

void FbBookPanel::OnOpenBook(wxCommandEvent & event)
{
	BookTreeItemData * data = GetSelectedBook();
	if (data) FbManager::OpenBook(data->GetId(), data->file_type);
}

class FbFolderUpdateThread: public wxThread
{
	public:
		FbFolderUpdateThread(const wxString &sql, const int folder, const FbFolderType type, const wxString &sql2 = wxEmptyString)
			:m_sql(sql), m_folder(folder), m_type(type), m_sql2(sql2) {};
	protected:
		static wxCriticalSection sm_queue;
		virtual void ExecSQL(FbDatabase &database, const wxString &sql);
		void * Entry();
		wxString m_sql;
		int m_folder;
		FbFolderType m_type;
		wxString m_sql2;
};

class FbCreateDownloadThread: public FbFolderUpdateThread
{
	public:
		FbCreateDownloadThread(const wxString &sql, const int folder, const FbFolderType type, const wxString &sql2 = wxEmptyString)
			:FbFolderUpdateThread(sql, folder, type, sql2) {};
	protected:
		void * Entry();
};

wxCriticalSection FbFolderUpdateThread::sm_queue;

void * FbFolderUpdateThread::Entry()
{
	wxCriticalSectionLocker locker(sm_queue);

	FbCommonDatabase database;
	database.AttachConfig();

	ExecSQL(database, m_sql);
	if (!m_sql2.IsEmpty()) ExecSQL(database, m_sql2);

	FbFolderEvent(ID_UPDATE_FOLDER, m_folder, m_type).Post();

	return NULL;
}

void FbFolderUpdateThread::ExecSQL(FbDatabase &database, const wxString &sql)
{
	try {
		database.ExecuteUpdate(sql);
	} catch (wxSQLite3Exception & e) {
		wxLogError(e.GetMessage());
	}
}


class FbIncrementFunction : public wxSQLite3ScalarFunction
{
	public:
		FbIncrementFunction(): m_increment(0) {};
		virtual void Execute(wxSQLite3FunctionContext& ctx);
	private:
		int m_increment;
};

void FbIncrementFunction::Execute(wxSQLite3FunctionContext& ctx)
{
	int argCount = ctx.GetArgCount();
	if (argCount != 1) {
		ctx.SetResultError(wxString::Format(_("INCREMENT called with wrong number of arguments: %d."), argCount));
		return;
	}

	m_increment++;
	int id = ctx.GetInt(0);
	id += m_increment;
	ctx.SetResult(id);
}

void * FbCreateDownloadThread::Entry()
{
	wxCriticalSectionLocker locker(sm_queue);

	FbCommonDatabase database;
	FbIncrementFunction function;
	database.CreateFunction(wxT("INCREMENT"), 1, function);
	database.AttachConfig();

	ExecSQL(database, m_sql);
	if (!m_sql2.IsEmpty()) ExecSQL(database, m_sql2);

	FbFolderEvent(ID_UPDATE_FOLDER, m_folder, m_type).Post();

	return NULL;
}

void FbBookPanel::OnFavoritesAdd(wxCommandEvent & event)
{
	DoFolderAdd( 0 );
}

void FbBookPanel::OnFolderAdd(wxCommandEvent& event)
{
	DoFolderAdd( FbBookMenu::GetFolder(event.GetId()) );
}

void FbBookPanel::DoFolderAdd(const int folder)
{
	wxString sel = m_BookList->GetSelected();
	wxString sql = wxString::Format(wxT("\
		INSERT INTO favorites(id_folder,md5sum) \
		SELECT DISTINCT %d, md5sum FROM books WHERE id IN (%s) \
	"), folder, sel.c_str());

	wxThread * thread = new FbFolderUpdateThread( sql, folder, FT_FOLDER );
	if ( thread->Create() == wxTHREAD_NO_ERROR ) thread->Run();
}

int FbBookPanel::UpdateChildRating(wxTreeItemId parent, int iRating, const wxString &sRating)
{
	int result = 0;
	wxTreeItemIdValue cookie;
	wxTreeItemId child = m_BookList->GetFirstChild(parent, cookie);
	while (child.IsOk()) {
		if (m_BookList->GetItemImage(child) == 1) {
			BookTreeItemData * data = (BookTreeItemData*) m_BookList->GetItemData(child);
			if (data && data->GetId()) {
				m_BookList->SetItemText(child, GetRatingColumn(), sRating);
				data->rating = iRating;
				result++;
			}
		}
		result += UpdateChildRating(child, iRating, sRating);
		child = m_BookList->GetNextChild(parent, cookie);
	}
	return result;
}

int FbBookPanel::UpdateSelectionRating(int iRating, const wxString &sRating)
{
	wxArrayTreeItemIds items;
	size_t count = m_BookList->GetSelections(items);
	for (size_t i=0; i<count; ++i) {
		BookTreeItemData * data = (BookTreeItemData*) m_BookList->GetItemData(items[i]);
		if (data && data->GetId()) {
			m_BookList->SetItemText(items[i], GetRatingColumn(), sRating);
			data->rating = iRating;
		}
	}
	return count;
}

void FbBookPanel::OnChangeRating(wxCommandEvent& event)
{
	int iRating = event.GetId() - ID_RATING_0;

	wxString sRating;
	if (iRating) sRating = wxT(" ") + strRating[iRating];
	int iUpdated = UpdateChildRating( m_BookList->GetRootItem(), iRating, sRating);
	if ( !iUpdated ) UpdateSelectionRating(iRating, sRating);

	m_BookList->Update();

	wxString sel = m_BookList->GetSelected();

	wxString sql1 = wxString::Format(wxT("\
		UPDATE states SET rating=%d WHERE md5sum IN \
		(SELECT DISTINCT md5sum FROM books WHERE id IN (%s)) \
	"), iRating, sel.c_str());

	wxString sql2 = wxString::Format(wxT("\
		INSERT INTO states(md5sum, rating) \
		SELECT DISTINCT md5sum, %d FROM books WHERE id IN (%s) \
		AND NOT EXISTS (SELECT rating FROM states WHERE states.md5sum = books.md5sum) \
	"), iRating, sel.c_str());

	wxThread * thread = new FbFolderUpdateThread( sql1, iRating, FT_RATING, sql2 );
	if ( thread->Create() == wxTHREAD_NO_ERROR ) thread->Run();
}

void FbBookPanel::DoDeleteDownload(const wxString &sel, const int folder)
{
	wxString sql1 = wxString::Format(wxT("\
		UPDATE states SET download=%d WHERE md5sum IN \
		(SELECT DISTINCT md5sum FROM books WHERE id>0 AND id IN (%s)) \
	"), folder, sel.c_str());

	wxThread * thread = new FbFolderUpdateThread( sql1, folder, FT_DOWNLOAD );
	if ( thread->Create() == wxTHREAD_NO_ERROR ) thread->Run();
}

void FbBookPanel::DoCreateDownload(const wxString &sel, const int folder)
{
	wxString sql1 = wxString::Format(wxT("\
		INSERT INTO states(md5sum, download) \
		SELECT DISTINCT md5sum, 1 FROM books WHERE id>0 AND id IN (%s) \
		AND NOT EXISTS (SELECT rating FROM states WHERE states.md5sum = books.md5sum) \
	"), sel.c_str());

	wxString sql2 = wxString::Format(wxT("\
		UPDATE states SET download=INCREMENT(%d) WHERE md5sum IN \
		(SELECT DISTINCT md5sum FROM books WHERE id>0 AND id IN (%s)) \
	"), folder, sel.c_str());

	wxThread * thread = new FbCreateDownloadThread( sql1, folder, FT_DOWNLOAD, sql2 );
	if ( thread->Create() == wxTHREAD_NO_ERROR ) thread->Run();
}

void FbBookPanel::OnDownloadBook(wxCommandEvent & event)
{
	wxString sel;
	size_t count = m_BookList->GetSelected(sel);
	if (count) {
		int folder = FbLocalDatabase().NewId(FB_NEW_DOWNLOAD, count) - count + 1;
		DoCreateDownload(sel, folder);
	}
}

void FbBookPanel::OnDeleteDownload(wxCommandEvent & event)
{
	wxString sel = m_BookList->GetSelected();
	DoDeleteDownload(sel, 0);
}

void FbBookPanel::OnEditComments(wxCommandEvent & event)
{
	wxTreeItemId selected = m_BookList->GetSelection();
	if (selected.IsOk()) {
		BookTreeItemData * data = (BookTreeItemData*)m_BookList->GetItemData(selected);
		if (data) new FbFrameHtml((wxAuiMDIParentFrame*)wxGetApp().GetTopWindow(), *data);
	}
}

void FbBookPanel::OnOpenAuthor(wxCommandEvent& event)
{
	int author = FbBookMenu::GetAuthor(event.GetId());
	if (author == 0) return;

	BookTreeItemData * data = GetSelectedBook();
	if (data) FbOpenEvent(ID_BOOK_AUTHOR, author, data->GetId()).Post();
}

void FbBookPanel::EmptyBooks(const int selected)
{
	m_AuthorItem = 0L;
	m_SequenceItem = 0L;
	if (selected) {
		m_selected = selected;
	} else {
		BookTreeItemData * data = GetSelectedBook();
		if (data) m_selected = data->GetId();
	}

	BookListUpdater updater(m_BookList);
	wxTreeItemId root = m_BookList->AddRoot(wxEmptyString);
	m_BookInfo->SetPage(wxEmptyString);
}

void FbBookPanel::AppendAuthor(const wxString title)
{
	FbTreeListUpdater updater(m_BookList);
	wxTreeItemId parent = m_BookList->GetRootItem();
	m_AuthorItem = m_BookList->AppendItem(parent, title, 0);
	m_BookList->SetItemBold(m_AuthorItem, true);
	m_BookList->Expand(parent);
}

void FbBookPanel::AppendSequence(const wxString title)
{
	wxString text = title.IsEmpty() ? strOtherSequence : title;
	FbTreeListUpdater updater(m_BookList);
	wxTreeItemId parent = m_AuthorItem.IsOk() ? m_AuthorItem : m_BookList->GetRootItem();
	m_SequenceItem = m_BookList->AppendItem(parent, text, 0);
	m_BookList->SetItemBold(m_SequenceItem, true);
	m_BookList->Expand(parent);
}

wxString FbBookPanel::F(const int number)
{
	int hi = number / 1000;
	int lo = number % 1000;
	if (hi)
		return F(hi) + wxT(" ") + wxString::Format(wxT("%03d"), lo);
	else
		return wxString::Format(wxT("%d"), lo);
}

void FbBookPanel::AppendBook(BookTreeItemData * data, const wxString & authors)
{
	FbTreeListUpdater updater(m_BookList);

	wxString file_type = data->file_type + wxT(" ");
	wxString file_size = F(data->file_size/1024) + wxT(" ");
	wxTreeItemId parent;
	wxString sRating;
	if (data->rating) sRating = wxT(" ") + strRating[data->rating];

	wxTreeItemId item;
	switch (m_ListMode) {
		case FB2_MODE_TREE: {
			parent = m_SequenceItem.IsOk() ? m_SequenceItem : ( m_AuthorItem.IsOk() ? m_AuthorItem : m_BookList->GetRootItem() );
			item = m_BookList->AppendItem(parent, data->title, 0, -1, data);
			m_BookList->SetItemText(item, 1, sRating);
			if (data->number) m_BookList->SetItemText(item, 2, wxString::Format(wxT(" %d "), data->number));
			m_BookList->SetItemText(item, 3, file_type);
			m_BookList->SetItemText(item, 4, file_size);
		} break;
		case FB2_MODE_LIST: {
			parent = m_BookList->GetRootItem();
			item = m_BookList->AppendItem(parent, data->title, 0, -1, data);
			m_BookList->SetItemText(item, 1, authors);
			m_BookList->SetItemText(item, 2, sRating);
			m_BookList->SetItemText(item, 3, file_type);
			m_BookList->SetItemText(item, 4, file_size);
		} break;
	}
	m_BookList->Expand(parent);
	if (data->GetId() == m_selected) m_BookList->SelectItem(item);
}

void FbBookPanel::CreateColumns(FbListMode mode)
{
	m_ListMode = mode;

	m_AuthorItem = 0L;
	m_SequenceItem = 0L;

	BookListUpdater updater(m_BookList);

	m_BookList->EmptyCols();

	switch (m_ListMode) {
		case FB2_MODE_TREE: {
			m_BookList->AddColumn (_("Заголовок"), 13, wxALIGN_LEFT);
			m_BookList->AddColumn (_("Рейтинг"), 3, wxALIGN_LEFT);
			m_BookList->AddColumn (_("№"), 2, wxALIGN_RIGHT);
			m_BookList->AddColumn (_("Тип"), 2, wxALIGN_RIGHT);
			m_BookList->AddColumn (_("Размер, Кб"), 3, wxALIGN_RIGHT);
		} break;
		case FB2_MODE_LIST: {
			m_BookList->AddColumn (_("Заголовок"), 9, wxALIGN_LEFT);
			m_BookList->AddColumn (_("Автор"), 6, wxALIGN_LEFT);
			m_BookList->AddColumn (_("Рейтинг"), 3, wxALIGN_LEFT);
			m_BookList->AddColumn (_("Тип"), 2, wxALIGN_RIGHT);
			m_BookList->AddColumn (_("Размер, Кб"), 3, wxALIGN_RIGHT);
		} break;
	}

	wxTreeItemId root = m_BookList->AddRoot(wxEmptyString);
	m_BookInfo->SetPage(wxEmptyString);
}

int FbBookPanel::GetRatingColumn()
{
	switch (m_ListMode) {
		case FB2_MODE_TREE: return 1;
		case FB2_MODE_LIST: return 2;
	}
	return 1;
}

void FbBookPanel::OnSystemDownload(wxCommandEvent & event)
{
	BookTreeItemData * data = GetSelectedBook();
	if (data && data->GetId()>0) {
		wxString url = FbDownloader::GetURL(data->GetId());
		wxLaunchDefaultBrowser(url);
	}
}