#include "fbgui.h"
#include "sysinfo.h"
#include "loggerengine.h"
#include "downloadmanager.h"
#include "javascriptinterface.h"
#include "sysinfolibsysfs.h"
#include <iostream>
#include <QtWebKit>
#include <QxtCore>
QString serialLocation("");
QString fileToTriggerURL("");
QString sessionID("");
QUrl baseURL("");
QString binPath("");
QString downloadPath("");
int updateInterval = -1;
int debugMode = -1;
//-------------------------------------------------------------------------------------------
fbgui::fbgui()
{
// test for libsys function
SysInfoLibsysfs* sil = new SysInfoLibsysfs();
sil->getInfoAboutNetworkInterface();
sil->getInfoAboutClassNet();
// start basic console debug log
qxtLog->disableLoggerEngine("DEFAULT");
qxtLog->enableLogLevels(QxtLogger::DebugLevel);
if (debugMode == 0 || debugMode == 1){
qxtLog->addLoggerEngine("std_logger", new LoggerEngine_std);
qxtLog->initLoggerEngine("std_logger");
qxtLog->setMinimumLevel("std_logger", QxtLogger::DebugLevel);
qxtLog->debug() << "Initializing fbgui...";
}
// setup layout of the gui: debug split or browser
_webView = new QWebView(this);
if (debugMode == 1) setupDebugSplit();
else setCentralWidget(_webView);
createActions();
// initialize javascript interface
JavascriptInterface* jsi = new JavascriptInterface(_webView->page()->mainFrame());
QObject::connect(jsi, SIGNAL(quitFbgui()), this, SLOT(close()));
QObject::connect(_webView->page()->mainFrame(), SIGNAL(javaScriptWindowObjectCleared()),
jsi, SLOT(attachToDOM()));
// initialize download manager
DownloadManager* dm = new DownloadManager();
QObject::connect(dm, SIGNAL(downloadInfo(const QString&, const double&)),
jsi, SLOT(downloadInfo(const QString&, const double&)));
QObject::connect(dm, SIGNAL(notify(const QString&)),
jsi, SLOT(notify(const QString&)));
QObject::connect(jsi, SIGNAL(requestFile(const QString&)), dm, SLOT(downloadFile(const QString&)));
QObject::connect(dm, SIGNAL(updateProgress(const int&, const double&, const QString&)),
jsi, SLOT(updateProgressBar(const int&, const double&, const QString&)));
QObject::connect(dm, SIGNAL(downloadQueueEmpty()), jsi, SLOT(callbackOnFinished()));
// load filler page & start watching for fileToTriggerURL
_webView->load(QUrl("qrc:/html/loadAbout.html"));
watchForTrigger();
// set properties
setWindowTitle("fbgui");
setAttribute(Qt::WA_QuitOnClose, true);
setWindowFlags(Qt::FramelessWindowHint);
showFullScreen();
}
//-------------------------------------------------------------------------------------------
void fbgui::createActions()
{
// CTRL + X to kill the gui
_quit = new QAction(tr("&quit"), this);
_quit->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_X));
this->addAction(_quit);
connect(_quit, SIGNAL(triggered()), this, SLOT(close()));
}
//-------------------------------------------------------------------------------------------
// File system watching
//-------------------------------------------------------------------------------------------
void fbgui::watchForTrigger()
{
// check if the directory to fileToTriggerURL exists
QFileInfo fi(fileToTriggerURL);
if (!fi.absoluteDir().exists()){
qxtLog->debug() << "[watcher] " << fi.absolutePath() << " does not exists!";
if (QDir::home().mkdir(fi.absolutePath()))
qxtLog->debug() << "[watcher] Successfully created " << fi.absolutePath();
else
qxtLog->debug() << "[watcher] Failed to create " << fi.absolutePath();
}
else
qxtLog->debug() << "[watcher] " << fi.absolutePath() << " exists!";
// check if fileToTriggerURL already exists
if (fi.exists()){
qxtLog->debug() << "[watcher] " << fileToTriggerURL << " exists already!";
// try to remove it
QFile file(fi.absoluteFilePath());
if (file.remove())
qxtLog->debug() << "[watcher] Purged: " << file.fileName();
else{
qxtLog->debug() << "[watcher] Could not remove: " << file.fileName();
exit(EXIT_FAILURE);
}
}
else {
// watch the path where trigger file is expected
qxtLog->debug() << "[watcher] Watching " << fi.absolutePath() << " for file: " << fi.fileName();
QStringList pathToWatch(fi.absolutePath());
_watcher = new QFileSystemWatcher(pathToWatch, this);
QObject::connect(_watcher, SIGNAL(directoryChanged(const QString&)), this, SLOT(checkForTrigger(const QString&)));
}
}
//-------------------------------------------------------------------------------------------
void fbgui::checkForTrigger(const QString& dirname)
{
// check if fileToTriggerURL exists in the directory where the change occured
QFileInfo tfi(fileToTriggerURL);
QFileInfo fi(dirname + "/" + tfi.fileName());
if (fi.exists()){
qxtLog->debug() << "[watcher] " << fileToTriggerURL << " detected.";
loadURL();
}
else
// do nothing / keep watching
qxtLog->debug() << "[watcher] weird file!";
}
//-------------------------------------------------------------------------------------------
// preparing URL loading
//-------------------------------------------------------------------------------------------
void fbgui::loadURL(){
// load if host is valid
if (checkHost()){
qxtLog->debug() << "[gui] Loading URL...";
QByteArray postData = generatePOSTData();
qxtLog->debug() << "[gui] POST data: " << postData;
QNetworkRequest req(baseURL);
_webView->load(req, QNetworkAccessManager::PostOperation, postData);
}
// disconnect _watcher, his job is done
qxtLog->debug() << "[watcher] see you soon!";
_watcher->disconnect(this);
}
//-------------------------------------------------------------------------------------------
bool fbgui::checkHost() const
{
QHostInfo hostInfo = QHostInfo::fromName(baseURL.host());
if (hostInfo.error() != QHostInfo::NoError){
qxtLog->debug() << "[gui] Lookup of " << baseURL.host() << "failed. Exiting...";
return false;
}
else{
qxtLog->debug() << "[gui] Lookup of " << baseURL.host() << " succeeded.";
return true;
}
}
//-------------------------------------------------------------------------------------------
QByteArray fbgui::generatePOSTData(){
qxtLog->debug() << "[gui] Generating POST data...";
// hash testing
SysInfo si;
QByteArray data(si.getInfo("mac").toUtf8() + "|" + si.getInfo("ip").toUtf8());
qxtLog->debug() << "[post] Hashing: " << data;
QByteArray hash = QCryptographicHash::hash(data, QCryptographicHash::Md5);
qxtLog->debug() << "[post] MD5 Hash: " << hash.toHex();
// fetch serial number from usb rootfs
QByteArray serial;
QFile file(serialLocation);
if (!file.open(QIODevice::ReadOnly)){
qxtLog->debug() << "[post] No such file: " << file.fileName();
serial = "10-23-43-55-67"; // tests
}
// everything ok, read data
serial = file.readAll();
file.close();
serial.chop(1); // chop EOF
qxtLog->debug() << "[post] Serial number is: " << serial;
// construct final byte array
QByteArray postData("bootisoID=1");
postData.append("&mac=" + si.getInfo("mac"));
postData.append("&hardwarehash=" + hash.toHex());
postData.append("&serial=" + serial);
return postData;
}
//-------------------------------------------------------------------------------------------
// setup debug console widget
//-------------------------------------------------------------------------------------------
void fbgui::setupDebugSplit()
{
_debugConsole = new QTextEdit(this);
_debugConsole->setWindowFlags(Qt::FramelessWindowHint);
QPalette pal;
pal.setColor(QPalette::Base, Qt::black);
_debugConsole->setPalette(pal);
_debugConsole->setTextColor(Qt::white);
// enable custom logger engine
qxtLog->addLoggerEngine("fb_logger", new LoggerEngine_fb(_debugConsole));
//qxtLog->initLoggerEngine("fb_logger");
qxtLog->setMinimumLevel("fb_logger", QxtLogger::DebugLevel);
// display browser and debug in a splitter
_splitter = new QSplitter(Qt::Vertical, this);
_splitter->addWidget(_webView);
_splitter->addWidget(_debugConsole);
// CTRL + D toggles debug window
_toggleDebug = new QAction(tr("&toggleDebug"), this);
_toggleDebug->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_D));
addAction(_toggleDebug);
connect(_toggleDebug, SIGNAL(triggered()), this, SLOT(toggleDebug()));
//_webView->load(QUrl("qrc:/html/loadAbout.html"));
setCentralWidget(_splitter);
}
//-------------------------------------------------------------------------------------------
void fbgui::toggleDebug()
{
if (_debugConsole->isVisible())
_debugConsole->hide();
else
_debugConsole->show();
}