From 472e9bc8c6c3a033fec8ba35d47dbfff9eb867ab Mon Sep 17 00:00:00 2001
From: Suphon Thanakornpakapong <paphonb@gmail.com>
Date: Tue, 12 Oct 2021 22:11:21 +0700
Subject: [PATCH] Fallback to apps search if there's no search activity

---
 lawnchair/src/app/lawnchair/qsb/QsbLayout.kt | 28 +++++++++++++++-----
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/lawnchair/src/app/lawnchair/qsb/QsbLayout.kt b/lawnchair/src/app/lawnchair/qsb/QsbLayout.kt
index 87b248b8f9..107604d88c 100644
--- a/lawnchair/src/app/lawnchair/qsb/QsbLayout.kt
+++ b/lawnchair/src/app/lawnchair/qsb/QsbLayout.kt
@@ -9,9 +9,12 @@ import android.widget.ImageView
 import androidx.core.view.ViewCompat
 import androidx.core.view.children
 import androidx.core.view.isVisible
+import app.lawnchair.launcher
 import com.android.launcher3.BaseActivity
 import com.android.launcher3.DeviceProfile
+import com.android.launcher3.LauncherState
 import com.android.launcher3.R
+import com.android.launcher3.anim.AnimatorListeners.forSuccessCallback
 import com.android.launcher3.qsb.QsbContainerView
 import com.android.launcher3.util.Themes
 import com.android.launcher3.views.ActivityContext
@@ -26,15 +29,9 @@ class QsbLayout(context: Context, attrs: AttributeSet?) : FrameLayout(context, a
         super.onFinishInflate()
         assistantIcon = ViewCompat.requireViewById(this, R.id.mic_icon)
         lensIcon = ViewCompat.requireViewById(this, R.id.lens_icon)
+        setupMainSearch()
 
         val searchPackage = QsbContainerView.getSearchWidgetPackageName(context)
-        setOnClickListener {
-            val intent = Intent("android.search.action.GLOBAL_SEARCH")
-                .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_CLEAR_TASK)
-                .setPackage(searchPackage)
-            context.startActivity(intent)
-        }
-
         val isGoogle = searchPackage == GOOGLE_PACKAGE
         assistantIcon.setIcon(isGoogle)
         if (isGoogle) {
@@ -65,6 +62,23 @@ class QsbLayout(context: Context, attrs: AttributeSet?) : FrameLayout(context, a
         }
     }
 
+    private fun setupMainSearch() {
+        setOnClickListener {
+            val searchPackage = QsbContainerView.getSearchWidgetPackageName(context)
+            val intent = Intent("android.search.action.GLOBAL_SEARCH")
+                .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_CLEAR_TASK)
+                .setPackage(searchPackage)
+            if (context.packageManager.resolveActivity(intent, 0) != null) {
+                context.startActivity(intent)
+            } else {
+                val launcher = context.launcher
+                launcher.stateManager.goToState(LauncherState.ALL_APPS, true, forSuccessCallback {
+                    launcher.appsView.searchUiManager.editText?.showKeyboard()
+                })
+            }
+        }
+    }
+
     private fun setupLensIcon() {
         val lensIntent = Intent.makeMainActivity(ComponentName(LENS_PACKAGE, LENS_ACTIVITY))
             .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_RESET_TASK_IF_NEEDED)