workspace/plasma/applets/kickoff/core/leavemodel.cpp

   1 /*
   2     Copyright 2007 Robert Knight <robertknight@gmail.com>
   3
   4     This library is free software; you can redistribute it and/or
   5     modify it under the terms of the GNU Library General Public
   6     License as published by the Free Software Foundation; either
   7     version 2 of the License, or (at your option) any later version.
   8
   9     This library is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12     Library General Public License for more details.
  13
  14     You should have received a copy of the GNU Library General Public License
  15     along with this library; see the file COPYING.LIB.  If not, write to
  16     the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  17     Boston, MA 02110-1301, USA.
  18 */
  19
  20 // Own
  21 #include "core/leavemodel.h"
  22
  23 // Qt
  24 #include <QFileInfo>
  25
  26 // KDE
  27 #include <KConfigGroup>
  28 #include <KDebug>
  29 #include <KLocalizedString>
  30 #include <KIcon>
  31 #include <solid/control/powermanager.h>
  32 #include <kworkspace/kworkspace.h>
  33
  34 // Local
  35 #include "core/models.h"
  36
  37 using namespace Kickoff;
  38
  39 class LeaveModel::Private
  40 {
  41 };
  42
  43 QStandardItem* LeaveModel::createStandardItem(const QString& url)
  44 {
  45     //Q_ASSERT(KUrl(url).scheme() == "leave");
  46     QStandardItem *item = new QStandardItem();
  47     const QString basename = QFileInfo(url).baseName();
  48     if (basename == "logoutonly") {
  49         item->setText(i18n("Logout"));
  50         item->setIcon(KIcon("system-log-out"));
  51         item->setData(i18n("End session"), Kickoff::SubTitleRole);
  52     } else if (basename == "lock") {
  53         item->setText(i18n("Lock"));
  54         item->setIcon(KIcon("system-lock-screen"));
  55         item->setData(i18n("Lock the screen"), Kickoff::SubTitleRole);
  56     } else if (basename == "switch") {
  57         item->setText(i18n("Switch User"));
  58         item->setIcon(KIcon("system-switch-user"));
  59         item->setData(i18n("Start a parallel session as a different user"), Kickoff::SubTitleRole);
  60     } else if (basename == "shutdown") {
  61         item->setText(i18n("Shutdown"));
  62         item->setIcon(KIcon("system-shutdown"));
  63         item->setData(i18n("Turn off computer"), Kickoff::SubTitleRole);
  64     } else if (basename == "restart") {
  65         item->setText(i18nc("Restart computer", "Restart"));
  66         item->setIcon(KIcon("system-restart"));
  67         item->setData(i18n("Restart computer"), Kickoff::SubTitleRole);
  68     } else if (basename == "savesession") {
  69         item->setText(i18n("Save Session"));
  70         item->setIcon(KIcon("document-save"));
  71         item->setData(i18n("Save current session for next login"), Kickoff::SubTitleRole);
  72     } else if (basename == "standby") {
  73         item->setText(i18nc("Puts the system on standby", "Standby"));
  74         item->setIcon(KIcon("system-suspend"));
  75         item->setData(i18n("Pause without logging out"), Kickoff::SubTitleRole);
  76     } else if (basename == "suspenddisk") {
  77         item->setText(i18n("Suspend to Disk"));
  78         item->setIcon(KIcon("system-suspend-hibernate"));
  79         item->setData(i18n("Pause without logging out"), Kickoff::SubTitleRole);
  80     } else if (basename == "suspendram") {
  81         item->setText(i18n("Suspend to RAM"));
  82         item->setIcon(KIcon("system-suspend-hibernate"));
  83         item->setData(i18n("Pause without logging out"), Kickoff::SubTitleRole);
  84     } else {
  85         item->setText(basename);
  86         item->setData(url, Kickoff::SubTitleRole);
  87     }
  88     item->setData(url, Kickoff::UrlRole);
  89     return item;
  90 }
  91
  92 LeaveModel::LeaveModel(QObject *parent)
  93         : QStandardItemModel(parent)
  94         , d(0)
  95 {
  96 }
  97
  98 void LeaveModel::updateModel()
  99 {
 100     clear();
 101
 102     // Session Options
 103     QStandardItem *sessionOptions = new QStandardItem(i18n("Session"));
 104
 105     // Logout
 106     QStandardItem *logoutOption = createStandardItem("leave:/logoutonly");
 107     sessionOptions->appendRow(logoutOption);
 108
 109     // Lock
 110     QStandardItem *lockOption = createStandardItem("leave:/lock");
 111     sessionOptions->appendRow(lockOption);
 112
 113     // Save Session
 114     KConfigGroup c(KSharedConfig::openConfig("ksmserverrc", KConfig::NoGlobals), "General");
 115     if (c.readEntry("loginMode") == "restoreSavedSession") {
 116         QStandardItem *saveSessionOption = createStandardItem("leave:/savesession");
 117         sessionOptions->appendRow(saveSessionOption);
 118     }
 119
 120     // Switch User
 121     QStandardItem *switchUserOption = createStandardItem("leave:/switch");
 122     sessionOptions->appendRow(switchUserOption);
 123
 124     // System Options
 125     QStandardItem *systemOptions = new QStandardItem(i18n("System"));
 126     bool addSystemSession = false;
 127
 128 //FIXME: the proper fix is to implement the KWorkSpace methods for Windows
 129 #ifndef Q_WS_WIN
 130     Solid::Control::PowerManager::SuspendMethods spdMethods = Solid::Control::PowerManager::supportedSuspendMethods();
 131     if (spdMethods & Solid::Control::PowerManager::Standby) {
 132         QStandardItem *standbyOption = createStandardItem("leave:/standby");
 133         systemOptions->appendRow(standbyOption);
 134         addSystemSession = true;
 135     }
 136
 137     if (spdMethods & Solid::Control::PowerManager::ToRam) {
 138         QStandardItem *suspendramOption = createStandardItem("leave:/suspendram");
 139         systemOptions->appendRow(suspendramOption);
 140         addSystemSession = true;
 141     }
 142
 143     if (spdMethods & Solid::Control::PowerManager::ToDisk) {
 144         QStandardItem *suspenddiskOption = createStandardItem("leave:/suspenddisk");
 145         systemOptions->appendRow(suspenddiskOption);
 146         addSystemSession = true;
 147     }
 148
 149     if (KWorkSpace::canShutDown(KWorkSpace::ShutdownConfirmDefault, KWorkSpace::ShutdownTypeReboot)) {
 150         // Restart
 151         QStandardItem *restartOption = createStandardItem("leave:/restart");
 152         systemOptions->appendRow(restartOption);
 153         addSystemSession = true;
 154     }
 155
 156     if (KWorkSpace::canShutDown(KWorkSpace::ShutdownConfirmDefault, KWorkSpace::ShutdownTypeHalt)) {
 157         // Shutdown
 158         QStandardItem *shutDownOption = createStandardItem("leave:/shutdown");
 159         systemOptions->appendRow(shutDownOption);
 160         addSystemSession = true;
 161     }
 162 #endif
 163
 164     appendRow(sessionOptions);
 165     if (addSystemSession) {
 166         appendRow(systemOptions);
 167     } else {
 168         delete systemOptions;
 169     }
 170 }
 171
 172 LeaveModel::~LeaveModel()
 173 {
 174     delete d;
 175 }
 176
 177 #include "leavemodel.moc"
 178