#include "formmanager.h"
#include "services/config.h"
#include "services/pluginmanager.h"
#include "sqlitestudio.h"
#include "uiloader.h"
#include "common/unused.h"
#include "common/global.h"
#include <QFile>
#include <QDir>
#include <QRegularExpression>
#include <QApplication>
#include <QDebug>

FormManager::FormManager()
{
 init();
}

FormManager::~FormManager()
{
 if (uiLoader)
 {
 delete uiLoader;
 uiLoader = nullptr;
 }
}

QWidget* FormManager::createWidget(const QString& name)
{
 if (!widgetNameToFullPath.contains(name))
 {
 qCritical() << "Asked for widget name which isn't managed by FormManager:" << name << ", while available widgets are:"
 << widgetNameToFullPath.keys();
 return nullptr;
 }
 return createWidgetByFullPath(widgetNameToFullPath[name]);
}

bool FormManager::hasWidget(const QString& name)
{
 return widgetNameToFullPath.contains(name);
}

QStringList FormManager::getAvailableForms() const
{
 return widgetNameToFullPath.keys();
}

QStringList FormManager::getFormDirs() const
{
 return formDirs;
}

QWidget* FormManager::createWidgetByFullPath(const QString& path)
{
 QWidget* widget = uiLoader->load(path);
 if (!widget)
 {
 qCritical() << "Error occured while loading ui file:" << path << ". Error message: "
 << uiLoader->errorString();
 return nullptr;
 }
 return widget;
}

void FormManager::rescanResources(Plugin* plugin, PluginType* pluginType)
{
 UNUSED(pluginType);
 rescanResources(plugin->getName());
}

void FormManager::rescanResources(const QString& pluginName)
{
 if (PLUGINS->isBuiltIn(pluginName))
 return;

 for (const QString& widgetName : resourceForms)
 widgetNameToFullPath.remove(widgetName);

 resourceForms.clear();
 loadRecurently(":/forms", "");
}

void FormManager::pluginsAboutToMassUnload()
{
 disconnect(PLUGINS, SIGNAL(loaded(Plugin*,PluginType*)), this, SLOT(rescanResources(Plugin*,PluginType*)));
 disconnect(PLUGINS, SIGNAL(unloaded(QString,PluginType*)), this, SLOT(rescanResources(QString)));
}

void FormManager::pluginsInitiallyLoaded()
{
 load();

 connect(PLUGINS, SIGNAL(loaded(Plugin*,PluginType*)), this, SLOT(rescanResources(Plugin*,PluginType*)));
 connect(PLUGINS, SIGNAL(unloaded(QString,PluginType*)), this, SLOT(rescanResources(QString)));
 connect(PLUGINS, SIGNAL(aboutToQuit()), this, SLOT(pluginsAboutToMassUnload()));
 disconnect(PLUGINS, SIGNAL(pluginsInitiallyLoaded()), this, SLOT(pluginsInitiallyLoaded()));
}

void FormManager::init()
{
 uiLoader = new UiLoader();

 if (PLUGINS->arePluginsInitiallyLoaded())
 pluginsInitiallyLoaded();
 else
 connect(PLUGINS, SIGNAL(pluginsInitiallyLoaded()), this, SLOT(pluginsInitiallyLoaded()));
}

void FormManager::load()
{
 formDirs += qApp->applicationDirPath() + "/forms";
 formDirs += ":/forms";
 formDirs += QDir(CFG->getConfigDir()).absoluteFilePath("forms");

 QString envDirs = SQLITESTUDIO->getEnv("SQLITESTUDIO_FORMS");
 if (!envDirs.isNull())
 formDirs += envDirs.split(PATH_LIST_SEPARATOR);

 formDirs += PLUGINS->getPluginDirs();

#ifdef FORMS_DIR
 formDirs += STRINGIFY(FORMS_DIR);
#endif

 for (QString dirPath : formDirs)
 loadRecurently(dirPath, "");
}

void FormManager::loadRecurently(const QString& path, const QString& prefix)
{
 static const QStringList fileExtensions = {"*.ui", "*.UI"};

 QDir dir(path);
 QString fullPath;
 QString widgetName;
 for (const QFileInfo& entry : dir.entryInfoList(fileExtensions, QDir::AllDirs|QDir::Files|QDir::NoDotAndDotDot|QDir::Readable))
 {
 fullPath = entry.absoluteFilePath();
 if (entry.isDir())
 {
 loadRecurently(fullPath, prefix+entry.fileName()+"_");
 continue;
 }

 qDebug() << "Loading form file:" << fullPath;

 widgetName = getWidgetName(fullPath);
 if (widgetName.isNull())
 continue;

 if (widgetNameToFullPath.contains(widgetName))
 {
 qCritical() << "Widget named" << widgetName << "was already loaded by FormManager from file" << widgetNameToFullPath[widgetName]
 << "therefore file" << fullPath << "will be ignored";
 continue;
 }

 widgetNameToFullPath[widgetName] = fullPath;
 if (fullPath.startsWith(":/"))
 resourceForms << widgetName;
 }
}

QString FormManager::getWidgetName(const QString& path)
{
 static const QRegularExpression re(R"(<widget class\=\"\w+\" name\=\"(\w+)\">)");

 QFile file(path);
 if (!file.open(QIODevice::ReadOnly))
 {
 qWarning() << "Could not open" << path << "for reading. Form file ignored.";
 return QString::null;
 }

 QString contents = file.readAll();
 file.close();

 QRegularExpressionMatch match = re.match(contents);
 if (!match.hasMatch())
 {
 qWarning() << "Could not match widget in" << path << " document. File ignored.";
 return QString::null;
 }

 QString widgetName = match.captured(1);

 return widgetName;
}