// This file is part of RSS Guard.
//
// Copyright (C) 2011-2017 by Martin Rotter <rotter.martinos@gmail.com>
//
// RSS Guard is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// RSS Guard is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with RSS Guard. If not, see <http://www.gnu.org/licenses/>.

#include "services/abstract/recyclebin.h"

#include "miscellaneous/application.h"
#include "miscellaneous/iconfactory.h"
#include "miscellaneous/textfactory.h"
#include "miscellaneous/databasequeries.h"
#include "services/abstract/serviceroot.h"

#include <QThread>


RecycleBin::RecycleBin(RootItem* parent_item) : RootItem(parent_item), m_totalCount(0),
	m_unreadCount(0), m_contextMenu(QList<QAction*>()) {
	setKind(RootItemKind::Bin);
	setId(ID_RECYCLE_BIN);
	setIcon(qApp->icons()->fromTheme(QSL("user-trash")));
	setTitle(tr("Recycle bin"));
	setDescription(tr("Recycle bin contains all deleted messages from all feeds."));
	setCreationDate(QDateTime::currentDateTime());
}

RecycleBin::~RecycleBin() {
}

int RecycleBin::countOfUnreadMessages() const {
	return m_unreadCount;
}

int RecycleBin::countOfAllMessages() const {
	return m_totalCount;
}

void RecycleBin::updateCounts(bool update_total_count) {
	bool is_main_thread = QThread::currentThread() == qApp->thread();
	QSqlDatabase database = is_main_thread ?
	                        qApp->database()->connection(metaObject()->className(), DatabaseFactory::FromSettings) :
	                        qApp->database()->connection(QSL("feed_upd"), DatabaseFactory::FromSettings);
	m_unreadCount = DatabaseQueries::getMessageCountsForBin(database, getParentServiceRoot()->accountId(), false);

	if (update_total_count) {
		m_totalCount = DatabaseQueries::getMessageCountsForBin(database, getParentServiceRoot()->accountId(), true);
	}
}

QVariant RecycleBin::data(int column, int role) const {
	switch (role) {
		case Qt::ToolTipRole:
			return tr("Recycle bin\n\n%1").arg(tr("%n deleted message(s).", 0, countOfAllMessages()));

		default:
			return RootItem::data(column, role);
	}
}

QList<QAction*> RecycleBin::contextMenu() {
	if (m_contextMenu.isEmpty()) {
		QAction* restore_action = new QAction(qApp->icons()->fromTheme(QSL("recycle-bin-restore-all")),
		                                      tr("Restore recycle bin"),
		                                      this);
		QAction* empty_action = new QAction(qApp->icons()->fromTheme(QSL("recycle-bin-empty")),
		                                    tr("Empty recycle bin"),
		                                    this);
		connect(restore_action, &QAction::triggered, this, &RecycleBin::restore);
		connect(empty_action, &QAction::triggered, this, &RecycleBin::empty);
		m_contextMenu.append(restore_action);
		m_contextMenu.append(empty_action);
	}

	return m_contextMenu;
}

QList<Message> RecycleBin::undeletedMessages() const {
	const int account_id = getParentServiceRoot()->accountId();
	QSqlDatabase database = qApp->database()->connection(metaObject()->className(), DatabaseFactory::FromSettings);
	return DatabaseQueries::getUndeletedMessagesForBin(database, account_id);
}

bool RecycleBin::markAsReadUnread(RootItem::ReadStatus status) {
	QSqlDatabase database = qApp->database()->connection(metaObject()->className(), DatabaseFactory::FromSettings);
	ServiceRoot* parent_root = getParentServiceRoot();

	if (DatabaseQueries::markBinReadUnread(database, parent_root->accountId(), status)) {
		updateCounts(false);
		parent_root->itemChanged(QList<RootItem*>() << this);
		parent_root->requestReloadMessageList(status == RootItem::Read);
		return true;
	}

	else {
		return false;
	}
}

bool RecycleBin::cleanMessages(bool clear_only_read) {
	QSqlDatabase database = qApp->database()->connection(metaObject()->className(), DatabaseFactory::FromSettings);
	ServiceRoot* parent_root = getParentServiceRoot();

	if (DatabaseQueries::purgeMessagesFromBin(database, clear_only_read, parent_root->accountId())) {
		updateCounts(true);
		parent_root->itemChanged(QList<RootItem*>() << this);
		parent_root->requestReloadMessageList(true);
		return true;;
	}

	else {
		return false;
	}
}

bool RecycleBin::empty() {
	return cleanMessages(false);
}

bool RecycleBin::restore() {
	QSqlDatabase database = qApp->database()->connection(metaObject()->className(), DatabaseFactory::FromSettings);
	ServiceRoot* parent_root = getParentServiceRoot();

	if (DatabaseQueries::restoreBin(database, parent_root->accountId())) {
		parent_root->updateCounts(true);
		parent_root->itemChanged(parent_root->getSubTree());
		parent_root->requestReloadMessageList(true);
		return true;
	}

	else {
		return false;
	}
}