File indexing completed on 2024-04-14 14:08:59

 /*
     SPDX-FileCopyrightText: 2004 Jason Harris <kstars@30doradus.org>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "kswizard.h"
 
 #include "geolocation.h"
 #include "kspaths.h"
 #include "kstars.h"
 #include "kstarsdata.h"
 #include "ksnotification.h"
 #include "ksutils.h"
 #include "Options.h"
 #include "widgets/dmsbox.h"
 
 #include <kns3/downloaddialog.h>
 
 #include <QDesktopServices>
 #include <QFile>
 #include <QLineEdit>
 #include <QPixmap>
 #include <QPushButton>
 #include <QStackedWidget>
 #include <QStandardPaths>
 
 namespace
 {
 bool hasPrefix(QString str, QString prefix)
 {
     if (prefix.isEmpty())
         return true;
     return str.startsWith(prefix, Qt::CaseInsensitive);
 }
 }
 
 WizWelcomeUI::WizWelcomeUI(QWidget *parent) : QFrame(parent)
 {
     setupUi(this);
 }
 
 WizLocationUI::WizLocationUI(QWidget *parent) : QFrame(parent)
 {
     setupUi(this);
 }
 
 WizDownloadUI::WizDownloadUI(QWidget *parent) : QFrame(parent)
 {
     setupUi(this);
 }
 
 #ifdef Q_OS_OSX
 WizDataUI::WizDataUI(QWidget *parent) : QFrame(parent)
 {
     setupUi(this);
 }
 #endif
 
 KSWizard::KSWizard(QWidget *parent) : QDialog(parent)
 {
     wizardStack = new QStackedWidget(this);
     adjustSize();
 
     setWindowTitle(i18nc("@title:window", "Startup Wizard"));
 
     QVBoxLayout *mainLayout = new QVBoxLayout;
     mainLayout->addWidget(wizardStack);
     setLayout(mainLayout);
 
     buttonBox = new QDialogButtonBox(Qt::Horizontal);
     nextB     = new QPushButton(i18n("&Next >"));
     nextB->setDefault(true);
     backB = new QPushButton(i18n("< &Back"));
     backB->setEnabled(false);
     completeB = new QPushButton(i18n("Done"));
 
     buttonBox->addButton(backB, QDialogButtonBox::ActionRole);
     buttonBox->addButton(nextB, QDialogButtonBox::ActionRole);
     buttonBox->addButton(completeB, QDialogButtonBox::AcceptRole);
     completeB->setVisible(false);
 
     mainLayout->addWidget(buttonBox);
 
     welcome = new WizWelcomeUI(wizardStack);
 #ifdef Q_OS_OSX
     data       = new WizDataUI(wizardStack);
 #endif
     location                = new WizLocationUI(wizardStack);
     WizDownloadUI *download = new WizDownloadUI(wizardStack);
 
     wizardStack->addWidget(welcome);
 #ifdef Q_OS_OSX
     wizardStack->addWidget(data);
 #endif
     wizardStack->addWidget(location);
     wizardStack->addWidget(download);
     wizardStack->setCurrentWidget(welcome);
 
     //Load images into banner frames.
     QPixmap im;
     if (im.load(KSPaths::locate(QStandardPaths::AppLocalDataLocation, "wzstars.png")))
         welcome->Banner->setPixmap(im);
     else if (im.load(QDir(QCoreApplication::applicationDirPath() + "/../Resources/kstars").absolutePath() +
                      "/wzstars.png"))
         welcome->Banner->setPixmap(im);
     if (im.load(KSPaths::locate(QStandardPaths::AppLocalDataLocation, "wzgeo.png")))
         location->Banner->setPixmap(im);
     else if (im.load(QDir(QCoreApplication::applicationDirPath() + "/../Resources/kstars").absolutePath() + "/wzgeo.png"))
         location->Banner->setPixmap(im);
     if (im.load(KSPaths::locate(QStandardPaths::AppLocalDataLocation, "wzdownload.png")))
         download->Banner->setPixmap(im);
     else if (im.load(QDir(QCoreApplication::applicationDirPath() + "/../Resources/kstars").absolutePath() +
                      "/wzdownload.png"))
         download->Banner->setPixmap(im);
 
 #ifdef Q_OS_OSX
     if (im.load(KSPaths::locate(QStandardPaths::AppLocalDataLocation, "wzdownload.png")))
         data->Banner->setPixmap(im);
     else if (im.load(QDir(QCoreApplication::applicationDirPath() + "/../Resources/kstars").absolutePath() +
                      "/wzdownload.png"))
         data->Banner->setPixmap(im);
 
     data->dataPath->setText(
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, QString(), QStandardPaths::LocateDirectory) +
         "kstars");
     slotUpdateDataButtons();
 
     connect(data->copyKStarsData, SIGNAL(clicked()), this, SLOT(slotOpenOrCopyKStarsDataDirectory()));
     connect(data->installGSC, SIGNAL(clicked()), this, SLOT(slotInstallGSC()));
 
     gscMonitor = new QProgressIndicator(data);
     data->GSCLayout->addWidget(gscMonitor);
     data->downloadProgress->setValue(0);
 
     data->downloadProgress->setEnabled(false);
     data->downloadProgress->setVisible(false);
 
     data->gscInstallCancel->setVisible(false);
     data->gscInstallCancel->setEnabled(false);
 
 #endif
 
     //connect signals/slots
     connect(nextB, SIGNAL(clicked()), this, SLOT(slotNextPage()));
     connect(backB, SIGNAL(clicked()), this, SLOT(slotPrevPage()));
     connect(buttonBox, SIGNAL(accepted()), this, SLOT(accept()));
     connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
 
     connect(location->CityListBox, SIGNAL(itemSelectionChanged()), this, SLOT(slotChangeCity()));
     connect(location->CityFilter, SIGNAL(textChanged(QString)), this, SLOT(slotFilterCities()));
     connect(location->ProvinceFilter, SIGNAL(textChanged(QString)), this, SLOT(slotFilterCities()));
     connect(location->CountryFilter, SIGNAL(textChanged(QString)), this, SLOT(slotFilterCities()));
     connect(download->DownloadButton, SIGNAL(clicked()), this, SLOT(slotDownload()));
 
     //Initialize Geographic Location page
     if (KStars::Instance())
         initGeoPage();
 }
 
 void KSWizard::setButtonsEnabled()
 {
     nextB->setEnabled(wizardStack->currentIndex() < wizardStack->count() - 1);
     backB->setEnabled(wizardStack->currentIndex() > 0);
     completeB->setVisible(wizardStack->currentIndex() == wizardStack->count() - 1);
 
 #ifdef Q_OS_OSX
     if ((wizardStack->currentWidget() == data) && (!dataDirExists()))
     {
         nextB->setEnabled(false);
     }
 #endif
 }
 
 void KSWizard::slotNextPage()
 {
     wizardStack->setCurrentIndex(wizardStack->currentIndex() + 1);
     setButtonsEnabled();
 }
 
 void KSWizard::slotPrevPage()
 {
     wizardStack->setCurrentIndex(wizardStack->currentIndex() - 1);
     setButtonsEnabled();
 }
 
 void KSWizard::initGeoPage()
 {
     KStarsData *data = KStarsData::Instance();
     location->LongBox->setReadOnly(true);
     location->LatBox->setReadOnly(true);
 
     //Populate the CityListBox
     //flag the ID of the current City
     foreach (GeoLocation *loc, data->getGeoList())
     {
         location->CityListBox->addItem(loc->fullName());
         filteredCityList.append(loc);
         if (loc->fullName() == data->geo()->fullName())
         {
             Geo = loc;
         }
     }
 
     //Sort alphabetically
     location->CityListBox->sortItems();
     //preset to current city
     QList<QListWidgetItem*> locations = location->CityListBox->findItems(QString(data->geo()->fullName()), Qt::MatchExactly);
     if (locations.isEmpty() == false)
         location->CityListBox->setCurrentItem(locations[0]);
 }
 
 void KSWizard::slotChangeCity()
 {
     if (location->CityListBox->currentItem())
     {
         for (auto &city : filteredCityList)
         {
             if (city->fullName() == location->CityListBox->currentItem()->text())
             {
                 Geo = city;
                 break;
             }
         }
         location->LongBox->show(Geo->lng());
         location->LatBox->show(Geo->lat());
     }
 }
 
 void KSWizard::slotFilterCities()
 {
     location->CityListBox->clear();
     //Do NOT delete members of filteredCityList!
     filteredCityList.clear();
 
     foreach (GeoLocation *loc, KStarsData::Instance()->getGeoList())
     {
         if (hasPrefix(loc->translatedName(), location->CityFilter->text()) &&
                 hasPrefix(loc->translatedCountry(), location->CountryFilter->text()) &&
                 hasPrefix(loc->translatedProvince(), location->ProvinceFilter->text()))
         {
             location->CityListBox->addItem(loc->fullName());
             filteredCityList.append(loc);
         }
     }
     location->CityListBox->sortItems();
 
     if (location->CityListBox->count() > 0) // set first item in list as selected
         location->CityListBox->setCurrentItem(location->CityListBox->item(0));
 }
 
 void KSWizard::slotDownload()
 {
     KNS3::DownloadDialog dlg;
     dlg.exec();
 }
 
 void KSWizard::slotOpenOrCopyKStarsDataDirectory()
 {
 #ifdef Q_OS_OSX
     QString dataLocation =
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory);
     if (dataLocation.isEmpty())
     {
         QDir dataSourceDir = QDir(QCoreApplication::applicationDirPath() + "/../Resources/kstars").absolutePath();
         if (! dataSourceDir.exists()) //If there is no default data directory in the app bundle
         {
             KSNotification::sorry(i18n("There was no default data directory found in the app bundle."));
             return;
         }
         QDir writableDir(KSPaths::writableLocation(QStandardPaths::AppLocalDataLocation));
         writableDir.mkpath(".");
         dataLocation =
             QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory);
         if (dataLocation.isEmpty()) //If there *still* is not a kstars data directory
         {
             KSNotification::sorry(i18n("There was a problem creating the data directory ~/Library/Application Support/."));
             return;
         }
         KSUtils::copyRecursively(dataSourceDir.absolutePath(), dataLocation);
         //This will update the next, ok, and copy kstars dir buttons.
         slotUpdateDataButtons();
     }
     else
     {
         QUrl path = QUrl::fromLocalFile(
                         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory));
         QDesktopServices::openUrl(path);
     }
 
 #endif
 }
 
 void KSWizard::slotInstallGSC()
 {
 #ifdef Q_OS_OSX
 
     QNetworkAccessManager *manager = new QNetworkAccessManager();
 
     QString location =
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory);
     gscZipPath            = location + "/gsc.zip";
 
     data->downloadProgress->setVisible(true);
     data->downloadProgress->setEnabled(true);
 
     data->gscInstallCancel->setVisible(true);
     data->gscInstallCancel->setEnabled(true);
 
     QString gscURL = "http://www.indilib.org/jdownloads/Mac/gsc.zip";
 
     QNetworkReply *response = manager->get(QNetworkRequest(QUrl(gscURL)));
 
     QMetaObject::Connection *cancelConnection = new QMetaObject::Connection();
     QMetaObject::Connection *replyConnection = new QMetaObject::Connection();
     QMetaObject::Connection *percentConnection = new QMetaObject::Connection();
 
     *percentConnection = connect(response, &QNetworkReply::downloadProgress,
                                  [ = ](qint64 bytesReceived, qint64 bytesTotal)
     {
         data->downloadProgress->setValue(bytesReceived);
         data->downloadProgress->setMaximum(bytesTotal);
     });
 
     *cancelConnection = connect(data->gscInstallCancel, &QPushButton::clicked,
                                 [ = ]()
     {
         qDebug() << Q_FUNC_INFO << "Download Cancelled.";
 
         if(cancelConnection)
             disconnect(*cancelConnection);
         if(replyConnection)
             disconnect(*replyConnection);
 
         if(response)
         {
             response->abort();
             response->deleteLater();
         }
 
         data->downloadProgress->setVisible(false);
         data->downloadProgress->setEnabled(false);
 
         data->gscInstallCancel->setVisible(false);
         data->gscInstallCancel->setEnabled(false);
 
         if(manager)
             manager->deleteLater();
 
     });
 
     *replyConnection = connect(response, &QNetworkReply::finished, this,
                                [ = ]()
     {
         if(response)
         {
 
             if(cancelConnection)
                 disconnect(*cancelConnection);
             if(replyConnection)
                 disconnect(*replyConnection);
 
             data->downloadProgress->setVisible(false);
             data->downloadProgress->setEnabled(false);
 
             data->gscInstallCancel->setVisible(false);
             data->gscInstallCancel->setEnabled(false);
 
 
             response->deleteLater();
             if(manager)
                 manager->deleteLater();
             if (response->error() != QNetworkReply::NoError)
                 return;
 
             QByteArray responseData = response->readAll();
 
             QFile file(gscZipPath);
             if (QFileInfo(QFileInfo(file).path()).isWritable())
             {
                 if (!file.open(QIODevice::WriteOnly))
                 {
                     KSNotification::error( i18n("File write error."));
                     return;
                 }
                 else
                 {
                     file.write(responseData.data(), responseData.size());
                     file.close();
                     slotExtractGSC();
                 }
             }
             else
             {
                 KSNotification::error( i18n("Data folder permissions error."));
             }
         }
     });
 
 #endif
 }
 
 void KSWizard::slotExtractGSC()
 {
 #ifdef Q_OS_OSX
     QString location =
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory);
     QProcess *gscExtractor = new QProcess();
     connect(gscExtractor, SIGNAL(finished(int)), this, SLOT(slotGSCInstallerFinished()));
     connect(gscExtractor, SIGNAL(finished(int)), this, SLOT(gscExtractor.deleteLater()));
     gscExtractor->setWorkingDirectory(location);
     gscExtractor->start("unzip", QStringList() << "-ao"
                         << "gsc.zip");
     gscMonitor->startAnimation();
 #endif
 }
 
 void KSWizard::slotGSCInstallerFinished()
 {
 #ifdef Q_OS_OSX
     if (downloadMonitor)
     {
         downloadMonitor->stop();
         delete downloadMonitor;
     }
     data->downloadProgress->setEnabled(false);
     data->downloadProgress->setValue(0);
     data->downloadProgress->setVisible(false);
     gscMonitor->stopAnimation();
     slotUpdateDataButtons();
     if (QFile(gscZipPath).exists())
         QFile(gscZipPath).remove();
 #endif
 }
 
 #ifdef Q_OS_OSX
 bool KSWizard::dataDirExists()
 {
     QString dataLocation =
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars", QStandardPaths::LocateDirectory);
     return !dataLocation.isEmpty();
 }
 
 bool KSWizard::GSCExists()
 {
     QString GSCLocation =
         QStandardPaths::locate(QStandardPaths::GenericDataLocation, "kstars/gsc", QStandardPaths::LocateDirectory);
     return !GSCLocation.isEmpty();
 }
 
 #endif
 
 void KSWizard::slotUpdateDataButtons()
 {
 #ifdef Q_OS_OSX
     data->dataDirFound->setChecked(dataDirExists());
     if (dataDirExists())
     {
         data->copyKStarsData->setText("Open KStars Data Directory");
         data->foundFeedback1->setText("The KStars Data Directory called kstars is located at:");
         data->foundFeedback2->setText("Your data directory was found.  If you have any problems with it, you can "
                                       "always delete this data directory and KStars will give you a new data "
                                       "directory.  You can click this button to open the data directory, just be "
                                       "careful not to delete any important files.");
     }
     else
     {
         data->foundFeedback1->setText("The KStars Data Directory called kstars should be located at:");
         data->foundFeedback2->setText("<html><head/><body><p>Your data directory was not found. You can click the "
                                       "button below to copy a default KStars data directory to the correct location, "
                                       "or if you have a KStars directory already some place else, you can exit KStars "
                                       "and copy it to that location yourself.</p></body></html>");
     }
     bool ifGSCExists = GSCExists();
     data->GSCFound->setChecked(ifGSCExists);
     data->installGSC->setDisabled(ifGSCExists || !dataDirExists());
     if (ifGSCExists)
         data->GSCFeedback->setText("GSC was found on your system.  To use it, just take an image in the CCD simulator. "
                                    "To uninstall, just delete the gsc folder from your data directory above.");
     setButtonsEnabled();
 #endif
 }