diff --git a/app/build.gradle b/app/build.gradle index f27d721f..67e5177e 100644 --- a/app/build.gradle +++ b/app/build.gradle @@ -74,6 +74,8 @@ dependencies { // Optional -- Mockito framework（可选，用于模拟一些依赖对象，以达到隔离依赖的效果） testImplementation 'org.mockito:mockito-core:5.12.0' testImplementation "org.robolectric:robolectric:4.13" + testImplementation 'cn.hutool:hutool-all:5.8.26' + implementation "org.jetbrains.kotlinx:kotlinx-coroutines-core:1.8.1" //implementation 'wang.harlon.quickjs:wrapper-java:1.0.0' // implementation(ext: 'aar', name: 'quickjs', group: 'fongmi', version: 'release') diff --git a/app/src/main/java/com/github/catvod/api/TianYiHandler.java b/app/src/main/java/com/github/catvod/api/TianYiHandler.java new file mode 100644 index 00000000..e24a6cd8 --- /dev/null +++ b/app/src/main/java/com/github/catvod/api/TianYiHandler.java @@ -0,0 +1,257 @@ +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.uc.Cache; +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.apache.commons.lang3.time.DateUtils; + +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; + +import static org.apache.commons.lang3.time.DateFormatUtils.ISO_8601_EXTENDED_DATETIME_TIME_ZONE_FORMAT; + +public class TianYiHandler { + + private ScheduledExecutorService service; + private AlertDialog dialog; + private final Cache cache; + + public File getCache() { + return Path.tv("tianyi"); + } + + private String indexUrl = ""; + + public TianYiHandler() { + + cache = Cache.objectFrom(Path.read(getCache())); + } + + public JsonObject getUUID() throws IOException { + Map params = new HashMap<>(); + params.put("appId", "E_189"); + Map 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"); + + OkResult okResult = OkHttp.post("https://open.e.189.cn/api/logbox/oauth2/getUUID.do", params, headers); + if (okResult.getCode() == 200) { + return Json.safeObject(okResult.getBody()); + } + + return null; + } + + public byte[] downloadQRCode(String uuid, String cookie) throws IOException { + + + Map 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); + + HttpUrl url = HttpUrl.parse("https://open.e.189.cn/api/logbox/oauth2/image.do").newBuilder().addQueryParameter("uuid", uuid).addQueryParameter("REQID", reqId).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 checkLoginStatus(String uuid, String encryuuid, String reqId, String lt, String paramId, String returnUrl, String secondCookie) throws IOException { + Map params = new HashMap<>(); + params.put("appId", "E_189"); + 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", String.valueOf(System.currentTimeMillis())); + params.put("cb_SaveName", "0"); + params.put("isOauth2", "false"); + params.put("state", ""); + params.put("paramId", paramId); + Map 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("origin", "https,//open.e.189.cn"); + headers.put("referer", indexUrl); + headers.put("Reqid", reqId); + OkResult okResult = OkHttp.post("https://open.e.189.cn/api/logbox/oauth2/qrcodeLoginState.do", params, headers); + JsonObject obj = Json.safeObject(okResult.getBody()).getAsJsonObject(); + if (okResult.getCode() == 200 && obj.get("status").getAsInt() == 0) { + + String redirectUrl = obj.get("redirectUrl").getAsString(); + Map result = new HashMap<>(); + fetchUserInfo(redirectUrl, secondCookie, result); + } + + + return null; + } + + private void fetchUserInfo(String redirectUrl, String secondCookie, Map result) throws IOException { + + Map params = new HashMap<>(); + Map headers = new HashMap<>(); + headers.put("Cookie", secondCookie); + Map> okResult = OkHttp.getLocationHeader(redirectUrl, headers); + /* if (okResult.getCode() == 200) { + okResult.getBody(); + }*/ + return ; + + } + + + public byte[] startScan() throws Exception { + + OkResult okResult1 = OkHttp.get("https://ux.21cn.com/api/htmlReportRest/getJs.js?pid=25577E0DEEDF48ADBD4459911F5825E4", new HashMap<>(), new HashMap<>()); + List cookie = okResult1.getResp().get("Set-Cookie"); + List cookieList = new ArrayList<>(); + for (String s : cookie) { + String[] split = s.split(";"); + String cookie1 = split[0]; + cookieList.add(cookie1); + } + String firstCookie = StringUtils.join(cookieList, ";"); + 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", firstCookie)); + Map> resHeaderMap = OkHttp.getLocationHeader(index, new HashMap<>()); + indexUrl =resHeaderMap.get("Location").get(0); + + cookieList.clear(); + for (String s : resHeaderMap.get("Set-Cookie")) { + String[] split = s.split(";"); + String cookie1 = split[0]; + cookieList.add(cookie1); + } + String secondCookie = StringUtils.join(cookieList, ";"); + HttpUrl httpParams = HttpUrl.parse(indexUrl); + String reqId = httpParams.queryParameter("reqId"); + String lt = httpParams.queryParameter("lt"); + Map 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", "https://open.e.189.cn"); + tHeaders.put("Lt", lt); + tHeaders.put("Reqid", reqId); + tHeaders.put("Cookie", secondCookie); + Map param = new HashMap<>(); + + param.put("version", "2.0"); + param.put("appKey", "cloud"); + String paramId; + String returnUrl; + OkResult okResult = OkHttp.post("https://open.e.189.cn/api/logbox/oauth2/appConf.do", param, tHeaders); + if (okResult.getCode() == 200) { + paramId = Json.safeObject(okResult.getBody()).get("data").getAsJsonObject().get("paramId").getAsString(); + returnUrl = Json.safeObject(okResult.getBody()).get("data").getAsJsonObject().get("returnUrl").getAsString(); + } else { + paramId = ""; + returnUrl = ""; + } + + + // Step 1: Get UUID + JsonObject uuidInfo = getUUID(); + String uuid = uuidInfo.get("uuid").getAsString(); + String encryuuid = uuidInfo.get("encryuuid").getAsString(); + + // Step 2: Get QR Code + byte[] byteStr = downloadQRCode(encryuuid, secondCookie); + + Init.run(() -> showQRCode(byteStr)); + // Step 3: Check login status + // return + Init.execute(() -> startService(uuid, encryuuid, reqId, lt, paramId, returnUrl, firstCookie)); + /*Map result = new HashMap<>(); + result.put("qrcode", "data:image/png;base64," + qrCode); + result.put("status", "NEW");*/ + return byteStr; + + } + + + /** + * 显示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 (IOException e) { + throw new RuntimeException(e); + } + + }, 1, 3, TimeUnit.SECONDS); + } + +} \ No newline at end of file diff --git a/app/src/main/java/com/github/catvod/net/OkHttp.java b/app/src/main/java/com/github/catvod/net/OkHttp.java index 029d6c94..94d66daf 100644 --- a/app/src/main/java/com/github/catvod/net/OkHttp.java +++ b/app/src/main/java/com/github/catvod/net/OkHttp.java @@ -77,6 +77,9 @@ public class OkHttp { public static String getLocation(String url, Map header) throws IOException { return getLocation(client().newBuilder().followRedirects(false).followSslRedirects(false).build().newCall(new Request.Builder().url(url).headers(Headers.of(header)).build()).execute().headers().toMultimap()); } + public static Map> getLocationHeader(String url, Map header) throws IOException { + return client().newBuilder().followRedirects(false).followSslRedirects(false).build().newCall(new Request.Builder().url(url).headers(Headers.of(header)).build()).execute().headers().toMultimap(); + } public static String getLocation(Map> headers) { if (headers == null) return null; diff --git a/app/src/main/java/com/github/catvod/utils/QRCode.java b/app/src/main/java/com/github/catvod/utils/QRCode.java index 20bdaf70..e711d8b7 100644 --- a/app/src/main/java/com/github/catvod/utils/QRCode.java +++ b/app/src/main/java/com/github/catvod/utils/QRCode.java @@ -43,6 +43,14 @@ public class QRCode { } } + public static Bitmap Bytes2Bimap(byte[] b) { + if (b.length != 0) { + return BitmapFactory.decodeByteArray(b, 0, b.length); + } else { + return null; + } + } + public static Bitmap base64StringToImage(String strBase64) { try { byte[] arr = android.util.Base64.decode(strBase64, Base64.NO_WRAP); diff --git a/app/src/test/java/com/github/catvod/api/TianYiHandlerTest.java b/app/src/test/java/com/github/catvod/api/TianYiHandlerTest.java new file mode 100644 index 00000000..d1f0c6ae --- /dev/null +++ b/app/src/test/java/com/github/catvod/api/TianYiHandlerTest.java @@ -0,0 +1,60 @@ +package com.github.catvod.api; + +import cn.hutool.core.io.FileUtil; +import com.google.gson.JsonObject; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.robolectric.RobolectricTestRunner; + +@RunWith(RobolectricTestRunner.class) +public class TianYiHandlerTest { + + private TianYiHandler tianYiHandler; + + + @Before + public void setUp() { + tianYiHandler = new TianYiHandler(); + + } + + @Test + public void startScan() throws Exception { + // Mock the OkHttp.get method to return a predefined OkResult + // Execute the method under test + FileUtil.writeBytes(tianYiHandler.startScan(), "c://qrcode.png"); + + while (true) { + + } + + } + + @Test + public void download() throws Exception { + // Mock the OkHttp.get method to return a predefined OkResult + // Execute the method under test + + + } + + @Test + public void testgetUUID() throws Exception { + JsonObject uuid = tianYiHandler.getUUID(); + System.out.println(uuid); + } + + @Test + public void testdownloadQRCode() throws Exception { +/* + JsonObject uuidInfo = tianYiHandler.getUUID(); + String uuid = uuidInfo.get("uuid").getAsString(); + byte[] qrCode = tianYiHandler.downloadQRCode(uuid); + FileUtil.writeBytes(qrCode, "c://qrcode.png"); + + System.out.println(uuid);*/ + } + + +} \ No newline at end of file