/*
 * @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 <memory>
#include <string>

#include "scheduler/PostgresSchedDB/PostgresSchedDB.hpp"
#include "common/log/Logger.hpp"

namespace cta::catalogue {

class Catalogue;
}

namespace cta {

class PostgresSchedDBQCR {
// QueueCleanupRunner
public:

  PostgresSchedDBQCR(catalogue::Catalogue &catalogue, PostgresSchedDB &pgs) { }
  void runOnePass(log::LogContext & lc) { }
};

class PostgresSchedDBGC {
// GarbageCollector
public:

  PostgresSchedDBGC(void *pgstuff, catalogue::Catalogue& catalogue) { }
  void runOnePass(log::LogContext & lc) { }
};

class PostgresSchedDBInit
{
public:
  PostgresSchedDBInit(const std::string& client_process, const std::string& db_conn_str, log::Logger& log,
    bool leaveNonEmptyAgentsBehind = false)
  {
    connStr = db_conn_str;
    clientProc = client_process;
    login = rdbms::Login::parseString(connStr);
    if (login.dbType != rdbms::Login::DBTYPE_POSTGRESQL) {
      std::runtime_error("scheduler dbconnect string must specify postgres.");
    }
  }

  std::unique_ptr<PostgresSchedDB> getSchedDB(catalogue::Catalogue& catalogue, log::Logger& log) {
    const uint64_t nbConns = 2;
    return std::make_unique<PostgresSchedDB>( clientProc, log, catalogue, login, nbConns);
  }

  PostgresSchedDBGC getGarbageCollector(catalogue::Catalogue& catalogue) {
    return PostgresSchedDBGC(nullptr, catalogue);
  }

  PostgresSchedDBQCR getQueueCleanupRunner(catalogue::Catalogue& catalogue, PostgresSchedDB &pgs) {
    return PostgresSchedDBQCR(catalogue, pgs);
  }

private:
  std::string connStr;
  std::string clientProc;
  rdbms::Login login;
};

typedef PostgresSchedDBInit      SchedulerDBInit_t;
typedef PostgresSchedDB          SchedulerDB_t;

} // namespace cta