/********************************************************************
 KWin - the KDE window manager
 This file is part of the KDE project.

Copyright (C) 2017 Martin Flöser <mgraesslin@kde.org>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, 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.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*********************************************************************/
#include "orientation_sensor.h"

#include <QOrientationSensor>
#include <QOrientationReading>

#include <KStatusNotifierItem>
#include <KLocalizedString>

namespace KWin
{

OrientationSensor::OrientationSensor(QObject *parent)
    : QObject(parent)
    , m_sensor(new QOrientationSensor(this))
{
    connect(m_sensor, &QOrientationSensor::readingChanged, this,
        [this] {
            auto toOrientation = [] (auto reading) {
                switch (reading->orientation()) {
                case QOrientationReading::Undefined:
                    return OrientationSensor::Orientation::Undefined;
                case QOrientationReading::TopUp:
                    return OrientationSensor::Orientation::TopUp;
                case QOrientationReading::TopDown:
                    return OrientationSensor::Orientation::TopDown;
                case QOrientationReading::LeftUp:
                    return OrientationSensor::Orientation::LeftUp;
                case QOrientationReading::RightUp:
                    return OrientationSensor::Orientation::RightUp;
                case QOrientationReading::FaceUp:
                    return OrientationSensor::Orientation::FaceUp;
                case QOrientationReading::FaceDown:
                    return OrientationSensor::Orientation::FaceDown;
                default:
                    Q_UNREACHABLE();
                }
            };
            const auto orientation = toOrientation(m_sensor->reading());
            if (m_orientation != orientation) {
                m_orientation = orientation;
                emit orientationChanged();
            }
        }
    );
    connect(m_sensor, &QOrientationSensor::activeChanged, this,
        [this] {
            if (!m_sni) {
                return;
            }
            if (m_sensor->isActive()) {
                m_sni->setToolTipTitle(i18n("Automatic screen rotation is enabled"));
            } else {
                m_sni->setToolTipTitle(i18n("Automatic screen rotation is disabled"));
            }
        }
    );
}

OrientationSensor::~OrientationSensor() = default;

void OrientationSensor::setEnabled(bool enabled)
{
    if (m_enabled == enabled) {
        return;
    }
    m_enabled = enabled;
    if (m_enabled) {
        setupStatusNotifier();
    } else {
        delete m_sni;
        m_sni = nullptr;
    }
    startStopSensor();
}

void OrientationSensor::setupStatusNotifier()
{
    if (m_sni) {
        return;
    }
    m_sni = new KStatusNotifierItem(QStringLiteral("kwin-automatic-rotation"), this);
    m_sni->setStandardActionsEnabled(false);
    m_sni->setCategory(KStatusNotifierItem::Hardware);
    m_sni->setStatus(KStatusNotifierItem::Passive);
    m_sni->setTitle(i18n("Automatic Screen Rotation"));
    // TODO: proper icon with state
    m_sni->setIconByName(QStringLiteral("preferences-desktop-display"));
    // we start disabled, it gets updated when the sensor becomes active
    m_sni->setToolTipTitle(i18n("Automatic screen rotation is disabled"));
    connect(m_sni, &KStatusNotifierItem::activateRequested, this,
        [this] {
            m_userEnabled = !m_userEnabled;
            startStopSensor();
        }
    );
}

void OrientationSensor::startStopSensor()
{
    if (m_enabled && m_userEnabled) {
        m_sensor->start();
    } else {
        m_sensor->stop();
    }
}

}