src/mainwnd.cpp

   1 /*
   2  * abby Copyright (C) 2009 Toni Gundogdu.
   3  * This file is part of abby.
   4  *
   5  * abby is free software: you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation, either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * abby is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18 #include <QMainWindow>
  19 #include <QSettings>
  20 #include <QCloseEvent>
  21 #include <QMessageBox>
  22 #include <QFileDialog>
  23 #include <QIcon>
  24
  25 #include "mainwnd.h"
  26 #include "prefsdlg.h"
  27 #include "passdlg.h"
  28 #include "aboutdlg.h"
  29
  30 MainWindow::MainWindow():
  31     cancelled(false)
  32 {
  33
  34
  35     /* TRANSLATOR
  36
  37     The word "English" is not meant to be translated. Instead,
  38     enter the name of the target language in native language.
  39     For example: Suomi, Deutsch, Svenska, Turkce, Francais, etc.
  40
  41     The only purpose of this string is to be able to list the
  42     available languages in the Preferences dialog. */
  43     const QString lang = tr("English");
  44
  45     setupUi(this);
  46     setWindowIcon(QIcon(":/rc/abby.png"));
  47     readSettings();
  48
  49     prefs = new PreferencesDialog(this);
  50
  51     connect(&process, SIGNAL(started()),
  52         this, SLOT(onProcStarted()));
  53     connect(&process, SIGNAL(error(QProcess::ProcessError)),
  54         this, SLOT(onProcError(QProcess::ProcessError)));
  55     connect(&process, SIGNAL(readyReadStandardError()),
  56         this, SLOT(onProcStderrReady()));
  57     connect(&process, SIGNAL(finished(int,QProcess::ExitStatus)),
  58         this, SLOT(onProcFinished(int,QProcess::ExitStatus)));
  59 }
  60
  61 void
  62 MainWindow::closeEvent(QCloseEvent *event) {
  63     writeSettings();
  64     event->accept();
  65 }
  66
  67 void
  68 MainWindow::writeSettings() {
  69     QSettings s;
  70     s.beginGroup("MainWindow");
  71     s.setValue("size",size());
  72     s.setValue("pos",pos());
  73     s.setValue("formatCombo",formatCombo->currentIndex());
  74     s.endGroup();
  75 }
  76
  77 void
  78 MainWindow::readSettings() {
  79     QSettings s;
  80     s.beginGroup("MainWindow");
  81     resize(s.value("size",QSize(400,400)).toSize());
  82     move(s.value("pos",QPoint(200,200)).toPoint());
  83     formatCombo->setCurrentIndex(s.value("formatCombo").toInt());
  84     s.endGroup();
  85 }
  86
  87 void
  88 MainWindow::updateLog(QString newText) {
  89     QString text = logEdit->toPlainText() + newText;
  90     logEdit->setPlainText(text);
  91 }
  92
  93
  94 // Slots
  95
  96 void
  97 MainWindow::onPreferences() {
  98     prefs->exec();
  99 }
 100
 101 void
 102 MainWindow::onSaveasStateChanged(int state) {
 103     saveasEdit->setEnabled(state != 0);
 104 }
 105
 106 void
 107 MainWindow::onStreamStateChanged(int state) {
 108     streamSpin->setEnabled(state != 0);
 109 }
 110
 111 void
 112 MainWindow::onSaveasBrowse() {
 113     QString fname = QFileDialog::getSaveFileName(this,tr("Save as"));
 114     if (!fname.isEmpty())
 115         saveasEdit->setText(fname);
 116 }
 117
 118 void
 119 MainWindow::onStart() {
 120     // Check video URL
 121
 122     QString url = urlEdit->text();
 123
 124     if (url.isEmpty()) {
 125         statusBar()->showMessage(tr("Enter a video link."));
 126         return;
 127     }
 128
 129     url = url.trimmed();
 130
 131     if (!url.startsWith("http://",Qt::CaseInsensitive))
 132         url.insert(0,"http://");
 133
 134     // Check cclive
 135
 136     QString cclive = prefs->ccliveEdit->text();
 137     if (cclive.isEmpty()) {
 138         QMessageBox::information(this,QCoreApplication::applicationName(),
 139             tr("Path to cclive command undefined. See preferences."));
 140         onPreferences();
 141         return;
 142     }
 143
 144     // Check video save directory
 145
 146     // NOTE: cclive itself does not support this concept so we
 147     // work around it by changing the working directory to the
 148     // save directory.
 149
 150     QString savedir = prefs->savedirEdit->text();
 151     if (savedir.isEmpty()) {
 152         QMessageBox::information(this,QCoreApplication::applicationName(),
 153             tr("Save directory undefined. See preferences."));
 154         onPreferences();
 155         return;
 156     }
 157     process.setWorkingDirectory(savedir);
 158
 159     // Construct cclive args
 160
 161     QStringList args;
 162     args << "--print-fname";
 163
 164     QString s = prefs->additionalEdit->text();
 165     if (!s.isEmpty())
 166         args << s;
 167
 168     s = prefs->streamEdit->text();
 169     if (!s.isEmpty() && streamBox->isChecked()) {
 170         args << QString("--stream-exec=%1").arg(s);
 171         args << QString("--stream=%1").arg(streamSpin->value());
 172     }
 173
 174     s = prefs->commandEdit->text();
 175     if (!s.isEmpty() && commandBox->isChecked()) {
 176         if (!s.endsWith(";"))
 177             s += ";";
 178         args << QString("--exec=%1").arg(s);
 179     }
 180
 181     if (prefs->proxyCombo->currentIndex() == 0)
 182         args << "--no-proxy";
 183     else {
 184         s = prefs->proxyEdit->text();
 185         if (!s.isEmpty())
 186             args << QString("--proxy=%1").arg(s);
 187     }
 188
 189     if (prefs->limitBox->checkState()) {
 190         int n = prefs->limitSpin->value();
 191         args << QString("--limit-rate=%1").arg(n);
 192     }
 193
 194     if (prefs->youtubeGroup->isChecked()) {
 195         QString user = prefs->ytuserEdit->text();
 196         QString pass = prefs->ytpassEdit->text();
 197         if (pass.isEmpty()) {
 198             PasswordDialog pwd(this);
 199             if (pwd.exec())
 200                 pass = pwd.passEdit->text();
 201         }
 202         if (!user.isEmpty() && !pass.isEmpty()) {
 203             args << QString("--youtube-user=%1").arg(user);
 204             args << QString("--youtube-pass=%1").arg(pass);
 205         }
 206     }
 207
 208     s = saveasEdit->text();
 209     if (!s.isEmpty())
 210         args << QString("--output-video=%1").arg(s);
 211
 212     if (continueBox->isChecked())
 213         args << "--continue";
 214
 215     args << QString("--download=%1").arg(formatCombo->currentText());
 216     args << QString("%1").arg(url);
 217
 218     // Prepare log
 219
 220     logEdit->clear();
 221     updateLog("% cclive " +args.join(" ")+ "\n");
 222
 223     // And finally start the process
 224
 225     cancelled = false;
 226     process.start(cclive,args);
 227 }
 228
 229 void
 230 MainWindow::onCancel() {
 231     cancelled = true;
 232     process.kill();
 233 }
 234
 235 void
 236 MainWindow::onAbout() {
 237     AboutDialog about(this,prefs->ccliveEdit->text());
 238     about.exec();
 239 }
 240
 241 void
 242 MainWindow::onURLEditingFinished() {
 243
 244     // Change format combobox contents dynamically based on the video URL.
 245
 246     QString url = urlEdit->text();
 247     if (url.isEmpty())
 248         return;
 249
 250     struct lookup_s {
 251         const char *host;
 252         const char *formats;
 253     };
 254     static const struct lookup_s lookup[] = {
 255         {"youtube.com",     "mp4|xflv|3gpp"},
 256         {"video.google.",   "mp4"},
 257         {"dailymotion.com", "spak-mini|vp6-hq|vp6-hd|vp6|h264"},
 258     };
 259
 260     const int c = sizeof(lookup)/sizeof(struct lookup_s);
 261
 262     QStringList formats;
 263     formats << "flv";
 264
 265     for (int i=0; i<c; ++i) {
 266         if (url.contains(lookup[i].host)) {
 267             QString s = lookup[i].formats;
 268             formats << s.split("|");
 269             break;
 270         }
 271     }
 272
 273     formatCombo->clear();
 274     formatCombo->addItems(formats);
 275 }
 276
 277 void
 278 MainWindow::onFormatStateChanged(int) {
 279
 280     // Disable --continue for the specified hosts if format is "flv".
 281     // Make changes based on the video URL.
 282
 283     QString url = urlEdit->text();
 284     if (url.isEmpty())
 285         return;
 286
 287     struct lookup_s {
 288         const char *host;
 289     };
 290     static const struct lookup_s lookup[] = {
 291         {"youtube.com"},
 292         {"video.google."},
 293     };
 294
 295     const int c = sizeof(lookup)/sizeof(struct lookup_s);
 296     bool enable = true;
 297
 298     for (int i=0; i<c; ++i) {
 299         if (url.contains(lookup[i].host)) {
 300             if (formatCombo->currentText() == "flv") {
 301                 enable = false;
 302                 break;
 303             }
 304         }
 305     }
 306
 307     continueBox->setEnabled(enable);
 308
 309     if (continueBox->isChecked() && !enable)
 310         continueBox->setCheckState(Qt::Unchecked);
 311 }
 312
 313 void
 314 MainWindow::onProcStarted() {
 315     statusBar() ->clearMessage();
 316     fileLabel   ->setText("-");
 317     sizeLabel   ->setText("-- / --");
 318     rateLabel   ->setText("--.-");
 319     etaLabel    ->setText("--:--");
 320     progressBar ->setValue(0);
 321
 322     startButton ->setEnabled(false);
 323     cancelButton->setEnabled(true);
 324 }
 325
 326 void
 327 MainWindow::onProcError(QProcess::ProcessError err) {
 328     if (err == QProcess::FailedToStart) {
 329         QString msg = tr("error: failed to start process");
 330         updateLog(msg);
 331         statusBar()->showMessage(msg);
 332     }
 333 }
 334
 335 void
 336 MainWindow::onProcStderrReady() {
 337     QString newText =
 338         QString::fromLocal8Bit(process.readAllStandardError());
 339
 340     QString status;
 341     if (newText.contains("fetch"))
 342         status = tr("Fetching link...");
 343     else if (newText.contains("verify"))
 344         status = tr("Verifying video link...");
 345     else if (newText.contains("error:")) {
 346         // Probably better to connect finished etc. signal instead
 347         status = tr("error: see log for details");
 348     }
 349     else if (newText.contains("file:")) {
 350         status = tr("Extracting video...");
 351         QStringList lst = newText.split(" ",QString::SkipEmptyParts);
 352         fileLabel->setText(lst[1].remove("\n"));
 353     }
 354
 355     if (!status.isEmpty())
 356         statusBar()->showMessage(status.remove("\n"));
 357
 358     if (newText.contains("%")) {
 359         newText.replace("\n", " ");
 360
 361         QStringList lst =
 362             newText.split(" ",QString::SkipEmptyParts);
 363
 364         QString percent = lst[1].remove(QChar('%'));
 365         QString now     = lst[2];
 366         QString expected= lst[4];
 367         QString rate    = lst[5];
 368         QString eta     = lst[6];
 369
 370         sizeLabel->setText(now +" / "+ expected);
 371         rateLabel->setText(rate);
 372         etaLabel->setText(eta);
 373         progressBar->setValue(percent.toInt());
 374     } else {
 375         updateLog(newText);
 376     }
 377 }
 378
 379 void
 380 MainWindow::onProcFinished(int exitCode, QProcess::ExitStatus exitStatus) {
 381     QString status;
 382     if (exitStatus == QProcess::NormalExit) {
 383         if (exitCode != 0)
 384             status = tr("Process exited with an error; see log");
 385         else
 386             status = tr("Process exited normally");
 387     } else {
 388         status = cancelled
 389             ? tr("Process terminated")
 390             : tr("Process crashed; see log");
 391     }
 392
 393     if (!statusBar()->currentMessage().contains(tr("error:")))
 394         statusBar()->showMessage(status);
 395
 396     updateLog(status + ".");
 397
 398     startButton ->setEnabled(true);
 399     cancelButton->setEnabled(false);
 400 }