plugins/Unity/Launcher/launchermodel.cpp · unity8

git.iko.soy/unity8

on 45cd840 help sign in

unity8/plugins/Unity/Launcher/launchermodel.cpp

353 lines 11 kB history raw

1/*

3 *

4 * This program is free software; you can redistribute it and/or modify

5 * it under the terms of the GNU Lesser General Public License as published by

6 * the Free Software Foundation; version 3.

7 *

8 * This program is distributed in the hope that it will be useful,

9 * but WITHOUT ANY WARRANTY; without even the implied warranty of

10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

11 * GNU Lesser General Public License for more details.

12 *

13 * You should have received a copy of the GNU Lesser General Public License

14 * along with this program. If not, see <http://www.gnu.org/licenses/>.

15 *

16 * Authors:

17 * Michael Zanetti <michael.zanetti@canonical.com>

18 */

20#include "launchermodel.h"

21#include "launcheritem.h"

22#include "backend/launcherbackend.h"

24#include <unity/shell/application/ApplicationInfoInterface.h>

26#include <QDebug>

28using namespace unity::shell::application;

30LauncherModel::LauncherModel(QObject *parent):

31 LauncherModelInterface(parent),

32 m_backend(new LauncherBackend(this)),

33 m_appManager(0)

34{

35 connect(m_backend, SIGNAL(countChanged(QString,int)), SLOT(countChanged(QString,int)));

36 connect(m_backend, SIGNAL(progressChanged(QString,int)), SLOT(progressChanged(QString,int)));

38 Q_FOREACH (const QString &entry, m_backend->storedApplications()) {

39 LauncherItem *item = new LauncherItem(entry,

40 m_backend->displayName(entry),

41 m_backend->icon(entry),

42 this);

43 item->setPinned(true);

44 m_list.append(item);

45 }

46}

48LauncherModel::~LauncherModel()

49{

50 while (!m_list.empty()) {

51 m_list.takeFirst()->deleteLater();

52 }

53}

55int LauncherModel::rowCount(const QModelIndex &parent) const

56{

57 Q_UNUSED(parent)

58 return m_list.count();

59}

61QVariant LauncherModel::data(const QModelIndex &index, int role) const

62{

63 LauncherItem *item = m_list.at(index.row());

64 switch(role) {

65 case RoleAppId:

66 return item->appId();

67 case RoleName:

68 return item->name();

69 case RoleIcon:

70 return item->icon();

71 case RolePinned:

72 return item->pinned();

73 case RoleCount:

74 return item->count();

75 case RoleProgress:

76 return item->progress();

77 case RoleFocused:

78 return item->focused();

79 }

81 return QVariant();

82}

84unity::shell::launcher::LauncherItemInterface *LauncherModel::get(int index) const

85{

86 if (index < 0 || index >= m_list.count()) {

87 return 0;

88 }

89 return m_list.at(index);

90}

92void LauncherModel::move(int oldIndex, int newIndex)

93{

94 // Make sure its not moved outside the lists

95 if (newIndex < 0) {

96 newIndex = 0;

97 }

98 if (newIndex >= m_list.count()) {

99 newIndex = m_list.count()-1;

100 }

101

102 // Nothing to do?

103 if (oldIndex == newIndex) {

104 return;

105 }

106

107 // QList's and QAbstractItemModel's move implementation differ when moving an item up the list :/

108 // While QList needs the index in the resulting list, beginMoveRows expects it to be in the current list

109 // adjust the model's index by +1 in case we're moving upwards

110 int newModelIndex = newIndex > oldIndex ? newIndex+1 : newIndex;

111

112 beginMoveRows(QModelIndex(), oldIndex, oldIndex, QModelIndex(), newModelIndex);

113 m_list.move(oldIndex, newIndex);

114 endMoveRows();

115

116 if (!m_list.at(newIndex)->pinned()) {

117 pin(m_list.at(newIndex)->appId());

118 } else {

119 storeAppList();

120 }

121}

122

123void LauncherModel::pin(const QString &appId, int index)

124{

125 int currentIndex = findApplication(appId);

126

127 if (currentIndex >= 0) {

128 if (index == -1 || index == currentIndex) {

129 m_list.at(currentIndex)->setPinned(true);

130 QModelIndex modelIndex = this->index(currentIndex);

131 Q_EMIT dataChanged(modelIndex, modelIndex);

132 } else {

133 move(currentIndex, index);

134 // move() will store the list to the backend itself, so just exit at this point.

135 return;

136 }

137 } else {

138 if (index == -1) {

139 index = m_list.count();

140 }

141 beginInsertRows(QModelIndex(), index, index);

142 LauncherItem *item = new LauncherItem(appId,

143 m_backend->displayName(appId),

144 m_backend->icon(appId));

145 item->setPinned(true);

146 m_list.insert(index, item);

147 endInsertRows();

148 }

149

150 storeAppList();

151}

152

153void LauncherModel::requestRemove(const QString &appId)

154{

155 int index = findApplication(appId);

156 if (index < 0) {

157 return;

158 }

159

160 if (m_appManager->findApplication(appId)) {

161 m_list.at(index)->setPinned(false);

162 return;

163 }

164

165 beginRemoveRows(QModelIndex(), index, index);

166 m_list.takeAt(index)->deleteLater();

167 endRemoveRows();

168

169 storeAppList();

170}

171

172void LauncherModel::quickListActionInvoked(const QString &appId, int actionIndex)

173{

174 int index = findApplication(appId);

175 if (index < 0) {

176 return;

177 }

178

179 LauncherItem *item = m_list.at(index);

180 QuickListModel *model = qobject_cast<QuickListModel*>(item->quickList());

181 if (model) {

182 QString actionId = model->get(actionIndex).actionId();

183

184 // Check if this is one of the launcher actions we handle ourselves

185 if (actionId == "pin_item") {

186 if (item->pinned()) {

187 requestRemove(appId);

188 } else {

189 pin(appId);

190 }

191

192 // Nope, we don't know this action, let the backend forward it to the application

193 } else {

194 m_backend->triggerQuickListAction(appId, actionId);

195 }

196 }

197}

198

199void LauncherModel::setUser(const QString &username)

200{

201 m_backend->setUser(username);

202}

203

204ApplicationManagerInterface *LauncherModel::applicationManager() const

205{

206 return m_appManager;

207}

208

209void LauncherModel::setApplicationManager(unity::shell::application::ApplicationManagerInterface *appManager)

210{

211 // Is there already another appmanager set?

212 if (m_appManager) {

213 // Disconnect any signals

214 disconnect(this, SLOT(applicationAdded(QModelIndex,int)));

215 disconnect(this, SLOT(applicationRemoved(QModelIndex,int)));

216 disconnect(this, SLOT(focusedAppIdChanged()));

217

218 // remove any recent/running apps from the launcher

219 QList<int> recentAppIndices;

220 for (int i = 0; i < m_list.count(); ++i) {

221 if (m_list.at(i)->recent()) {

222 recentAppIndices << i;

223 }

224 }

225 int run = 0;

226 while (recentAppIndices.count() > 0) {

227 beginRemoveRows(QModelIndex(), recentAppIndices.first() - run, recentAppIndices.first() - run);

228 m_list.takeAt(recentAppIndices.first() - run)->deleteLater();

229 endRemoveRows();

230 recentAppIndices.takeFirst();

231 ++run;

232 }

233 }

234

235 m_appManager = appManager;

236 connect(m_appManager, SIGNAL(rowsInserted(QModelIndex, int, int)), SLOT(applicationAdded(QModelIndex,int)));

237 connect(m_appManager, SIGNAL(rowsAboutToBeRemoved(QModelIndex,int,int)), SLOT(applicationRemoved(QModelIndex,int)));

238 connect(m_appManager, SIGNAL(focusedApplicationIdChanged()), SLOT(focusedAppIdChanged()));

239

240 Q_EMIT applicationManagerChanged();

241

242 for (int i = 0; i < appManager->count(); ++i) {

243 applicationAdded(QModelIndex(), i);

244 }

245}

246

247

248void LauncherModel::storeAppList()

249{

250 QStringList appIds;

251 Q_FOREACH(LauncherItem *item, m_list) {

252 if (item->pinned()) {

253 appIds << item->appId();

254 }

255 }

256 m_backend->setStoredApplications(appIds);

257}

258

259int LauncherModel::findApplication(const QString &appId)

260{

261 for (int i = 0; i < m_list.count(); ++i) {

262 LauncherItem *item = m_list.at(i);

263 if (item->appId() == appId) {

264 return i;

265 }

266 }

267 return -1;

268}

269

270void LauncherModel::progressChanged(const QString &appId, int progress)

271{

272 int idx = findApplication(appId);

273 if (idx >= 0) {

274 LauncherItem *item = m_list.at(idx);

275 item->setProgress(progress);

276 Q_EMIT dataChanged(index(idx), index(idx), QVector<int>() << RoleProgress);

277 }

278}

279

280

281void LauncherModel::countChanged(const QString &appId, int count)

282{

283 int idx = findApplication(appId);

284 if (idx >= 0) {

285 LauncherItem *item = m_list.at(idx);

286 item->setCount(count);

287 Q_EMIT dataChanged(index(idx), index(idx), QVector<int>() << RoleCount);

288 }

289}

290

291void LauncherModel::applicationAdded(const QModelIndex &parent, int row)

292{

293 Q_UNUSED(parent);

294

295 ApplicationInfoInterface *app = m_appManager->get(row);

296 if (!app) {

297 qDebug() << "Huh? Can't get newly added Application from ApplicationManager... Ignoring this app...";

298 return;

299 }

300

301 bool found = false;

302 Q_FOREACH(LauncherItem *item, m_list) {

303 if (app->appId() == item->appId()) {

304 found = true;

305 break;

306 }

307 }

308 if (found) {

309 // Shall we paint some running/recent app highlight? If yes, do it here.

310 } else {

311 LauncherItem *item = new LauncherItem(app->appId(), app->name(), app->icon().toString());

312 item->setRecent(true);

313 item->setFocused(app->focused());

314

315 beginInsertRows(QModelIndex(), m_list.count(), m_list.count());

316 m_list.append(item);

317 endInsertRows();

318 }

319}

320

321void LauncherModel::applicationRemoved(const QModelIndex &parent, int row)

322{

323 Q_UNUSED(parent)

324

325 int appIndex = -1;

326 for (int i = 0; i < m_list.count(); ++i) {

327 if (m_list.at(i)->appId() == m_appManager->get(row)->appId()) {

328 appIndex = i;

329 break;

330 }

331 }

332

333 if (appIndex > -1 && !m_list.at(appIndex)->pinned()) {

334 beginRemoveRows(QModelIndex(), appIndex, appIndex);

335 m_list.takeAt(appIndex)->deleteLater();

336 endRemoveRows();

337 }

338}

339

340void LauncherModel::focusedAppIdChanged()

341{

342 QString appId = m_appManager->focusedApplicationId();

343 for (int i = 0; i < m_list.count(); ++i) {

344 LauncherItem *item = m_list.at(i);

345 if (!item->focused() && item->appId() == appId) {

346 item->setFocused(true);

347 Q_EMIT dataChanged(index(i), index(i), QVector<int>() << RoleFocused);

348 } else if (item->focused() && item->appId() != appId) {

349 item->setFocused(false);

350 Q_EMIT dataChanged(index(i), index(i), QVector<int>() << RoleFocused);

351 }

352 }

353}