// SPDX-FileCopyrightText: 2019-2024 Connor McLaughlin <stenzek@gmail.com>
// SPDX-License-Identifier: CC-BY-NC-ND-4.0

#pragma once

#include "common/types.h"

#include "ui_autoupdaterwindow.h"

#include <memory>
#include <string>
#include <string_view>

#include <QtCore/QDateTime>
#include <QtCore/QStringList>
#include <QtCore/QTimer>
#include <QtWidgets/QDialog>

class Error;
class HTTPDownloader;

class EmuThread;

class AutoUpdaterWindow final : public QWidget
{
  Q_OBJECT

public:
  explicit AutoUpdaterWindow(QWidget* parent = nullptr);
  ~AutoUpdaterWindow();

  void queueUpdateCheck(bool display_errors);
  void queueGetLatestRelease();

  static bool isSupported();
  static bool canInstallUpdate();
  static QStringList getTagList();
  static std::string getDefaultTag();
  static void cleanupAfterUpdate();
  static bool isOfficialBuild();
  static void warnAboutUnofficialBuild();

Q_SIGNALS:
  void updateCheckCompleted();

protected:
  void closeEvent(QCloseEvent* event) override;

private:
  void httpPollTimerPoll();

  void downloadUpdateClicked();
  void skipThisUpdateClicked();
  void remindMeLaterClicked();

  void reportError(const std::string_view msg);

  bool ensureHttpReady();

  bool updateNeeded() const;
  std::string getCurrentUpdateTag() const;

  void getLatestTagComplete(s32 status_code, const Error& error, std::vector<u8> response, bool display_errors);
  void getLatestReleaseComplete(s32 status_code, const Error& error, std::vector<u8> response);

  void queueGetChanges();
  void getChangesComplete(s32 status_code, const Error& error, std::vector<u8> response);

  bool processUpdate(const std::vector<u8>& update_data);

#ifdef _WIN32
  bool doesUpdaterNeedElevation(const std::string& application_dir) const;
  bool doUpdate(const std::string& application_dir, const std::string& zip_path, const std::string& updater_path);
  bool extractUpdater(const std::string& zip_path, const std::string& destination_path, Error* error);
#endif

  Ui::AutoUpdaterWindow m_ui;

  std::unique_ptr<HTTPDownloader> m_http;
  QTimer* m_http_poll_timer = nullptr;
  QString m_latest_sha;
  QString m_download_url;
  int m_download_size = 0;

  bool m_update_will_break_save_states = false;
};
