mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "QDebug"
#include "QFileDialog"
#include "QFileInfo"
#include "QMessageBox"
#include "QProgressDialog"
#include "QtSql/QSqlDriver"
#include "QtSql/QSqlDatabase"
#include "QtSql/QSqlError"
#include "QtSql/QSqlQuery"
#include "QThread"

const QList<QString> QLIST_DB_FIELD_NAMES = { "id", "account_id", "guid", "rating",
																							"view_offset", "view_count", "last_viewed_at",
																							"created_at", "updated_at", "skip_count",
																							"last_skipped_at", "changed_at", "extra_data",
																							"last_rated_at" };

MainWindow::MainWindow(QWidget *parent)
  : QMainWindow(parent)
  , ui(new Ui::MainWindow)
{
  ui->setupUi(this);
}

MainWindow::~MainWindow()
{
  delete ui;
}

bool MainWindow::createDBConnection(QString dbFile)
{
  bool success = false;
  if (dbFile.isEmpty() == false)
  {
      // check if dbFile is an existant file
      if (QFileInfo::exists(dbFile) && QFileInfo(dbFile).isFile())
      {
          // dbFile exists and is a file (not a directory) so attempt to db connection to it
          qDebug() << "source:" << ui->leSource->text();
          // for some reason, the addDatabase call has to be a static call on the next line
          qsqlDB = QSqlDatabase::addDatabase("QSQLITE");
					qsqlDB.setDatabaseName(dbFile);
          bool success = qsqlDB.open();
          qDebug() << "db connection was:" << success;
					return success;
      }
  }
  else
  {
      // dbFile argument passed was empty
      qDebug() << "Argument dbFile was empty: '" << dbFile <<"'";
  }
  return success;
}

void MainWindow::on_btnSource_clicked()
{
  QString qstrSource = QFileDialog::getOpenFileName(this,
      tr("Select Source file..."), QDir::toNativeSeparators(QDir::homePath()), tr("Plex Database File (*.plugins.library.db*);;Any File (*.*)"));
  ui->leSource->setText(qstrSource);
  qDebug() << "qstrSource:" << qstrSource;
}

void MainWindow::on_btnDest_clicked()
{
	// dev-note: QFileDialog::getSaveFileName can be used to 'select' a non-existant file using a standard file selection dialog
	// reference: https://stackoverflow.com/questions/29907625/qfiledialog-using-getopenfilename-allow-for-non-existent-files/29907690#29907690
	QString qstrDest = QFileDialog::getSaveFileName(this,
			tr("Select Destination file..."),
			QDir::toNativeSeparators(QDir::homePath()),
			tr("Plex Database File (*.plugins.library.db*);;Any File (*.*)"),
			nullptr,
			QFileDialog::DontConfirmOverwrite);
  ui->leDest->setText(qstrDest);
  qDebug() << "qstrDest:" << qstrDest;
}

void MainWindow::on_btnLaunch_clicked()
{
// check the value of ui->coboAction and act accordingly
  if (ui->coboAction->currentText() == "Export from Source to Destination")
  {
			// exporting from database, attempt connection
			if (MainWindow::createDBConnection(ui->leSource->text()))
			{
					int iRowsTotal = 0;
					QSqlQuery query;
					query.prepare("SELECT * FROM metadata_item_settings");
					if (query.exec())
					{
						QFile qfileOutFile(ui->leDest->text());
						if (qfileOutFile.open(QIODevice::WriteOnly | QIODevice::Text))
						{
							QTextStream qtstOut(&qfileOutFile);
							qDebug() << "db query succeeded, numRowsAffected:" << query.numRowsAffected();
							// loop through the database query results
							query.first();
							do
							{
								 bool okCheck;
								 iRowsTotal++;
								 QString o = "INSERT OR REPLACE INTO metadata_item_settings VALUES (";
								 // loop through the fieldNames
								 for (int i = 0; i < QLIST_DB_FIELD_NAMES.size(); ++i)
								 {
									 QString qstrFieldValue = query.value(i).toString();
									 if (qstrFieldValue.toInt(&okCheck, 10))
									 {
											 o.append(qstrFieldValue + ", ");
									 }
									 else if (qstrFieldValue.isEmpty())
									 {
										 o.append("NULL, ");
									 }
									 else
									 {
										 o.append("'" + qstrFieldValue + "', ");
									 }
								 }
								 o.chop(2);
								 o.append(");");
								 qtstOut << o << "\n";
							} while (query.next());
							qfileOutFile.close();
							QString qstrCompleteMsg =	"Total records exported: " + QString::number(iRowsTotal);
							// the qbytearray is used to cast the QString to a char string so that we can use it in the tr() call for the QMessageBox
							QByteArray ba = qstrCompleteMsg.toLocal8Bit();
							const char *cstrCompleteMsg = ba.data();
							QMessageBox::information(this, tr("Export Complete"),
																						 tr(cstrCompleteMsg),
																						 QMessageBox::Ok);
						}
						else
						{
							qDebug() << "Oops! Could not open output file '" << ui->leDest->text() << "' for writing.";
							QMessageBox::warning(	this,
																		tr("Open file failed"),
																		tr("Unable to open the destination file for writing.<br/>Please check the file's permissions or existence."),
																		QMessageBox::Ok);
						}
					}
					else
					{
							qDebug() << "db query failed, error was: " << query.lastError();
							QMessageBox::warning(	this,
																		tr("Database query failed"),
																		tr("Query to the SQLite database failed. No data could be read from the database file so there's nothing to export.<br/>Please check the file's permissions, format or existence."),
																		QMessageBox::Ok);
					}
					qDebug() << "Total Rows Processed:" << iRowsTotal;
			}
			else
			{
				// database connection failed - should prolly let the user nkow about that?
				QMessageBox::warning(	this,
															tr("Database connection failed"),
															tr("Unable to connect to the SQLite database.<br/>Please check the file's permissions, format or existence."),
															QMessageBox::Ok);
			}
  }
  else if (ui->coboAction->currentText() == "Import from Source to Destination")
  {
// importing from database, attempt connection
			if (MainWindow::createDBConnection(ui->leDest->text()))
			{
				int iRowsTotal	= 0;
				int iRowsOK		= 0;
				int iRowsFail	= 0;
				QSqlQuery query = QSqlQuery(qsqlDB);
				QFile qfileInFile(ui->leSource->text());
				qint64 filesize = qfileInFile.size();
				int readBytes = 0;

// open the sourceFile and loop it, pushing contents into destFile
				if (qfileInFile.open(QIODevice::ReadOnly | QIODevice::Text))
				{
					QByteArray line;
					QProgressDialog progress("Importing data...", "Cancel", 0, 100, this);
					progress.setWindowModality(Qt::WindowModal);

					while (!qfileInFile.atEnd())
					{
						iRowsTotal++;
						line = qfileInFile.readLine();
						readBytes += line.length();
						qDebug() << "Query: " + line;
						query.prepare(line);
						if (query.exec(line))
						{
							iRowsOK++;
						}
						else
						{
							iRowsFail++;
							qDebug() << query.lastError();
						}
						progress.setValue((int)((readBytes * 100) / filesize));
						if (progress.wasCanceled())
							break;
					}
					QString qstrCompleteMsg =	"Total lines processed: " + QString::number(iRowsTotal) + "\n" +
																		"Records imported: " + QString::number(iRowsOK) + "\n" +
																		"Lines failed: " + QString::number(iRowsFail);
					// the qbytearray is used to cast the QString to a char string so that we can use it in the tr() call for the QMessageBox
					QByteArray ba = qstrCompleteMsg.toLocal8Bit();
					const char *cstrCompleteMsg = ba.data();
					QMessageBox::information(this, tr("Import Complete"),
																				 tr(cstrCompleteMsg),
																				 QMessageBox::Ok);
				}
				else
				{
					// file open for reading failed - should prolly let user know about that?
					QMessageBox::warning(	this,
																tr("Open file failed"),
																tr("Unable to open the source file.<br/>Please check the file's permissions, format or existence."),
																QMessageBox::Ok);
				}

			}
			else
			{
				// database connection failed - should prolly let the user nkow about that?
				QMessageBox::warning(	this,
															tr("Database connection failed"),
															tr("Unable to connect to the SQLite database.<br/>Please check the file's permissions, format or existence."),
															QMessageBox::Ok);
			}
  }
  else
  {
      // this should never happen unless there is a bug
      qDebug() << "some how this happened... ???";
  }
	qsqlDB.close();
}

void MainWindow::on_actionAbout_triggered()
{
		QMessageBox::about(this, "PMSIE::About", "PMSIE is a small utility to export/import watched media metadata from/to Plex Media Server's database.<br/><br/>Written by Cabji, 2020.<br/><br/><a href=\"https://github.com/Cabji/qtPMSIE\">https://github.com/Cabji/qtPMSIE</a><br/><br/>Free for all, provided as is, no warranty or liability.");
}

void MainWindow::on_actionHelp_triggered()
{
		QMessageBox::about(this, "PMSIE::Help", "For help, please see <a href=\"http://www.github.com/Cabji/qtPMSIE/help\">PMSIE Help Online</a>.");
}