nearby/sharing/linux/qml_tray_app/file_share_state.h

#ifndef SHARING_LINUX_QML_TRAY_APP_FILE_SHARE_STATE_H_
#define SHARING_LINUX_QML_TRAY_APP_FILE_SHARE_STATE_H_

#include <QHash>
#include <QString>
#include <QStringList>
#include <QVariantList>
#include <QVariantMap>
#include <QSet>
#include "share_target_model.h"
#include "transfer_model.h"

class FileShareState {
 public:
  FileShareState();

  // Getters
  QString mode() const { return mode_; }
  QString deviceName() const { return device_name_; }
  QString statusMessage() const { return status_message_; }
  bool running() const { return running_; }
  bool autoAcceptIncoming() const { return auto_accept_incoming_; }

  QString pendingSendFileName() const { return pending_send_file_name_; }
  QString pendingSendFilePath() const { return pending_send_file_path_; }
  qlonglong pendingSendTargetId() const { return pending_send_target_id_; }

  QVariantList discoveredTargets() const { return discovered_targets_; }
  QVariantList transfers() const { return transfers_; }

  QString qrCodeUrl() const { return qr_code_url_; }
  QStringList qrCodeRows() const { return qr_code_rows_; }
  int qrCodeSize() const { return qr_code_size_; }

  QString logPath() const { return log_path_; }

  // Setters
  void SetMode(const QString& mode) { mode_ = mode; }
  void SetDeviceName(const QString& name) { device_name_ = name; }
  void SetStatusMessage(const QString& message) { status_message_ = message; }
  void SetRunning(bool running) { running_ = running; }
  void SetAutoAcceptIncoming(bool enabled) { auto_accept_incoming_ = enabled; }

  void SetPendingSendFile(const QString& file_path, const QString& file_name,
                          qlonglong target_id) {
    pending_send_file_path_ = file_path;
    pending_send_file_name_ = file_name;
    pending_send_target_id_ = target_id;
  }

  void ClearPendingSendFile() {
    pending_send_file_path_.clear();
    pending_send_file_name_.clear();
    pending_send_target_id_ = 0;
  }

  void SetQrCodeData(const QString& url, const QStringList& rows, int size) {
    qr_code_url_ = url;
    qr_code_rows_ = rows;
    qr_code_size_ = size;
  }

  void SetLogPath(const QString& path) { log_path_ = path; }

  // Target management
  void AddOrUpdateTarget(qlonglong id, const QString& name, bool is_incoming);
  void RemoveTarget(qlonglong id);
  QString GetTargetName(qlonglong id) const;
  bool HasTarget(qlonglong id) const;

  // Transfer management
  void AddOrUpdateTransfer(qlonglong target_id, const QString& target_name,
                           const QString& status, double progress,
                           qulonglong transferred_bytes,
                           const QString& direction, const QString& file_name,
                           const QString& file_path);
  void RemoveTransfer(qlonglong target_id);
  bool HasActiveTransferForTarget(qlonglong target_id) const;
  bool HasActiveTransfers() const;

  // Pending target removal management
  void AddPendingTargetRemoval(qlonglong id);
  void RemovePendingTargetRemoval(qlonglong id);
  bool IsPendingTargetRemoval(qlonglong id) const;

  void ClearAll();

 private:
  QString mode_ = QStringLiteral("Receive");
  QString device_name_ = QStringLiteral("NearbyLinux");
  QString status_message_ = QStringLiteral("Idle");
  bool running_ = false;
  bool auto_accept_incoming_ = true;

  // QR Code
  QString qr_code_url_;
  QStringList qr_code_rows_;
  int qr_code_size_ = 0;
  QString log_path_ = QStringLiteral("/tmp/nearby_qml_file_tray.log");

  // Pending send state
  QString pending_send_file_path_;
  QString pending_send_file_name_;
  qlonglong pending_send_target_id_ = 0;

  // Discovered targets
  QVariantList discovered_targets_;
  QHash<qlonglong, int> discovered_row_by_target_;
  QHash<qlonglong, QString> target_names_;

  // Transfers
  QVariantList transfers_;
  QHash<qlonglong, int> transfer_row_by_target_;

  // Pending removals
  QSet<qlonglong> pending_target_removals_;
};

#endif  // SHARING_LINUX_QML_TRAY_APP_FILE_SHARE_STATE_H_