// This file is part of RSS Guard.
//
// Copyright (C) 2011-2017 by Martin Rotter <rotter.martinos@gmail.com>
// Copyright (C) 2010-2014 by David Rosca <nowrep@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 "network-web/adblock/adblockdialog.h"
#include "network-web/adblock/adblockmanager.h"
#include "network-web/adblock/adblocksubscription.h"
#include "network-web/adblock/adblocktreewidget.h"
#include "network-web/adblock/adblockaddsubscriptiondialog.h"

#include "network-web/webfactory.h"

#include <QMenu>
#include <QTimer>
#include <QMessageBox>
#include <QInputDialog>


AdBlockDialog::AdBlockDialog(QWidget* parent)
	: QWidget(parent), m_ui(new Ui::AdBlockDialog), m_manager(AdBlockManager::instance()), m_currentTreeWidget(0), m_currentSubscription(0), m_loaded(false) {
	setAttribute(Qt::WA_DeleteOnClose);
	m_ui->setupUi(this);
#ifdef Q_OS_MACOS
	m_ui->tabWidget->setDocumentMode(false);
#endif
	m_ui->adblockCheckBox->setChecked(m_manager->isEnabled());
	QMenu* menu = new QMenu(m_ui->buttonOptions);
	m_actionAddRule = menu->addAction(tr("Add rule"), this, SLOT(addRule()));
	m_actionRemoveRule = menu->addAction(tr("Remove rule"), this, SLOT(removeRule()));
	menu->addSeparator();
	m_actionAddSubscription = menu->addAction(tr("Add subscription"), this, SLOT(addSubscription()));
	m_actionRemoveSubscription = menu->addAction(tr("Remove subscription"), this, SLOT(removeSubscription()));
	menu->addAction(tr("Update subscriptions"), m_manager, SLOT(updateAllSubscriptions()));
	menu->addSeparator();
	menu->addAction(tr("Learn about writing rules..."), this, SLOT(learnAboutRules()));
	m_ui->buttonOptions->setMenu(menu);
	connect(menu, SIGNAL(aboutToShow()), this, SLOT(aboutToShowMenu()));
	connect(m_ui->adblockCheckBox, SIGNAL(toggled(bool)), this, SLOT(enableAdBlock(bool)));
	connect(m_ui->tabWidget, SIGNAL(currentChanged(int)), this, SLOT(currentChanged(int)));
	connect(m_ui->buttonBox, &QDialogButtonBox::clicked, this, &AdBlockDialog::close);
	load();
	m_ui->buttonBox->setFocus();
}

void AdBlockDialog::showRule(const AdBlockRule* rule) const {
	AdBlockSubscription* subscription = rule->subscription();

	if (!subscription) {
		return;
	}

	for (int i = 0; i < m_ui->tabWidget->count(); ++i) {
		AdBlockTreeWidget* treeWidget = qobject_cast<AdBlockTreeWidget*>(m_ui->tabWidget->widget(i));

		if (subscription == treeWidget->subscription()) {
			treeWidget->showRule(rule);
			m_ui->tabWidget->setCurrentIndex(i);
			break;
		}
	}
}

void AdBlockDialog::addRule() {
	m_currentTreeWidget->addRule();
}

void AdBlockDialog::removeRule() {
	m_currentTreeWidget->removeRule();
}

void AdBlockDialog::addSubscription() {
	AdBlockAddSubscriptionDialog dialog(this);

	if (dialog.exec() != QDialog::Accepted) {
		return;
	}

	QString title = dialog.title();
	QString url = dialog.url();

	if (AdBlockSubscription* subscription = m_manager->addSubscription(title, url)) {
		AdBlockTreeWidget* tree = new AdBlockTreeWidget(subscription, m_ui->tabWidget);
		int index = m_ui->tabWidget->insertTab(m_ui->tabWidget->count() - 1, tree, subscription->title());
		m_ui->tabWidget->setCurrentIndex(index);
	}
}

void AdBlockDialog::removeSubscription() {
	if (m_manager->removeSubscription(m_currentSubscription)) {
		delete m_currentTreeWidget;
	}
}

void AdBlockDialog::currentChanged(int index) {
	if (index != -1) {
		m_currentTreeWidget = qobject_cast<AdBlockTreeWidget*>(m_ui->tabWidget->widget(index));
		m_currentSubscription = m_currentTreeWidget->subscription();
	}
}

void AdBlockDialog::enableAdBlock(bool state) {
	m_manager->setEnabled(state);

	if (state) {
		load();
	}
}

void AdBlockDialog::aboutToShowMenu() {
	bool subscriptionEditable = m_currentSubscription && m_currentSubscription->canEditRules();
	bool subscriptionRemovable = m_currentSubscription && m_currentSubscription->canBeRemoved();
	m_actionAddRule->setEnabled(subscriptionEditable);
	m_actionRemoveRule->setEnabled(subscriptionEditable);
	m_actionRemoveSubscription->setEnabled(subscriptionRemovable);
}

void AdBlockDialog::learnAboutRules() {
	WebFactory::instance()->openUrlInExternalBrowser(QSL("http://adblockplus.org/en/filters"));
}

void AdBlockDialog::loadSubscriptions() {
	for (int i = 0; i < m_ui->tabWidget->count(); ++i) {
		AdBlockTreeWidget* treeWidget = qobject_cast<AdBlockTreeWidget*>(m_ui->tabWidget->widget(i));
		treeWidget->refresh();
	}
}

void AdBlockDialog::load() {
	if (m_loaded || !m_ui->adblockCheckBox->isChecked()) {
		return;
	}

	foreach (AdBlockSubscription* subscription, m_manager->subscriptions()) {
		AdBlockTreeWidget* tree = new AdBlockTreeWidget(subscription, m_ui->tabWidget);
		m_ui->tabWidget->addTab(tree, subscription->title());
	}

	m_loaded = true;
	QTimer::singleShot(50, this, SLOT(loadSubscriptions()));
}