From f335fae90084567b141298a6de2e12d2ef00cdbd Mon Sep 17 00:00:00 2001
From: RickyRister <ricky.rister.wang@gmail.com>
Date: Tue, 24 Dec 2024 20:29:02 -0800
Subject: [PATCH] add new method to replay tree widget

---
 .../src/server/remote/remote_replay_list_tree_widget.cpp | 9 +++++++++
 .../src/server/remote/remote_replay_list_tree_widget.h   | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/cockatrice/src/server/remote/remote_replay_list_tree_widget.cpp b/cockatrice/src/server/remote/remote_replay_list_tree_widget.cpp
index ee24aa817..838524519 100644
--- a/cockatrice/src/server/remote/remote_replay_list_tree_widget.cpp
+++ b/cockatrice/src/server/remote/remote_replay_list_tree_widget.cpp
@@ -309,6 +309,15 @@ RemoteReplayList_TreeWidget::RemoteReplayList_TreeWidget(AbstractClient *_client
     setSelectionMode(QAbstractItemView::ExtendedSelection);
 }
 
+/**
+ * Gets the replay at the given index
+ * @return The replay. Returns nullptr if there is no replay at the index.
+ */
+ServerInfo_Replay const *RemoteReplayList_TreeWidget::getReplay(const QModelIndex &ind) const
+{
+    return treeModel->getReplay(proxyModel->mapToSource(ind));
+}
+
 /**
  * Gets all currently selected replays.
  * Any selection that isn't a replay file (e.g. a folder) will appear as a nullptr in the list.
diff --git a/cockatrice/src/server/remote/remote_replay_list_tree_widget.h b/cockatrice/src/server/remote/remote_replay_list_tree_widget.h
index d05957ecd..a93e693fa 100644
--- a/cockatrice/src/server/remote/remote_replay_list_tree_widget.h
+++ b/cockatrice/src/server/remote/remote_replay_list_tree_widget.h
@@ -107,10 +107,10 @@ class RemoteReplayList_TreeWidget : public QTreeView
 private:
     RemoteReplayList_TreeModel *treeModel;
     QSortFilterProxyModel *proxyModel;
-    ServerInfo_Replay const *getNode(const QModelIndex &ind) const;
 
 public:
     RemoteReplayList_TreeWidget(AbstractClient *_client, QWidget *parent = nullptr);
+    ServerInfo_Replay const *getReplay(const QModelIndex &ind) const;
     QList<ServerInfo_Replay const *> getSelectedReplays() const;
     QSet<ServerInfo_ReplayMatch const *> getSelectedReplayMatches() const;
     void refreshTree();