/* 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.search import android.Manifest import android.content.Context import android.content.DialogInterface import android.graphics.Typeface.BOLD import android.graphics.Typeface.ITALIC import android.os.Bundle import android.text.style.StyleSpan import android.view.LayoutInflater import android.view.View import android.view.ViewGroup import androidx.appcompat.app.AlertDialog import androidx.appcompat.app.AppCompatActivity import androidx.coordinatorlayout.widget.CoordinatorLayout import androidx.fragment.app.Fragment import androidx.navigation.Navigation import kotlinx.android.synthetic.main.fragment_search.* import kotlinx.android.synthetic.main.fragment_search.view.* import mozilla.components.browser.search.SearchEngine import mozilla.components.feature.qr.QrFeature import mozilla.components.support.base.feature.BackHandler import mozilla.components.support.base.feature.ViewBoundFeatureWrapper import mozilla.components.support.ktx.android.content.hasCamera import mozilla.components.support.ktx.android.content.isPermissionGranted import mozilla.components.support.ktx.kotlin.isUrl import org.mozilla.fenix.BrowserDirection import org.mozilla.fenix.FenixViewModelProvider import org.mozilla.fenix.HomeActivity import org.mozilla.fenix.R import org.mozilla.fenix.components.metrics.Event import org.mozilla.fenix.components.toolbar.SearchAction import org.mozilla.fenix.components.toolbar.SearchChange import org.mozilla.fenix.components.toolbar.SearchState import org.mozilla.fenix.components.toolbar.ToolbarComponent import org.mozilla.fenix.components.toolbar.ToolbarUIView import org.mozilla.fenix.components.toolbar.ToolbarViewModel import org.mozilla.fenix.ext.getSpannable import org.mozilla.fenix.ext.requireComponents import org.mozilla.fenix.mvi.ActionBusFactory import org.mozilla.fenix.mvi.getAutoDisposeObservable import org.mozilla.fenix.mvi.getManagedEmitter import org.mozilla.fenix.search.awesomebar.AwesomeBarAction import org.mozilla.fenix.search.awesomebar.AwesomeBarChange import org.mozilla.fenix.search.awesomebar.AwesomeBarComponent import org.mozilla.fenix.search.awesomebar.AwesomeBarState import org.mozilla.fenix.search.awesomebar.AwesomeBarUIView import org.mozilla.fenix.search.awesomebar.AwesomeBarViewModel @Suppress("TooManyFunctions") class SearchFragment : Fragment(), BackHandler { private lateinit var toolbarComponent: ToolbarComponent private lateinit var awesomeBarComponent: AwesomeBarComponent private var sessionId: String? = null private var isPrivate = false private val qrFeature = ViewBoundFeatureWrapper() private var permissionDidUpdate = false override fun onCreate(savedInstanceState: Bundle?) { super.onCreate(savedInstanceState) // Disabled while awaiting a better solution to #3209 // postponeEnterTransition() // sharedElementEnterTransition = // TransitionInflater.from(context).inflateTransition(android.R.transition.move).setDuration( // SHARED_TRANSITION_MS // ) requireComponents.analytics.metrics.track(Event.InteractWithSearchURLArea) } override fun onCreateView( inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle? ): View? { sessionId = SearchFragmentArgs.fromBundle(arguments!!).sessionId isPrivate = (activity as HomeActivity).browsingModeManager.isPrivate val session = sessionId?.let { requireComponents.core.sessionManager.findSessionById(it) } val view = inflater.inflate(R.layout.fragment_search, container, false) val url = session?.url ?: "" toolbarComponent = ToolbarComponent( view.toolbar_component_wrapper, ActionBusFactory.get(this), sessionId, isPrivate, true, view.search_engine_icon, FenixViewModelProvider.create( this, ToolbarViewModel::class.java ) { ToolbarViewModel(SearchState(url, session?.searchTerms ?: "", isEditing = true)) } ).also { // Remove background from toolbar view since it conflicts with the search UI. it.uiView.view.background = null it.uiView.view.layoutParams.height = CoordinatorLayout.LayoutParams.MATCH_PARENT } awesomeBarComponent = AwesomeBarComponent( view.search_layout, ActionBusFactory.get(this), FenixViewModelProvider.create( this, AwesomeBarViewModel::class.java ) { AwesomeBarViewModel(AwesomeBarState("", false)) } ) ActionBusFactory.get(this).logMergedObservables() return view } override fun onViewCreated(view: View, savedInstanceState: Bundle?) { super.onViewCreated(view, savedInstanceState) search_scan_button.visibility = if (context?.hasCamera() == true) View.VISIBLE else View.GONE layoutComponents(view.search_layout) qrFeature.set( QrFeature( requireContext(), fragmentManager = requireFragmentManager(), onNeedToRequestPermissions = { permissions -> requestPermissions(permissions, REQUEST_CODE_CAMERA_PERMISSIONS) }, onScanResult = { result -> search_scan_button.isChecked = false activity?.let { AlertDialog.Builder(it).apply { val spannable = resources.getSpannable( R.string.qr_scanner_confirmation_dialog_message, listOf( getString(R.string.app_name) to listOf(StyleSpan(BOLD)), result to listOf(StyleSpan(ITALIC)) ) ) setMessage(spannable) setNegativeButton(R.string.qr_scanner_dialog_negative) { dialog: DialogInterface, _ -> requireComponents.analytics.metrics.track(Event.QRScannerNavigationDenied) dialog.cancel() } setPositiveButton(R.string.qr_scanner_dialog_positive) { dialog: DialogInterface, _ -> requireComponents.analytics.metrics.track(Event.QRScannerNavigationAllowed) (activity as HomeActivity) .openToBrowserAndLoad( searchTermOrURL = result, newTab = sessionId == null, from = BrowserDirection.FromSearch ) dialog.dismiss() } create() }.show() requireComponents.analytics.metrics.track(Event.QRScannerPromptDisplayed) } }), owner = this, view = view ) view.search_scan_button.setOnClickListener { getManagedEmitter().onNext(SearchChange.ToolbarClearedFocus) requireComponents.analytics.metrics.track(Event.QRScannerOpened) qrFeature.get()?.scan(R.id.container) } lifecycle.addObserver((toolbarComponent.uiView as ToolbarUIView).toolbarIntegration) view.toolbar_wrapper.clipToOutline = false search_shortcuts_button.setOnClickListener { val isOpen = (awesomeBarComponent.uiView as AwesomeBarUIView).state?.showShortcutEnginePicker ?: false getManagedEmitter().onNext(AwesomeBarChange.SearchShortcutEnginePicker(!isOpen)) if (isOpen) { requireComponents.analytics.metrics.track(Event.SearchShortcutMenuClosed) } else { requireComponents.analytics.metrics.track(Event.SearchShortcutMenuOpened) } } startPostponedEnterTransition() } override fun onResume() { super.onResume() subscribeToSearchActions() subscribeToAwesomeBarActions() if (!permissionDidUpdate) { getManagedEmitter().onNext(SearchChange.ToolbarRequestedFocus) } permissionDidUpdate = false (activity as AppCompatActivity).supportActionBar?.hide() } override fun onPause() { super.onPause() getManagedEmitter().onNext(SearchChange.ToolbarClearedFocus) } override fun onBackPressed(): Boolean { return when { qrFeature.onBackPressed() -> { view?.search_scan_button?.isChecked = false getManagedEmitter().onNext(SearchChange.ToolbarRequestedFocus) true } else -> false } } private fun subscribeToSearchActions() { getAutoDisposeObservable() .subscribe { when (it) { is SearchAction.UrlCommitted -> { if (it.url.isNotBlank()) { (activity as HomeActivity).openToBrowserAndLoad( searchTermOrURL = it.url, newTab = sessionId == null, from = BrowserDirection.FromSearch, engine = it.engine ) val event = if (it.url.isUrl()) { Event.EnteredUrl(false) } else { val engine = it.engine ?: requireComponents .search.searchEngineManager.getDefaultSearchEngine(requireContext()) createSearchEvent(engine, false) } requireComponents.analytics.metrics.track(event) } } is SearchAction.TextChanged -> { getManagedEmitter().onNext(SearchChange.QueryTextChanged(it.query)) getManagedEmitter().onNext(AwesomeBarChange.UpdateQuery(it.query)) } is SearchAction.EditingCanceled -> { Navigation.findNavController(toolbar_wrapper).navigateUp() } } } } private fun subscribeToAwesomeBarActions() { getAutoDisposeObservable() .subscribe { when (it) { is AwesomeBarAction.URLTapped -> { (activity as HomeActivity).openToBrowserAndLoad( searchTermOrURL = it.url, newTab = sessionId == null, from = BrowserDirection.FromSearch ) requireComponents.analytics.metrics.track(Event.EnteredUrl(false)) } is AwesomeBarAction.SearchTermsTapped -> { (activity as HomeActivity).openToBrowserAndLoad( searchTermOrURL = it.searchTerms, newTab = sessionId == null, from = BrowserDirection.FromSearch, engine = it.engine, forceSearch = true ) val engine = it.engine ?: requireComponents .search.searchEngineManager.getDefaultSearchEngine(requireContext()) val event = createSearchEvent(engine, true) requireComponents.analytics.metrics.track(event) } is AwesomeBarAction.SearchShortcutEngineSelected -> { getManagedEmitter() .onNext(AwesomeBarChange.SearchShortcutEngineSelected(it.engine)) getManagedEmitter() .onNext(SearchChange.SearchShortcutEngineSelected(it.engine)) requireComponents.analytics.metrics.track(Event.SearchShortcutSelected(it.engine.name)) } } } } private fun createSearchEvent(engine: SearchEngine, isSuggestion: Boolean): Event.PerformedSearch { val isShortcut = engine != requireComponents.search.searchEngineManager.defaultSearchEngine val engineSource = if (isShortcut) Event.PerformedSearch.EngineSource.Shortcut(engine) else Event.PerformedSearch.EngineSource.Default(engine) val source = if (isSuggestion) Event.PerformedSearch.EventSource.Suggestion(engineSource) else Event.PerformedSearch.EventSource.Action(engineSource) return Event.PerformedSearch(source) } override fun onRequestPermissionsResult(requestCode: Int, permissions: Array, grantResults: IntArray) { when (requestCode) { REQUEST_CODE_CAMERA_PERMISSIONS -> qrFeature.withFeature { it.onPermissionsResult(permissions, grantResults) context?.let { context: Context -> if (context.isPermissionGranted(Manifest.permission.CAMERA)) { permissionDidUpdate = true } else { view?.search_scan_button?.isChecked = false } } } else -> super.onRequestPermissionsResult(requestCode, permissions, grantResults) } } companion object { private const val SHARED_TRANSITION_MS = 150L private const val REQUEST_CODE_CAMERA_PERMISSIONS = 1 } }