blob: 570b16aa906db56f927c3a9b91b8ef945bcee7c0 [file] [log] [blame]
Dees_Troya13d74f2013-03-24 08:54:55 -05001/*
2 Copyright 2013 bigbiff/Dees_Troy TeamWin
3 This file is part of TWRP/TeamWin Recovery Project.
4
5 TWRP is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 TWRP is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with TWRP. If not, see <http://www.gnu.org/licenses/>.
17*/
18
Dees_Troya13d74f2013-03-24 08:54:55 -050019#include <string.h>
Dees_Troya13d74f2013-03-24 08:54:55 -050020#include <sys/stat.h>
Dees_Troya13d74f2013-03-24 08:54:55 -050021#include <dirent.h>
Dees_Troya13d74f2013-03-24 08:54:55 -050022
23extern "C" {
Dees_Troy2673cec2013-04-02 20:22:16 +000024#include "../twcommon.h"
Dees_Troya13d74f2013-03-24 08:54:55 -050025}
Ethan Yonkerfbb43532015-12-28 21:54:50 +010026#include "../minuitwrp/minui.h"
Dees_Troya13d74f2013-03-24 08:54:55 -050027
28#include "rapidxml.hpp"
29#include "objects.hpp"
30#include "../data.hpp"
Dees_Troya13d74f2013-03-24 08:54:55 -050031#include "../partitions.hpp"
32
Ethan Yonker0a3a98f2015-02-05 00:48:28 +010033GUIPartitionList::GUIPartitionList(xml_node<>* node) : GUIScrollList(node)
Dees_Troya13d74f2013-03-24 08:54:55 -050034{
35 xml_attribute<>* attr;
36 xml_node<>* child;
Dees_Troya13d74f2013-03-24 08:54:55 -050037
Ethan Yonker0a3a98f2015-02-05 00:48:28 +010038 mIconSelected = mIconUnselected = NULL;
Dees_Troya13d74f2013-03-24 08:54:55 -050039 mUpdate = 0;
Dees_Troya13d74f2013-03-24 08:54:55 -050040 updateList = false;
Dees_Troya13d74f2013-03-24 08:54:55 -050041
Ethan Yonker21ff02a2015-02-18 14:35:00 -060042 child = FindNode(node, "icon");
Dees_Troya13d74f2013-03-24 08:54:55 -050043 if (child)
44 {
thatf6ed8fc2015-02-14 20:23:16 +010045 mIconSelected = LoadAttrImage(child, "selected");
46 mIconUnselected = LoadAttrImage(child, "unselected");
Dees_Troya13d74f2013-03-24 08:54:55 -050047 }
Dees_Troya13d74f2013-03-24 08:54:55 -050048
49 // Handle the result variable
Ethan Yonker21ff02a2015-02-18 14:35:00 -060050 child = FindNode(node, "data");
Dees_Troya13d74f2013-03-24 08:54:55 -050051 if (child)
52 {
53 attr = child->first_attribute("name");
54 if (attr)
55 mVariable = attr->value();
56 attr = child->first_attribute("selectedlist");
57 if (attr)
58 selectedList = attr->value();
59 }
60
Ethan Yonker58f21322018-08-24 11:17:36 -050061 int iconWidth = 0, iconHeight = 0;
62 if (mIconSelected && mIconSelected->GetResource() && mIconUnselected && mIconUnselected->GetResource()) {
63 iconWidth = std::max(mIconSelected->GetWidth(), mIconUnselected->GetWidth());
64 iconHeight = std::max(mIconSelected->GetHeight(), mIconUnselected->GetHeight());
65 } else if (mIconSelected && mIconSelected->GetResource()) {
66 iconWidth = mIconSelected->GetWidth();
67 iconHeight = mIconSelected->GetHeight();
68 } else if (mIconUnselected && mIconUnselected->GetResource()) {
69 iconWidth = mIconUnselected->GetWidth();
70 iconHeight = mIconUnselected->GetHeight();
71 }
thatf6ed8fc2015-02-14 20:23:16 +010072 SetMaxIconSize(iconWidth, iconHeight);
Dees_Troya13d74f2013-03-24 08:54:55 -050073
Ethan Yonker21ff02a2015-02-18 14:35:00 -060074 child = FindNode(node, "listtype");
Ethan Yonker0a3a98f2015-02-05 00:48:28 +010075 if (child && (attr = child->first_attribute("name"))) {
76 ListType = attr->value();
77 updateList = true;
Dees_Troya13d74f2013-03-24 08:54:55 -050078 } else {
79 mList.clear();
Dees_Troy2673cec2013-04-02 20:22:16 +000080 LOGERR("No partition listtype specified for partitionlist GUI element\n");
Dees_Troya13d74f2013-03-24 08:54:55 -050081 return;
82 }
83}
84
85GUIPartitionList::~GUIPartitionList()
86{
Dees_Troya13d74f2013-03-24 08:54:55 -050087}
88
89int GUIPartitionList::Update(void)
90{
Matt Mowera8a89d12016-12-30 18:10:37 -060091 if (!isConditionTrue())
Vojtech Bocekede51c52014-02-07 23:58:09 +010092 return 0;
93
Dees_Troya13d74f2013-03-24 08:54:55 -050094 // Check for changes in mount points if the list type is mount and update the list and render if needed
95 if (ListType == "mount") {
96 int listSize = mList.size();
97 for (int i = 0; i < listSize; i++) {
98 if (PartitionManager.Is_Mounted_By_Path(mList.at(i).Mount_Point) && !mList.at(i).selected) {
99 mList.at(i).selected = 1;
100 mUpdate = 1;
101 } else if (!PartitionManager.Is_Mounted_By_Path(mList.at(i).Mount_Point) && mList.at(i).selected) {
102 mList.at(i).selected = 0;
103 mUpdate = 1;
104 }
105 }
106 }
107
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100108 GUIScrollList::Update();
109
110 if (updateList) {
Matt Mowera8a89d12016-12-30 18:10:37 -0600111 // Completely update the list if needed -- Used primarily for
112 // restore as the list for restore will change depending on what
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100113 // partitions were backed up
114 mList.clear();
115 PartitionManager.Get_Partition_List(ListType, &mList);
116 SetVisibleListLocation(0);
117 updateList = false;
118 mUpdate = 1;
Ethan Yonkercfd65092015-02-14 10:39:21 -0600119 if (ListType == "backup" || ListType == "flashimg")
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100120 MatchList();
121 }
122
123 if (mUpdate) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500124 mUpdate = 0;
125 if (Render() == 0)
126 return 2;
127 }
128
Dees_Troya13d74f2013-03-24 08:54:55 -0500129 return 0;
130}
131
Vojtech Bocek07220562014-02-08 02:05:33 +0100132int GUIPartitionList::NotifyVarChange(const std::string& varName, const std::string& value)
Dees_Troya13d74f2013-03-24 08:54:55 -0500133{
thatfa30aca2015-02-13 01:22:22 +0100134 GUIScrollList::NotifyVarChange(varName, value);
135
Matt Mowera8a89d12016-12-30 18:10:37 -0600136 if (!isConditionTrue())
Vojtech Bocekede51c52014-02-07 23:58:09 +0100137 return 0;
138
Dees_Troya13d74f2013-03-24 08:54:55 -0500139 if (varName == mVariable && !mUpdate)
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200140 {
141 if (ListType == "storage") {
Dees_Troya13d74f2013-03-24 08:54:55 -0500142 currentValue = value;
Ethan Yonkercfd65092015-02-14 10:39:21 -0600143 SetPosition();
Dees_Troya13d74f2013-03-24 08:54:55 -0500144 } else if (ListType == "backup") {
bigbiffee7b7ff2020-03-23 15:08:27 -0400145 updateList = true;
Dees_Troya13d74f2013-03-24 08:54:55 -0500146 MatchList();
147 } else if (ListType == "restore") {
148 updateList = true;
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100149 SetVisibleListLocation(0);
Dees_Troya13d74f2013-03-24 08:54:55 -0500150 }
151
152 mUpdate = 1;
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200153 return 0;
154 }
Dees_Troya13d74f2013-03-24 08:54:55 -0500155 return 0;
156}
157
Dees_Troya13d74f2013-03-24 08:54:55 -0500158void GUIPartitionList::SetPageFocus(int inFocus)
159{
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100160 GUIScrollList::SetPageFocus(inFocus);
Dees_Troya13d74f2013-03-24 08:54:55 -0500161 if (inFocus) {
Ethan Yonkercfd65092015-02-14 10:39:21 -0600162 if (ListType == "storage" || ListType == "flashimg") {
Dees_Troya13d74f2013-03-24 08:54:55 -0500163 DataManager::GetValue(mVariable, currentValue);
Ethan Yonkercfd65092015-02-14 10:39:21 -0600164 SetPosition();
Dees_Troya13d74f2013-03-24 08:54:55 -0500165 }
166 updateList = true;
167 mUpdate = 1;
168 }
169}
170
171void GUIPartitionList::MatchList(void) {
172 int i, listSize = mList.size();
173 string variablelist, searchvalue;
174 size_t pos;
175
176 DataManager::GetValue(mVariable, variablelist);
177
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100178 for (i = 0; i < listSize; i++) {
Dees_Troya13d74f2013-03-24 08:54:55 -0500179 searchvalue = mList.at(i).Mount_Point + ";";
180 pos = variablelist.find(searchvalue);
181 if (pos != string::npos) {
182 mList.at(i).selected = 1;
183 } else {
184 mList.at(i).selected = 0;
185 }
186 }
187}
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100188
Ethan Yonkercfd65092015-02-14 10:39:21 -0600189void GUIPartitionList::SetPosition() {
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100190 int listSize = mList.size();
191
Ethan Yonkercfd65092015-02-14 10:39:21 -0600192 SetVisibleListLocation(0);
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100193 for (int i = 0; i < listSize; i++) {
194 if (mList.at(i).Mount_Point == currentValue) {
195 mList.at(i).selected = 1;
196 SetVisibleListLocation(i);
197 } else {
198 mList.at(i).selected = 0;
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100199 }
200 }
201}
202
203size_t GUIPartitionList::GetItemCount()
204{
205 return mList.size();
206}
207
that0af77952015-02-25 08:52:19 +0100208void GUIPartitionList::RenderItem(size_t itemindex, int yPos, bool selected)
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100209{
that0af77952015-02-25 08:52:19 +0100210 // note: the "selected" parameter above is for the currently touched item
211 // don't confuse it with the more persistent "selected" flag per list item used below
212 ImageResource* icon = mList.at(itemindex).selected ? mIconSelected : mIconUnselected;
213 const std::string& text = mList.at(itemindex).Display_Name;
214
215 RenderStdItem(yPos, selected, icon, text.c_str());
Ethan Yonker0a3a98f2015-02-05 00:48:28 +0100216}
217
218void GUIPartitionList::NotifySelect(size_t item_selected)
219{
220 if (item_selected < mList.size()) {
221 int listSize = mList.size();
222 if (ListType == "mount") {
223 if (!mList.at(item_selected).selected) {
224 if (PartitionManager.Mount_By_Path(mList.at(item_selected).Mount_Point, true)) {
225 mList.at(item_selected).selected = 1;
226 PartitionManager.Add_MTP_Storage(mList.at(item_selected).Mount_Point);
227 mUpdate = 1;
228 }
229 } else {
230 if (PartitionManager.UnMount_By_Path(mList.at(item_selected).Mount_Point, true)) {
231 mList.at(item_selected).selected = 0;
232 mUpdate = 1;
233 }
234 }
235 } else if (!mVariable.empty()) {
236 if (ListType == "storage") {
237 int i;
238 std::string str = mList.at(item_selected).Mount_Point;
239 bool update_size = false;
240 TWPartition* Part = PartitionManager.Find_Partition_By_Path(str);
241 if (Part == NULL) {
242 LOGERR("Unable to locate partition for '%s'\n", str.c_str());
243 return;
244 }
245 if (!Part->Is_Mounted() && Part->Removable)
246 update_size = true;
247 if (!Part->Mount(true)) {
248 // Do Nothing
249 } else if (update_size && !Part->Update_Size(true)) {
250 // Do Nothing
251 } else {
252 for (i=0; i<listSize; i++)
253 mList.at(i).selected = 0;
254
255 if (update_size) {
256 char free_space[255];
257 sprintf(free_space, "%llu", Part->Free / 1024 / 1024);
258 mList.at(item_selected).Display_Name = Part->Storage_Name + " (";
259 mList.at(item_selected).Display_Name += free_space;
260 mList.at(item_selected).Display_Name += "MB)";
261 }
262 mList.at(item_selected).selected = 1;
263 mUpdate = 1;
264
265 DataManager::SetValue(mVariable, str);
266 }
267 } else {
268 if (ListType == "flashimg") { // only one item can be selected for flashing images
269 for (int i=0; i<listSize; i++)
270 mList.at(i).selected = 0;
271 }
272 if (mList.at(item_selected).selected)
273 mList.at(item_selected).selected = 0;
274 else
275 mList.at(item_selected).selected = 1;
276
277 int i;
278 string variablelist;
279 for (i=0; i<listSize; i++) {
280 if (mList.at(i).selected) {
281 variablelist += mList.at(i).Mount_Point + ";";
282 }
283 }
284
285 mUpdate = 1;
286 if (selectedList.empty())
287 DataManager::SetValue(mVariable, variablelist);
288 else
289 DataManager::SetValue(selectedList, variablelist);
290 }
291 }
292 }
293}