git.sur5r.net Git - minitube/blob - src/jsfunctions.cpp

   1 /* $BEGIN_LICENSE
   2
   3 This file is part of Minitube.
   4 Copyright 2009, Flavio Tordini <flavio.tordini@gmail.com>
   5
   6 Minitube is free software: you can redistribute it and/or modify
   7 it under the terms of the GNU General Public License as published by
   8 the Free Software Foundation, either version 3 of the License, or
   9 (at your option) any later version.
  10
  11 Minitube is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with Minitube.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 $END_LICENSE */
  20
  21 #include "jsfunctions.h"
  22 #include "networkaccess.h"
  23 #include "constants.h"
  24 #include "compatibility/qurlqueryhelper.h"
  25 #include "compatibility/pathsservice.h"
  26
  27 namespace The {
  28 NetworkAccess* http();
  29 }
  30
  31 JsFunctions* JsFunctions::instance() {
  32     static JsFunctions *i = new JsFunctions(QLatin1String(Constants::WEBSITE) + "-ws/functions.js");
  33     return i;
  34 }
  35
  36 JsFunctions::JsFunctions(const QString &url, QObject *parent) : QObject(parent), url(url), engine(0) {
  37     QFile file(jsPath());
  38     if (file.exists()) {
  39         if (file.open(QIODevice::ReadOnly | QIODevice::Text))
  40             parseJs(QString::fromUtf8(file.readAll()));
  41         else
  42             qWarning() << "Cannot open" << file.errorString() << file.fileName();
  43         QFileInfo info(file);
  44         bool stale = info.size() == 0 || info.lastModified().toTime_t() < QDateTime::currentDateTime().toTime_t() - 1800;
  45         if (stale) loadJs();
  46     } else {
  47         QFile resFile(QLatin1String(":/") + jsFilename());
  48         resFile.open(QIODevice::ReadOnly | QIODevice::Text);
  49         parseJs(QString::fromUtf8(resFile.readAll()));
  50         loadJs();
  51     }
  52 }
  53
  54 void JsFunctions::parseJs(const QString &js) {
  55     if (js.isEmpty()) return;
  56     // qDebug() << "Parsing" << js;
  57     if (engine) delete engine;
  58     engine = new QScriptEngine(this);
  59     engine->evaluate(js);
  60     emit ready();
  61 }
  62
  63 QString JsFunctions::jsFilename() {
  64     return QFileInfo(url).fileName();
  65 }
  66
  67 QString JsFunctions::jsPath() {
  68     return Paths::getDataLocation() + "/" + jsFilename();
  69 }
  70
  71 void JsFunctions::loadJs() {
  72     QUrl url(this->url);
  73     {
  74         QUrlQueryHelper urlHelper(url);
  75         urlHelper.addQueryItem("v", Constants::VERSION);
  76     }
  77
  78     NetworkReply* reply = The::http()->get(url);
  79     connect(reply, SIGNAL(data(QByteArray)), SLOT(gotJs(QByteArray)));
  80     connect(reply, SIGNAL(error(QNetworkReply*)), SLOT(errorJs(QNetworkReply*)));
  81 }
  82
  83 void JsFunctions::gotJs(const QByteArray &bytes) {
  84     if (bytes.isEmpty()) {
  85         qWarning() << "Got empty js";
  86         return;
  87     }
  88     if (!QDir().mkpath(Paths::getDataLocation())) {
  89       qCritical() << "Failed to create" << Paths::getDataLocation();
  90       return;
  91     }
  92     QFile file(jsPath());
  93     if (!file.open(QIODevice::WriteOnly)) {
  94         qWarning() << "Cannot write" << file.errorString() << file.fileName();
  95         return;
  96     }
  97     QDataStream stream(&file);
  98     stream.writeRawData(bytes.constData(), bytes.size());
  99     parseJs(QString::fromUtf8(bytes));
 100 }
 101
 102 void JsFunctions::errorJs(QNetworkReply *reply) {
 103     qWarning() << "Cannot get" << jsFilename() << reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt()
 104                << reply->url().toString() << reply->errorString();
 105 }
 106
 107 QScriptValue JsFunctions::evaluate(const QString &js) {
 108     if (!engine) return QString();
 109     QScriptValue value = engine->evaluate(js);
 110     if (value.isUndefined())
 111         qWarning() << "Undefined result for" << js;
 112     if (value.isError())
 113         qWarning() << "Error in" << js << value.toString();
 114
 115     return value;
 116 }
 117
 118 QString JsFunctions::string(const QString &js) {
 119     return evaluate(js).toString();
 120 }
 121
 122 QStringList JsFunctions::stringArray(const QString &js) {
 123     QStringList items;
 124     QScriptValue array = evaluate(js);
 125     if (!array.isArray()) return items;
 126     QScriptValueIterator it(array);
 127     while (it.hasNext()) {
 128         it.next();
 129         QScriptValue value = it.value();
 130         if (!value.isString()) continue;
 131         items << value.toString();
 132     }
 133     return items;
 134 }
 135
 136 QString JsFunctions::decryptSignature(const QString &s) {
 137     return string("decryptSignature('" + s + "')");
 138 }
 139
 140 QString JsFunctions::decryptAgeSignature(const QString &s) {
 141     return string("decryptAgeSignature('" + s + "')");
 142 }
 143
 144 QString JsFunctions::videoIdRE() {
 145     return string("videoIdRE()");
 146 }
 147
 148 QString JsFunctions::videoTokenRE() {
 149     return string("videoTokenRE()");
 150 }
 151
 152 QString JsFunctions::videoInfoFmtMapRE() {
 153     return string("videoInfoFmtMapRE()");
 154 }
 155
 156 QString JsFunctions::webPageFmtMapRE() {
 157     return string("webPageFmtMapRE()");
 158 }
 159
 160 QString JsFunctions::ageGateRE() {
 161     return string("ageGateRE()");
 162 }
 163
 164 QString JsFunctions::jsPlayerRE() {
 165     return string("jsPlayerRE()");
 166 }
 167
 168 QString JsFunctions::signatureFunctionNameRE() {
 169     return string("signatureFunctionNameRE()");
 170 }
 171
 172 QStringList JsFunctions::apiKeys() {
 173     return stringArray("apiKeys()");
 174 }