#include "StudyRecord.h"

#include "StudySettings.h"
#include "../utils/TimeProvider.h"

ostream& operator<<(ostream& os, const StudyRecord& study)
{
    if(study.date <= QDateTime::currentDateTime())
    {
        const char* dateStr = study.date.toString(Qt::ISODate).toStdString().c_str();
        os << "(" << dateStr <<
            ", g" << study.grade << ", e" << study.easiness << ", " <<
            "i" << study.interval << ")";
    }
    else
        os << "(New card)";
    return os;
}

// Create "new" study record
StudyRecord::StudyRecord():
    date(QDateTime()),
    level(New),
    interval(0),
    grade(Unknown),
    easiness(StudySettings::inst()->initEasiness),
    recallTime(0),
    answerTime(0)
    {
    }

StudyRecord::StudyRecord(int level, int grade, double easiness, double interval):
    date(QDateTime()),
    recallTime(0),
    answerTime(0)
{
    this->level = level;
    this->interval = interval;
    this->grade = grade;
    this->easiness = easiness;
}

bool StudyRecord::operator==(const StudyRecord& aOther) const
    {
    return level == aOther.level &&
        date == aOther.date &&
        interval == aOther.interval &&
        grade == aOther.grade &&
        easiness == aOther.easiness;
    }

bool StudyRecord::timeTriggered() const
{
    if(date.isNull())
        return false;
    QDateTime nextRepetition = date.addSecs( (int)(interval * 60*60*24) );
    return nextRepetition <= TimeProvider::get();
}

int StudyRecord::getSecsToNextRepetition() const
{
    if(date.isNull())
        return 0;
    QDateTime nextRepetition = date.addSecs( (int)(interval * 60*60*24) );
    return TimeProvider::get().secsTo(nextRepetition);
}

bool StudyRecord::isOneDayOld() const
{
    return date.secsTo(TimeProvider::get()) / (60*60*24.) >=
            StudySettings::inst()->nextDayInterval;
}

bool StudyRecord::isLearning() const
{
    return (level == ShortLearning || level == LongLearning) &&
            !isOneDayOld();
}

int StudyRecord::getScheduledTodayReviews() const
{
    switch(level)
    {
    case New:
        return 3;
    case ShortLearning:
        return 2;
    case LongLearning:
        return 1;
    default:
        return 0;
    }
}

bool StudyRecord::isReviewedToday() const
{
    QDateTime recShiftedDate = shiftedDate(date);
    QDateTime curShiftedDate = shiftedDate(QDateTime::currentDateTime());
    return recShiftedDate.date() == curShiftedDate.date();
}

bool StudyRecord::isActivatedToday() const
{
    if(date.isNull())
        return false;
    QDateTime nextRepetition = date.addSecs( (int)(interval * 60*60*24) );
    QDateTime shiftedNextRep = shiftedDate(nextRepetition);
    QDateTime curShiftedDate = shiftedDate(QDateTime::currentDateTime());
    return shiftedNextRep.date() <= curShiftedDate.date();
}

QDateTime StudyRecord::shiftedDate(const QDateTime& aDate)
{
    return aDate.addSecs(-60 * 60 * StudySettings::inst()->dayShift);
}

void StudyRecord::setRecallTime(double time)
{
    recallTime = time;
    if(recallTime > MaxAnswerTime)
        recallTime = MaxAnswerTime;
}

void StudyRecord::setAnswerTime(double time)
{
    answerTime = time;
    if(answerTime > MaxAnswerTime)
        answerTime = MaxAnswerTime;
}