aboutsummaryrefslogtreecommitdiff
path: root/app/src/main/kotlin/com/pitchedapps/frost/dbflow/CookiesDb.kt
blob: 14d15bd6419ddbaa71d96c8d9e82b16d2581b385 (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
package com.pitchedapps.frost.dbflow

import android.os.Parcel
import android.os.Parcelable
import com.github.pwittchen.reactivenetwork.library.rx2.ReactiveNetwork
import com.pitchedapps.frost.facebook.FACEBOOK_COM
import com.pitchedapps.frost.facebook.FbTab
import com.pitchedapps.frost.utils.L
import com.pitchedapps.frost.utils.logFrostAnswers
import com.raizlabs.android.dbflow.annotation.ConflictAction
import com.raizlabs.android.dbflow.annotation.Database
import com.raizlabs.android.dbflow.annotation.PrimaryKey
import com.raizlabs.android.dbflow.annotation.Table
import com.raizlabs.android.dbflow.kotlinextensions.*
import com.raizlabs.android.dbflow.structure.BaseModel
import org.jetbrains.anko.doAsync
import org.jsoup.Jsoup
import paperparcel.PaperParcel
import java.net.UnknownHostException

/**
 * Created by Allan Wang on 2017-05-30.
 */

@Database(name = CookiesDb.NAME, version = CookiesDb.VERSION)
object CookiesDb {
    const val NAME = "Cookies"
    const val VERSION = 2
}

@PaperParcel
@Table(database = CookiesDb::class, allFields = true, primaryKeyConflict = ConflictAction.REPLACE)
data class CookieModel(@PrimaryKey var id: Long = -1L, var name: String? = null, var cookie: String? = null) : BaseModel(), Parcelable {
    companion object {
        @JvmField val CREATOR = PaperParcelCookieModel.CREATOR
    }

    override fun describeContents() = 0
    override fun writeToParcel(dest: Parcel, flags: Int) = PaperParcelCookieModel.writeToParcel(this, dest, flags)
}

fun loadFbCookie(id: Long): CookieModel? = (select from CookieModel::class where (CookieModel_Table.id eq id)).querySingle()
fun loadFbCookie(name: String): CookieModel? = (select from CookieModel::class where (CookieModel_Table.name eq name)).querySingle()

/**
 * Loads cookies sorted by name
 */
fun loadFbCookiesAsync(callback: (cookies: List<CookieModel>) -> Unit) {
    (select from CookieModel::class).orderBy(CookieModel_Table.name, true).async().queryListResultCallback { _, tResult -> callback(tResult) }.execute()
}

fun loadFbCookiesSync(): List<CookieModel> = (select from CookieModel::class).orderBy(CookieModel_Table.name, true).queryList()


fun saveFbCookie(cookie: CookieModel, callback: (() -> Unit)? = null) {
    cookie.async save {
        L.d("Fb cookie saved", cookie.toString())
        callback?.invoke()
    }
}

fun removeCookie(id: Long) {
    loadFbCookie(id)?.async?.delete({
        L.d("Fb cookie deleted", id.toString())
    })
}

fun CookieModel.fetchUsername(callback: (String) -> Unit) {
    ReactiveNetwork.checkInternetConnectivity().subscribe {
        yes, _ ->
        if (!yes) return@subscribe callback("")
        doAsync {
            var result = ""
            try {
                result = Jsoup.connect(FbTab.PROFILE.url)
                        .cookie(FACEBOOK_COM, cookie)
                        .get().title()
                L.d("Fetch username found", result)
            } catch (e: Exception) {
                if (e !is UnknownHostException)
                    e.logFrostAnswers("Fetch username failed")
            } finally {
                if (result.isBlank() && (name?.isNotBlank() ?: false)) {
                    callback(name!!)
                    return@doAsync
                }
                if (name != result) {
                    name = result
                    saveFbCookie(this@fetchUsername)
                }
                callback(result)
            }
        }

    }
}