fengyuecanzhu
3 years ago
parent
d4190694a9
commit
eb7b98b429
@ -0,0 +1,166 @@ |
|||||||
|
package xyz.fycz.myreader.model.third3 |
||||||
|
|
||||||
|
import android.util.Base64 |
||||||
|
import android.util.Log |
||||||
|
import xyz.fycz.myreader.common.APPCONST |
||||||
|
import xyz.fycz.myreader.greendao.service.CacheManager |
||||||
|
import xyz.fycz.myreader.greendao.service.CookieStore |
||||||
|
import xyz.fycz.myreader.model.third3.analyzeRule.JsExtensions |
||||||
|
import xyz.fycz.myreader.util.utils.EncoderUtils |
||||||
|
import xyz.fycz.myreader.util.utils.GSON |
||||||
|
import xyz.fycz.myreader.util.utils.fromJsonObject |
||||||
|
import javax.script.SimpleBindings |
||||||
|
|
||||||
|
/** |
||||||
|
* 可在js里调用,source.xxx() |
||||||
|
*/ |
||||||
|
@Suppress("unused") |
||||||
|
abstract class BaseSource : JsExtensions { |
||||||
|
|
||||||
|
open var concurrentRate: String? = null // 并发率 |
||||||
|
open var loginUrl: String? = null // 登录地址 |
||||||
|
//var loginUi: String? // 登录UI |
||||||
|
open var header: String? = null // 请求头 |
||||||
|
|
||||||
|
open fun getTag(): String = "" |
||||||
|
|
||||||
|
open fun getKey(): String = "" |
||||||
|
|
||||||
|
/*fun loginUi(): List<RowUi>? { |
||||||
|
return GSON.fromJsonArray(loginUi) |
||||||
|
}*/ |
||||||
|
|
||||||
|
fun getLoginJs(): String? { |
||||||
|
val loginJs = loginUrl |
||||||
|
return when { |
||||||
|
loginJs == null -> null |
||||||
|
loginJs.startsWith("@js:") -> loginJs.substring(4) |
||||||
|
loginJs.startsWith("<js>") -> |
||||||
|
loginJs.substring(4, loginJs.lastIndexOf("<")) |
||||||
|
else -> loginJs |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun login() { |
||||||
|
getLoginJs()?.let { |
||||||
|
evalJS(it) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 解析header规则 |
||||||
|
*/ |
||||||
|
fun getHeaderMap(hasLoginHeader: Boolean = false) = HashMap<String, String>().apply { |
||||||
|
this[APPCONST.UA_NAME] = APPCONST.DEFAULT_USER_AGENT |
||||||
|
header?.let { |
||||||
|
GSON.fromJsonObject<Map<String, String>>( |
||||||
|
when { |
||||||
|
it.startsWith("@js:", true) -> |
||||||
|
evalJS(it.substring(4)).toString() |
||||||
|
it.startsWith("<js>", true) -> |
||||||
|
evalJS(it.substring(4, it.lastIndexOf("<"))).toString() |
||||||
|
else -> it |
||||||
|
} |
||||||
|
)?.let { map -> |
||||||
|
putAll(map) |
||||||
|
} |
||||||
|
} |
||||||
|
if (hasLoginHeader) { |
||||||
|
getLoginHeaderMap()?.let { |
||||||
|
putAll(it) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取用于登录的头部信息 |
||||||
|
*/ |
||||||
|
fun getLoginHeader(): String? { |
||||||
|
return CacheManager.get("loginHeader_${getKey()}") |
||||||
|
} |
||||||
|
|
||||||
|
fun getLoginHeaderMap(): Map<String, String>? { |
||||||
|
val cache = getLoginHeader() ?: return null |
||||||
|
return GSON.fromJsonObject(cache) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 保存登录头部信息,map格式,访问时自动添加 |
||||||
|
*/ |
||||||
|
fun putLoginHeader(header: String) { |
||||||
|
CacheManager.put("loginHeader_${getKey()}", header) |
||||||
|
} |
||||||
|
|
||||||
|
fun removeLoginHeader() { |
||||||
|
CacheManager.delete("loginHeader_${getKey()}") |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取用户信息,可以用来登录 |
||||||
|
* 用户信息采用aes加密存储 |
||||||
|
*/ |
||||||
|
fun getLoginInfo(): String? { |
||||||
|
try { |
||||||
|
val key = APPCONST.androidId.encodeToByteArray(0, 8) |
||||||
|
val cache = CacheManager.get("userInfo_${getKey()}") ?: return null |
||||||
|
val encodeBytes = Base64.decode(cache, Base64.DEFAULT) |
||||||
|
val decodeBytes = EncoderUtils.decryptAES(encodeBytes, key) |
||||||
|
?: return null |
||||||
|
return String(decodeBytes) |
||||||
|
} catch (e: Exception) { |
||||||
|
Log.e("BaseSource","" + e.localizedMessage) |
||||||
|
return null |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun getLoginInfoMap(): Map<String, String>? { |
||||||
|
return GSON.fromJsonObject(getLoginInfo()) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 保存用户信息,aes加密 |
||||||
|
*/ |
||||||
|
fun putLoginInfo(info: String): Boolean { |
||||||
|
return try { |
||||||
|
val key = (APPCONST.androidId).encodeToByteArray(0, 8) |
||||||
|
val encodeBytes = EncoderUtils.encryptAES(info.toByteArray(), key) |
||||||
|
val encodeStr = Base64.encodeToString(encodeBytes, Base64.DEFAULT) |
||||||
|
CacheManager.put("userInfo_${getKey()}", encodeStr) |
||||||
|
true |
||||||
|
} catch (e: Exception) { |
||||||
|
Log.e("BaseSource","" + e.localizedMessage) |
||||||
|
false |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun removeLoginInfo() { |
||||||
|
CacheManager.delete("userInfo_${getKey()}") |
||||||
|
} |
||||||
|
|
||||||
|
fun setVariable(variable: String?) { |
||||||
|
if (variable != null) { |
||||||
|
CacheManager.put("sourceVariable_${getKey()}", variable) |
||||||
|
} else { |
||||||
|
CacheManager.delete("sourceVariable_${getKey()}") |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun getVariable(): String? { |
||||||
|
return CacheManager.get("sourceVariable_${getKey()}") |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 执行JS |
||||||
|
*/ |
||||||
|
@Throws(Exception::class) |
||||||
|
fun evalJS(jsStr: String, bindingsConfig: SimpleBindings.() -> Unit = {}): Any? { |
||||||
|
val bindings = SimpleBindings() |
||||||
|
bindings.apply(bindingsConfig) |
||||||
|
bindings["java"] = this |
||||||
|
bindings["source"] = this |
||||||
|
bindings["baseUrl"] = getKey() |
||||||
|
bindings["cookie"] = CookieStore |
||||||
|
bindings["cache"] = CacheManager |
||||||
|
return APPCONST.SCRIPT_ENGINE.eval(jsStr, bindings) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue