#include <QDesktopServices>

#include <QFileInfo>

#include <QUrl>

#include "RemoteDock.h"

#include "ui_RemoteDock.h"

#include "Settings.h"

#include "RemoteDatabase.h"

#include "RemoteModel.h"

#include "MainWindow.h"

#include "RemotePushDialog.h"

#include "PreferencesDialog.h"

RemoteDock::RemoteDock(MainWindow* parent)

: QDialog(parent),

ui(new Ui::RemoteDock),

mainWindow(parent),

remoteDatabase(parent->getRemote()),

remoteModel(new RemoteModel(this, parent->getRemote()))

{

ui->setupUi(this);

// Set up model

ui->treeStructure->setModel(remoteModel);

// Reload the directory tree when a database upload has finished

connect(&remoteDatabase, &RemoteDatabase::uploadFinished, this, &RemoteDock::setNewIdentity);

// Whenever a new directory listing has been parsed, check if it was a new root dir and, if so, open the user's directory

connect(remoteModel, &RemoteModel::directoryListingParsed, this, &RemoteDock::newDirectoryNode);

// When the Preferences link is clicked in the no-certificates-label, open the preferences dialog. For other links than the ones we know,

// just open them in a web browser

connect(ui->labelNoCert, &QLabel::linkActivated, [this](const QString& link) {

if(link == "#preferences")

{

PreferencesDialog dialog(mainWindow, PreferencesDialog::TabRemote);

if(dialog.exec())

mainWindow->reloadSettings();

} else {

QDesktopServices::openUrl(QUrl(link));

}

});

// Initial setup

reloadSettings();

}

RemoteDock::~RemoteDock()

{

delete ui;

}

void RemoteDock::reloadSettings()

{

// Load list of client certs

ui->comboUser->clear();

QStringList client_certs = Settings::getValue("remote", "client_certificates").toStringList();

for(const QString& file : client_certs)

{

auto certs = QSslCertificate::fromPath(file);

for(const QSslCertificate& cert : certs)

ui->comboUser->addItem(cert.subjectInfo(QSslCertificate::CommonName).at(0), file);

}

// Add public certificate for anonymous read-only access to dbhub.io

ui->comboUser->addItem(tr("Public"), ":/user_certs/public.cert.pem");

}

void RemoteDock::setNewIdentity()

{

// Get identity

QString identity = ui->comboUser->currentText();

if(identity.isEmpty())

return;

// Get certificate file name

QString cert = ui->comboUser->itemData(ui->comboUser->findText(identity), Qt::UserRole).toString();

if(cert.isEmpty())

return;

// Open root directory. Get host name from client cert

QString host = remoteDatabase.getInfoFromClientCert(cert, RemoteDatabase::CertInfoServer);

remoteModel->setNewRootDir(QString("https://%1:5550/").arg(host), cert);

// Enable buttons if necessary

enableButtons();

}

void RemoteDock::fetchDatabase(const QModelIndex& idx)

{

if(!idx.isValid())

return;

// Get item

const RemoteModelItem* item = remoteModel->modelIndexToItem(idx);

// Only open database file

if(item->value(RemoteModelColumnType).toString() == "database")

remoteDatabase.fetch(item->value(RemoteModelColumnUrl).toString(), RemoteDatabase::RequestTypeDatabase, remoteModel->currentClientCertificate());

}

void RemoteDock::enableButtons()

{

bool db_opened = mainWindow->getDb().isOpen();

bool logged_in = !remoteModel->currentClientCertificate().isEmpty();

ui->buttonPushDatabase->setEnabled(db_opened && logged_in);

}

void RemoteDock::pushDatabase()

{

// If the currently active identity is the read-only public access to dbhub.io, don't show the Push Database dialog because it won't work anyway.

// Instead switch to an explanation offering some advice to create and import a proper certificate.

if(remoteModel->currentClientCertificate() == ":/user_certs/public.cert.pem")

{

ui->stack->setCurrentIndex(1);

return;

}

// The default suggestion for a database name is the local file name. If it is a remote file (like when it initially was fetched using DB4S),

// the extra bit of information at the end of the name gets removed first.

QString name = QFileInfo(mainWindow->getDb().currentFile()).fileName();

name = name.remove(QRegExp("_[0-9]+.remotedb$"));

// Show the user a dialog for setting all the commit details

QString host = QString("https://%1:5550/").arg(remoteDatabase.getInfoFromClientCert(remoteModel->currentClientCertificate(), RemoteDatabase::CertInfoServer));

RemotePushDialog pushDialog(this, remoteDatabase, host, remoteModel->currentClientCertificate(), name);

if(pushDialog.exec() != QDialog::Accepted)

return;

// Build push URL

QString url = host;

url.append(remoteDatabase.getInfoFromClientCert(remoteModel->currentClientCertificate(), RemoteDatabase::CertInfoUser));

url.append("/");

url.append(pushDialog.name());

// Push database

remoteDatabase.push(mainWindow->getDb().currentFile(), url, remoteModel->currentClientCertificate(), pushDialog.name(),

pushDialog.commitMessage(), pushDialog.licence(), pushDialog.isPublic(), pushDialog.branch(), pushDialog.forcePush());

}

void RemoteDock::newDirectoryNode(const QModelIndex& parent)

{

// Was this a new root dir?

if(!parent.isValid())

{

// Then check if there is a directory with the current user name

// Get current user name

QString user = remoteDatabase.getInfoFromClientCert(remoteModel->currentClientCertificate(), RemoteDatabase::CertInfoUser);

for(int i=0;i<remoteModel->rowCount(parent);i++)

{

QModelIndex child = remoteModel->index(i, RemoteModelColumnName, parent);

if(child.data().toString() == user)

ui->treeStructure->expand(child);

}

}

}

void RemoteDock::reject()

{

// We override this, to ensure the Escape key doesn't make this dialog

// dock go away

return;

}

void RemoteDock::switchToMainView()

{

ui->stack->setCurrentIndex(0);

}