aboutsummaryrefslogtreecommitdiff
path: root/app/src/main/kotlin/com/pitchedapps/frost/web/FrostWebViewSearch.kt
blob: bcadf32af301556c9b635fea0a0ee7d09469c677 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
package com.pitchedapps.frost.web

import android.annotation.SuppressLint
import android.content.Context
import android.view.View
import android.webkit.*
import ca.allanwang.kau.searchview.SearchItem
import ca.allanwang.kau.utils.gone
import com.pitchedapps.frost.facebook.FbTab
import com.pitchedapps.frost.facebook.USER_AGENT_BASIC
import com.pitchedapps.frost.injectors.JsAssets
import com.pitchedapps.frost.injectors.JsBuilder
import com.pitchedapps.frost.injectors.jsInject
import com.pitchedapps.frost.utils.L
import com.pitchedapps.frost.utils.Prefs
import io.reactivex.schedulers.Schedulers
import io.reactivex.subjects.PublishSubject
import org.jetbrains.anko.runOnUiThread
import org.jsoup.Jsoup
import java.util.concurrent.TimeUnit

@SuppressLint("ViewConstructor")
/**
 * Created by Allan Wang on 2017-06-25.
 *
 * A bare bone search view meant solely to extract data from the web
 * This should be hidden
 * Having a single webview allows us to avoid loading the whole page with each query
 */
class FrostWebViewSearch(context: Context, val contract: SearchContract) : WebView(context) {

    val searchSubject = PublishSubject.create<String>()

    init {
        gone()
        setupWebview()
    }

    /**
     * Basic info of last search results, so we can check if the list has actually changed
     * Contains the last item's href (search more) as well as the number of items found
     * This holder is synchronized
     */
    var previousResult: Pair<String?, Int> = Pair(null, 0)

    fun saveResultFrame(result: List<Pair<List<String>, String>>) {
        synchronized(previousResult) {
            previousResult = Pair(result.lastOrNull()?.second, result.size)
        }
    }

    @SuppressLint("SetJavaScriptEnabled")
    fun setupWebview() {
        settings.javaScriptEnabled = true
        settings.userAgentString = USER_AGENT_BASIC
        setLayerType(View.LAYER_TYPE_HARDWARE, null)
        webViewClient = SearchWebViewClient()
        webChromeClient = SearchChromeClient()
        addJavascriptInterface(SearchJSI(), "Frost")
        searchSubject.debounce(300, TimeUnit.MILLISECONDS).subscribeOn(Schedulers.newThread())
                .map {
                    Jsoup.parse(it).select("a:not([rel*='keywords(']):not([href=#])[rel]").map {
                        element ->
                        //split text into separate items
                        L.v("Search element ${element.attr("href")}")
                        val texts = element.select("div").map { (it.text()) }.filter { it.isNotBlank() }
                        val pair = Pair(texts, element.attr("href"))
                        L.v("Search element potential $pair")
                        pair
                    }.filter { it.first.isNotEmpty() }
                }
                .filter { content -> Pair(content.lastOrNull()?.second, content.size) != previousResult }
                .subscribe {
                    content: List<Pair<List<String>, String>> ->
                    saveResultFrame(content)
                    L.d("Search element count ${content.size}")
                    contract.emitSearchResponse(content.map {
                        (texts, href) ->
                        SearchItem(href, texts[0], texts.getOrNull(1))
                    })
                }
        reload()
    }

    /**
     * Toggles web activity
     * Should be done in conjunction with showing/hiding the search view
     */
    var pauseLoad: Boolean
        get() = settings.blockNetworkLoads
        set(value) {
            context.runOnUiThread { settings.blockNetworkLoads = value }
        }

    override fun reload() {
        super.loadUrl(FbTab.SEARCH.url)
    }

    /**
     * Sets the input to have our given text, then dispatches the input event so the webpage recognizes it
     */
    fun query(input: String) {
        pauseLoad = false
        L.d("Searching attempt", input)
        JsBuilder().js("var e=document.getElementById('main-search-input');if(e){e.value='$input';var n=new Event('input',{bubbles:!0,cancelable:!0});e.dispatchEvent(n),e.dispatchEvent(new Event('focus'))}else console.log('Input field not found');").build().inject(this)
    }

    /**
     * Created by Allan Wang on 2017-05-31.
     *
     * Barebones client that does what [FrostWebViewSearch] needs
     */
    inner class SearchWebViewClient : BaseWebViewClient() {

        override fun onPageFinished(view: WebView, url: String) {
            super.onPageFinished(view, url)
            L.i("Search Page finished $url")
            view.jsInject(JsAssets.SEARCH)
        }

        override fun shouldInterceptRequest(view: WebView, request: WebResourceRequest): WebResourceResponse?
                = super.shouldInterceptRequest(view, request).filterCss(request)
    }

    class SearchChromeClient : WebChromeClient() {

        //mute console
        override fun onConsoleMessage(consoleMessage: ConsoleMessage) = true
    }

    inner class SearchJSI {
        @JavascriptInterface
        fun handleHtml(html: String) {
            L.d("Search received response ${contract.isSearchOpened}")
            if (!contract.isSearchOpened) pauseLoad = true
            searchSubject.onNext(html)
        }

        @JavascriptInterface
        fun emit(flag: Int) {
            when (flag) {
                0 -> {
                    L.d("Search loaded successfully")
                }
                1 -> { //something is not found in the search view; this is effectively useless
                    L.eThrow("Search subject error; reverting to full overlay")
                    Prefs.searchBar = false
                    searchSubject.onComplete()
                    contract.searchOverlayDispose()
                }
            }
        }
    }

    /**
     * Clear up some components
     */
    fun dispose() {
        searchSubject.onComplete()
    }

    interface SearchContract {
        fun searchOverlayDispose()
        fun emitSearchResponse(items: List<SearchItem>)
        val isSearchOpened: Boolean
    }
}