1 files changed, 127 insertions, 0 deletions
diff --git a/src/base_tab/BaseGameTab.cpp b/src/base_tab/BaseGameTab.cpp
new file mode 100644
index 0000000..79f28c1
--- /dev/null
+++ b/src/base_tab/BaseGameTab.cpp
@@ -0,0 +1,127 @@
+#include "BaseGameTab.hpp"
+#include "AppendGameDialog.hpp"
+#include <wx/filename.h>
+
+BaseGameTab::BaseGameTab(wxFrame *parent, std::string base_file)
+    : TabBase_TabGames(parent), base_file(base_file),
+      base(NULL) {
+
+  game_list->InsertColumn(0, L"id", wxLIST_FORMAT_LEFT, 50);
+  game_list->InsertColumn(1, L"White", wxLIST_FORMAT_LEFT, 200);
+  game_list->InsertColumn(2, L"Black", wxLIST_FORMAT_LEFT, 200);
+  game_list->InsertColumn(3, L"Event", wxLIST_FORMAT_LEFT, 150);
+  game_list->InsertColumn(4, L"Round", wxLIST_FORMAT_LEFT, 100);
+  game_list->InsertColumn(5, L"Result", wxLIST_FORMAT_LEFT, 200);
+  game_list->InsertColumn(6, L"ECO", wxLIST_FORMAT_LEFT, 200);
+
+  this->Bind(wxEVT_BUTTON, &BaseGameTab::OnDelete, this, ID_DELETE_BUTTON);
+  this->Bind(wxEVT_BUTTON, &BaseGameTab::OnSave, this, ID_SAVE_BUTTON);
+  this->Bind(wxEVT_BUTTON, &BaseGameTab::OnExport, this, ID_EXPORT_BUTTON);
+  this->Bind(wxEVT_LIST_ITEM_ACTIVATED, &BaseGameTab::OnOpenGame, this, wxID_ANY);
+  this->Bind(wxEVT_BUTTON, &BaseGameTab::OnImport, this, ID_IMPORT_BUTTON);
+
+  current_base->SetLabel(base_file);
+  LoadFile();
+}
+
+void BaseGameTab::OnImport(wxCommandEvent &event) {
+  AppendGameDialog *dia = new AppendGameDialog(this, base);
+  dia->ShowModal();
+  game_list->DeleteAllItems();
+  deleted.clear();
+  edited.clear();
+  LoadFile();
+}
+
+void BaseGameTab::OnDelete(wxCommandEvent &event) {
+  long selected = -1;
+
+  while ((selected = game_list->GetNextItem(selected, wxLIST_NEXT_ALL,
+                                            wxLIST_STATE_SELECTED)) !=
+         wxNOT_FOUND) {
+    wxListItem listItem;
+    listItem.m_itemId = selected; // sets row
+    listItem.m_col = 0;           // sets column
+    game_list->GetItem(listItem); // gets item
+    deleted.push_back(std::stoi(listItem.GetText().ToStdString()));
+    for (std::uint32_t &i : deleted) {
+      wxLogDebug("%d", i);
+    }
+    game_list->SetItemBackgroundColour(selected, *wxRED);
+  }
+}
+
+void BaseGameTab::OnSave(wxCommandEvent &event) {
+  std::vector<std::shared_ptr<GameBase>> new_games_bases;
+  std::vector<std::shared_ptr<Game>> new_games;
+  new_games.insert(
+      new_games.end(), edited.begin(),
+      edited.end()); // Add edited game (since they are also deleted)
+  base->Save(deleted, new_games_bases, new_games);
+  game_list->DeleteAllItems();
+  edited.clear();
+  deleted.clear();
+  LoadFile();
+}
+
+void BaseGameTab::OnOpenGame(wxListEvent &event) {
+  wxLogDebug("Open!");
+  long id = std::stoi(event.GetItem().GetText().ToStdString());
+  std::shared_ptr<Game> *g = new std::shared_ptr<Game>(base->GetGame(id));
+  if (g != NULL) {
+    edited.push_back(*g);
+    deleted.push_back(id);
+    game_list->SetItemBackgroundColour(event.GetIndex(), *wxGREEN);
+    wxCommandEvent newGameEvent(NEW_GAME_EVENT, GetId());
+    newGameEvent.SetEventObject(this);
+    newGameEvent.SetClientData(g);
+    ProcessEvent(newGameEvent);
+  }
+}
+
+void BaseGameTab::ApplyPreferences() {}
+
+void BaseGameTab::OnExport(wxCommandEvent &event) {
+  wxFileDialog openFileDialog(this, _("Export database"), "", "",
+                              "Database files (*.pgn)|*.pgn",
+                              wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
+  if (openFileDialog.ShowModal() != wxID_CANCEL) {
+    std::string path = openFileDialog.GetPath().ToStdString();
+    wxFileName file(base_file);
+    wxString ext = file.GetExt().Lower();
+    GameBase *base;
+    if (ext == "pgn") {
+      base = new PGNGameBase(path);
+      base->Export(this->base);
+      delete base;
+    }
+  }
+}
+
+void BaseGameTab::LoadFile() {
+  wxFileName file(base_file);
+  wxString ext = file.GetExt().Lower();
+  if (ext == "pgn") {
+    base = std::shared_ptr<GameBase>(new PGNGameBase(base_file));
+    SetLabel(file.GetName() + "(PGN)");
+  }
+
+  if (base != NULL) {
+    long id = 0;
+    while (base->NextGame()) {
+      long index =
+          game_list->InsertItem(0, std::to_string(id)); // want this for col. 1
+      game_list->SetItem(index, 1, base->GetTag("White"));
+      game_list->SetItem(index, 2, base->GetTag("Black"));
+      game_list->SetItem(index, 3, base->GetTag("Event"));
+      game_list->SetItem(index, 4, base->GetTag("Round"));
+      game_list->SetItem(index, 5, base->GetTag("Result"));
+      game_list->SetItem(index, 6, base->GetTag("ECO"));
+      id++;
+    }
+  }
+
+  wxCommandEvent event(REFRESH_TAB_TITLE, GetId());
+  event.SetEventObject(this);
+  ProcessEvent(event);
+}