From 7ba519d82dbab72a328c2bed53cd0347ecca96bc Mon Sep 17 00:00:00 2001
From: tibbi <tibor@kaputa.sk>
Date: Tue, 27 Sep 2022 15:38:56 +0200
Subject: [PATCH] start listening for events with widget host view on start

---
 .../launcher/activities/MainActivity.kt              | 10 ++++++++++
 .../launcher/views/HomeScreenGrid.kt                 | 12 ++++++++----
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/app/src/main/kotlin/com/simplemobiletools/launcher/activities/MainActivity.kt b/app/src/main/kotlin/com/simplemobiletools/launcher/activities/MainActivity.kt
index f484b88..f2f7efa 100644
--- a/app/src/main/kotlin/com/simplemobiletools/launcher/activities/MainActivity.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/launcher/activities/MainActivity.kt
@@ -91,6 +91,16 @@ class MainActivity : SimpleActivity(), FlingListener {
         }
     }
 
+    override fun onStart() {
+        super.onStart()
+        home_screen_grid.appWidgetHost.startListening()
+    }
+
+    override fun onStop() {
+        super.onStop()
+        home_screen_grid.appWidgetHost.stopListening()
+    }
+
     override fun onBackPressed() {
         if (isAllAppsFragmentExpanded()) {
             hideFragment(all_apps_fragment)
diff --git a/app/src/main/kotlin/com/simplemobiletools/launcher/views/HomeScreenGrid.kt b/app/src/main/kotlin/com/simplemobiletools/launcher/views/HomeScreenGrid.kt
index d9708f0..d4500de 100644
--- a/app/src/main/kotlin/com/simplemobiletools/launcher/views/HomeScreenGrid.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/launcher/views/HomeScreenGrid.kt
@@ -42,11 +42,15 @@ class HomeScreenGrid(context: Context, attrs: AttributeSet, defStyle: Int) : Rel
     private var rowHeight = 0
     private var iconSize = 0
 
+    // apply fake margins at the home screen. Real ones would cause the icons be cut at dragging at screen sides
+    private var sideMargins = Rect()
+
     private var gridItems = ArrayList<HomeScreenGridItem>()
     private var gridCenters = ArrayList<Pair<Int, Int>>()
     private var draggedItemCurrentCoords = Pair(-1, -1)
 
-    private var sideMargins = Rect()   // apply fake margins at the home screen. Real ones would cause the icons be cut at dragging at screen sides
+    val appWidgetHost = MyAppWidgetHost(context, WIDGET_HOST_ID)
+    private val appWidgetManager = AppWidgetManager.getInstance(context)
 
     init {
         textPaint = TextPaint(Paint.ANTI_ALIAS_FLAG).apply {
@@ -259,18 +263,18 @@ class HomeScreenGrid(context: Context, attrs: AttributeSet, defStyle: Int) : Rel
     }
 
     private fun bindWidget(item: HomeScreenGridItem, isInitialDrawAfterLaunch: Boolean) {
-        val infoList = AppWidgetManager.getInstance(context).installedProviders
+        val infoList = appWidgetManager.installedProviders
         val appWidgetProviderInfo = infoList.firstOrNull { it.provider.shortClassName == item.shortClassName }
         if (appWidgetProviderInfo != null) {
-            val appWidgetHost = MyAppWidgetHost(context, WIDGET_HOST_ID)
             val appWidgetId = appWidgetHost.allocateAppWidgetId()
-            val appWidgetManager = AppWidgetManager.getInstance(context)
             val canCreateWidget = appWidgetManager.bindAppWidgetIdIfAllowed(appWidgetId, appWidgetProviderInfo.provider)
             if (canCreateWidget) {
                 if (appWidgetProviderInfo.configure != null && !isInitialDrawAfterLaunch) {
                     appWidgetHost.startAppWidgetConfigureActivityForResult(context as MainActivity, appWidgetId, 0, REQUEST_CONFIGURE_WIDGET, null)
                 } else {
                     val widgetView = appWidgetHost.createView(context, appWidgetId, appWidgetProviderInfo) as MyAppWidgetHostView
+                    widgetView.setAppWidget(appWidgetId, appWidgetProviderInfo)
+
                     widgetView.x = item.left * rowWidth + sideMargins.left.toFloat()
                     widgetView.y = item.top * rowHeight + sideMargins.top.toFloat()
                     val widgetWidth = item.widthCells * rowWidth