diff --git a/chopperbot-live/src/main/java/org/example/utils/HttpClientUtil.java b/chopperbot-live/src/main/java/org/example/utils/HttpClientUtil.java index c728806..e6093ee 100644 --- a/chopperbot-live/src/main/java/org/example/utils/HttpClientUtil.java +++ b/chopperbot-live/src/main/java/org/example/utils/HttpClientUtil.java @@ -9,7 +9,6 @@ import org.apache.http.entity.StringEntity; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; -import org.example.pool.HttpClientPool; import java.io.IOException; import java.nio.charset.StandardCharsets; @@ -51,6 +50,10 @@ public class HttpClientUtil { return executeRequest(httpPost); } + public static String put(String url) { + return executeRequest(new HttpPut(url)); + } + public static String put(String url, String json) { HttpPut httpPut = new HttpPut(url); httpPut.setEntity(new StringEntity(json, ContentType.APPLICATION_JSON)); @@ -102,4 +105,5 @@ public class HttpClientUtil { throw new RuntimeException("Response entity is null"); } } + } diff --git a/chopperbot-test/src/test/java/org/example/video/BilibiliVideoTest.java b/chopperbot-test/src/test/java/org/example/video/BilibiliVideoTest.java new file mode 100644 index 0000000..5282b17 --- /dev/null +++ b/chopperbot-test/src/test/java/org/example/video/BilibiliVideoTest.java @@ -0,0 +1,195 @@ +package org.example.video; + +import com.alibaba.fastjson2.JSONArray; +import com.alibaba.fastjson2.JSONObject; +import org.apache.http.client.methods.CloseableHttpResponse; +import org.apache.http.client.methods.HttpPut; +import org.apache.http.entity.ByteArrayEntity; +import org.apache.http.impl.client.CloseableHttpClient; +import org.apache.http.impl.client.HttpClients; +import org.apache.http.util.EntityUtils; +import org.aspectj.weaver.ast.ITestVisitor; +import org.example.utils.HttpClientUtil; +import org.openqa.selenium.devtools.v113.network.model.TrustTokenOperationDone; + +import java.io.*; +import java.time.LocalTime; +import java.util.HashMap; +import java.util.Map; + +/** + * @author dhx + * @date 2023/9/9 0:47 + */ +public class BilibiliVideoTest { + // 视频路径 + public static String inputVideoPath = ""; + + // 视频分段暂存路径 + public static String outputDirectory = ""; + + // B站Cookie + public static String Cookie = ""; + public static void main(String[] args) { + String jctPart = Cookie.substring(Cookie.indexOf("bili_jct")); + String csrf; + try{ + csrf = jctPart.substring(jctPart.indexOf("=")+1,jctPart.indexOf(";")); + System.out.println(csrf); + }catch (Exception e){ + System.out.println("Cookie错误"); + e.printStackTrace(); + return; + } + Map header = new HashMap<>(); + int filesize = 0; + FileInputStream fileInputStream1 = null; + try { + File file = new File(inputVideoPath); + if (file.exists() && file.isFile()) { + fileInputStream1 = new FileInputStream(file); + filesize = fileInputStream1.available(); + } + } catch (Exception e) { + e.printStackTrace(); + } + header.put("Cookie", Cookie); + header.put("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36 Edg/116.0.1938.69"); + header.put("Referer", "https://member.bilibili.com/platform/upload/video/frame?spm_id_from=333.1007.top_bar.upload"); + String res = HttpClientUtil.get("https://member.bilibili.com/preupload?probe_version=20221109&upcdn=bda2&zone=cs&name=test.mp4&r=upos&profile=ugcfx%2Fbup&ssl=0&version=2.14.0.0&build=2140000&size=" + filesize + "&webVersion=2.14.0" + , header); + JSONObject mp4Obj = JSONObject.parseObject(res); + String mp4_upos_uri = mp4Obj.getString("upos_uri"); + Long biz_id = mp4Obj.getLong("biz_id"); + String endpoint = mp4Obj.getString("endpoint"); + String put_query = mp4Obj.getString("put_query"); + Long chunk_size = mp4Obj.getLong("chunk_size"); + String auth = mp4Obj.getString("auth"); + System.out.println(mp4_upos_uri); + String res2 = HttpClientUtil.get("https://member.bilibili.com/preupload?name=file_meta.txt&size=2000&r=upos&profile=fxmeta%2Fbup&ssl=0&version=2.14.0.0&build=2140000&webVersion=2.14.0" + , header); + JSONObject txtObj = JSONObject.parseObject(res2); + String meta_upos_uri = txtObj.getString("upos_uri"); + System.out.println(meta_upos_uri); + String preUploadUrl = String.format("https:%s%s?uploads&output=json%s&filesize=%s&partsize=%s&meta_upos_uri=%s&biz_id=%s", endpoint, mp4_upos_uri.substring(mp4_upos_uri.indexOf('/') + 1), put_query.substring(put_query.indexOf('&')), String.valueOf(filesize), chunk_size, meta_upos_uri, biz_id); + System.out.println(preUploadUrl); + Map header2 = new HashMap<>(); + header2.put("X-Upos-Auth", auth); + String res3 = HttpClientUtil.post(preUploadUrl, "{}", header2); + System.out.println(res3); + JSONObject uploadObj = JSONObject.parseObject(res3); + String upload_id = uploadObj.getString("upload_id"); + + + // 分割大小，这里设置为10MB + + int chunkNumber = 1; + + try { + File inputFile = new File(inputVideoPath); + FileInputStream fileInputStream = new FileInputStream(inputFile); + byte[] buffer = new byte[Math.toIntExact(chunk_size)]; + + int bytesRead; + + while ((bytesRead = fileInputStream.read(buffer)) != -1) { + String outputFilePath = outputDirectory + "chunk_" + chunkNumber + ".bin"; + File outputFile = new File(outputFilePath); + FileOutputStream fileOutputStream = new FileOutputStream(outputFile); + fileOutputStream.write(buffer, 0, bytesRead); + fileOutputStream.close(); + + chunkNumber++; + } + chunkNumber--; + + fileInputStream.close(); + } catch (IOException e) { + e.printStackTrace(); + } + + for (int i = 1; i <= chunkNumber; i++) { + long end = i == chunkNumber ? filesize : i * chunk_size; + long start = (i-1)*chunk_size; + File file = new File(outputDirectory + "chunk_" + i +".bin"); + try (CloseableHttpClient httpClient = HttpClients.createDefault()) { + HttpPut httpPut = new HttpPut(String.format("https:%s%s?partNumber=%s&uploadId=%s&chunk=%s&chunks=%s&size=%s&start=%s&end=%s&total=%s", + endpoint,mp4_upos_uri.substring(mp4_upos_uri.indexOf('/') + 1),i,upload_id,i-1,chunkNumber, end - start,start, end,filesize)); // 替换为实际的目标URL + + byte[] binaryData = new byte[(int) file.length()]; + + FileInputStream fileInputStream = new FileInputStream(file); + fileInputStream.read(binaryData); + fileInputStream.close(); + + ByteArrayEntity entity = new ByteArrayEntity(binaryData); + httpPut.setEntity(entity); + httpPut.setHeader("Content-Type", "application/octet-stream"); + httpPut.setHeader("X-Upos-Auth",auth); + + CloseableHttpResponse response = httpClient.execute(httpPut); + int statusCode = response.getStatusLine().getStatusCode(); + String responseBody = EntityUtils.toString(response.getEntity()); + if(statusCode==200){ + System.out.println("上传中："+i*100/chunkNumber+"%"); + } + } catch (IOException e) { + e.printStackTrace(); + } + } + JSONObject body = new JSONObject(); + JSONArray parts = new JSONArray(); + for(int i= 1;i<=chunkNumber;i++){ + JSONObject item = new JSONObject(); + item.put("partNumber",i); + item.put("eTag","etag"); + parts.add(item); + } + body.put("parts",parts); + System.out.println(body.toString()); + String res4 = HttpClientUtil.post( + String.format( + "https:%s%s?output=json&name=test.mp4%s&uploadId=%s&biz_id=%s", + endpoint,mp4_upos_uri.substring(mp4_upos_uri.indexOf('/') + 1), put_query.substring(put_query.indexOf('&')),upload_id,biz_id + ),body.toString(),header2 + ); + System.out.println(res4); + JSONObject body1 = new JSONObject(); + JSONObject subtitle = new JSONObject(); + JSONArray videos = new JSONArray(); + subtitle.put("open",0); + subtitle.put("lan",""); + JSONObject video = new JSONObject(); + video.put("filename",mp4_upos_uri.substring(mp4_upos_uri.lastIndexOf('/')+1,mp4_upos_uri.indexOf('.'))); + video.put("title","test"); + video.put("desc",""); + video.put("cid",biz_id); + videos.add(video); + body1.put("act_reserve_create",0); + body1.put("copyright",1); + body1.put("cover","https://i0.hdslb.com/bfs/archive/2e8c380c85c6970712ea8ede8b26cdc23f0a5b91.jpg"); + //TODO 封面上传 + body1.put("csrf",csrf); + body1.put("desc",""); + body1.put("desc_format_id",0); + body1.put("dolby",0); + body1.put("dynamic",""); + body1.put("interactive",0); + body1.put("lossless_music",0); + body1.put("no_disturbance",0); + body1.put("no_reprint",1); + body1.put("recreate",-1); + body1.put("tag","助眠,音乐"); + body1.put("tid",130); + body1.put("title","test"); + body1.put("up_close_danmu",false); + body1.put("up_close_reply",false); + body1.put("up_selection_reply",false); + body1.put("web_os",1); + body1.put("subtitle",subtitle); + body1.put("videos",videos); + System.out.println(body1); + String res5 = HttpClientUtil.post(String.format("https://member.bilibili.com/x/vu/web/add/v3?t=%s&csrf=%s", LocalTime.now(),csrf),body1.toString(),header); + System.out.println(res5); + } +}