implement album export, finish WinUI styles, cleanup code

2026-06-21 19:35:49 +08:00 · 2026-03-06 04:47:23 +03:00
parent 35c5985f47
commit 9043350731
67 changed files with 1340 additions and 587 deletions
@@ -0,0 +1,191 @@
+/*
+ * iDescriptor: A free and open-source idevice management tool.
+ *
+ * Copyright (C) 2025 Uncore <https://github.com/uncor3>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published
+ * by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include "appdownload.h"
+#include <QDesktopServices>
+#include <QDir>
+#include <QFutureWatcher>
+#include <QLabel>
+#include <QMessageBox>
+#include <QPointer>
+#include <QProgressBar>
+#include <QPushButton>
+#include <QVBoxLayout>
+#include <QtConcurrent/QtConcurrent>
+
+void AppDownloadBaseDialog::updateProgressBar(int percentage)
+{
+
+    if (m_progressBar) {
+        m_progressBar->setValue(percentage);
+    }
+}
+
+void AppDownloadBaseDialog::addProgressBar(int index)
+{
+    m_progressBar = new QProgressBar();
+    m_progressBar->setRange(0, 100);
+    m_progressBar->setValue(0);
+    m_progressBar->setTextVisible(true);
+    m_progressBar->setFixedHeight(25);
+    m_progressBar->setStyleSheet(
+        "QProgressBar { border-radius: 6px; background: #eee; } "
+        "QProgressBar::chunk { background: #34C759; }");
+    m_layout->insertWidget(index, m_progressBar);
+}
+
+AppDownloadBaseDialog::AppDownloadBaseDialog(const QString &appName,
+                                             const QString &bundleId,
+                                             QWidget *parent)
+    : QDialog(parent), m_appName(appName), m_downloadProcess(nullptr),
+      m_progressTimer(nullptr), m_bundleId(bundleId)
+{
+    m_layout = new QVBoxLayout(this);
+    m_layout->setContentsMargins(20, 20, 20, 20);
+#ifdef WIN32
+    setupWinWindow(this);
+#endif
+}
+
+void AppDownloadBaseDialog::startDownloadProcess(const QString &bundleId,
+                                                 const QString &outputDir,
+                                                 int index,
+                                                 bool promptToOpenDir,
+                                                 bool close)
+{
+
+    if (bundleId.isEmpty()) {
+        QMessageBox::critical(this, "Error", "Bundle ID not provided.");
+        reject();
+        return;
+    }
+
+    addProgressBar(index);
+    if (m_actionButton)
+        m_actionButton->setEnabled(false);
+
+    m_operationInProgress = true;
+    tryToDownload(bundleId, outputDir, promptToOpenDir, close);
+}
+
+void AppDownloadBaseDialog::tryToDownload(const QString &bundleId,
+                                          const QString &outputDir,
+                                          bool promptToOpenDir, bool close)
+{
+    AppStoreManager *manager = AppStoreManager::sharedInstance();
+    if (!manager) {
+        QMessageBox::critical(this, "Error",
+                              "Failed to initialize App Store manager.");
+        reject();
+        return;
+    }
+    bool acquireLicense = true;
+    m_operationInProgress = true;
+    QPointer<AppDownloadBaseDialog> safeThis = this;
+
+    auto progressCallback = [safeThis](long long current, long long total) {
+        if (!safeThis) {
+            return;
+        }
+        int percentage = 0;
+        if (total > 0) {
+            percentage = static_cast<int>((current * 100) / total);
+        }
+        safeThis->updateProgressBar(percentage);
+    };
+
+    manager->downloadApp(
+        bundleId, outputDir, "", acquireLicense,
+        [safeThis, promptToOpenDir, outputDir, close](int result) {
+            if (!safeThis) {
+                return;
+            }
+            safeThis->m_operationInProgress = false;
+            if (result == 0) { // Success
+                emit safeThis->downloadFinished(true, "Success");
+                if (safeThis->m_progressBar)
+                    safeThis->m_progressBar->setValue(100);
+                if (promptToOpenDir) {
+
+                    if (QMessageBox::Yes ==
+                        QMessageBox::question(
+                            safeThis, "Download Successful",
+                            QString("Successfully downloaded. Would you like "
+                                    "to open the output directory: %1?")
+                                .arg(outputDir))) {
+                        QDir dir(outputDir);
+                        if (!dir.exists()) {
+                            QMessageBox::warning(
+                                safeThis, "Directory Not Found",
+                                QString("The directory %1 does not exist.")
+                                    .arg(outputDir));
+                        } else {
+                            QDesktopServices::openUrl(
+                                QUrl::fromLocalFile(outputDir));
+                        }
+                    }
+                }
+                if (close)
+                    safeThis->accept();
+            } else { // Failure
+                // 3 attempts
+                if (safeThis->m_tries < 3) {
+                    safeThis->m_tries++;
+                    qDebug()
+                        << "Retrying download for" + safeThis->m_bundleId +
+                               "Attempt:" + QString::number(safeThis->m_tries);
+                    safeThis->tryToDownload(safeThis->m_bundleId, outputDir,
+                                            promptToOpenDir);
+                    return;
+                }
+                emit safeThis->downloadFinished(false, "Failed");
+                // if (promptToOpenDir)
+                QMessageBox::critical(
+                    safeThis, "Download Failed",
+                    QString("Failed to download %1. Try signing out and back "
+                            "in. Error code: %2")
+                        .arg(safeThis->m_appName)
+                        .arg(result));
+                if (close)
+                    safeThis->reject();
+            }
+        },
+        progressCallback);
+}
+
+void AppDownloadBaseDialog::reject()
+{
+    if (m_operationInProgress) {
+        AppStoreManager *manager = AppStoreManager::sharedInstance();
+        if (manager) {
+            manager->cancelDownload(m_bundleId);
+        }
+        m_operationInProgress = false;
+    }
+
+    cleanup();
+    QDialog::reject();
+}
+
+void AppDownloadBaseDialog::cleanup()
+{
+    if (m_progressTimer && m_progressTimer->isActive()) {
+        m_progressTimer->stop();
+    }
+}