content: /*
   * Copyright (C) 2015 The Android Open Source Project
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package com.android.launcher3.util;
  
  import android.content.Context;
  import android.content.SharedPreferences;
  import android.content.pm.LauncherActivityInfo;
  import android.os.Handler;
  import android.os.Process;
  import android.os.UserHandle;
  
  import com.android.launcher3.FolderInfo;
  import com.android.launcher3.InstallShortcutReceiver;
  import com.android.launcher3.ItemInfo;
  import com.android.launcher3.LauncherFiles;
  import com.android.launcher3.LauncherModel;
  import com.android.launcher3.MainThreadExecutor;
  import com.android.launcher3.R;
  import com.android.launcher3.SessionCommitReceiver;
  import com.android.launcher3.ShortcutInfo;
  import com.android.launcher3.Utilities;
  import com.android.launcher3.compat.UserManagerCompat;
  import com.android.launcher3.model.BgDataModel;
  import com.android.launcher3.model.ModelWriter;
  
  import java.util.ArrayList;
  import java.util.HashSet;
  import java.util.List;
  
  /**
   * Handles addition of app shortcuts for managed profiles.
   * Methods of class should only be called on {@link LauncherModel#sWorkerThread}.
   */
  public class ManagedProfileHeuristic {
  
      private static final String USER_FOLDER_ID_PREFIX = "user_folder_";
  
      /**
       * Duration (in milliseconds) for which app shortcuts will be added to work folder.
       */
      private static final long AUTO_ADD_TO_FOLDER_DURATION = 8 * 60 * 60 * 1000;
  
      public static void onAllAppsLoaded(final Context context,
              List<LauncherActivityInfo> apps, UserHandle user) {
          if (Process.myUserHandle().equals(user)) {
              return;
          }
  
          UserFolderInfo ufi = new UserFolderInfo(context, user, null);
          // We only handle folder creation once. Later icon additions are handled using package
          // or session events.
          if (ufi.folderAlreadyCreated) {
              return;
          }
  
          if (Utilities.ATLEAST_OREO && !SessionCommitReceiver.isEnabled(context)) {
              // Just mark the folder id preference to avoid new folder creation later.
              ufi.prefs.edit().putLong(ufi.folderIdKey, ItemInfo.NO_ID).apply();
              return;
          }
  
          InstallShortcutReceiver.enableInstallQueue(InstallShortcutReceiver.FLAG_BULK_ADD);
          for (LauncherActivityInfo app : apps) {
              // Queue all items which should go in the work folder.
              if (app.getFirstInstallTime() < ufi.addIconToFolderTime) {
                  InstallShortcutReceiver.queueActivityInfo(app, context);
              }
          }
          // Post the queue update on next frame, so that the loader gets finished.
          new Handler(LauncherModel.getWorkerLooper()).post(new Runnable() {
              @Override
              public void run() {
                  InstallShortcutReceiver.disableAndFlushInstallQueue(
                          InstallShortcutReceiver.FLAG_BULK_ADD, context);
              }
          });
      }
  
  
      /**
       * Utility class to help workspace icon addition.
       */
      public static class UserFolderInfo {
  
          final ArrayList<ShortcutInfo> pendingShortcuts = new ArrayList<>();
  
          final UserHandle user;
  
          final long userSerial;
          // Time until which icons will be added to folder instead.
          final long addIconToFolderTime;
  
          final String folderIdKey;
          final SharedPreferences prefs;
  
          final boolean folderAlreadyCreated;
          final FolderInfo folderInfo;
  
          boolean folderPendingAddition;
  
          public UserFolderInfo(Context context, UserHandle user, BgDataModel dataModel) {
              this.user = user;
  
              UserManagerCompat um = UserManagerCompat.getInstance(context);
              userSerial = um.getSerialNumberForUser(user);
              addIconToFolderTime = um.getUserCreationTime(user) + AUTO_ADD_TO_FOLDER_DURATION;
  
              folderIdKey = USER_FOLDER_ID_PREFIX + userSerial;
              prefs = prefs(context);
  
              folderAlreadyCreated = prefs.contains(folderIdKey);
              if (dataModel != null) {
                  if (folderAlreadyCreated) {
                      long folderId = prefs.getLong(folderIdKey, ItemInfo.NO_ID);
                      folderInfo = dataModel.folders.get(folderId);
                  } else {
                      folderInfo = new FolderInfo();
                      folderInfo.title = context.getText(R.string.work_folder_name);
                      folderInfo.setOption(FolderInfo.FLAG_WORK_FOLDER, true, null);
                      folderPendingAddition = true;
                  }
              } else {
                  folderInfo = null;
              }
          }
  
          /**
           * Returns the ItemInfo which should be added to the workspace. In case the the provided
           * {@link ShortcutInfo} or a wrapped {@link FolderInfo} or null.
           */
          public ItemInfo convertToWorkspaceItem(
                  ShortcutInfo shortcut, LauncherActivityInfo activityInfo) {
              if (activityInfo.getFirstInstallTime() >= addIconToFolderTime) {
                  return shortcut;
              }
  
              if (folderAlreadyCreated) {
                  if (folderInfo == null) {
                      // Work folder was deleted by user, add icon to home screen.
                      return shortcut;
                  } else {
                      // Add item to work folder instead. Nothing needs to be added
                      // on the homescreen.
                      pendingShortcuts.add(shortcut);
                      return null;
                  }
              }
  
              pendingShortcuts.add(shortcut);
              folderInfo.add(shortcut, false);
              if (folderPendingAddition) {
                  folderPendingAddition = false;
                  return folderInfo;
              } else {
                  // WorkFolder already requested to be added. Nothing new needs to be added.
                  return null;
              }
          }
  
          public void applyPendingState(ModelWriter writer) {
              if (folderInfo == null) {
                  return;
              }
  
              int startingRank = 0;
              if (folderAlreadyCreated) {
                  startingRank = folderInfo.contents.size();
              }
  
              for (ShortcutInfo info : pendingShortcuts) {
                  info.rank = startingRank++;
                  writer.addItemToDatabase(info, folderInfo.id, 0, 0, 0);
              }
  
              if (folderAlreadyCreated) {
                  // FolderInfo could already be bound. We need to add shortcuts on the UI thread.
                  new MainThreadExecutor().execute(new Runnable() {
  
                      @Override
                      public void run() {
                          folderInfo.prepareAutoUpdate();
                          for (ShortcutInfo info : pendingShortcuts) {
                              folderInfo.add(info, false);
                          }
                      }
                  });
              } else {
                  prefs.edit().putLong(folderIdKey, folderInfo.id).apply();
              }
          }
      }
  
      /**
       * Verifies that entries corresponding to {@param users} exist and removes all invalid entries.
       */
      public static void processAllUsers(List<UserHandle> users, Context context) {
          UserManagerCompat userManager = UserManagerCompat.getInstance(context);
          HashSet<String> validKeys = new HashSet<>();
          for (UserHandle user : users) {
              validKeys.add(USER_FOLDER_ID_PREFIX + userManager.getSerialNumberForUser(user));
          }
  
          SharedPreferences prefs = prefs(context);
          SharedPreferences.Editor editor = prefs.edit();
          for (String key : prefs.getAll().keySet()) {
              if (!validKeys.contains(key)) {
                  editor.remove(key);
              }
          }
          editor.apply();
      }
  
      /**
       * For each user, if a work folder has not been created, mark it such that the folder will
       * never get created.
       */
      public static void markExistingUsersForNoFolderCreation(Context context) {
          UserManagerCompat userManager = UserManagerCompat.getInstance(context);
          UserHandle myUser = Process.myUserHandle();
  
          SharedPreferences prefs = null;
          for (UserHandle user : userManager.getUserProfiles()) {
              if (myUser.equals(user)) {
                  continue;
              }
              if (prefs == null) {
                  prefs = prefs(context);
              }
              String folderIdKey = USER_FOLDER_ID_PREFIX + userManager.getSerialNumberForUser(user);
              if (!prefs.contains(folderIdKey)) {
                  prefs.edit().putLong(folderIdKey, ItemInfo.NO_ID).apply();
              }
          }
      }
  
      public static SharedPreferences prefs(Context context) {
          return context.getSharedPreferences(
                  LauncherFiles.MANAGED_USER_PREFERENCES_KEY, Context.MODE_PRIVATE);
      }
  }
  