#include "DetectorsDCS/RunStatusChecker.h"

#include "CCDB/BasicCCDBManager.h"

#include "CCDB/CCDBTimeStampUtils.h"

using namespace o2::dcs;

const o2::parameters::GRPECSObject* RunStatusChecker::check(long ts)

{

if (ts < 0) {

ts = o2::ccdb::getCurrentTimestamp();

}

if (ts <= mLastTimeStampChecked) {

LOGP(alarm, "RunStatusChecker::check was called with decreasing timestamp {}, previous was {}", ts, mLastTimeStampChecked);

return nullptr;

}

mLastTimeStampChecked = ts;

auto& mgr = o2::ccdb::BasicCCDBManager::instance();

bool fatalOn = mgr.getFatalWhenNull();

mgr.setFatalWhenNull(false);

if (mRunStatus == RunStatus::STOP) { // the STOP was detected at previous check

mRunFollowed = -1;

mRunStatus = RunStatus::NONE;

}

std::map<std::string, std::string> md;

if (mRunFollowed > 0) { // run start was seen

md["runNumber"] = std::to_string(mRunFollowed);

}

const auto* grp = mgr.getSpecific<o2::parameters::GRPECSObject>("GLO/Config/GRPECS", ts, md);

if (grp) { // some object was returned

if (mRunFollowed > 0) {

if ((ts > grp->getTimeEnd()) && (grp->getTimeEnd() > grp->getTimeStart())) { // this means that the EOR was registered

mRunStatus = RunStatus::STOP;

} else { // run still continues

mRunStatus = RunStatus::ONGOING;

}

} else { // we were not following detector run, check if the current one has asked detectors

if ((grp->getDetsReadOut() & mDetMask) == mDetMask) { // we start following this run

if (grp->getTimeEnd() > grp->getTimeStart()) {

if (ts < grp->getTimeEnd()) { // only in tests with ad hoc ts the ts_EOR can be seen > ts

mRunStatus = RunStatus::START;

mRunFollowed = grp->getRun();

}

} else {

mRunStatus = RunStatus::START;

mRunFollowed = grp->getRun();

}

}

}

} else { // query did not return any GRP -> we are certainly not in the wanted detectors run

if (mRunFollowed > 0) { // normally this should not happen

LOGP(warning, "We were following {} run {} but the query at {} did not return any GRP, problem with EOR?", o2::detectors::DetID::getNames(mDetMask), mRunFollowed, ts);

mRunStatus = RunStatus::STOP;

}

}

mgr.setFatalWhenNull(fatalOn);

return mRunStatus == RunStatus::NONE ? nullptr : grp;

}