qBittorrent/src/headlessloader.h

/*
 * Bittorrent Client using Qt4 and libtorrent.
 * Copyright (C) 2006  Christophe Dumez, Frédéric Lassabe
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link this program with the OpenSSL project's "OpenSSL" library (or with
 * modified versions of it that use the same license as the "OpenSSL" library),
 * and distribute the linked executables. You must obey the GNU General Public
 * License in all respects for all of the code used other than "OpenSSL".  If you
 * modify file(s), you may extend this exception to your version of the file(s),
 * but you are not obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * Contact : chris@qbittorrent.org
 */

#ifndef HEADLESSLOADER_H
#define HEADLESSLOADER_H

#include <QObject>
#include <QCoreApplication>
#include <QLocalServer>
#include <QLocalSocket>
#include "preferences.h"
#include "bittorrent.h"

class HeadlessLoader: QObject {
  Q_OBJECT

  private:
    QLocalServer *localServer;
    Bittorrent *BTSession;

  public:
    HeadlessLoader(QStringList torrentCmdLine) {
      // Enable Web UI
      Preferences::setWebUiEnabled(true);
      // Instanciate Bittorrent Object
      BTSession = new Bittorrent();
      connect(BTSession, SIGNAL(newConsoleMessage(QString)), this, SLOT(displayConsoleMessage(QString)));
      // Resume unfinished torrents
      BTSession->startUpTorrents();
      // Process command line parameters
      processParams(torrentCmdLine);
      // Use a tcp server to allow only one instance of qBittorrent
      localServer = new QLocalServer();
      QString uid = QString::number(getuid());
#ifdef Q_WS_X11
      if(QFile::exists(QDir::tempPath()+QDir::separator()+QString("qBittorrent-")+uid)) {
        // Socket was not closed cleanly
        std::cerr << "Warning: Local domain socket was not closed cleanly, deleting file...\n";
        QFile::remove(QDir::tempPath()+QDir::separator()+QString("qBittorrent-")+uid);
      }
#endif
      if (!localServer->listen("qBittorrent-"+uid)) {
        std::cerr  << "Couldn't create socket, single instance mode won't work...\n";
      }
      connect(localServer, SIGNAL(newConnection()), this, SLOT(acceptConnection()));
      // Display some information to the user
      std::cout << std::endl << "******** " << tr("Information").toLocal8Bit().data() << " ********" << std::endl;
      std::cout << tr("To control qBittorrent, access the Web UI at http://localhost:%1").arg(QString::number(Preferences::getWebUiPort())).toLocal8Bit().data() << std::endl;
      std::cout << tr("The Web UI administrator user name is: %1").arg(Preferences::getWebUiUsername()).toLocal8Bit().data() << std::endl;
      if(Preferences::getWebUiPassword() == "f6fdffe48c908deb0f4c3bd36c032e72") {
        std::cout << tr("The Web UI administrator password is still the default one: %1").arg("adminadmin").toLocal8Bit().data() << std::endl;
        std::cout << tr("This is a security risk, please consider changing your password from program preferences.").toLocal8Bit().data() << std::endl;
      }
    }

    ~HeadlessLoader() {
      delete BTSession;
    }

  public slots:
    // Call this function to exit qBittorrent headless loader
    // and return to prompt (object will be deleted by main)
    void exit() {
     qApp->quit();
    }

    void displayConsoleMessage(QString msg) {
      std::cout << msg.toLocal8Bit().data() << std::endl;
    }

    // As program parameters, we can get paths or urls.
    // This function parse the parameters and call
    // the right addTorrent function, considering
    // the parameter type.
    void processParams(const QStringList& params) {
      foreach(QString param, params) {
        param = param.trimmed();
        if(param.startsWith("--")) continue;
        if(param.startsWith(QString::fromUtf8("http://"), Qt::CaseInsensitive) || param.startsWith(QString::fromUtf8("ftp://"), Qt::CaseInsensitive) || param.startsWith(QString::fromUtf8("https://"), Qt::CaseInsensitive)) {
          BTSession->downloadFromUrl(param);
        }else{
          if(param.startsWith("magnet:", Qt::CaseInsensitive)) {
            BTSession->addMagnetUri(param);
          } else {
            BTSession->addTorrent(param);
          }
        }
      }
    }

    void acceptConnection() {
      QLocalSocket *clientConnection = localServer->nextPendingConnection();
      connect(clientConnection, SIGNAL(disconnected()), this, SLOT(readParamsOnSocket()));
      qDebug("accepted connection from another instance");
    }

    void readParamsOnSocket() {
      QLocalSocket *clientConnection = static_cast<QLocalSocket*>(sender());
      if(clientConnection) {
        QByteArray params = clientConnection->readAll();
        if(!params.isEmpty()) {
          processParams(QString::fromUtf8(params.data()).split(QString::fromUtf8("\n")));
          qDebug("Received parameters from another instance");
        }
        clientConnection->deleteLater();
      }
    }

};

#endif