1 files changed, 86 insertions, 0 deletions
diff --git a/mobile/android/fenix/app/src/main/java/org/mozilla/fenix/home/toolbar/SearchSelectorBinding.kt b/mobile/android/fenix/app/src/main/java/org/mozilla/fenix/home/toolbar/SearchSelectorBinding.kt
new file mode 100644
index 0000000000..6a6b92684c
--- /dev/null
+++ b/mobile/android/fenix/app/src/main/java/org/mozilla/fenix/home/toolbar/SearchSelectorBinding.kt
@@ -0,0 +1,86 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+package org.mozilla.fenix.home.toolbar
+
+import android.content.Context
+import android.graphics.drawable.BitmapDrawable
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.distinctUntilChanged
+import kotlinx.coroutines.flow.map
+import mozilla.components.browser.state.search.SearchEngine
+import mozilla.components.browser.state.state.BrowserState
+import mozilla.components.browser.state.state.selectedOrDefaultSearchEngine
+import mozilla.components.browser.state.store.BrowserStore
+import mozilla.components.concept.menu.Orientation
+import mozilla.components.lib.state.helpers.AbstractBinding
+import mozilla.components.service.glean.private.NoExtras
+import mozilla.components.support.ktx.android.content.getColorFromAttr
+import org.mozilla.fenix.GleanMetrics.UnifiedSearch
+import org.mozilla.fenix.R
+import org.mozilla.fenix.databinding.FragmentHomeBinding
+import org.mozilla.fenix.ext.increaseTapAreaVertically
+import org.mozilla.fenix.ext.settings
+import org.mozilla.fenix.search.toolbar.SearchSelectorMenu
+
+/**
+ * A binding that shows the search engine in the search selector button.
+ */
+class SearchSelectorBinding(
+    private val context: Context,
+    private val binding: FragmentHomeBinding,
+    private val searchSelectorMenu: SearchSelectorMenu,
+    browserStore: BrowserStore,
+) : AbstractBinding<BrowserState>(browserStore) {
+
+    override fun start() {
+        super.start()
+
+        binding.searchSelectorButton.apply {
+            setOnClickListener {
+                val orientation = if (context.settings().shouldUseBottomToolbar) {
+                    Orientation.UP
+                } else {
+                    Orientation.DOWN
+                }
+
+                UnifiedSearch.searchMenuTapped.record(NoExtras())
+
+                searchSelectorMenu.menuController.show(
+                    anchor = it.findViewById(R.id.search_selector),
+                    orientation = orientation,
+                )
+            }
+
+            increaseTapAreaVertically(SEARCH_SELECTOR_INCREASE_HEIGHT_DPS)
+        }
+    }
+
+    override suspend fun onState(flow: Flow<BrowserState>) {
+        flow.map { state -> state.search.selectedOrDefaultSearchEngine }
+            .distinctUntilChanged()
+            .collect { searchEngine ->
+                val name = searchEngine?.name
+                val icon = searchEngine?.let {
+                    val iconSize =
+                        context.resources.getDimensionPixelSize(R.dimen.preference_icon_drawable_size)
+                    BitmapDrawable(context.resources, searchEngine.icon).apply {
+                        setBounds(0, 0, iconSize, iconSize)
+                        // Setting tint manually for icons that were converted from Drawable
+                        // to Bitmap. Search Engine icons are stored as Bitmaps, hence
+                        // theming/attribute mechanism won't work.
+                        if (searchEngine.type == SearchEngine.Type.APPLICATION) {
+                            setTint(context.getColorFromAttr(R.attr.textPrimary))
+                        }
+                    }
+                }
+
+                binding.searchSelectorButton.setIcon(icon, name)
+            }
+    }
+
+    companion object {
+        const val SEARCH_SELECTOR_INCREASE_HEIGHT_DPS = 10
+    }
+}