1 /* 2 * Copyright (C) 2014 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.launcher3; 18 19 import android.appwidget.AppWidgetHost; 20 import android.content.ComponentName; 21 import android.content.ContentValues; 22 import android.content.Context; 23 import android.content.Intent; 24 import android.content.pm.ActivityInfo; 25 import android.content.pm.PackageManager; 26 import android.content.res.Resources; 27 import android.content.res.XmlResourceParser; 28 import android.database.sqlite.SQLiteDatabase; 29 import android.graphics.drawable.Drawable; 30 import android.net.Uri; 31 import android.os.Build; 32 import android.os.Build.VERSION; 33 import android.os.Bundle; 34 import android.os.Process; 35 import android.text.TextUtils; 36 import android.util.ArrayMap; 37 import android.util.Log; 38 import android.util.Pair; 39 import android.util.Patterns; 40 import com.android.launcher3.LauncherProvider.SqlArguments; 41 import com.android.launcher3.LauncherSettings.Favorites; 42 import com.android.launcher3.config.FeatureFlags; 43 import com.android.launcher3.graphics.LauncherIcons; 44 import com.android.launcher3.util.Thunk; 45 import java.io.IOException; 46 import java.util.ArrayList; 47 import java.util.Locale; 48 import org.xmlpull.v1.XmlPullParser; 49 import org.xmlpull.v1.XmlPullParserException; 50 51 /** 52 * Layout parsing code for auto installs layout 53 */ 54 public class AutoInstallsLayout { 55 private static final String TAG = "AutoInstalls"; 56 private static final boolean LOGD = false; 57 58 /** Marker action used to discover a package which defines launcher customization */ 59 static final String ACTION_LAUNCHER_CUSTOMIZATION = 60 "android.autoinstalls.config.action.PLAY_AUTO_INSTALL"; 61 62 /** 63 * Layout resource which also includes grid size and hotseat count, e.g., default_layout_6x6_h5 64 */ 65 private static final String FORMATTED_LAYOUT_RES_WITH_HOSTEAT = "default_layout_%dx%d_h%s"; 66 private static final String FORMATTED_LAYOUT_RES = "default_layout_%dx%d"; 67 private static final String LAYOUT_RES = "default_layout"; 68 get(Context context, AppWidgetHost appWidgetHost, LayoutParserCallback callback)69 static AutoInstallsLayout get(Context context, AppWidgetHost appWidgetHost, 70 LayoutParserCallback callback) { 71 Pair<String, Resources> customizationApkInfo = Utilities.findSystemApk( 72 ACTION_LAUNCHER_CUSTOMIZATION, context.getPackageManager()); 73 if (customizationApkInfo == null) { 74 return null; 75 } 76 return get(context, customizationApkInfo.first, customizationApkInfo.second, 77 appWidgetHost, callback); 78 } 79 get(Context context, String pkg, Resources targetRes, AppWidgetHost appWidgetHost, LayoutParserCallback callback)80 static AutoInstallsLayout get(Context context, String pkg, Resources targetRes, 81 AppWidgetHost appWidgetHost, LayoutParserCallback callback) { 82 InvariantDeviceProfile grid = LauncherAppState.getIDP(context); 83 84 // Try with grid size and hotseat count 85 String layoutName = String.format(Locale.ENGLISH, FORMATTED_LAYOUT_RES_WITH_HOSTEAT, 86 grid.numColumns, grid.numRows, grid.numHotseatIcons); 87 int layoutId = targetRes.getIdentifier(layoutName, "xml", pkg); 88 89 // Try with only grid size 90 if (layoutId == 0) { 91 Log.d(TAG, "Formatted layout: " + layoutName 92 + " not found. Trying layout without hosteat"); 93 layoutName = String.format(Locale.ENGLISH, FORMATTED_LAYOUT_RES, 94 grid.numColumns, grid.numRows); 95 layoutId = targetRes.getIdentifier(layoutName, "xml", pkg); 96 } 97 98 // Try the default layout 99 if (layoutId == 0) { 100 Log.d(TAG, "Formatted layout: " + layoutName + " not found. Trying the default layout"); 101 layoutId = targetRes.getIdentifier(LAYOUT_RES, "xml", pkg); 102 } 103 104 if (layoutId == 0) { 105 Log.e(TAG, "Layout definition not found in package: " + pkg); 106 return null; 107 } 108 return new AutoInstallsLayout(context, appWidgetHost, callback, targetRes, layoutId, 109 TAG_WORKSPACE); 110 } 111 112 // Object Tags 113 private static final String TAG_INCLUDE = "include"; 114 private static final String TAG_WORKSPACE = "workspace"; 115 private static final String TAG_APP_ICON = "appicon"; 116 private static final String TAG_AUTO_INSTALL = "autoinstall"; 117 private static final String TAG_FOLDER = "folder"; 118 private static final String TAG_APPWIDGET = "appwidget"; 119 private static final String TAG_SHORTCUT = "shortcut"; 120 private static final String TAG_EXTRA = "extra"; 121 122 private static final String ATTR_CONTAINER = "container"; 123 private static final String ATTR_RANK = "rank"; 124 125 private static final String ATTR_PACKAGE_NAME = "packageName"; 126 private static final String ATTR_CLASS_NAME = "className"; 127 private static final String ATTR_TITLE = "title"; 128 private static final String ATTR_SCREEN = "screen"; 129 130 // x and y can be specified as negative integers, in which case -1 represents the 131 // last row / column, -2 represents the second last, and so on. 132 private static final String ATTR_X = "x"; 133 private static final String ATTR_Y = "y"; 134 135 private static final String ATTR_SPAN_X = "spanX"; 136 private static final String ATTR_SPAN_Y = "spanY"; 137 private static final String ATTR_ICON = "icon"; 138 private static final String ATTR_URL = "url"; 139 140 // Attrs for "Include" 141 private static final String ATTR_WORKSPACE = "workspace"; 142 143 // Style attrs -- "Extra" 144 private static final String ATTR_KEY = "key"; 145 private static final String ATTR_VALUE = "value"; 146 147 private static final String HOTSEAT_CONTAINER_NAME = 148 Favorites.containerToString(Favorites.CONTAINER_HOTSEAT); 149 150 private static final String ACTION_APPWIDGET_DEFAULT_WORKSPACE_CONFIGURE = 151 "com.android.launcher.action.APPWIDGET_DEFAULT_WORKSPACE_CONFIGURE"; 152 153 @Thunk final Context mContext; 154 @Thunk final AppWidgetHost mAppWidgetHost; 155 protected final LayoutParserCallback mCallback; 156 157 protected final PackageManager mPackageManager; 158 protected final Resources mSourceRes; 159 protected final int mLayoutId; 160 161 private final InvariantDeviceProfile mIdp; 162 private final int mRowCount; 163 private final int mColumnCount; 164 165 private final long[] mTemp = new long[2]; 166 @Thunk final ContentValues mValues; 167 protected final String mRootTag; 168 169 protected SQLiteDatabase mDb; 170 AutoInstallsLayout(Context context, AppWidgetHost appWidgetHost, LayoutParserCallback callback, Resources res, int layoutId, String rootTag)171 public AutoInstallsLayout(Context context, AppWidgetHost appWidgetHost, 172 LayoutParserCallback callback, Resources res, 173 int layoutId, String rootTag) { 174 mContext = context; 175 mAppWidgetHost = appWidgetHost; 176 mCallback = callback; 177 178 mPackageManager = context.getPackageManager(); 179 mValues = new ContentValues(); 180 mRootTag = rootTag; 181 182 mSourceRes = res; 183 mLayoutId = layoutId; 184 185 mIdp = LauncherAppState.getIDP(context); 186 mRowCount = mIdp.numRows; 187 mColumnCount = mIdp.numColumns; 188 } 189 190 /** 191 * Loads the layout in the db and returns the number of entries added on the desktop. 192 */ loadLayout(SQLiteDatabase db, ArrayList<Long> screenIds)193 public int loadLayout(SQLiteDatabase db, ArrayList<Long> screenIds) { 194 mDb = db; 195 try { 196 return parseLayout(mLayoutId, screenIds); 197 } catch (Exception e) { 198 Log.e(TAG, "Error parsing layout: " + e); 199 return -1; 200 } 201 } 202 203 /** 204 * Parses the layout and returns the number of elements added on the homescreen. 205 */ parseLayout(int layoutId, ArrayList<Long> screenIds)206 protected int parseLayout(int layoutId, ArrayList<Long> screenIds) 207 throws XmlPullParserException, IOException { 208 XmlResourceParser parser = mSourceRes.getXml(layoutId); 209 beginDocument(parser, mRootTag); 210 final int depth = parser.getDepth(); 211 int type; 212 ArrayMap<String, TagParser> tagParserMap = getLayoutElementsMap(); 213 int count = 0; 214 215 while (((type = parser.next()) != XmlPullParser.END_TAG || 216 parser.getDepth() > depth) && type != XmlPullParser.END_DOCUMENT) { 217 if (type != XmlPullParser.START_TAG) { 218 continue; 219 } 220 count += parseAndAddNode(parser, tagParserMap, screenIds); 221 } 222 return count; 223 } 224 225 /** 226 * Parses container and screenId attribute from the current tag, and puts it in the out. 227 * @param out array of size 2. 228 */ parseContainerAndScreen(XmlResourceParser parser, long[] out)229 protected void parseContainerAndScreen(XmlResourceParser parser, long[] out) { 230 if (HOTSEAT_CONTAINER_NAME.equals(getAttributeValue(parser, ATTR_CONTAINER))) { 231 out[0] = Favorites.CONTAINER_HOTSEAT; 232 // Hack: hotseat items are stored using screen ids 233 long rank = Long.parseLong(getAttributeValue(parser, ATTR_RANK)); 234 out[1] = (FeatureFlags.NO_ALL_APPS_ICON || rank < mIdp.getAllAppsButtonRank()) 235 ? rank : (rank + 1); 236 } else { 237 out[0] = Favorites.CONTAINER_DESKTOP; 238 out[1] = Long.parseLong(getAttributeValue(parser, ATTR_SCREEN)); 239 } 240 } 241 242 /** 243 * Parses the current node and returns the number of elements added. 244 */ parseAndAddNode( XmlResourceParser parser, ArrayMap<String, TagParser> tagParserMap, ArrayList<Long> screenIds)245 protected int parseAndAddNode( 246 XmlResourceParser parser, 247 ArrayMap<String, TagParser> tagParserMap, 248 ArrayList<Long> screenIds) 249 throws XmlPullParserException, IOException { 250 251 if (TAG_INCLUDE.equals(parser.getName())) { 252 final int resId = getAttributeResourceValue(parser, ATTR_WORKSPACE, 0); 253 if (resId != 0) { 254 // recursively load some more favorites, why not? 255 return parseLayout(resId, screenIds); 256 } else { 257 return 0; 258 } 259 } 260 261 mValues.clear(); 262 parseContainerAndScreen(parser, mTemp); 263 final long container = mTemp[0]; 264 final long screenId = mTemp[1]; 265 266 mValues.put(Favorites.CONTAINER, container); 267 mValues.put(Favorites.SCREEN, screenId); 268 269 mValues.put(Favorites.CELLX, 270 convertToDistanceFromEnd(getAttributeValue(parser, ATTR_X), mColumnCount)); 271 mValues.put(Favorites.CELLY, 272 convertToDistanceFromEnd(getAttributeValue(parser, ATTR_Y), mRowCount)); 273 274 TagParser tagParser = tagParserMap.get(parser.getName()); 275 if (tagParser == null) { 276 if (LOGD) Log.d(TAG, "Ignoring unknown element tag: " + parser.getName()); 277 return 0; 278 } 279 long newElementId = tagParser.parseAndAdd(parser); 280 if (newElementId >= 0) { 281 // Keep track of the set of screens which need to be added to the db. 282 if (!screenIds.contains(screenId) && 283 container == Favorites.CONTAINER_DESKTOP) { 284 screenIds.add(screenId); 285 } 286 return 1; 287 } 288 return 0; 289 } 290 addShortcut(String title, Intent intent, int type)291 protected long addShortcut(String title, Intent intent, int type) { 292 long id = mCallback.generateNewItemId(); 293 mValues.put(Favorites.INTENT, intent.toUri(0)); 294 mValues.put(Favorites.TITLE, title); 295 mValues.put(Favorites.ITEM_TYPE, type); 296 mValues.put(Favorites.SPANX, 1); 297 mValues.put(Favorites.SPANY, 1); 298 mValues.put(Favorites._ID, id); 299 if (mCallback.insertAndCheck(mDb, mValues) < 0) { 300 return -1; 301 } else { 302 return id; 303 } 304 } 305 getFolderElementsMap()306 protected ArrayMap<String, TagParser> getFolderElementsMap() { 307 ArrayMap<String, TagParser> parsers = new ArrayMap<>(); 308 parsers.put(TAG_APP_ICON, new AppShortcutParser()); 309 parsers.put(TAG_AUTO_INSTALL, new AutoInstallParser()); 310 parsers.put(TAG_SHORTCUT, new ShortcutParser(mSourceRes)); 311 return parsers; 312 } 313 getLayoutElementsMap()314 protected ArrayMap<String, TagParser> getLayoutElementsMap() { 315 ArrayMap<String, TagParser> parsers = new ArrayMap<>(); 316 parsers.put(TAG_APP_ICON, new AppShortcutParser()); 317 parsers.put(TAG_AUTO_INSTALL, new AutoInstallParser()); 318 parsers.put(TAG_FOLDER, new FolderParser()); 319 parsers.put(TAG_APPWIDGET, new PendingWidgetParser()); 320 parsers.put(TAG_SHORTCUT, new ShortcutParser(mSourceRes)); 321 return parsers; 322 } 323 324 protected interface TagParser { 325 /** 326 * Parses the tag and adds to the db 327 * @return the id of the row added or -1; 328 */ parseAndAdd(XmlResourceParser parser)329 long parseAndAdd(XmlResourceParser parser) 330 throws XmlPullParserException, IOException; 331 } 332 333 /** 334 * App shortcuts: required attributes packageName and className 335 */ 336 protected class AppShortcutParser implements TagParser { 337 338 @Override parseAndAdd(XmlResourceParser parser)339 public long parseAndAdd(XmlResourceParser parser) { 340 final String packageName = getAttributeValue(parser, ATTR_PACKAGE_NAME); 341 final String className = getAttributeValue(parser, ATTR_CLASS_NAME); 342 343 if (!TextUtils.isEmpty(packageName) && !TextUtils.isEmpty(className)) { 344 ActivityInfo info; 345 try { 346 ComponentName cn; 347 try { 348 cn = new ComponentName(packageName, className); 349 info = mPackageManager.getActivityInfo(cn, 0); 350 } catch (PackageManager.NameNotFoundException nnfe) { 351 String[] packages = mPackageManager.currentToCanonicalPackageNames( 352 new String[] { packageName }); 353 cn = new ComponentName(packages[0], className); 354 info = mPackageManager.getActivityInfo(cn, 0); 355 } 356 final Intent intent = new Intent(Intent.ACTION_MAIN, null) 357 .addCategory(Intent.CATEGORY_LAUNCHER) 358 .setComponent(cn) 359 .setFlags(Intent.FLAG_ACTIVITY_NEW_TASK | 360 Intent.FLAG_ACTIVITY_RESET_TASK_IF_NEEDED); 361 362 return addShortcut(info.loadLabel(mPackageManager).toString(), 363 intent, Favorites.ITEM_TYPE_APPLICATION); 364 } catch (PackageManager.NameNotFoundException e) { 365 Log.e(TAG, "Favorite not found: " + packageName + "/" + className); 366 } 367 return -1; 368 } else { 369 return invalidPackageOrClass(parser); 370 } 371 } 372 373 /** 374 * Helper method to allow extending the parser capabilities 375 */ invalidPackageOrClass(XmlResourceParser parser)376 protected long invalidPackageOrClass(XmlResourceParser parser) { 377 Log.w(TAG, "Skipping invalid <favorite> with no component"); 378 return -1; 379 } 380 } 381 382 /** 383 * AutoInstall: required attributes packageName and className 384 */ 385 protected class AutoInstallParser implements TagParser { 386 387 @Override parseAndAdd(XmlResourceParser parser)388 public long parseAndAdd(XmlResourceParser parser) { 389 final String packageName = getAttributeValue(parser, ATTR_PACKAGE_NAME); 390 final String className = getAttributeValue(parser, ATTR_CLASS_NAME); 391 if (TextUtils.isEmpty(packageName) || TextUtils.isEmpty(className)) { 392 if (LOGD) Log.d(TAG, "Skipping invalid <favorite> with no component"); 393 return -1; 394 } 395 396 mValues.put(Favorites.RESTORED, ShortcutInfo.FLAG_AUTOINSTALL_ICON); 397 final Intent intent = new Intent(Intent.ACTION_MAIN, null) 398 .addCategory(Intent.CATEGORY_LAUNCHER) 399 .setComponent(new ComponentName(packageName, className)) 400 .setFlags(Intent.FLAG_ACTIVITY_NEW_TASK | 401 Intent.FLAG_ACTIVITY_RESET_TASK_IF_NEEDED); 402 return addShortcut(mContext.getString(R.string.package_state_unknown), intent, 403 Favorites.ITEM_TYPE_APPLICATION); 404 } 405 } 406 407 /** 408 * Parses a web shortcut. Required attributes url, icon, title 409 */ 410 protected class ShortcutParser implements TagParser { 411 412 private final Resources mIconRes; 413 ShortcutParser(Resources iconRes)414 public ShortcutParser(Resources iconRes) { 415 mIconRes = iconRes; 416 } 417 418 @Override parseAndAdd(XmlResourceParser parser)419 public long parseAndAdd(XmlResourceParser parser) { 420 final int titleResId = getAttributeResourceValue(parser, ATTR_TITLE, 0); 421 final int iconId = getAttributeResourceValue(parser, ATTR_ICON, 0); 422 423 if (titleResId == 0 || iconId == 0) { 424 if (LOGD) Log.d(TAG, "Ignoring shortcut"); 425 return -1; 426 } 427 428 final Intent intent = parseIntent(parser); 429 if (intent == null) { 430 return -1; 431 } 432 433 Drawable icon = mIconRes.getDrawable(iconId); 434 if (icon == null) { 435 if (LOGD) Log.d(TAG, "Ignoring shortcut, can't load icon"); 436 return -1; 437 } 438 439 // Auto installs should always support the current platform version. 440 LauncherIcons li = LauncherIcons.obtain(mContext); 441 mValues.put(LauncherSettings.Favorites.ICON, Utilities.flattenBitmap( 442 li.createBadgedIconBitmap(icon, Process.myUserHandle(), VERSION.SDK_INT).icon)); 443 li.recycle(); 444 445 mValues.put(Favorites.ICON_PACKAGE, mIconRes.getResourcePackageName(iconId)); 446 mValues.put(Favorites.ICON_RESOURCE, mIconRes.getResourceName(iconId)); 447 448 intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK | 449 Intent.FLAG_ACTIVITY_RESET_TASK_IF_NEEDED); 450 return addShortcut(mSourceRes.getString(titleResId), 451 intent, Favorites.ITEM_TYPE_SHORTCUT); 452 } 453 parseIntent(XmlResourceParser parser)454 protected Intent parseIntent(XmlResourceParser parser) { 455 final String url = getAttributeValue(parser, ATTR_URL); 456 if (TextUtils.isEmpty(url) || !Patterns.WEB_URL.matcher(url).matches()) { 457 if (LOGD) Log.d(TAG, "Ignoring shortcut, invalid url: " + url); 458 return null; 459 } 460 return new Intent(Intent.ACTION_VIEW, null).setData(Uri.parse(url)); 461 } 462 } 463 464 /** 465 * AppWidget parser: Required attributes packageName, className, spanX and spanY. 466 * Options child nodes: <extra key=... value=... /> 467 * It adds a pending widget which allows the widget to come later. If there are extras, those 468 * are passed to widget options during bind. 469 * The config activity for the widget (if present) is not shown, so any optional configurations 470 * should be passed as extras and the widget should support reading these widget options. 471 */ 472 protected class PendingWidgetParser implements TagParser { 473 474 @Override parseAndAdd(XmlResourceParser parser)475 public long parseAndAdd(XmlResourceParser parser) 476 throws XmlPullParserException, IOException { 477 final String packageName = getAttributeValue(parser, ATTR_PACKAGE_NAME); 478 final String className = getAttributeValue(parser, ATTR_CLASS_NAME); 479 if (TextUtils.isEmpty(packageName) || TextUtils.isEmpty(className)) { 480 if (LOGD) Log.d(TAG, "Skipping invalid <appwidget> with no component"); 481 return -1; 482 } 483 484 mValues.put(Favorites.SPANX, getAttributeValue(parser, ATTR_SPAN_X)); 485 mValues.put(Favorites.SPANY, getAttributeValue(parser, ATTR_SPAN_Y)); 486 mValues.put(Favorites.ITEM_TYPE, Favorites.ITEM_TYPE_APPWIDGET); 487 488 // Read the extras 489 Bundle extras = new Bundle(); 490 int widgetDepth = parser.getDepth(); 491 int type; 492 while ((type = parser.next()) != XmlPullParser.END_TAG || 493 parser.getDepth() > widgetDepth) { 494 if (type != XmlPullParser.START_TAG) { 495 continue; 496 } 497 498 if (TAG_EXTRA.equals(parser.getName())) { 499 String key = getAttributeValue(parser, ATTR_KEY); 500 String value = getAttributeValue(parser, ATTR_VALUE); 501 if (key != null && value != null) { 502 extras.putString(key, value); 503 } else { 504 throw new RuntimeException("Widget extras must have a key and value"); 505 } 506 } else { 507 throw new RuntimeException("Widgets can contain only extras"); 508 } 509 } 510 511 return verifyAndInsert(new ComponentName(packageName, className), extras); 512 } 513 verifyAndInsert(ComponentName cn, Bundle extras)514 protected long verifyAndInsert(ComponentName cn, Bundle extras) { 515 mValues.put(Favorites.APPWIDGET_PROVIDER, cn.flattenToString()); 516 mValues.put(Favorites.RESTORED, 517 LauncherAppWidgetInfo.FLAG_ID_NOT_VALID | 518 LauncherAppWidgetInfo.FLAG_PROVIDER_NOT_READY | 519 LauncherAppWidgetInfo.FLAG_DIRECT_CONFIG); 520 mValues.put(Favorites._ID, mCallback.generateNewItemId()); 521 if (!extras.isEmpty()) { 522 mValues.put(Favorites.INTENT, new Intent().putExtras(extras).toUri(0)); 523 } 524 525 long insertedId = mCallback.insertAndCheck(mDb, mValues); 526 if (insertedId < 0) { 527 return -1; 528 } else { 529 return insertedId; 530 } 531 } 532 } 533 534 protected class FolderParser implements TagParser { 535 private final ArrayMap<String, TagParser> mFolderElements; 536 FolderParser()537 public FolderParser() { 538 this(getFolderElementsMap()); 539 } 540 FolderParser(ArrayMap<String, TagParser> elements)541 public FolderParser(ArrayMap<String, TagParser> elements) { 542 mFolderElements = elements; 543 } 544 545 @Override parseAndAdd(XmlResourceParser parser)546 public long parseAndAdd(XmlResourceParser parser) 547 throws XmlPullParserException, IOException { 548 final String title; 549 final int titleResId = getAttributeResourceValue(parser, ATTR_TITLE, 0); 550 if (titleResId != 0) { 551 title = mSourceRes.getString(titleResId); 552 } else { 553 title = mContext.getResources().getString(R.string.folder_name); 554 } 555 556 mValues.put(Favorites.TITLE, title); 557 mValues.put(Favorites.ITEM_TYPE, Favorites.ITEM_TYPE_FOLDER); 558 mValues.put(Favorites.SPANX, 1); 559 mValues.put(Favorites.SPANY, 1); 560 mValues.put(Favorites._ID, mCallback.generateNewItemId()); 561 long folderId = mCallback.insertAndCheck(mDb, mValues); 562 if (folderId < 0) { 563 if (LOGD) Log.e(TAG, "Unable to add folder"); 564 return -1; 565 } 566 567 final ContentValues myValues = new ContentValues(mValues); 568 ArrayList<Long> folderItems = new ArrayList<>(); 569 570 int type; 571 int folderDepth = parser.getDepth(); 572 int rank = 0; 573 while ((type = parser.next()) != XmlPullParser.END_TAG || 574 parser.getDepth() > folderDepth) { 575 if (type != XmlPullParser.START_TAG) { 576 continue; 577 } 578 mValues.clear(); 579 mValues.put(Favorites.CONTAINER, folderId); 580 mValues.put(Favorites.RANK, rank); 581 582 TagParser tagParser = mFolderElements.get(parser.getName()); 583 if (tagParser != null) { 584 final long id = tagParser.parseAndAdd(parser); 585 if (id >= 0) { 586 folderItems.add(id); 587 rank++; 588 } 589 } else { 590 throw new RuntimeException("Invalid folder item " + parser.getName()); 591 } 592 } 593 594 long addedId = folderId; 595 596 // We can only have folders with >= 2 items, so we need to remove the 597 // folder and clean up if less than 2 items were included, or some 598 // failed to add, and less than 2 were actually added 599 if (folderItems.size() < 2) { 600 // Delete the folder 601 Uri uri = Favorites.getContentUri(folderId); 602 SqlArguments args = new SqlArguments(uri, null, null); 603 mDb.delete(args.table, args.where, args.args); 604 addedId = -1; 605 606 // If we have a single item, promote it to where the folder 607 // would have been. 608 if (folderItems.size() == 1) { 609 final ContentValues childValues = new ContentValues(); 610 copyInteger(myValues, childValues, Favorites.CONTAINER); 611 copyInteger(myValues, childValues, Favorites.SCREEN); 612 copyInteger(myValues, childValues, Favorites.CELLX); 613 copyInteger(myValues, childValues, Favorites.CELLY); 614 615 addedId = folderItems.get(0); 616 mDb.update(Favorites.TABLE_NAME, childValues, 617 Favorites._ID + "=" + addedId, null); 618 } 619 } 620 return addedId; 621 } 622 } 623 beginDocument(XmlPullParser parser, String firstElementName)624 protected static void beginDocument(XmlPullParser parser, String firstElementName) 625 throws XmlPullParserException, IOException { 626 int type; 627 while ((type = parser.next()) != XmlPullParser.START_TAG 628 && type != XmlPullParser.END_DOCUMENT); 629 630 if (type != XmlPullParser.START_TAG) { 631 throw new XmlPullParserException("No start tag found"); 632 } 633 634 if (!parser.getName().equals(firstElementName)) { 635 throw new XmlPullParserException("Unexpected start tag: found " + parser.getName() + 636 ", expected " + firstElementName); 637 } 638 } 639 convertToDistanceFromEnd(String value, int endValue)640 private static String convertToDistanceFromEnd(String value, int endValue) { 641 if (!TextUtils.isEmpty(value)) { 642 int x = Integer.parseInt(value); 643 if (x < 0) { 644 return Integer.toString(endValue + x); 645 } 646 } 647 return value; 648 } 649 650 /** 651 * Return attribute value, attempting launcher-specific namespace first 652 * before falling back to anonymous attribute. 653 */ getAttributeValue(XmlResourceParser parser, String attribute)654 protected static String getAttributeValue(XmlResourceParser parser, String attribute) { 655 String value = parser.getAttributeValue( 656 "http://schemas.android.com/apk/res-auto/com.android.launcher3", attribute); 657 if (value == null) { 658 value = parser.getAttributeValue(null, attribute); 659 } 660 return value; 661 } 662 663 /** 664 * Return attribute resource value, attempting launcher-specific namespace 665 * first before falling back to anonymous attribute. 666 */ getAttributeResourceValue(XmlResourceParser parser, String attribute, int defaultValue)667 protected static int getAttributeResourceValue(XmlResourceParser parser, String attribute, 668 int defaultValue) { 669 int value = parser.getAttributeResourceValue( 670 "http://schemas.android.com/apk/res-auto/com.android.launcher3", attribute, 671 defaultValue); 672 if (value == defaultValue) { 673 value = parser.getAttributeResourceValue(null, attribute, defaultValue); 674 } 675 return value; 676 } 677 678 public interface LayoutParserCallback { generateNewItemId()679 long generateNewItemId(); 680 insertAndCheck(SQLiteDatabase db, ContentValues values)681 long insertAndCheck(SQLiteDatabase db, ContentValues values); 682 } 683 copyInteger(ContentValues from, ContentValues to, String key)684 @Thunk static void copyInteger(ContentValues from, ContentValues to, String key) { 685 to.put(key, from.getAsInteger(key)); 686 } 687 } 688