src/com/android/launcher3/util/ManagedProfileHeuristic.java - LeafOS-Project/android_packages_apps_Launcher3 - Gitiles

 /*
  * 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);
     }
 }
	/*
	* 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);
	}
	}