aboutsummaryrefslogtreecommitdiffstats
path: root/SQLiteStudio3/coreSQLiteStudio/plugins/scriptingqt.h
blob: a7e156c201c35932b59a806216250c168bcc6cc6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#ifndef SCRIPTINGQT_H
#define SCRIPTINGQT_H

#include "builtinplugin.h"
#include "scriptingplugin.h"
#include <QHash>
#include <QVariant>
#include <QCache>
#include <QJSValue>
#include <QThreadStorage>

class QMutex;
class ScriptingQtDbProxy;
class ScriptingQtConsole;

class ScriptingQt : public BuiltInPlugin, public DbAwareScriptingPlugin
{
    Q_OBJECT

    SQLITESTUDIO_PLUGIN_TITLE("JavaScript scripting")
    SQLITESTUDIO_PLUGIN_DESC("JavaScript scripting support.")
    SQLITESTUDIO_PLUGIN_VERSION(10100)
    SQLITESTUDIO_PLUGIN_AUTHOR("sqlitestudio.pl")

    public:
        ScriptingQt();
        ~ScriptingQt();

        static QJSValueList toValueList(QJSEngine* engine, const QList<QVariant>& values);
        static QVariant convertVariant(const QVariant& value, bool wrapStrings = false);

        QString getLanguage() const;
        Context* createContext();
        void releaseContext(Context* context);
        void resetContext(Context* context);
        QVariant evaluate(const QString& code, const FunctionInfo& funcInfo, const QList<QVariant>& args, Db* db, bool locking = false, QString* errorMessage = nullptr);
        QVariant evaluate(Context* context, const QString& code, const FunctionInfo& funcInfo, const QList<QVariant>& args, Db* db, bool locking = false);
        void setVariable(Context* context, const QString& name, const QVariant& value);
        QVariant getVariable(Context* context, const QString& name);
        bool hasError(Context* context) const;
        QString getErrorMessage(Context* context) const;
        QString getIconPath() const;
        bool init();
        void deinit();

    private:
        using DbAwareScriptingPlugin::evaluate;

        class ContextQt : public ScriptingPlugin::Context
        {
            public:
                ContextQt();
                ~ContextQt();

                QJSEngine* engine = nullptr;
                QCache<QString, QJSValue> scriptCache;
                QString error;
                ScriptingQtDbProxy* dbProxy = nullptr;
                ScriptingQtConsole* console = nullptr;
                QJSValue dbProxyScriptValue;
        };

        ContextQt* getContext(ScriptingPlugin::Context* context) const;
        QJSValue getFunctionValue(ContextQt* ctx, const QString& code, const FunctionInfo& funcInfo);
        QVariant evaluate(ContextQt* ctx, const QString& code, const FunctionInfo& funcInfo, const QList<QVariant>& args, Db* db, bool locking);
        ContextQt* getMainContext();

        static const constexpr int cacheSize = 5;

        QThreadStorage<ContextQt*> mainContext;
        QList<Context*> contexts;
        QList<ContextQt*> managedMainContexts;
        QMutex* managedMainContextsMutex = nullptr;
};

class ScriptingQtConsole : public QObject
{
    Q_OBJECT

    public:
        ScriptingQtConsole(QJSEngine* engine);

    private:
        QJSEngine* engine = nullptr;

    public slots:
        QJSValue log(const QJSValue& value);
};

#endif // SCRIPTINGQT_H