/*
 * @project      The CERN Tape Archive (CTA)
 * @copyright    Copyright © 2022 CERN
 * @license      This program is free software, distributed under the terms of the GNU General Public
 *               Licence version 3 (GPL Version 3), copied verbatim in the file "COPYING". You can
 *               redistribute it and/or modify it under the terms of the GPL Version 3, or (at your
 *               option) any later version.
 *
 *               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.
 *
 *               In applying this licence, CERN does not waive the privileges and immunities
 *               granted to it by virtue of its status as an Intergovernmental Organization or
 *               submit itself to any jurisdiction.
 */

#pragma once

#include "scheduler/PostgresSchedDB/PostgresSchedDB.hpp"
#include "common/dataStructures/LabelFormat.hpp"
#include "common/dataStructures/MountType.hpp"
#include "scheduler/PostgresSchedDB/sql/Transaction.hpp"
#include "catalogue/TapeDrivesCatalogueState.hpp"

#include <memory>
#include <optional>
#include <string>
#include <cstdint>
#include <time.h>

namespace cta::postgresscheddb {

class TapeMountDecisionInfo : public SchedulerDatabase::TapeMountDecisionInfo {
 friend class cta::PostgresSchedDB;

 public:
   explicit TapeMountDecisionInfo(PostgresSchedDB &pdb, rdbms::ConnPool &cp, const std::string &ownerId, TapeDrivesCatalogueState *drivesState, log::Logger &logger);

   std::unique_ptr<SchedulerDatabase::ArchiveMount> createArchiveMount(const cta::SchedulerDatabase::PotentialMount& mount,
                                                                       const catalogue::TapeForWriting& tape,
                                                                       const std::string& driveName,
                                                                       const std::string& logicalLibrary,
                                                                       const std::string& hostName) override;

   std::unique_ptr<SchedulerDatabase::RetrieveMount> createRetrieveMount(const cta::SchedulerDatabase::PotentialMount& mount,
                                                                         const std::string& driveName,
                                                                         const std::string& logicalLibrary,
                                                                         const std::string& hostName) override;

  private:
    /** Acquire scheduler global lock */
    void lock();
    /** Commit decision and release scheduler global lock */
    void commit();

    PostgresSchedDB& m_PostgresSchedDB;
    Transaction m_txn;
    std::string m_ownerId;
    bool m_lockTaken = false;
    log::Logger&           m_logger;
    TapeDrivesCatalogueState *m_tapeDrivesState = nullptr;
};

} // namespace cta::postgresscheddb