Fix bug where work profile widgets are restored with main profile provider.

import android.util.Log;

import com.android.launcher3.LauncherSettings.Favorites;

import com.android.launcher3.compat.UserManagerCompat;

import com.android.launcher3.config.FeatureFlags;

import com.android.launcher3.model.LoaderTask;

import com.android.launcher3.provider.RestoreDbTask;

import androidx.annotation.WorkerThread;

import static android.os.Process.myUserHandle;

public class AppWidgetsRestoredReceiver extends BroadcastReceiver {

    private static final String TAG = "AWRestoredReceiver";

                state = LauncherAppWidgetInfo.FLAG_PROVIDER_NOT_READY;

            }

            String[] widgetIdParams = new String[] { Integer.toString(oldWidgetIds[i]) };

            // b/135926478: Work profile widget restore is broken in platform. This forces us to

            // recreate the widget during loading with the correct host provider.

            long mainProfileId = UserManagerCompat.getInstance(context)

                    .getSerialNumberForUser(myUserHandle());

            String oldWidgetId = Integer.toString(oldWidgetIds[i]);

            int result = new ContentWriter(context, new ContentWriter.CommitParams(

                    "appWidgetId=? and (restored & 1) = 1", widgetIdParams))

                    "appWidgetId=? and (restored & 1) = 1 and profileId=?",

                    new String[] { oldWidgetId, Long.toString(mainProfileId) }))

                    .put(LauncherSettings.Favorites.APPWIDGET_ID, newWidgetIds[i])

                    .put(LauncherSettings.Favorites.RESTORED, state)

                    .commit();

            if (result == 0) {

                Cursor cursor = cr.query(Favorites.CONTENT_URI,

                        new String[] {Favorites.APPWIDGET_ID},

                        "appWidgetId=?", widgetIdParams, null);

                        "appWidgetId=?", new String[] { oldWidgetId }, null);

                try {

                    if (!cursor.moveToFirst()) {

                        // The widget no long exists.