Adaptix-Framework-AdaptixC2/AdaptixClient/Source/Client/AxScript/BridgeForm.cpp

#include <Client/AxScript/BridgeForm.h>
#include <Client/AxScript/AxElementWrappers.h>
#include <Client/AxScript/AxScriptEngine.h>
#include <QMessageBox>
#include <QMetaMethod>
#include <QJsonDocument>

BridgeForm::BridgeForm(AxScriptEngine* scriptEngine, QObject* parent) : QObject(parent), scriptEngine(scriptEngine), widget(new QWidget()) {}

BridgeForm::~BridgeForm() { delete widget; }

void BridgeForm::connect(QObject* sender, const QString& signalName, const QJSValue& handler)
{
    if (!sender || !handler.isCallable()) {
        emit scriptError("connect -> Invalid sender or handler");
        return;
    }

    const QMetaObject* meta = sender->metaObject();
    for (int i = 0; i < meta->methodCount(); ++i) {
        QMetaMethod method = meta->method(i);
        if (method.methodType() != QMetaMethod::Signal)
            continue;

        QString name = QString::fromLatin1(method.methodSignature());
        if (!name.startsWith(signalName))
            continue;

        int paramCount = method.parameterCount();
        auto* proxy = new SignalProxy(scriptEngine->engine(), handler, sender);

        bool connected = false;

        if (paramCount == 0) {
            connected = QObject::connect(sender, method, proxy, proxy->metaObject()->method(proxy->metaObject()->indexOfSlot("call()")));
        }
        else if (paramCount == 1) {
            QByteArray typeName = method.parameterTypes().value(0);
            if (typeName == "QString")
                connected = QObject::connect(sender, method, proxy, proxy->metaObject()->method(proxy->metaObject()->indexOfSlot("callWithArg(QString)")));
            else if (typeName == "int")
                connected = QObject::connect(sender, method, proxy, proxy->metaObject()->method(proxy->metaObject()->indexOfSlot("callWithArg(int)")));
            else if (typeName == "bool")
                connected = QObject::connect(sender, method, proxy, proxy->metaObject()->method(proxy->metaObject()->indexOfSlot("callWithArg(bool)")));
        }
        else {
            emit scriptError("connect -> Signal " + signalName + " has too many parameters (not supported)");
            return;
        }

        if (!connected)
            emit scriptError("connect -> Failed to connect signal " + method.methodSignature());

        return;
    }

    emit scriptError("connect -> Signal " + signalName + " not found");
}

/// Elements

QObject* BridgeForm::create_vlayout()
{
    auto* wrapper = new AxBoxLayoutWrapper(QBoxLayout::TopToBottom, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_hlayout()
{
    auto* wrapper = new AxBoxLayoutWrapper(QBoxLayout::LeftToRight, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_gridlayout()
{
    auto* wrapper = new AxGridLayoutWrapper(this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_vline()
{
    auto* wrapper = new AxLineWrapper(QFrame::VLine, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_hline()
{
    auto* wrapper = new AxLineWrapper(QFrame::HLine, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_vspacer()
{
    auto* wrapper = new AxSpacerWrapper(20, 40, QSizePolicy::Minimum, QSizePolicy::Expanding, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_hspacer()
{
    auto* wrapper = new AxSpacerWrapper(40, 20, QSizePolicy::Expanding, QSizePolicy::Minimum, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_label(const QString& text)
{
    auto* label = new QLabel(text, widget);
    auto* wrapper = new AxLabelWrapper(label, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_textline(const QString &text)
{
    auto* edit = new QLineEdit(text, widget);
    auto* wrapper = new AxTextLineWrapper(edit, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_combo()
{
    auto* combo = new QComboBox(widget);
    auto* wrapper = new AxComboBoxWrapper(combo, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_check(const QString& label)
{
    auto* check = new QCheckBox(label, widget);
    auto* wrapper = new AxCheckBoxWrapper(check, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_spin()
{
    auto* spin = new QSpinBox(widget);
    auto* wrapper = new AxSpinBoxWrapper(spin, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_dateline(const QString& format)
{
    auto* date_widget = new QDateEdit(widget);
    auto* wrapper = new AxDateEditWrapper(date_widget, format, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_timeline(const QString& format)
{
    auto* time_widget = new QTimeEdit(widget);
    auto* wrapper = new AxTimeEditWrapper(time_widget, format, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_button(const QString& text)
{
    auto* btn = new QPushButton(text, widget);
    auto* wrapper = new AxButtonWrapper(btn, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_textmulti(const QString& text)
{
    auto* textEdit = new QPlainTextEdit(text, widget);
    auto* wrapper = new AxTextMultiWrapper(textEdit, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_file_selector()
{
    auto fileSelector = new FileSelector(widget);
    auto* wrapper = new AxFileSelectorWrapper(fileSelector, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_tabs()
{
    auto* tabWidget = new QTabWidget(widget);
    auto* wrapper = new AxTabWrapper(tabWidget, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_panel()
{
    auto* panel = new QWidget(widget);
    auto* wrapper = new AxPanelWrapper(panel, this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_container()
{
    auto* wrapper = new AxContainerWrapper(scriptEngine->engine(), this);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}

QObject* BridgeForm::create_dialog(const QString& title) const
{
    auto* wrapper = new AxDialogWrapper(title, widget);
    scriptEngine->registerObject(wrapper);
    return wrapper;
}