quickshell/src/cpp/rootwrapper.cpp

#include "rootwrapper.hpp"
#include <cstdlib>
#include <utility>

#include <qlogging.h>
#include <qobject.h>
#include <qqmlcomponent.h>
#include <qqmlengine.h>
#include <qurl.h>

#include "scavenge.hpp"
#include "shell.hpp"

RootWrapper::RootWrapper(QUrl rootUrl):
    QObject(nullptr), rootUrl(std::move(rootUrl)), engine(this) {
	this->reloadGraph(true);

	if (this->activeRoot == nullptr) {
		qCritical() << "could not create scene graph, exiting";
		exit(-1); // NOLINT
	}
}

void RootWrapper::reloadGraph(bool hard) {
	if (this->activeRoot != nullptr) {
		this->engine.clearComponentCache();
	}

	auto component = QQmlComponent(&this->engine, this->rootUrl);

	SCAVENGE_PARENT = hard ? nullptr : this;
	auto* obj = component.beginCreate(this->engine.rootContext());
	SCAVENGE_PARENT = nullptr;

	if (obj == nullptr) {
		qWarning() << component.errorString().toStdString().c_str();
		qWarning() << "failed to create root component";
		return;
	}

	auto* newRoot = qobject_cast<QtShell*>(obj);
	if (newRoot == nullptr) {
		qWarning() << "root component was not a QtShell";
		delete obj;
		return;
	}

	component.completeCreate();

	if (this->activeRoot != nullptr) {
		this->activeRoot->deleteLater();
		this->activeRoot = nullptr;
	}

	this->activeRoot = newRoot;
}

void RootWrapper::changeRoot(QtShell* newRoot) {
	if (this->activeRoot != nullptr) {
		QObject::disconnect(this->destroyConnection);
		this->activeRoot->deleteLater();
	}

	if (newRoot != nullptr) {
		this->activeRoot = newRoot;
		QObject::connect(this->activeRoot, &QtShell::destroyed, this, &RootWrapper::destroy);
	}
}

QObject* RootWrapper::scavengeTargetFor(QObject* /* child */) { return this->activeRoot; }

void RootWrapper::destroy() { this->deleteLater(); }