qlocalsocket/lackey/main.cpp

/****************************************************************************
**
** Copyright (C) 2015 The Qt Company Ltd.
** Contact: http://www.qt.io/licensing/
**
** This file is part of the test suite of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:LGPL$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see http://www.qt.io/terms-conditions. For further
** information use the contact form at http://www.qt.io/contact-us.
**
** GNU Lesser General Public License Usage
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 or version 3 as published by the Free
** Software Foundation and appearing in the file LICENSE.LGPLv21 and
** LICENSE.LGPLv3 included in the packaging of this file. Please review the
** following information to ensure the GNU Lesser General Public License
** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** As a special exception, The Qt Company gives you certain additional
** rights. These rights are described in The Qt Company LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3.0 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU General Public License version 3.0 requirements will be
** met: http://www.gnu.org/copyleft/gpl.html.
**
** $QT_END_LICENSE$
**
****************************************************************************/


#include <qscriptengine.h>
 #include <QFile>
#include <QTest>

#include <qlocalsocket.h>
#include <qlocalserver.h>

class QScriptLocalSocket : public QObject
{
    Q_OBJECT
    Q_PROPERTY(QString serverName WRITE connectToServer READ serverName)

public:
    QScriptLocalSocket(QObject *parent = 0) : QObject(parent)
    {
        lc = new QLocalSocket(this);
    }

public slots:
    QString serverName()
    {
        return lc->serverName();
    }

    void connectToServer(const QString &name) {
        lc->connectToServer(name);
    }

    void sleep(int x) const
    {
        QTest::qSleep(x);
    }

    bool isConnected() {
        return (lc->state() == QLocalSocket::ConnectedState);
    }

    void open() {
        lc->open(QIODevice::ReadWrite);
    }

    bool waitForConnected() {
        return lc->waitForConnected(100000);
    }
    void waitForReadyRead() {
        lc->waitForReadyRead();
    }

    void write(const QString &string) {
        QTextStream out(lc);
        out << string << endl;
    }

    bool waitForBytesWritten(int t = 3000) {
        return lc->waitForBytesWritten(t);
    }

    QString readLine() {
        QTextStream in(lc);
        return in.readLine();
    }

    QString errorString() {
        return lc->errorString();
    }

    void close() {
        lc->close();
    }

public:
    QLocalSocket *lc;
};

class QScriptLocalServer : public QLocalServer
{
    Q_OBJECT
    Q_PROPERTY(int maxPendingConnections WRITE setMaxPendingConnections READ maxPendingConnections)
    Q_PROPERTY(QString name WRITE listen READ serverName)
    Q_PROPERTY(bool listening READ isListening)

public:
    QScriptLocalServer(QObject *parent = 0) : QLocalServer(parent)
    {
    }

public slots:
    bool listen(const QString &name) {
        if (!QLocalServer::listen(name)) {
            if (serverError() == QAbstractSocket::AddressInUseError) {
                QFile::remove(serverName());
                return QLocalServer::listen(name);
            }
            return false;
        }
        return true;
    }

    QScriptLocalSocket *nextConnection() {
        QLocalSocket *other = nextPendingConnection();
        QScriptLocalSocket *s = new QScriptLocalSocket(this);
        delete s->lc;
        s->lc = other;
        return s;
    }

    bool waitForNewConnection() {
        return QLocalServer::waitForNewConnection(30000);
    }

    QString errorString() {
        return QLocalServer::errorString();
    }


};

template <typename T>
static QScriptValue _q_ScriptValueFromQObject(QScriptEngine *engine, T* const &in)
{
    return engine->newQObject(in);
}
template <typename T>
static void _q_ScriptValueToQObject(const QScriptValue &v, T* &out)
{    out = qobject_cast<T*>(v.toQObject());
}
template <typename T>
static int _q_ScriptRegisterQObjectMetaType(QScriptEngine *engine, const QScriptValue &prototype)
{
    return qScriptRegisterMetaType<T*>(engine, _q_ScriptValueFromQObject<T>, _q_ScriptValueToQObject<T>, prototype);
}

QT_BEGIN_NAMESPACE
Q_SCRIPT_DECLARE_QMETAOBJECT(QScriptLocalSocket, QObject*);
Q_SCRIPT_DECLARE_QMETAOBJECT(QScriptLocalServer, QObject*);
QT_END_NAMESPACE

static void interactive(QScriptEngine &eng)
{
    QTextStream qin(stdin, QFile::ReadOnly);

    const char *qscript_prompt = "qs> ";
    const char *dot_prompt = ".... ";
    const char *prompt = qscript_prompt;

    QString code;

    forever {
        QString line;

        printf("%s", prompt);
        fflush(stdout);

        line = qin.readLine();
        if (line.isNull())
        break;

        code += line;
        code += QLatin1Char('\n');

        if (line.trimmed().isEmpty()) {
            continue;

        } else if (! eng.canEvaluate(code)) {
            prompt = dot_prompt;

        } else {
            QScriptValue result = eng.evaluate(code);
            code.clear();
            prompt = qscript_prompt;
            if (!result.isUndefined())
                fprintf(stderr, "%s\n", qPrintable(result.toString()));
        }
    }
}
Q_DECLARE_METATYPE(QScriptLocalSocket*)
Q_DECLARE_METATYPE(QScriptLocalServer*)
int main(int argc, char *argv[])
{
    QCoreApplication app(argc, argv);
    QScriptEngine eng;
    QScriptValue globalObject = eng.globalObject();

    _q_ScriptRegisterQObjectMetaType<QScriptLocalServer>(&eng, QScriptValue());

    QScriptValue lss = qScriptValueFromQMetaObject<QScriptLocalServer>(&eng);
    eng.globalObject().setProperty("QScriptLocalServer", lss);

    _q_ScriptRegisterQObjectMetaType<QScriptLocalSocket>(&eng, QScriptValue());

    QScriptValue lsc = qScriptValueFromQMetaObject<QScriptLocalSocket>(&eng);
    eng.globalObject().setProperty("QScriptLocalSocket", lsc);

    if (! *++argv) {
        interactive(eng);
        return EXIT_SUCCESS;
    }

    QStringList arguments;
    for (int i = 0; i < argc - 1; ++i)
        arguments << QString::fromLocal8Bit(argv[i]);

    while (!arguments.isEmpty()) {
        QString fn = arguments.takeFirst();

        if (fn == QLatin1String("-i")) {
            interactive(eng);
            break;
        }

        QString contents;

        if (fn == QLatin1String("-")) {
            QTextStream stream(stdin, QFile::ReadOnly);
            contents = stream.readAll();
        } else {
            QFile file(fn);
	    if (!file.exists()) {
                fprintf(stderr, "%s doesn't exists\n", qPrintable(fn));
	        return EXIT_FAILURE;
	    }
            if (file.open(QFile::ReadOnly)) {
                QTextStream stream(&file);
                contents = stream.readAll();
                file.close();
            }
        }

        if (contents.isEmpty())
            continue;

        if (contents[0] == '#') {
            contents.prepend("//");
            QScriptValue args = eng.newArray();
            args.setProperty("0", QScriptValue(&eng, fn));
            int i = 1;
            while (!arguments.isEmpty())
                args.setProperty(i++, QScriptValue(&eng, arguments.takeFirst()));
            eng.currentContext()->activationObject().setProperty("args", args);
        }
        QScriptValue r = eng.evaluate(contents);
        if (eng.hasUncaughtException()) {
            int line = eng.uncaughtExceptionLineNumber();
            fprintf(stderr, "%d: %s\n\t%s\n\n", line, qPrintable(fn), qPrintable(r.toString()));
            return EXIT_FAILURE;
        }
        if (r.isNumber())
            return r.toInt32();
    }

    return EXIT_SUCCESS;
}

#include "main.moc"