summaryrefslogtreecommitdiffstats
path: root/framework/src/domain/daylongeventmodel.cpp
blob: c6d5776e8bd58f3a6e8404970efe62e62223e18f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
    Copyright (c) 2018 Michael Bohlender <michael.bohlender@kdemail.net>
    Copyright (c) 2018 Christian Mollekopf <mollekopf@kolabsys.com>
    Copyright (c) 2018 Rémi Nicole <minijackson@riseup.net>

    This library is free software; you can redistribute it and/or modify it
    under the terms of the GNU Library General Public License as published by
    the Free Software Foundation; either version 2 of the License, or (at your
    option) any later version.

    This library 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 Library General Public
    License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to the
    Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
    02110-1301, USA.
*/

#include "daylongeventmodel.h"

#include <sink/log.h>
#include <sink/query.h>
#include <sink/store.h>

DayLongEventModel::DayLongEventModel(QObject *parent) : QSortFilterProxyModel(parent)
{
    Sink::Query query;
    query.setFlags(Sink::Query::LiveQuery);
    query.request<Event::Summary>();
    query.request<Event::Description>();
    query.request<Event::StartTime>();
    query.request<Event::EndTime>();

    query.filter<Event::AllDay>(true);

    eventModel = Sink::Store::loadModel<Event>(query);

    setSourceModel(eventModel.data());
}

QHash<int, QByteArray> DayLongEventModel::roleNames() const
{
    return {
        {Summary, "summary"},
        {Description, "description"},
        {StartDate, "starts"},
        {Duration, "duration"},
    };
}

QVariant DayLongEventModel::data(const QModelIndex &idx, int role) const
{
    auto srcIdx = mapToSource(idx);
    auto event  = srcIdx.data(Sink::Store::DomainObjectRole).value<Event::Ptr>();

    switch (role) {
        case Summary:
            return event->getSummary();
        case Description:
            return event->getDescription();
        case StartDate: {
            auto dayIndex = mPeriodStart.daysTo(event->getStartTime().date());
            if (dayIndex < 0) {
                return 0;
            }

            return dayIndex;
        }
        case Duration:
            return event->getStartTime().date().daysTo(event->getEndTime().date());
    }

    return QSortFilterProxyModel::data(idx, role);
}

bool DayLongEventModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
{
    auto idx   = sourceModel()->index(sourceRow, 0, sourceParent);
    auto event = idx.data(Sink::Store::DomainObjectRole).value<Event::Ptr>();

    auto eventStart = event->getStartTime().date();
    auto eventEnd   = event->getEndTime().date();

    if (!eventStart.isValid() || !eventEnd.isValid()) {
        SinkWarning() << "Invalid date in the eventModel, ignoring...";
        return false;
    }

    return eventStart < mPeriodStart.addDays(mPeriodLength) && eventEnd >= mPeriodStart;
}

QDate DayLongEventModel::periodStart() const
{
    return mPeriodStart;
}

void DayLongEventModel::setPeriodStart(const QDate &start)
{
    if (!start.isValid()) {
        SinkWarning() << "Passed an invalid starting date in setPeriodStart, ignoring...";
        return;
    }

    mPeriodStart = start;
}

void DayLongEventModel::setPeriodStart(const QVariant &start)
{
    setPeriodStart(start.toDate());
}

int DayLongEventModel::periodLength() const
{
    return mPeriodLength;
}

void DayLongEventModel::setPeriodLength(int length)
{
    mPeriodLength = length;
}