kde-extraapps/kdeplasma-addons/runners/datetime/datetimerunner.cpp

/*
 *   Copyright (C) 2006 Aaron Seigo <aseigo@kde.org>
 *   Copyright (C) 2010 Marco MArtin <notmart@gmail.com>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU Library General Public License version 2 as
 *   published by the Free Software Foundation
 *
 *   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
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this program; if not, write to the
 *   Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include "datetimerunner.h"

#include <KIcon>

#include <QDateTime>
#include <KDebug>
#include <KGlobal>
#include <KLocale>
#include <KSystemTimeZones>
#include <KTimeZone>

#include <Plasma/Applet>

static const char* dateWordChars = "date";
static const QString dateWord = i18nc("Note this is a KRunner keyword", dateWordChars);
static const char* timeWordChars = "time";
static const QString timeWord = i18nc("Note this is a KRunner keyword", timeWordChars);

DateTimeRunner::DateTimeRunner(QObject *parent, const QVariantList &args)
    : Plasma::AbstractRunner(parent, args)
{
    setObjectName(QLatin1String( "DataTimeRunner" ));

    addSyntax(Plasma::RunnerSyntax(dateWord, i18n("Displays the current date")));
    addSyntax(Plasma::RunnerSyntax(dateWord + QLatin1String( " :q:" ), i18n("Displays the current date in a given timezone")));
    addSyntax(Plasma::RunnerSyntax(timeWord, i18n("Displays the current time")));
    addSyntax(Plasma::RunnerSyntax(timeWord + QLatin1String( " :q:" ), i18n("Displays the current time in a given timezone")));
}

DateTimeRunner::~DateTimeRunner()
{
}

void DateTimeRunner::match(Plasma::RunnerContext &context)
{
    const QString term = context.query();
    if (term.compare(dateWord, Qt::CaseInsensitive) == 0 ||
        term.compare(QLatin1String(dateWordChars), Qt::CaseInsensitive) == 0) {
        const QString date = KGlobal::locale()->formatDate(QDate::currentDate());
        addMatch(i18n("Today's date is %1", date), date, context);
    } else if (term.startsWith(dateWord + QLatin1Char(' '), Qt::CaseInsensitive)) {
        QString tzName;
        const QString tz = term.right(term.length() - term.indexOf(QLatin1Char(' ')) - 1);
        QDateTime dt = datetime(tz, tzName);
        if (dt.isValid()) {
            const QString date = KGlobal::locale()->formatDate(dt.date());
            addMatch(i18n("The date in %1 is %2", tzName, date), date, context);
        }
    } else if (term.compare(timeWord, Qt::CaseInsensitive) == 0 ||
        term.compare(QLatin1String(timeWordChars), Qt::CaseInsensitive) == 0) {
        const QString time = KGlobal::locale()->formatTime(QTime::currentTime());
        addMatch(i18n("The current time is %1", time), time, context);
    } else if (term.startsWith(timeWord + QLatin1Char(' '), Qt::CaseInsensitive) ||
        term.startsWith(QLatin1String(timeWordChars) + QLatin1Char(' '), Qt::CaseInsensitive)) {
        QString tzName;
        const QString tz = term.right(term.length() - term.indexOf(QLatin1Char(' ')) - 1);
        QDateTime dt = datetime(tz, tzName);
        if (dt.isValid()) {
            const QString time = KGlobal::locale()->formatTime(dt.time());
            addMatch(i18n("The current time in %1 is %2", tzName, time), time, context);
        }
    }
}

QDateTime DateTimeRunner::datetime(const QString &tz, QString &tzName)
{
    QDateTime dt;
    const QDateTime cdt = QDateTime::currentDateTimeUtc();

    if (tz.compare(QLatin1String("UTC"), Qt::CaseInsensitive) == 0) {
        tzName = QLatin1String( "UTC" );
        dt = cdt;
        return dt;
    }

    foreach (const KTimeZone &zone, KSystemTimeZones::zones()) {
        const QString zoneName = zone.name();
        const QDateTime zoneDateTime = zone.toZoneTime(cdt);
        if (zoneName.compare(tz, Qt::CaseInsensitive) == 0) {
            tzName = zoneName;
            dt = zoneDateTime;
            break;
        }

        if (zoneName.contains(tz, Qt::CaseInsensitive)) {
            tzName = zoneName;
            dt = zoneDateTime;
        } else {
            foreach (const QByteArray &abbrev, zone.abbreviations()) {
                if (QString(abbrev).contains(tz, Qt::CaseInsensitive)) {
                    tzName = abbrev;
                    dt = zoneDateTime;
                }
            }
        }
        if (dt.isValid()) {
            break;
        }
    }

    return dt;
}

void DateTimeRunner::addMatch(const QString &text, const QString &clipboardText, Plasma::RunnerContext &context)
{
    Plasma::QueryMatch match(this);
    match.setText(text);
    match.setData(clipboardText);
    match.setType(Plasma::QueryMatch::InformationalMatch);
    match.setIcon(KIcon(QLatin1String("clock")));

    QList<Plasma::QueryMatch> matches;
    matches << match;
    context.addMatches(context.query(), matches);
}

K_EXPORT_PLASMA_RUNNER(datetime, DateTimeRunner)

#include "moc_datetimerunner.cpp"