RefreshCollection.h

/*
 * Copyright (C) 2024-2025 Yury Bobylev <bobilev_yury@mail.ru>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation, version 3.
 *
 * 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 General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program. If not, see <https://www.gnu.org/licenses/>.
 */
 
#ifndef REFRESHCOLLECTION_H
#define REFRESHCOLLECTION_H
 
#include <AuxFunc.h>
#include <BaseKeeper.h>
#include <BookBaseEntry.h>
#include <BookMarks.h>
#include <CreateCollection.h>
#include <FileParseEntry.h>
#include <cstdint>
#include <filesystem>
#include <functional>
#include <memory>
#include <string>
#include <vector>
 
#ifndef USE_OPENMP
#include <atomic>
#include <mutex>
#else
#include <omp.h>
#endif

class RefreshCollection : public CreateCollection
{
public:
  RefreshCollection(const std::shared_ptr<AuxFunc> &af,
                    const std::string &collection_name, const int &num_threads,
                    const bool &remove_empty, const bool &refresh_bookmarks,
                    const bool &fast_refresh,
                    const std::shared_ptr<BookMarks> &bookmarks);

  virtual ~RefreshCollection();

  std::function<void(const double &total_hash)> total_bytes_to_hash;

  std::function<void(const double &hashed)> bytes_hashed;

  void
  refreshCollection();

  void
  refreshFile(const BookBaseEntry &bbe);

  bool
  editBook(const BookBaseEntry &bbe_old, const BookBaseEntry &bbe_new);

  bool
  refreshBook(const BookBaseEntry &bbe);

  void
  set_rar_support(const bool &rar_support);
 
private:
  std::filesystem::path
  getBasePath(const std::string &collection_name);
 
  std::filesystem::path
  getBooksPath();
 
  void
  compareVectors(std::vector<FileParseEntry> &base,
                 std::vector<std::filesystem::path> &books_files);
 
  bool
  compareFunction1(const std::filesystem::path &book_path,
                   const FileParseEntry &ent);
 
  bool
  compareFunction2(const FileParseEntry &ent,
                   const std::filesystem::path &book_path);
 
  void
  checkHashes(std::vector<FileParseEntry> *base,
              std::vector<std::filesystem::path> *books_files);
 
  void
  hashThread(const std::filesystem::path &file_to_hash,
             std::vector<FileParseEntry> *base);
 
  void
  refreshBookMarks(const std::shared_ptr<BaseKeeper> &bk);
 
  void
  getFilenamesFromArchives(const std::filesystem::path &arch_path,
                           const std::string &prefix,
                           std::vector<std::string> &result, void *la);
 
  std::shared_ptr<AuxFunc> af;
  int num_threads = 1;
  bool remove_empty = false;
  bool fast_refresh = false;
  bool refresh_bookmarks = false;
  std::shared_ptr<BookMarks> bookmarks;
 
  std::string collection_name;
 
#ifndef USE_OPENMP
  std::atomic<uintmax_t> bytes_summ;
 
  std::mutex basemtx;
  std::mutex already_hashedmtx;
  std::mutex need_to_parsemtx;
 
  std::mutex newthrmtx;
  std::condition_variable continue_hashing;
  std::vector<std::tuple<unsigned, bool>> thr_pool;
#else
  uintmax_t bytes_summ = 0;
  omp_lock_t basemtx;
  omp_lock_t already_hashedmtx;
  omp_lock_t need_to_parsemtx;
#endif
};
 
#endif // REFRESHCOLLECTION_H