379 lines
15 KiB
Java
379 lines
15 KiB
Java
package com.github.catvod.api;
|
||
|
||
import android.app.AlertDialog;
|
||
import android.content.DialogInterface;
|
||
import android.graphics.Color;
|
||
import android.graphics.drawable.ColorDrawable;
|
||
import android.view.Gravity;
|
||
import android.widget.FrameLayout;
|
||
import android.widget.ImageView;
|
||
import com.github.catvod.bean.tianyi.Cache;
|
||
import com.github.catvod.bean.tianyi.User;
|
||
import com.github.catvod.crawler.SpiderDebug;
|
||
import com.github.catvod.net.OkHttp;
|
||
import com.github.catvod.net.OkResult;
|
||
import com.github.catvod.spider.Init;
|
||
import com.github.catvod.utils.*;
|
||
import com.google.gson.JsonObject;
|
||
import okhttp3.Headers;
|
||
import okhttp3.HttpUrl;
|
||
import okhttp3.Request;
|
||
import okhttp3.Response;
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.apache.commons.lang3.time.DateFormatUtils;
|
||
import org.jetbrains.annotations.NotNull;
|
||
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.net.URLEncoder;
|
||
import java.util.*;
|
||
import java.util.concurrent.Executors;
|
||
import java.util.concurrent.ScheduledExecutorService;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
public class TianYiHandler {
|
||
|
||
public static final String API_URL = "https://open.e.189.cn";
|
||
private ScheduledExecutorService service;
|
||
private AlertDialog dialog;
|
||
private final Cache cache;
|
||
|
||
public File getCache() {
|
||
return Path.tv("tianyi");
|
||
}
|
||
|
||
private String indexUrl = "";
|
||
|
||
private String reqId;
|
||
private String lt;
|
||
private Map<String, String> cookieMap;
|
||
private String cookie;
|
||
|
||
public TianYiHandler() {
|
||
cookie = "";
|
||
cookieMap = new HashMap<>();
|
||
cache = Cache.objectFrom(Path.read(getCache()));
|
||
}
|
||
|
||
public void refreshCookie(String cookie) {
|
||
Map<String, String> header = new HashMap<>();
|
||
header.put("Cookie", cookie);
|
||
String url = "https://cloud.189.cn/api/portal/loginUrl.action?redirectURL=https%3A%2F%2Fcloud.189.cn%2Fweb%2Fredirect.html&defaultSaveName=3&defaultSaeNameCheck=uncheck&browserId=dff95dced0b03d9d972d920f03ddd05e";
|
||
OkResult okResult = OkHttp.get(url, new HashMap<>(), header);
|
||
SpiderDebug.log("refreshCookie:" + okResult.getBody());
|
||
SpiderDebug.log("refreshCookie header:" + Json.toJson(okResult.getResp()));
|
||
if (okResult.getResp().containsKey("set-cookie")) {
|
||
List<String> cookieList = getCookieList(okResult.getResp().get("set-cookie"));
|
||
cache.setTianyiUser(User.objectFrom(StringUtils.join(cookieList, ";")));
|
||
SpiderDebug.log("获取cookie成功:" + StringUtils.join(cookieList, ";"));
|
||
|
||
}
|
||
}
|
||
|
||
public byte[] startScan() throws Exception {
|
||
|
||
OkResult okResult1 = OkHttp.get("https://ux.21cn.com/api/htmlReportRest/getJs.js?pid=25577E0DEEDF48ADBD4459911F5825E4", new HashMap<>(), new HashMap<>());
|
||
List<String> cookie = okResult1.getResp().get("Set-Cookie");
|
||
List<String> cookieList = getCookieList(cookie);
|
||
//String firstCookie = StringUtils.join(cookieList, ";");
|
||
SpiderDebug.log("index ori: " + "https://cloud.189.cn/api/portal/loginUrl.action?redirectURL=https%3A%2F%2Fcloud.189.cn%2Fweb%2Fredirect.html&defaultSaveName=3&defaultSaveNameCheck=uncheck&browserId=dff95dced0b03d9d972d920f03ddd05e");
|
||
String index = OkHttp.getLocation("https://cloud.189.cn/api/portal/loginUrl.action?redirectURL=https://cloud.189.cn/web/redirect.html&defaultSaveName=3&defaultSaveNameCheck=uncheck&browserId=8d38da4344fba4699d13d6e6854319d7", Map.of("Cookie", ""));
|
||
SpiderDebug.log("index red: " + index);
|
||
Map<String, List<String>> resHeaderMap = OkHttp.getLocationHeader(index, new HashMap<>());
|
||
indexUrl = resHeaderMap.get("Location").get(0);
|
||
SpiderDebug.log("indexUrl red: " + indexUrl);
|
||
|
||
cookieList.clear();
|
||
|
||
cookieList = getCookieList(resHeaderMap.get("Set-Cookie"));
|
||
String secondCookie = StringUtils.join(cookieList, ";");
|
||
SpiderDebug.log("secondCookie: " + secondCookie);
|
||
|
||
HttpUrl httpParams = HttpUrl.parse(indexUrl);
|
||
reqId = httpParams.queryParameter("reqId");
|
||
lt = httpParams.queryParameter("lt");
|
||
|
||
Result result = appConf(secondCookie);
|
||
|
||
// Step 1: Get UUID
|
||
JsonObject uuidInfo = getUUID();
|
||
String uuid = uuidInfo.get("uuid").getAsString();
|
||
String encryuuid = uuidInfo.get("encryuuid").getAsString();
|
||
String encodeuuid = uuidInfo.get("encodeuuid").getAsString();
|
||
|
||
// Step 2: Get QR Code
|
||
byte[] byteStr = downloadQRCode(encodeuuid, reqId, secondCookie);
|
||
|
||
Init.run(() -> showQRCode(byteStr));
|
||
// Step 3: Check login status
|
||
// return
|
||
Init.execute(() -> startService(uuid, encryuuid, reqId, lt, result.paramId, result.returnUrl, secondCookie));
|
||
/*Map<String, Object> result = new HashMap<>();
|
||
result.put("qrcode", "data:image/png;base64," + qrCode);
|
||
result.put("status", "NEW");*/
|
||
return byteStr;
|
||
|
||
}
|
||
|
||
|
||
private String api(String url, Map<String, String> params, Map<String, String> headers, Integer retry, String method) throws InterruptedException {
|
||
|
||
|
||
int leftRetry = retry != null ? retry : 3;
|
||
|
||
OkResult okResult;
|
||
if ("GET".equals(method)) {
|
||
okResult = OkHttp.get(this.API_URL + url, params, headers);
|
||
} else {
|
||
okResult = OkHttp.post(this.API_URL + url, params, headers);
|
||
}
|
||
if (okResult.getResp().get("Set-Cookie") != null) {
|
||
getCookieMap(okResult.getResp().get("Set-Cookie"));
|
||
this.cookie = mapToCookie(cookieMap);
|
||
SpiderDebug.log("cookie: " + this.cookie);
|
||
}
|
||
|
||
if (okResult.getCode() != 200 && leftRetry > 0) {
|
||
SpiderDebug.log("请求" + url + " failed;");
|
||
Thread.sleep(1000);
|
||
return api(url, params, headers, leftRetry - 1, method);
|
||
}
|
||
SpiderDebug.log("请求" + url + " 成功;" + "返回结果:" + okResult.getBody());
|
||
return okResult.getBody();
|
||
}
|
||
|
||
/**
|
||
* 获取appConf
|
||
*
|
||
* @param secondCookie
|
||
* @return
|
||
*/
|
||
|
||
private @NotNull Result appConf(String secondCookie) throws Exception {
|
||
Map<String, String> tHeaders = new HashMap<>();
|
||
tHeaders.put("Content-Type", "application/x-www-form-urlencoded");
|
||
tHeaders.put("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:74.0) Gecko/20100101 Firefox/76.0");
|
||
tHeaders.put("Referer", indexUrl);
|
||
tHeaders.put("origin", API_URL);
|
||
tHeaders.put("Lt", lt);
|
||
tHeaders.put("Reqid", reqId);
|
||
tHeaders.put("Cookie", secondCookie);
|
||
Map<String, String> param = new HashMap<>();
|
||
|
||
param.put("version", "2.0");
|
||
param.put("appKey", "cloud");
|
||
String paramId;
|
||
String returnUrl;
|
||
String body = api("/api/logbox/oauth2/appConf.do", param, tHeaders, 3, "POST");
|
||
|
||
paramId = Json.safeObject(body).get("data").getAsJsonObject().get("paramId").getAsString();
|
||
returnUrl = Json.safeObject(body).get("data").getAsJsonObject().get("returnUrl").getAsString();
|
||
|
||
SpiderDebug.log("paramId: " + paramId);
|
||
SpiderDebug.log("returnUrl: " + returnUrl);
|
||
return new Result(paramId, returnUrl);
|
||
}
|
||
|
||
private static class Result {
|
||
public final String paramId;
|
||
public final String returnUrl;
|
||
|
||
public Result(String paramId, String returnUrl) {
|
||
this.paramId = paramId;
|
||
this.returnUrl = returnUrl;
|
||
}
|
||
}
|
||
|
||
|
||
private static @NotNull List<String> getCookieList(List<String> cookie) {
|
||
List<String> cookieList = new ArrayList<>();
|
||
for (String s : cookie) {
|
||
String[] split = s.split(";");
|
||
String cookie1 = split[0];
|
||
cookieList.add(cookie1);
|
||
}
|
||
return cookieList;
|
||
}
|
||
|
||
// 现有方法:List转Map(已优化)
|
||
private void getCookieMap(List<String> cookie) {
|
||
|
||
for (String s : cookie) {
|
||
String[] split = s.split(";");
|
||
String cookieItem = split[0].trim();
|
||
int equalsIndex = cookieItem.indexOf('=');
|
||
if (equalsIndex > 0) {
|
||
String key = cookieItem.substring(0, equalsIndex);
|
||
String value = equalsIndex < cookieItem.length() - 1 ? cookieItem.substring(equalsIndex + 1) : "";
|
||
cookieMap.put(key, value);
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
// 新增方法:Map转Cookie字符串
|
||
private String mapToCookie(Map<String, String> map) {
|
||
if (map == null || map.isEmpty()) return "";
|
||
List<String> joiner = new ArrayList<>();
|
||
for (Map.Entry<String, String> entry : map.entrySet()) {
|
||
joiner.add(entry.getKey() + "=" + entry.getValue());
|
||
}
|
||
return StringUtils.join(joiner, ";");
|
||
}
|
||
|
||
public JsonObject getUUID() throws InterruptedException {
|
||
Map<String, String> params = new HashMap<>();
|
||
params.put("appId", "cloud");
|
||
Map<String, String> headers = new HashMap<>();
|
||
headers.put("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36");
|
||
headers.put("lt", lt);
|
||
headers.put("reqId", reqId);
|
||
headers.put("referer", indexUrl);
|
||
|
||
|
||
String body = api("/api/logbox/oauth2/getUUID.do", params, headers, 3, "POST");
|
||
return Json.safeObject(body);
|
||
|
||
}
|
||
|
||
public byte[] downloadQRCode(String uuid, String reqId, String cookie) throws IOException {
|
||
|
||
|
||
Map<String, String> headers = new HashMap<>();
|
||
headers.put("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36");
|
||
|
||
headers.put("referer", indexUrl);
|
||
headers.put("cookie", cookie);
|
||
// OkResult okResult = OkHttp.get("https://open.e.189.cn/api/logbox/oauth2/image.do", params, headers);
|
||
//.addQueryParameter("uuid", uuid).addQueryParameter("REQID", reqId)
|
||
HttpUrl url = HttpUrl.parse(API_URL + "/api/logbox/oauth2/image.do?uuid=" + uuid + "&REQID=" + reqId).newBuilder().build();
|
||
|
||
Request request = new Request.Builder().url(url).headers(Headers.of(headers)).build();
|
||
Response response = OkHttp.newCall(request);
|
||
if (response.code() == 200) {
|
||
return response.body().bytes();
|
||
}
|
||
return null;
|
||
}
|
||
|
||
|
||
private Map<String, Object> checkLoginStatus(String uuid, String encryuuid, String reqId, String lt, String paramId, String returnUrl, String secondCookie) throws Exception {
|
||
Map<String, String> params = new HashMap<>();
|
||
params.put("appId", "cloud");
|
||
params.put("encryuuid", encryuuid);
|
||
params.put("uuid", uuid);
|
||
params.put("date", DateFormatUtils.format(new Date(), "yyyy-MM-ddHH:mm:ss") + new Random().nextInt(24));
|
||
params.put("returnUrl", URLEncoder.encode(returnUrl, "UTF-8"));
|
||
params.put("clientType", "1");
|
||
params.put("timeStamp", (System.currentTimeMillis() / 1000 + 1) + "000");
|
||
params.put("cb_SaveName", "0");
|
||
params.put("isOauth2", "false");
|
||
params.put("state", "");
|
||
params.put("paramId", paramId);
|
||
Map<String, String> headers = new HashMap<>();
|
||
headers.put("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36");
|
||
headers.put("referer", indexUrl);
|
||
headers.put("Reqid", reqId);
|
||
headers.put("cookie", secondCookie);
|
||
String body = api("/api/logbox/oauth2/qrcodeLoginState.do", params, headers, 3, "POST");
|
||
// OkResult okResult = OkHttp.post(API_URL + "/api/logbox/oauth2/qrcodeLoginState.do", params, headers);
|
||
SpiderDebug.log("qrcodeLoginState result------" + body);
|
||
|
||
JsonObject obj = Json.safeObject(body).getAsJsonObject();
|
||
if (Objects.nonNull(obj.get("status")) && obj.get("status").getAsInt() == 0) {
|
||
|
||
SpiderDebug.log("扫码成功------" + obj.get("redirectUrl").getAsString());
|
||
String redirectUrl = obj.get("redirectUrl").getAsString();
|
||
Map<String, Object> result = new HashMap<>();
|
||
|
||
fetchUserInfo(redirectUrl, secondCookie, result);
|
||
|
||
|
||
} else {
|
||
SpiderDebug.log("扫码失败------" + body);
|
||
}
|
||
|
||
|
||
return null;
|
||
}
|
||
|
||
private void fetchUserInfo(String redirectUrl, String secondCookie, Map<String, Object> result) throws IOException {
|
||
|
||
Map<String, String> params = new HashMap<>();
|
||
Map<String, String> headers = new HashMap<>();
|
||
headers.put("Cookie", secondCookie);
|
||
Map<String, List<String>> okResult = OkHttp.getLocationHeader(redirectUrl, headers);
|
||
SpiderDebug.log("扫码返回数据:" + Json.toJson(okResult));
|
||
if (okResult.containsKey("set-cookie")) {
|
||
List<String> cookieList = getCookieList(okResult.get("set-cookie"));
|
||
cache.setTianyiUser(User.objectFrom(StringUtils.join(cookieList, ";")));
|
||
SpiderDebug.log("获取cookie成功:" + StringUtils.join(cookieList, ";"));
|
||
//停止检验线程,关闭弹窗
|
||
stopService();
|
||
}
|
||
|
||
|
||
/* if (okResult.getCode() == 200) {
|
||
okResult.getBody();
|
||
}*/
|
||
return;
|
||
|
||
}
|
||
|
||
|
||
/**
|
||
* 显示qrcode
|
||
*
|
||
* @param bytes
|
||
*/
|
||
public void showQRCode(byte[] bytes) {
|
||
try {
|
||
int size = ResUtil.dp2px(240);
|
||
FrameLayout.LayoutParams params = new FrameLayout.LayoutParams(size, size);
|
||
ImageView image = new ImageView(Init.context());
|
||
image.setScaleType(ImageView.ScaleType.CENTER_CROP);
|
||
image.setImageBitmap(QRCode.Bytes2Bimap(bytes));
|
||
FrameLayout frame = new FrameLayout(Init.context());
|
||
params.gravity = Gravity.CENTER;
|
||
frame.addView(image, params);
|
||
dialog = new AlertDialog.Builder(Init.getActivity()).setView(frame).setOnCancelListener(this::dismiss).setOnDismissListener(this::dismiss).show();
|
||
dialog.getWindow().setBackgroundDrawable(new ColorDrawable(Color.TRANSPARENT));
|
||
Notify.show("请使用天翼网盘App扫描二维码");
|
||
} catch (Exception ignored) {
|
||
}
|
||
}
|
||
|
||
private void dismiss() {
|
||
try {
|
||
if (dialog != null) dialog.dismiss();
|
||
} catch (Exception ignored) {
|
||
}
|
||
}
|
||
|
||
private void dismiss(DialogInterface dialog) {
|
||
stopService();
|
||
}
|
||
|
||
private void stopService() {
|
||
if (service != null) service.shutdownNow();
|
||
Init.run(this::dismiss);
|
||
}
|
||
|
||
public void startService(String uuid, String encryuuid, String reqId, String lt, String paramId, String returnUrl, String secondCookie) {
|
||
SpiderDebug.log("----start checkLoginStatus service");
|
||
|
||
service = Executors.newScheduledThreadPool(1);
|
||
|
||
service.scheduleWithFixedDelay(() -> {
|
||
SpiderDebug.log("----checkLoginStatus ing....");
|
||
try {
|
||
checkLoginStatus(uuid, encryuuid, reqId, lt, paramId, returnUrl, secondCookie);
|
||
} catch (Exception e) {
|
||
SpiderDebug.log("----checkLoginStatus error" + e.getMessage());
|
||
throw new RuntimeException(e);
|
||
}
|
||
|
||
}, 1, 3, TimeUnit.SECONDS);
|
||
}
|
||
|
||
} |