Greasy Fork 还支持 简体中文。

Wenku Doc Downloader

对文档截图,合并为纯图片PDF。有限地支持(1)豆丁网(2)道客巴巴(3)360个人图书馆(4)得力文库(5)MBA智库(6)爱问文库(7)原创力文档(8)读根网(9)国标网(10)安全文库网(11)人人文库(12)云展网(13)360文库(14)技工教育网(15)文库吧(16)中国社会科学文库(17)金锄头(18)自然资源标准。预览多少页,导出多少页。额外支持(1)食典通(2)JJG 计量技术规范,详见下方说明。

目前為 2023-07-31 提交的版本,檢視 最新版本

  1. // ==UserScript==
  2. // @name Wenku Doc Downloader
  3. // @namespace http://tampermonkey.net/
  4. // @version 1.9.2
  5. // @description 对文档截图,合并为纯图片PDF。有限地支持(1)豆丁网(2)道客巴巴(3)360个人图书馆(4)得力文库(5)MBA智库(6)爱问文库(7)原创力文档(8)读根网(9)国标网(10)安全文库网(11)人人文库(12)云展网(13)360文库(14)技工教育网(15)文库吧(16)中国社会科学文库(17)金锄头(18)自然资源标准。预览多少页,导出多少页。额外支持(1)食典通(2)JJG 计量技术规范,详见下方说明。
  6. // @author 2690874578@qq.com
  7. // @match *://*.docin.com/p-*
  8. // @match *://docimg1.docin.com/?wk=true
  9. // @match *://ishare.iask.sina.com.cn/f/*
  10. // @match *://ishare.iask.com/f/*
  11. // @match *://swf.ishare.down.sina.com.cn/?path=*
  12. // @match *://swf.ishare.down.sina.com.cn/?wk=true
  13. // @match *://www.deliwenku.com/p-*
  14. // @match *://file.deliwenku.com/?num=*
  15. // @match *://file3.deliwenku.com/?num=*
  16. // @match *://www.doc88.com/p-*
  17. // @match *://www.360doc.com/content/*
  18. // @match *://doc.mbalib.com/view/*
  19. // @match *://www.dugen.com/p-*
  20. // @match *://max.book118.com/html/*
  21. // @match *://openapi.book118.com/?*
  22. // @match *://view-cache.book118.com/pptView.html?*
  23. // @match *://*.book118.com/?readpage=*
  24. // @match *://c.gb688.cn/bzgk/gb/showGb?*
  25. // @match *://www.safewk.com/p-*
  26. // @match *://www.renrendoc.com/paper/*
  27. // @match *://www.renrendoc.com/p-*
  28. // @match *://www.yunzhan365.com/basic/*
  29. // @match *://book.yunzhan365.com/*index.html*
  30. // @match *://www.bing.com/search?q=Bing+AI&showconv=1*
  31. // @match *://wenku.so.com/d/*
  32. // @match *://jg.class.com.cn/cms/resourcedetail.htm?contentUid=*
  33. // @match *://preview.imm.aliyuncs.com/index.html?url=*/jgjyw/*
  34. // @match *://www.wenkub.com/p-*.html*
  35. // @match *://www.sklib.cn/manuscripts/?*
  36. // @match *://www-sklib-cn-s.z.library.sh.cn/booklib/bookPdfRead?*
  37. // @match *://www.jinchutou.com/shtml/view-*
  38. // @match *://www.jinchutou.com/p-*
  39. // @match *://www.nrsis.org.cn/*/read/*
  40. // @require https://cdn.staticfile.org/jspdf/2.5.1/jspdf.umd.min.js
  41. // @require https://cdn.staticfile.org/html2canvas/1.4.1/html2canvas.min.js
  42. // @icon https://s2.loli.net/2022/01/12/wc9je8RX7HELbYQ.png
  43. // @icon64 https://s2.loli.net/2022/01/12/tmFeSKDf8UkNMjC.png
  44. // @grant none
  45. // @run-at document-idle
  46. // @license GPL-3.0-only
  47. // @create 2021-11-22
  48. // @note 1. 暴露主函数到全局,签名:`window.wk_main(host: string) -> void`
  49. // @note 2. 社会科学文库代码修改,更适于临时调整,暴露了 `window.wk_sklib_url` 用于覆盖基础 url,其中必须包含 `"{id}"` 和 `"{index}"`
  50. // @note 3. 人人文库适配PPT
  51. // ==/UserScript==
  52.  
  53.  
  54. (function () {
  55. 'use strict';
  56.  
  57. /**
  58. * 基于 window.postMessage 通信的套接字对象
  59. */
  60. class Socket {
  61. /**
  62. * 创建套接字对象
  63. * @param {Window} target 目标窗口
  64. */
  65. constructor(target) {
  66. if (!(target.window && (target === target.window))) {
  67. console.log(target);
  68. throw new Error(`target is not a [Window Object]`);
  69. }
  70. this.target = target;
  71. this.connected = false;
  72. this.listeners = new Set();
  73. }
  74.  
  75. get [Symbol.toStringTag]() { return "Socket"; }
  76.  
  77. /**
  78. * 向目标窗口发消息
  79. * @param {*} message
  80. */
  81. talk(message) {
  82. if (!this.target) {
  83. throw new TypeError(
  84. `socket.target is not a window: ${this.target}`
  85. );
  86. }
  87. this.target.postMessage(message, "*");
  88. }
  89.  
  90. /**
  91. * 添加捕获型监听器,返回实际添加的监听器
  92. * @param {Function} listener (e: MessageEvent) => {...}
  93. * @param {boolean} once 是否在执行后自动销毁,默认 false;如为 true 则使用自动包装过的监听器
  94. * @returns {Function} listener
  95. */
  96. listen(listener, once=false) {
  97. if (this.listeners.has(listener)) {
  98. return;
  99. }
  100.  
  101. let real_listener = listener;
  102. // 包装监听器
  103. if (once) {
  104. const self = this;
  105. function wrapped(e) {
  106. listener(e);
  107. self.notListen(wrapped);
  108. }
  109. real_listener = wrapped;
  110. }
  111. // 添加监听器
  112. this.listeners.add(real_listener);
  113. window.addEventListener(
  114. "message", real_listener, true
  115. );
  116. return real_listener;
  117. }
  118.  
  119. /**
  120. * 移除socket上的捕获型监听器
  121. * @param {Function} listener (e: MessageEvent) => {...}
  122. */
  123. notListen(listener) {
  124. console.log(listener);
  125. console.log(
  126. "listener delete operation:",
  127. this.listeners.delete(listener)
  128. );
  129. window.removeEventListener("message", listener, true);
  130. }
  131.  
  132. /**
  133. * 检查对方来信是否为pong消息
  134. * @param {MessageEvent} e
  135. * @param {Function} resolve
  136. */
  137. _on_pong(e, resolve) {
  138. // 收到pong消息
  139. if (e.data.pong) {
  140. this.connected = true;
  141. this.listeners.forEach(
  142. listener => listener.ping ? this.notListen(listener) : 0
  143. );
  144. console.log("Client: Connected!\n" + new Date());
  145. resolve(this);
  146. }
  147. }
  148.  
  149. /**
  150. * 向对方发送ping消息
  151. * @returns {Promise<Socket>}
  152. */
  153. _ping() {
  154. return new Promise((resolve, reject) => {
  155. // 绑定pong检查监听器
  156. const listener = this.listen(
  157. e => this._on_pong(e, resolve)
  158. );
  159. listener.ping = true;
  160.  
  161. // 5分钟后超时
  162. setTimeout(
  163. () => reject(new Error(`Timeout Error during receiving pong (>5min)`)),
  164. 5 * 60 * 1000
  165. );
  166. // 发送ping消息
  167. this.talk({ ping: true });
  168. });
  169. }
  170.  
  171. /**
  172. * 检查对方来信是否为ping消息
  173. * @param {MessageEvent} e
  174. * @param {Function} resolve
  175. */
  176. _on_ping(e, resolve) {
  177. // 收到ping消息
  178. if (e.data.ping) {
  179. this.target = e.source;
  180. this.connected = true;
  181. this.listeners.forEach(
  182. listener => listener.pong ? this.notListen(listener) : 0
  183. );
  184. console.log("Server: Connected!\n" + new Date());
  185. // resolve 后期约状态无法回退
  186. // 但后续代码仍可执行
  187. resolve(this);
  188. // 回应pong消息
  189. this.talk({ pong: true });
  190. }
  191. }
  192.  
  193. /**
  194. * 当对方来信是为ping消息时回应pong消息
  195. * @returns {Promise<Socket>}
  196. */
  197. _pong() {
  198. return new Promise(resolve => {
  199. // 绑定ping检查监听器
  200. const listener = this.listen(
  201. e => this._on_ping(e, resolve)
  202. );
  203. listener.pong = true;
  204. });
  205. }
  206.  
  207. /**
  208. * 连接至目标窗口
  209. * @param {boolean} talk_first 是否先发送ping消息
  210. * @param {Window} target 目标窗口
  211. * @returns {Promise<Socket>}
  212. */
  213. connect(talk_first) {
  214. // 先发起握手
  215. if (talk_first) {
  216. return this._ping();
  217. }
  218. // 后发起握手
  219. return this._pong();
  220. }
  221. }
  222.  
  223.  
  224. const base = {
  225. Socket,
  226.  
  227. init_gbk_encoder() {
  228.  
  229. let table;
  230.  
  231. function initGbkTable() {
  232. // https://en.wikipedia.org/wiki/GBK_(character_encoding)#Encoding
  233. const ranges = [
  234. [0xA1, 0xA9, 0xA1, 0xFE],
  235. [0xB0, 0xF7, 0xA1, 0xFE],
  236. [0x81, 0xA0, 0x40, 0xFE],
  237. [0xAA, 0xFE, 0x40, 0xA0],
  238. [0xA8, 0xA9, 0x40, 0xA0],
  239. [0xAA, 0xAF, 0xA1, 0xFE],
  240. [0xF8, 0xFE, 0xA1, 0xFE],
  241. [0xA1, 0xA7, 0x40, 0xA0],
  242. ];
  243. const codes = new Uint16Array(23940);
  244. let i = 0;
  245.  
  246. for (const [b1Begin, b1End, b2Begin, b2End] of ranges) {
  247. for (let b2 = b2Begin; b2 <= b2End; b2++) {
  248. if (b2 !== 0x7F) {
  249. for (let b1 = b1Begin; b1 <= b1End; b1++) {
  250. codes[i++] = b2 << 8 | b1;
  251. }
  252. }
  253. }
  254. }
  255. table = new Uint16Array(65536);
  256. table.fill(0xFFFF);
  257.  
  258. const str = new TextDecoder('gbk').decode(codes);
  259. for (let i = 0; i < str.length; i++) {
  260. table[str.charCodeAt(i)] = codes[i];
  261. }
  262. }
  263.  
  264. const defaultOnAlloc = (len) => new Uint8Array(len);
  265. const defaultOnError = () => 63; // '?'
  266.  
  267. /**
  268. * 字符串编码为gbk字节串
  269. * @param {string} str
  270. * @param {Function} onError 处理编码失败时返回字符替代值的函数,默认是返回 63('?') 的函数
  271. * @returns {Uint8Array}
  272. */
  273. return function(str, onError=null) {
  274. if (!table) {
  275. initGbkTable();
  276. }
  277. const onAlloc = defaultOnAlloc;
  278. onError = onError === null ? defaultOnError : onError;
  279.  
  280. const buf = onAlloc(str.length * 2);
  281. let n = 0;
  282.  
  283. for (let i = 0; i < str.length; i++) {
  284. const code = str.charCodeAt(i);
  285. if (code < 0x80) {
  286. buf[n++] = code;
  287. continue;
  288. }
  289.  
  290. const gbk = table[code];
  291.  
  292. if (gbk !== 0xFFFF) {
  293. buf[n++] = gbk;
  294. buf[n++] = gbk >> 8;
  295. }
  296. else if (code === 8364) {
  297. // 8364 == '€'.charCodeAt(0)
  298. // Code Page 936 has a single-byte euro sign at 0x80
  299. buf[n++] = 0x80;
  300. }
  301. else {
  302. const ret = onError(i, str);
  303. if (ret === -1) {
  304. break;
  305. }
  306. if (ret > 0xFF) {
  307. buf[n++] = ret;
  308. buf[n++] = ret >> 8;
  309. } else {
  310. buf[n++] = ret;
  311. }
  312. }
  313. }
  314. return buf.subarray(0, n)
  315. }
  316. },
  317.  
  318. /**
  319. * Construct a table with table[i] as the length of the longest prefix of the substring 0..i
  320. * @param {Array<number>} arr
  321. * @returns {Array<number>}
  322. */
  323. longest_prefix: function(arr) {
  324.  
  325. // create a table of size equal to the length of `str`
  326. // table[i] will store the prefix of the longest prefix of the substring str[0..i]
  327. let table = new Array(arr.length);
  328. let maxPrefix = 0;
  329. // the longest prefix of the substring str[0] has length
  330. table[0] = 0;
  331.  
  332. // for the substrings the following substrings, we have two cases
  333. for (let i = 1; i < arr.length; i++) {
  334. // case 1. the current character doesn't match the last character of the longest prefix
  335. while (maxPrefix > 0 && arr[i] !== arr[maxPrefix]) {
  336. // if that is the case, we have to backtrack, and try find a character that will be equal to the current character
  337. // if we reach 0, then we couldn't find a chracter
  338. maxPrefix = table[maxPrefix - 1];
  339. }
  340. // case 2. The last character of the longest prefix matches the current character in `str`
  341. if (arr[maxPrefix] === arr[i]) {
  342. // if that is the case, we know that the longest prefix at position i has one more character.
  343. // for example consider `-` be any character not contained in the set [a-c]
  344. // str = abc----abc
  345. // consider `i` to be the last character `c` in `str`
  346. // maxPrefix = will be 2 (the first `c` in `str`)
  347. // maxPrefix now will be 3
  348. maxPrefix++;
  349. // so the max prefix for table[9] is 3
  350. }
  351. table[i] = maxPrefix;
  352. }
  353. return table;
  354. },
  355.  
  356. // 用于取得一次列表中所有迭代器的值
  357. getAllValus: function(iterators) {
  358. if (iterators.length === 0) {
  359. return [true, []];
  360. }
  361. let values = [];
  362. for (let iterator of iterators) {
  363. let {value, done} = iterator.next();
  364. if (done) {
  365. return [true, []];
  366. }
  367. values.push(value);
  368. }
  369. return [false, values];
  370. },
  371.  
  372. /**
  373. * 使用过时的execCommand复制文字
  374. * @param {string} text
  375. */
  376. oldCopy: function(text) {
  377. const input = document.createElement("input");
  378. input.value = text;
  379. document.body.appendChild(input);
  380. input.select();
  381. document.execCommand("copy");
  382. input.remove();
  383. },
  384.  
  385. b64ToUint6: function(nChr) {
  386. return nChr > 64 && nChr < 91 ?
  387. nChr - 65
  388. : nChr > 96 && nChr < 123 ?
  389. nChr - 71
  390. : nChr > 47 && nChr < 58 ?
  391. nChr + 4
  392. : nChr === 43 ?
  393. 62
  394. : nChr === 47 ?
  395. 63
  396. :
  397. 0;
  398. },
  399.  
  400. /**
  401. * 元素选择器
  402. * @param {string} selector 选择器
  403. * @returns {Array<HTMLElement>} 元素列表
  404. */
  405. $: function(selector) {
  406. const self = this?.querySelectorAll ? this : document;
  407. return [...self.querySelectorAll(selector)];
  408. },
  409.  
  410. /**
  411. * 安全元素选择器,直到元素存在时才返回元素列表,最多等待5秒
  412. * @param {string} selector 选择器
  413. * @returns {Promise<Array<HTMLElement>>} 元素列表
  414. */
  415. $$: async function(selector) {
  416. const self = this?.querySelectorAll ? this : document;
  417.  
  418. for (let i = 0; i < 10; i++) {
  419. let elems = [...self.querySelectorAll(selector)];
  420. if (elems.length > 0) {
  421. return elems;
  422. }
  423. await new Promise(r => setTimeout(r, 500));
  424. }
  425. throw Error(`"${selector}" not found in 5s`);
  426. },
  427.  
  428. /**
  429. * 将2个及以上的空白字符(除了换行符)替换成一个空格
  430. * @param {string} text
  431. * @returns {string}
  432. */
  433. stripBlanks: function(text) {
  434. return text
  435. .replace(/([^\r\n])(\s{2,})(?=[^\r\n])/g, "$1 ")
  436. .replace(/\n{2,}/, "\n");
  437. },
  438.  
  439. /**
  440. * 复制属性(含访问器)到 target
  441. * @param {Object} target
  442. * @param {...Object} sources
  443. * @returns
  444. */
  445. superAssign: function(target, ...sources) {
  446. sources.forEach(source =>
  447. Object.defineProperties(
  448. target, Object.getOwnPropertyDescriptors(source)
  449. )
  450. );
  451. return target;
  452. },
  453.  
  454. makeCRC32: function() {
  455. function makeCRCTable() {
  456. let c;
  457. let crcTable = [];
  458. for(var n =0; n < 256; n++){
  459. c = n;
  460. for(var k =0; k < 8; k++){
  461. c = ((c&1) ? (0xEDB88320 ^ (c >>> 1)) : (c >>> 1));
  462. }
  463. crcTable[n] = c;
  464. }
  465. return crcTable;
  466. }
  467.  
  468. const crcTable = makeCRCTable();
  469.  
  470. /**
  471. * @param {string} str
  472. * @returns {number}
  473. */
  474. return function(str) {
  475. let crc = 0 ^ (-1);
  476. for (var i = 0; i < str.length; i++ ) {
  477. crc = (crc >>> 8) ^ crcTable[(crc ^ str.charCodeAt(i)) & 0xFF];
  478. }
  479. return (crc ^ (-1)) >>> 0;
  480. };
  481. }
  482. };
  483.  
  484. const box = `
  485. <div class="wk-box">
  486. <section class="btns-sec">
  487. <p class="logo_tit">Wenku Doc Downloader</p>
  488. <button class="btn-1">展开文档 😈</button>
  489. <button class="btn-2">空按钮 2</button>
  490. <button class="btn-3">空按钮 3</button>
  491. <button class="btn-4">空按钮 4</button>
  492. <button class="btn-5">空按钮 5</button>
  493. </section>
  494. <p class="wk-fold-btn unfold"></p>
  495. </div>
  496. `;
  497.  
  498. const style = `
  499. <style class="wk-style">
  500. .wk-fold-btn {
  501. position: fixed;
  502. left: 151px;
  503. top: 36%;
  504. user-select: none;
  505. font-size: large;
  506. z-index: 1001;
  507. }
  508.  
  509. .wk-fold-btn::after {
  510. content: "🐵";
  511. }
  512. .wk-fold-btn.folded {
  513. left: 20px;
  514. }
  515. .wk-fold-btn.folded::after {
  516. content: "🙈";
  517. }
  518.  
  519. .wk-box {
  520. position: fixed;
  521. width: 154px;
  522. left: 10px;
  523. top: 32%;
  524. z-index: 1000;
  525. }
  526.  
  527. .btns-sec {
  528. background: #E7F1FF;
  529. border: 2px solid #1676FF;
  530. padding: 0px 0px 10px 0px;
  531. font-weight: 600;
  532. border-radius: 2px;
  533. font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'PingFang SC', 'Hiragino Sans GB',
  534. 'Microsoft YaHei', 'Helvetica Neue', Helvetica, Arial, sans-serif, 'Apple Color Emoji',
  535. 'Segoe UI Emoji', 'Segoe UI Symbol';
  536. }
  537.  
  538. .btns-sec.folded {
  539. display: none;
  540. }
  541.  
  542. .logo_tit {
  543. width: 100%;
  544. background: #1676FF;
  545. text-align: center;
  546. font-size: 12px;
  547. color: #E7F1FF;
  548. line-height: 40px;
  549. height: 40px;
  550. margin: 0 0 16px 0;
  551. }
  552.  
  553. .btn-1 {
  554. display: block;
  555. width: 128px;
  556. height: 28px;
  557. background: linear-gradient(180deg, #00E7F7 0%, #FEB800 0.01%, #FF8700 100%);
  558. border-radius: 4px;
  559. color: #fff;
  560. font-size: 12px;
  561. border: none;
  562. outline: none;
  563. margin: 8px auto;
  564. font-weight: bold;
  565. cursor: pointer;
  566. opacity: .9;
  567. }
  568.  
  569. .btn-2 {
  570. display: none;
  571. width: 128px;
  572. height: 28px;
  573. background: #07C160;
  574. border-radius: 4px;
  575. color: #fff;
  576. font-size: 12px;
  577. border: none;
  578. outline: none;
  579. margin: 8px auto;
  580. font-weight: bold;
  581. cursor: pointer;
  582. opacity: .9;
  583. }
  584.  
  585. .btn-3 {
  586. display: none;
  587. width: 128px;
  588. height: 28px;
  589. background: #FA5151;
  590. border-radius: 4px;
  591. color: #fff;
  592. font-size: 12px;
  593. border: none;
  594. outline: none;
  595. margin: 8px auto;
  596. font-weight: bold;
  597. cursor: pointer;
  598. opacity: .9;
  599. }
  600.  
  601. .btn-4 {
  602. display: none;
  603. width: 128px;
  604. height: 28px;
  605. background: #1676FF;
  606. border-radius: 4px;
  607. color: #fff;
  608. font-size: 12px;
  609. border: none;
  610. outline: none;
  611. margin: 8px auto;
  612. font-weight: bold;
  613. cursor: pointer;
  614. opacity: .9;
  615. }
  616.  
  617. .btn-5 {
  618. display: none;
  619. width: 128px;
  620. height: 28px;
  621. background: #ff6600;
  622. border-radius: 4px;
  623. color: #fff;
  624. font-size: 12px;
  625. border: none;
  626. outline: none;
  627. margin: 8px auto;
  628. font-weight: bold;
  629. cursor: pointer;
  630. opacity: .9;
  631. }
  632.  
  633. .btn-1:hover,
  634. .btn-2:hover,
  635. .btn-3:hover,
  636. .btn-4:hover,
  637. .btn-5:hover {
  638. opacity: .8;
  639. }
  640.  
  641. .btn-1:active,
  642. .btn-2:active,
  643. .btn-3:active,
  644. .btn-4:active,
  645. .btn-5:active {
  646. opacity: 1;
  647. }
  648.  
  649. .wk-popup-container {
  650. height: 100vh;
  651. width: 100vw;
  652. display: flex;
  653. flex-direction: column;
  654. justify-content: space-around;
  655. z-index: 999;
  656. background: 0 0;
  657. }
  658.  
  659. .wk-popup-head {
  660. font-size: 1.5em;
  661. margin-bottom: 12px
  662. }
  663.  
  664. .wk-card {
  665. background: #fff;
  666. background-image: linear-gradient(48deg, #fff 0, #e5efe9 100%);
  667. border-top-right-radius: 16px;
  668. border-bottom-left-radius: 16px;
  669. box-shadow: -20px 20px 35px 1px rgba(10, 49, 86, .18);
  670. display: flex;
  671. flex-direction: column;
  672. padding: 32px;
  673. margin: 0;
  674. max-width: 400px;
  675. width: 100%
  676. }
  677.  
  678. .content-wrapper {
  679. font-size: 1.1em;
  680. margin-bottom: 44px
  681. }
  682.  
  683. .content-wrapper:last-child {
  684. margin-bottom: 0
  685. }
  686.  
  687. .wk-button {
  688. align-items: center;
  689. background: #e5efe9;
  690. border: 1px solid #5a72b5;
  691. border-radius: 4px;
  692. color: #121943;
  693. cursor: pointer;
  694. display: flex;
  695. font-size: 1em;
  696. font-weight: 700;
  697. height: 40px;
  698. justify-content: center;
  699. width: 150px
  700. }
  701.  
  702. .wk-button:focus {
  703. border: 2px solid transparent;
  704. box-shadow: 0 0 0 2px #121943;
  705. outline: solid 4px transparent
  706. }
  707.  
  708. .link {
  709. color: #121943
  710. }
  711.  
  712. .link:focus {
  713. box-shadow: 0 0 0 2px #121943
  714. }
  715.  
  716. .input-wrapper {
  717. display: flex;
  718. flex-direction: column
  719. }
  720.  
  721. .input-wrapper .label {
  722. align-items: baseline;
  723. display: flex;
  724. font-weight: 700;
  725. justify-content: space-between;
  726. margin-bottom: 8px
  727. }
  728.  
  729. .input-wrapper .optional {
  730. color: #5a72b5;
  731. font-size: .9em
  732. }
  733.  
  734. .input-wrapper .input {
  735. border: 1px solid #5a72b5;
  736. border-radius: 4px;
  737. height: 40px;
  738. padding: 8px
  739. }
  740.  
  741. .modal-header {
  742. align-items: baseline;
  743. display: flex;
  744. justify-content: space-between
  745. }
  746.  
  747. .close {
  748. background: 0 0;
  749. border: none;
  750. cursor: pointer;
  751. display: flex;
  752. height: 16px;
  753. text-decoration: none;
  754. width: 16px
  755. }
  756.  
  757. .close svg {
  758. width: 16px
  759. }
  760.  
  761. .modal-wrapper {
  762. background: rgba(0, 0, 0, .7);
  763. }
  764.  
  765. #wk-popup {
  766. opacity: 0;
  767. transition: opacity .25s ease-in-out;
  768. display: none;
  769. flex-direction: row;
  770. justify-content: space-around;
  771. }
  772.  
  773. #wk-popup:target {
  774. opacity: 1;
  775. display: flex;
  776. }
  777.  
  778. #wk-popup:target .modal-body {
  779. opacity: 1;
  780. transform: translateY(1);
  781. }
  782.  
  783. #wk-popup .modal-body {
  784. max-width: 500px;
  785. opacity: 0;
  786. transform: translateY(-3vh);
  787. transition: opacity .25s ease-in-out;
  788. width: 100%;
  789. z-index: 1
  790. }
  791.  
  792. .outside-trigger {
  793. bottom: 0;
  794. cursor: default;
  795. left: 0;
  796. position: fixed;
  797. right: 0;
  798. top: 0;
  799. }
  800. </style>
  801. `;
  802.  
  803. const popup = `
  804. <div class="wk-popup-container">
  805. <div class='modal-wrapper' id='wk-popup'>
  806. <div class='modal-body wk-card'>
  807. <div class='modal-header'>
  808. <h2 class='wk-popup-head'>下载进度条</h2>
  809. <a href='#!' role='wk-button' class='close' aria-label='close this modal'>
  810. <svg viewBox='0 0 24 24'>
  811. <path
  812. d='M24 20.188l-8.315-8.209 8.2-8.282-3.697-3.697-8.212 8.318-8.31-8.203-3.666 3.666 8.321 8.24-8.206 8.313 3.666 3.666 8.237-8.318 8.285 8.203z'>
  813. </path>
  814. </svg>
  815. </a>
  816. </div>
  817. <p class='wk-popup-body'>正在初始化内容...</p>
  818. </div>
  819. <a href='#!' class='outside-trigger'></a>
  820. </div>
  821. </div>
  822. `;
  823.  
  824. globalThis.wk$ = base.$;
  825. globalThis.wk$$ = base.$$;
  826.  
  827.  
  828. const utils = {
  829. Socket: base.Socket,
  830.  
  831. PDF_LIB_URL: "https://cdn.staticfile.org/pdf-lib/1.17.1/pdf-lib.min.js",
  832.  
  833. encode_to_gbk: base.init_gbk_encoder(),
  834.  
  835. /**
  836. * 以指定原因弹窗提示并抛出错误
  837. * @param {string} reason
  838. */
  839. raise: function(reason) {
  840. alert(reason);
  841. throw new Error(reason);
  842. },
  843.  
  844. /**
  845. * 合并多个PDF
  846. * @param {Array<ArrayBuffer | Uint8Array>} pdfs
  847. * @returns {Promise<Uint8Array>}
  848. */
  849. join_pdfs: async function(pdfs) {
  850. if (!window.PDFLib) {
  851. await this.load_web_script(this.PDF_LIB_URL);
  852. }
  853.  
  854. const combined = await PDFLib.PDFDocument.create();
  855.  
  856. for (const [i, buffer] of this.enumerate(pdfs)) {
  857. const pdf = await PDFLib.PDFDocument.load(buffer);
  858. const pages = await combined.copyPages(
  859. pdf, pdf.getPageIndices()
  860. );
  861.  
  862. for (const page of pages) {
  863. combined.addPage(page);
  864. }
  865. this.update_popup(`已经合并 ${i + 1} 组`);
  866. }
  867.  
  868. return await combined.save();
  869. },
  870.  
  871. /**
  872. * raise an error for status which is not in [200, 299]
  873. * @param {Response} response
  874. */
  875. raise_for_status(response) {
  876. if (!response.ok) {
  877. throw new Error(
  878. `Fetch Error with status code: ${response.status}`
  879. );
  880. }
  881. },
  882.  
  883. /**
  884. * 计算 str 的 CRC32 摘要(number)
  885. * @param {string} str
  886. * @returns {number}
  887. */
  888. crc32: base.makeCRC32(),
  889.  
  890. /**
  891. * 返回函数参数定义
  892. * @param {Function} fn
  893. * @param {boolean} print 是否打印到控制台,默认 true
  894. * @returns {string | undefined}
  895. */
  896. help: function(fn, print=true) {
  897. if (!(fn instanceof Function))
  898. throw new Error(`fn must be a function`);
  899.  
  900. const
  901. _fn = fn.__func__ || fn,
  902. ARROW_ARG = /^([^(]+?)=>/,
  903. FN_ARGS = /^[^(]*\(\s*([^)]*)\)/m,
  904. STRIP_COMMENTS = /((\/\/.*$)|(\/\*[\s\S]*?\*\/))/mg,
  905. fn_text = Function.prototype.toString.call(_fn).replace(STRIP_COMMENTS, ''),
  906. args = fn_text.match(ARROW_ARG) || fn_text.match(FN_ARGS),
  907. // 如果自带 doc,优先使用,否则使用源码
  908. doc = fn.__doc__ ? fn.__doc__ : args[0];
  909. if (!print) return base.stripBlanks(doc);
  910.  
  911. const color = (window.matchMedia &&
  912. window.matchMedia('(prefers-color-scheme: dark)').matches
  913. ) ;
  914. console.log("%c" + doc, `color: ${color}; font: small italic`);
  915. },
  916.  
  917. /**
  918. * 字节数组转十六进制字符串
  919. * @param {Uint8Array} arr
  920. * @returns {string}
  921. */
  922. hex_bytes: function(arr) {
  923. return Array.from(arr)
  924. .map(byte => byte.toString(16).padStart(2, "0"))
  925. .join("");
  926. },
  927.  
  928. /**
  929. * 取得对象类型
  930. * @param {*} obj
  931. * @returns {string} class
  932. */
  933. classof: function(obj) {
  934. return Object
  935. .prototype
  936. .toString
  937. .call(obj)
  938. .slice(8, -1);
  939. },
  940.  
  941. /**
  942. * 随机改变字体颜色、大小、粗细
  943. * @param {HTMLElement} elem
  944. */
  945. emphasize_text: function(elem) {
  946. const rand = Math.random;
  947. elem.style.cssText = `
  948. font-weight: ${200 + parseInt(700 * rand())};
  949. font-size: ${(1 + rand()).toFixed(1)}em;
  950. color: hsl(${parseInt(360 * rand())}, ${parseInt(40 + 60 * rand())}%, ${parseInt(60 * rand())}%);
  951. background-color: yellow;`;
  952. },
  953.  
  954. /**
  955. * 等待直到 DOM 节点停止变化
  956. * @param {HTMLElement} elem 监听节点
  957. * @param {number} timeout 超时毫秒数
  958. * @returns {Promise<MutationObserver>} observer
  959. */
  960. until_stop: async function(elem, timeout=2000) {
  961. // 创建用于共享的监听器
  962. let observer;
  963. // 创建超时 Promise
  964. const timeout_promise = new Promise((_, reject) => {
  965. setTimeout(() => {
  966. // 停止监听、释放资源
  967. observer.disconnect();
  968. const error = new Error(
  969. `Timeout Error occured on listening DOM mutation (max ${timeout}ms)`,
  970. { cause: elem }
  971. );
  972. reject(error);
  973. }, timeout);
  974. });
  975. // 开始元素节点变动监听
  976. return Promise.race([
  977. new Promise(resolve => {
  978. // 创建监听器
  979. observer = new MutationObserver(
  980. (_, observer) => {
  981. // DOM 变动结束后终止监听、释放资源
  982. observer.disconnect();
  983. // 返回监听器
  984. resolve(observer);
  985. }
  986. );
  987. // 开始监听目标节点
  988. observer.observe(elem, {
  989. subtree: true,
  990. childList: true,
  991. attributes: true
  992. });
  993. }),
  994. timeout_promise,
  995. ])
  996. .catch(error => {
  997. if (`${error}`.includes("Timeout Error")) {
  998. return observer;
  999. }
  1000. console.error(error);
  1001. throw error;
  1002. });
  1003. },
  1004.  
  1005. /**
  1006. * Find all the patterns that matches in a given string `str`
  1007. * this algorithm is based on the Knuth–Morris–Pratt algorithm. Its beauty consists in that it performs the matching in O(n)
  1008. * @param {Array<number>} arr
  1009. * @param {Array<number>} sub_arr
  1010. * @returns {Array<number>}
  1011. */
  1012. kmp_matching: function(arr, sub_arr) {
  1013. // find the prefix table in O(n)
  1014. let prefixes = base.longest_prefix(sub_arr);
  1015. let matches = [];
  1016.  
  1017. // `j` is the index in `P`
  1018. let j = 0;
  1019. // `i` is the index in `S`
  1020. let i = 0;
  1021. while (i < arr.length) {
  1022. // Case 1. S[i] == P[j] so we move to the next index in `S` and `P`
  1023. if (arr[i] === sub_arr[j]) {
  1024. i++;
  1025. j++;
  1026. }
  1027. // Case 2. `j` is equal to the length of `P`
  1028. // that means that we reached the end of `P` and thus we found a match
  1029. if (j === sub_arr.length) {
  1030. matches.push(i - j);
  1031. // Next we have to update `j` because we want to save some time
  1032. // instead of updating to j = 0 , we can jump to the last character of the longest prefix well known so far.
  1033. // j-1 means the last character of `P` because j is actually `P.length`
  1034. // e.g.
  1035. // S = a b a b d e
  1036. // P = `a b`a b
  1037. // we will jump to `a b` and we will compare d and a in the next iteration
  1038. // a b a b `d` e
  1039. // a b `a` b
  1040. j = prefixes[j - 1];
  1041. }
  1042. // Case 3.
  1043. // S[i] != P[j] There's a mismatch!
  1044. else if (arr[i] !== sub_arr[j]) {
  1045. // if we have found at least a character in common, do the same thing as in case 2
  1046. if (j !== 0) {
  1047. j = prefixes[j - 1];
  1048. } else {
  1049. // otherwise, j = 0, and we can move to the next character S[i+1]
  1050. i++;
  1051. }
  1052. }
  1053. }
  1054.  
  1055. return matches;
  1056. },
  1057.  
  1058. /**
  1059. * 用文件头切断文件集合体
  1060. * @param {Uint8Array} bytes
  1061. * @param {Uint8Array} head 默认 null,即使用 data 前 8 字节
  1062. * @returns {Array<Uint8Array>}
  1063. */
  1064. split_files_by_head: function(bytes, head=null) {
  1065. const sub = bytes.subarray || bytes.slice;
  1066. head = head || sub.call(bytes, 0, 8);
  1067. const indexes = this.kmp_matching(bytes, head);
  1068. const size = indexes.length;
  1069. indexes.push(bytes.length);
  1070.  
  1071. const parts = new Array(size);
  1072. for (let i = 0; i < size; i++) {
  1073. parts[i] = sub.call(bytes, indexes[i], indexes[i+1]);
  1074. }
  1075. // 返回结果数组
  1076. return parts;
  1077. },
  1078.  
  1079. /**
  1080. * 函数装饰器:仅执行一次 func
  1081. */
  1082. once: function(fn) {
  1083. let used = false;
  1084. return function() {
  1085. if (!used) {
  1086. used = true;
  1087. return fn();
  1088. }
  1089. }
  1090. },
  1091.  
  1092. /**
  1093. * 返回一个包含计数器的迭代器, 其每次迭代值为 [index, value]
  1094. * @param {Iterable} iterable
  1095. * @returns
  1096. */
  1097. enumerate: function* (iterable) {
  1098. let i = 0;
  1099. for (let value of iterable) {
  1100. yield [i, value];
  1101. i++;
  1102. }
  1103. },
  1104.  
  1105. /**
  1106. * 同步的迭代若干可迭代对象
  1107. * @param {...Iterable} iterables
  1108. * @returns
  1109. */
  1110. zip: function* (...iterables) {
  1111. // 强制转为迭代器
  1112. let iterators = iterables.map(
  1113. iterable => iterable[Symbol.iterator]()
  1114. );
  1115.  
  1116. // 逐次迭代
  1117. while (true) {
  1118. let [done, values] = base.getAllValus(iterators);
  1119. if (done) {
  1120. return;
  1121. }
  1122. if (values.length === 1) {
  1123. yield values[0];
  1124. } else {
  1125. yield values;
  1126. }
  1127. }
  1128. },
  1129.  
  1130. /**
  1131. * 返回指定范围整数生成器
  1132. * @param {number} end 如果只提供 end, 则返回 [0, end)
  1133. * @param {number} end2 如果同时提供 end2, 则返回 [end, end2)
  1134. * @param {number} step 步长, 可以为负数,不能为 0
  1135. * @returns
  1136. */
  1137. range: function*(end, end2=null, step=1) {
  1138. // 参数合法性校验
  1139. if (step === 0) {
  1140. throw new RangeError("step can't be zero");
  1141. }
  1142. const len = end2 - end;
  1143. if (end2 && len && step && (len * step < 0)) {
  1144. throw new RangeError(`[${end}, ${end2}) with step ${step} is invalid`);
  1145. }
  1146.  
  1147. // 生成范围
  1148. end2 = end2 === null ? 0 : end2;
  1149. let [small, big] = [end, end2].sort((a, b) => a - b);
  1150. // 开始迭代
  1151. if (step > 0) {
  1152. for (let i = small; i < big; i += step) {
  1153. yield i;
  1154. }
  1155. } else {
  1156. for (let i = big; i > small; i += step) {
  1157. yield i;
  1158. }
  1159. } },
  1160.  
  1161. /**
  1162. * 获取整个文档的全部css样式
  1163. * @returns {string} css text
  1164. */
  1165. get_all_styles: function() {
  1166. let styles = [];
  1167. for (let sheet of document.styleSheets) {
  1168. let rules;
  1169. try {
  1170. rules = sheet.cssRules;
  1171. } catch(e) {
  1172. if (!(e instanceof DOMException)) {
  1173. console.error(e);
  1174. }
  1175. continue;
  1176. }
  1177.  
  1178. for (let rule of rules) {
  1179. styles.push(rule.cssText);
  1180. }
  1181. }
  1182. return styles.join("\n\n");
  1183. },
  1184.  
  1185. /**
  1186. * 复制text到剪贴板
  1187. * @param {string} text
  1188. * @returns
  1189. */
  1190. copy_text: function(text) {
  1191. // 输出到控制台和剪贴板
  1192. console.log(
  1193. text.length > 20 ?
  1194. text.slice(0, 21) + "..." : text
  1195. );
  1196. if (!navigator.clipboard) {
  1197. base.oldCopy(text);
  1198. return;
  1199. }
  1200. navigator.clipboard
  1201. .writeText(text)
  1202. .catch(_ => base.oldCopy(text));
  1203. },
  1204.  
  1205. /**
  1206. * 复制媒体到剪贴板
  1207. * @param {Blob} blob
  1208. */
  1209. copy: async function(blob) {
  1210. const data = [new ClipboardItem({ [blob.type]: blob })];
  1211. try {
  1212. await navigator.clipboard.write(data);
  1213. console.log(`${blob.type} 成功复制到剪贴板`);
  1214. } catch (err) {
  1215. console.error(err.name, err.message);
  1216. }
  1217. },
  1218.  
  1219. /**
  1220. * 创建并下载文件
  1221. * @param {string} file_name 文件名
  1222. * @param {ArrayBuffer | ArrayBufferView | Blob | string} content 内容
  1223. * @param {string} type 媒体类型,需要符合 MIME 标准
  1224. */
  1225. save: function(file_name, content, type="") {
  1226. const blob = new Blob(
  1227. [content], { type }
  1228. );
  1229. const size = (blob.size / 1024).toFixed(1);
  1230. console.log(`blob saved, size: ${size} kb, type: ${blob.type}`, blob);
  1231.  
  1232. const url = URL.createObjectURL(blob);
  1233. const a = document.createElement("a");
  1234. a.download = file_name || "未命名文件";
  1235. a.href = url;
  1236. a.click();
  1237. URL.revokeObjectURL(url);
  1238. },
  1239.  
  1240. /**
  1241. * 显示/隐藏按钮区
  1242. */
  1243. toggle_box: function() {
  1244. let sec = wk$(".wk-box")[0];
  1245. if (sec.style.display === "none") {
  1246. sec.style.display = "block";
  1247. return;
  1248. }
  1249. sec.style.display = "none";
  1250. },
  1251.  
  1252. /**
  1253. * 异步地睡眠 delay 毫秒, 可选 max_delay 控制波动范围
  1254. * @param {number} delay 等待毫秒
  1255. * @param {number} max_delay 最大等待毫秒, 默认为null
  1256. * @returns
  1257. */
  1258. sleep: async function(delay, max_delay=null) {
  1259. max_delay = max_delay === null ? delay : max_delay;
  1260. delay = delay + (max_delay - delay) * Math.random();
  1261. return new Promise(resolve => setTimeout(resolve, delay));
  1262. },
  1263.  
  1264. /**
  1265. * 允许打印页面
  1266. */
  1267. allow_print: function() {
  1268. const style = document.createElement("style");
  1269. style.innerHTML = `
  1270. @media print {
  1271. body { display: block; }
  1272. }`;
  1273. document.head.append(style);
  1274. },
  1275.  
  1276. /**
  1277. * 取得get参数key对应的value
  1278. * @param {string} key
  1279. * @returns {string} value
  1280. */
  1281. get_param: function(key) {
  1282. return new URL(location.href).searchParams.get(key);
  1283. },
  1284.  
  1285. /**
  1286. * 求main_set去除cut_set后的set
  1287. * @param {Iterable} main_set
  1288. * @param {Iterable} cut_set
  1289. * @returns 差集
  1290. */
  1291. diff: function(main_set, cut_set) {
  1292. const _diff = new Set(main_set);
  1293. for (let elem of cut_set) {
  1294. _diff.delete(elem);
  1295. }
  1296. return _diff;
  1297. },
  1298.  
  1299. /**
  1300. * 增强按钮(默认为蓝色按钮:展开文档)的点击效果
  1301. * @param {string} i 按钮序号
  1302. */
  1303. enhance_click: async function(i) {
  1304. let btn = this.btn(i);
  1305. const style = btn.getAttribute("style") || "";
  1306. // 变黑缩小
  1307. btn.setAttribute(
  1308. "style",
  1309. style + "color: black; font-weight: normal;"
  1310. );
  1311. await utils.sleep(500);
  1312. btn = this.btn(i);
  1313. // 复原加粗
  1314. btn.setAttribute("style", style);
  1315. },
  1316.  
  1317. /**
  1318. * 绑定事件处理函数到指定按钮,返回实际添加的事件处理函数
  1319. * @param {(event: PointerEvent) => void} listener click监听器,第一个参数是按钮的引用
  1320. * @param {number} i 按钮序号
  1321. * @param {string} new_text 按钮的新文本,为null则不替换
  1322. * @returns {Function} 事件处理函数
  1323. */
  1324. onclick: function(listener, i, new_text=null) {
  1325. const btn = this.btn(i);
  1326.  
  1327. // 如果需要,替换按钮内文本
  1328. if (new_text) {
  1329. btn.textContent = new_text;
  1330. }
  1331. // const _listener = async () => {
  1332. // const btn = this.btn(i);
  1333. // await listener(btn);
  1334. // await this.enhance_click(i);
  1335. // btn.textContent = new_text;
  1336. // };
  1337.  
  1338. // 绑定事件,添加到页面上
  1339. btn.onclick = listener;
  1340. return listener;
  1341. },
  1342.  
  1343. /**
  1344. * 返回第 index 个按钮引用
  1345. * @param {number} i
  1346. * @returns {HTMLButtonElement}
  1347. */
  1348. btn: function(i) {
  1349. return wk$(`.wk-box [class="btn-${i}"]`)[0];
  1350. },
  1351.  
  1352. /**
  1353. * 强制隐藏元素
  1354. * @param {string | Array<HTMLElement>} selector_or_elems
  1355. */
  1356. force_hide: function(selector_or_elems) {
  1357. const cls = "force-hide";
  1358. const elems = selector_or_elems instanceof Array ?
  1359. selector_or_elems : wk$(selector_or_elems);
  1360.  
  1361. elems.forEach(elem => {
  1362. elem.classList.add(cls);
  1363. });
  1364.  
  1365. // 判断css样式是否已经存在
  1366. let style = wk$(`style.${cls}`)[0];
  1367. // 如果已经存在,则无须重复创建
  1368. if (style) {
  1369. return;
  1370. }
  1371. // 否则创建
  1372. style = document.createElement("style");
  1373. style.innerHTML = `style.${cls} {
  1374. visibility: hidden !important;
  1375. display: none !important;
  1376. }`;
  1377. document.head.append(style);
  1378. },
  1379.  
  1380. /**
  1381. * 等待直到元素可见。最多等待5秒。
  1382. * @param {HTMLElement} elem 一个元素
  1383. * @returns {Promise<HTMLElement>} elem
  1384. */
  1385. until_visible: async function(elem) {
  1386. let [max, i] = [25, 0];
  1387. let style = getComputedStyle(elem);
  1388. // 如果不可见就等待0.2秒/轮
  1389. while (i <= max &&
  1390. (style.display === "none" ||
  1391. style.visibility !== "hidden")
  1392. ) {
  1393. i++;
  1394. style = getComputedStyle(elem);
  1395. await this.sleep(200);
  1396. }
  1397. return elem;
  1398. },
  1399.  
  1400. /**
  1401. * 等待直到函数返回true
  1402. * @param {Function} isReady 判断条件达成与否的函数
  1403. * @param {number} timeout 最大等待秒数, 默认5000毫秒
  1404. */
  1405. wait_until: async function(isReady, timeout=5000) {
  1406. const gap = 200;
  1407. let chances = parseInt(timeout / gap);
  1408. chances = chances < 1 ? 1 : chances;
  1409. while (! await isReady()) {
  1410. await this.sleep(200);
  1411. chances -= 1;
  1412. if (!chances) {
  1413. break;
  1414. }
  1415. }
  1416. },
  1417.  
  1418. /**
  1419. * 隐藏按钮,打印页面,显示按钮
  1420. */
  1421. print_page: function() {
  1422. // 隐藏按钮,然后打印页面
  1423. this.toggle_box();
  1424. setTimeout(print, 500);
  1425. setTimeout(this.toggle_box, 1000);
  1426. },
  1427.  
  1428. /**
  1429. * 切换按钮显示/隐藏状态
  1430. * @param {number} i 按钮序号
  1431. * @returns 按钮元素的引用
  1432. */
  1433. toggle_btn: function(i) {
  1434. const btn = this.btn(i);
  1435. const display = getComputedStyle(btn).display;
  1436. if (display === "none") {
  1437. btn.style.display = "block";
  1438. } else {
  1439. btn.style.display = "none";
  1440. }
  1441. return btn;
  1442. },
  1443.  
  1444. /**
  1445. * 用input框跳转到对应页码
  1446. * @param {HTMLInputElement} input 当前页码
  1447. * @param {string | number} page_num 目标页码
  1448. * @param {string} type 键盘事件类型:"keyup" | "keypress" | "keydown"
  1449. */
  1450. to_page: function(input, page_num, type) {
  1451. // 设置跳转页码为目标页码
  1452. input.value = `${page_num}`;
  1453. // 模拟回车事件来跳转
  1454. const enter = new KeyboardEvent(type, {
  1455. bubbles: true,
  1456. cancelable: true,
  1457. keyCode: 13
  1458. });
  1459. input.dispatchEvent(enter);
  1460. },
  1461.  
  1462. /**
  1463. * 判断给定的url是否与当前页面同源
  1464. * @param {string} url
  1465. * @returns {boolean}
  1466. */
  1467. is_same_origin: function(url) {
  1468. url = new URL(url);
  1469. if (url.protocol === "data:") {
  1470. return true;
  1471. }
  1472. if (location.protocol === url.protocol
  1473. && location.host === url.host
  1474. && location.port === url.port
  1475. ) {
  1476. return true;
  1477. }
  1478. return false;
  1479. },
  1480.  
  1481. /**
  1482. * 在新标签页打开链接,如果提供文件名则下载
  1483. * @param {string} url
  1484. * @param {string} fname 下载文件的名称,默认为空,代表不下载
  1485. */
  1486. open_in_new_tab: function(url, fname="") {
  1487. const a = document.createElement("a");
  1488. a.href = url;
  1489. a.target = "_blank";
  1490. if (fname && this.is_same_origin(url)) {
  1491. a.download = fname;
  1492. }
  1493. a.click();
  1494. },
  1495.  
  1496. /**
  1497. * 用try移除元素
  1498. * @param {HTMLElement | string} elem_or_selector
  1499. */
  1500. remove: function(elem_or_selector) {
  1501. try {
  1502. const cls = this.classof(elem_or_selector);
  1503. if (cls === "String") {
  1504. wk$(elem_or_selector).forEach(
  1505. elem => elem.remove()
  1506. );
  1507. }
  1508. else if (cls.endsWith("Element")) {
  1509. elem_or_selector.remove();
  1510. }
  1511. } catch (e) {
  1512. console.error(e);
  1513. }
  1514. },
  1515.  
  1516. /**
  1517. * 用try移除若干元素
  1518. * @param {Iterable<HTMLElement>} elements 要移除的元素列表
  1519. */
  1520. remove_multi: function(elements) {
  1521. for (const elem of elements) {
  1522. this.remove(elem);
  1523. }
  1524. },
  1525.  
  1526. /**
  1527. * 等待全部任务落定后返回值的列表
  1528. * @param {Array<Promise>} tasks
  1529. * @returns {Promise<Array>} values
  1530. */
  1531. gather: async function(tasks) {
  1532. const results = await Promise.allSettled(tasks);
  1533. return results
  1534. .filter(result => result.value)
  1535. .map(result => result.value);
  1536. },
  1537.  
  1538. /**
  1539. * html元素列表转为canvas列表
  1540. * @param {Array<HTMLElement>} elements
  1541. * @returns {Promise<Array<HTMLCanvasElement>>}
  1542. */
  1543. elems_to_canvases: async function(elements) {
  1544. if (!globalThis.html2canvas) {
  1545. await this.load_web_script(
  1546. "https://cdn.staticfile.org/html2canvas/1.4.1/html2canvas.min.js"
  1547. );
  1548. }
  1549.  
  1550. // 如果是空列表, 则抛出异常
  1551. if (elements.length === 0) {
  1552. throw new Error("htmlToCanvases 未得到任何html元素");
  1553. }
  1554.  
  1555. return this.gather(
  1556. elements.map(html2canvas)
  1557. );
  1558. },
  1559.  
  1560. /**
  1561. * 将html元素转为canvas再合并到pdf中,最后下载pdf
  1562. * @param {Array<HTMLElement>} elements 元素列表
  1563. * @param {string} title 文档标题
  1564. */
  1565. elems_to_pdf: async function(elements, title="文档") {
  1566. // 如果是空元素列表,终止函数
  1567. const canvases = await this.elems_to_canvases(elements);
  1568. // 控制台检查结果
  1569. console.log("生成的canvas元素如下:");
  1570. console.log(canvases);
  1571. // 合并为PDF
  1572. this.imgs_to_pdf(canvases, title);
  1573. },
  1574.  
  1575. /**
  1576. * 使用xhr异步GET请求目标url,返回响应体blob
  1577. * @param {string} url
  1578. * @returns {Promise<Blob>} blob
  1579. */
  1580. xhr_get_blob: async function(url) {
  1581. const xhr = new XMLHttpRequest();
  1582. xhr.open("GET", url);
  1583. xhr.responseType = "blob";
  1584. return new Promise((resolve, reject) => {
  1585. xhr.onload = () => {
  1586. const code = xhr.status;
  1587. if (code >= 200 && code <= 299) {
  1588. resolve(xhr.response);
  1589. }
  1590. else {
  1591. reject(new Error(`Network Error: ${code}`));
  1592. }
  1593. };
  1594. xhr.send();
  1595. });
  1596. },
  1597.  
  1598. /**
  1599. * 加载CDN脚本
  1600. * @param {string} url
  1601. */
  1602. load_web_script: async function(url) {
  1603. try {
  1604. // xhr+eval方式
  1605. const blob = await this.xhr_get_blob(url);
  1606. const code = await blob.text();
  1607. Function(code)();
  1608. } catch(e) {
  1609. console.error(e);
  1610. // 嵌入<script>方式
  1611. return new Promise(resolve => {
  1612. const script = document.createElement("script");
  1613. script.src = url;
  1614. script.onload = resolve;
  1615. document.body.append(script);
  1616. });
  1617. }
  1618. },
  1619.  
  1620. /**
  1621. * b64编码字符串转Uint8Array
  1622. * @param {string} sBase64 b64编码的字符串
  1623. * @param {number} nBlockSize 字节数
  1624. * @returns {Uint8Array} arr
  1625. */
  1626. b64_to_bytes: function(sBase64, nBlockSize=1) {
  1627. const
  1628. sB64Enc = sBase64.replace(/[^A-Za-z0-9\+\/]/g, ""), nInLen = sB64Enc.length,
  1629. nOutLen = nBlockSize ? Math.ceil((nInLen * 3 + 1 >>> 2) / nBlockSize) * nBlockSize : nInLen * 3 + 1 >>> 2, aBytes = new Uint8Array(nOutLen);
  1630.  
  1631. for (var nMod3, nMod4, nUint24 = 0, nOutIdx = 0, nInIdx = 0; nInIdx < nInLen; nInIdx++) {
  1632. nMod4 = nInIdx & 3;
  1633. nUint24 |= base.b64ToUint6(sB64Enc.charCodeAt(nInIdx)) << 18 - 6 * nMod4;
  1634. if (nMod4 === 3 || nInLen - nInIdx === 1) {
  1635. for (nMod3 = 0; nMod3 < 3 && nOutIdx < nOutLen; nMod3++, nOutIdx++) {
  1636. aBytes[nOutIdx] = nUint24 >>> (16 >>> nMod3 & 24) & 255;
  1637. }
  1638. nUint24 = 0;
  1639. }
  1640. }
  1641. return aBytes;
  1642. },
  1643.  
  1644. /**
  1645. * canvas转blob
  1646. * @param {HTMLCanvasElement} canvas
  1647. * @param {string} type
  1648. * @returns {Promise<Blob>}
  1649. */
  1650. canvas_to_blob: function(canvas, type="image/png") {
  1651. return new Promise(
  1652. resolve => canvas.toBlob(resolve, type, 1)
  1653. );
  1654. },
  1655.  
  1656. /**
  1657. * 合并blobs到压缩包,然后下载
  1658. * @param {Iterable<Blob>} blobs
  1659. * @param {string} base_name 文件名通用部分,如 page-1.jpg 中的 page
  1660. * @param {string} ext 扩展名,如 jpg
  1661. * @param {string} zip_name 压缩包名称
  1662. * @param {boolean} download 是否下载,可选,默认true,如果不下载则返回压缩包对象
  1663. * @returns {"Promise<JSZip | null>"}
  1664. */
  1665. blobs_to_zip: async function(blobs, base_name, ext, zip_name, download=true) {
  1666. const zip = new window.JSZip();
  1667. // 归档
  1668. for (const [i, blob] of this.enumerate(blobs)) {
  1669. zip.file(`${base_name}-${i+1}.${ext}`, blob, { binary: true });
  1670. }
  1671.  
  1672. // 导出
  1673. if (!download) {
  1674. return zip;
  1675. }
  1676.  
  1677. const zip_blob = await zip.generateAsync({ type: "blob" });
  1678. console.log(zip_blob);
  1679. this.save(`${zip_name}.zip`, zip_blob);
  1680. return null;
  1681. },
  1682.  
  1683. /**
  1684. * 存储所有canvas图形为png到一个压缩包
  1685. * @param {Iterable<HTMLCanvasElement>} canvases canvas元素列表
  1686. * @param {string} title 文档标题
  1687. */
  1688. canvases_to_zip: async function(canvases, title) {
  1689. // canvas元素转为png图像
  1690. // 所有png合并为一个zip压缩包
  1691. const tasks = [];
  1692. for (let canvas of canvases) {
  1693. tasks.push(this.canvas_to_blob(canvas));
  1694. }
  1695. const blobs = await this.gather(tasks);
  1696. this.blobs_to_zip(blobs, "page", "png", title);
  1697. },
  1698.  
  1699.  
  1700.  
  1701. /**
  1702. * 合并图像并导出PDF
  1703. * @param {Iterable<HTMLCanvasElement | Uint8Array | HTMLImageElement>} imgs 图像元素列表
  1704. * @param {string} title 文档标题
  1705. * @param {number} width (可选)页面宽度 默认 0
  1706. * @param {number} height (可选)页面高度 默认 0
  1707. * @param {boolean} blob (可选)是否返回 blob 默认 false
  1708. */
  1709. imgs_to_pdf: async function(imgs, title, width = 0, height = 0, blob=false) {
  1710. imgs = Array.from(imgs);
  1711. if (imgs.length === 0) {
  1712. this.raise("没有任何图像用于合并为PDF");
  1713. }
  1714.  
  1715. // 先获取第一个canvas用于判断竖向还是横向,以及得到页面长宽
  1716. const first = imgs[0];
  1717.  
  1718. // 如果没有手动指定canvas的长宽,则自动检测
  1719. if (!width && !height) {
  1720. // 如果是字节数组
  1721. if (first instanceof Uint8Array) {
  1722. const cover = await createImageBitmap(
  1723. new Blob([first])
  1724. );
  1725. [width, height] = [cover.width, cover.height];
  1726.  
  1727. // 如果是画布或图像元素
  1728. } else if (
  1729. first instanceof HTMLCanvasElement ||
  1730. first instanceof HTMLImageElement
  1731. ) {
  1732. if (first.width && parseInt(first.width) && parseInt(first.height)) {
  1733. [width, height] = [first.width, first.height];
  1734. } else {
  1735. const
  1736. width_str = first.style.width.replace(/(px)|(rem)|(em)/, ""),
  1737. height_str = first.style.height.replace(/(px)|(rem)|(em)/, "");
  1738. width = parseInt(width_str);
  1739. height = parseInt(height_str);
  1740. }
  1741. } else {
  1742. // 其他未知类型
  1743. throw TypeError("不能处理的画布元素类型:" + this.classof(first));
  1744. }
  1745. }
  1746. console.log(`canvas数据:宽: ${width}px,高: ${height}px`);
  1747. // 如果文档第一页的宽比长更大,则landscape,否则portrait
  1748. const orientation = width > height ? 'l' : 'p';
  1749. // jsPDF的第三个参数为format,当自定义时,参数为数字数组。
  1750. const pdf = new jspdf.jsPDF(orientation, 'px', [height, width]);
  1751.  
  1752. const last = imgs.pop();
  1753. const self = this;
  1754. // 保存每一页文档到每一页pdf
  1755. imgs.forEach((canvas, i) => {
  1756. pdf.addImage(canvas, 'png', 0, 0, width, height);
  1757. pdf.addPage();
  1758. self?.update_popup(`PDF 已经绘制 ${i + 1} 页`);
  1759. });
  1760. // 添加尾页
  1761. pdf.addImage(last, 'png', 0, 0, width, height);
  1762. // 导出文件
  1763. if (blob) {
  1764. return pdf.output("blob");
  1765. }
  1766. pdf.save(`${title}.pdf`);
  1767. },
  1768.  
  1769. /**
  1770. * imageBitMap转canvas
  1771. * @param {ImageBitmap} bmp
  1772. * @returns {HTMLCanvasElement} canvas
  1773. */
  1774. bmp_to_canvas: function(bmp) {
  1775. const canvas = document.createElement("canvas");
  1776. canvas.height = bmp.height;
  1777. canvas.width = bmp.width;
  1778. const ctx = canvas.getContext("bitmaprenderer");
  1779. ctx.transferFromImageBitmap(bmp);
  1780. return canvas;
  1781. },
  1782.  
  1783. /**
  1784. * 导出图片链接
  1785. * @param {Iterable<string>} urls
  1786. */
  1787. save_urls: function(urls) {
  1788. const _urls = Array
  1789. .from(urls)
  1790. .map((url) => {
  1791. const _url = url.trim();
  1792. if (url.startsWith("//"))
  1793. return "https:" + _url;
  1794. return _url;
  1795. })
  1796. .filter(url => url);
  1797.  
  1798. this.save("urls.csv", _urls.join("\n"), "text/csv");
  1799. },
  1800.  
  1801. /**
  1802. * 图片blobs合并并导出为单个PDF
  1803. * @param {Array<Blob>} blobs
  1804. * @param {string} title (可选)文档名称, 不含后缀, 默认为"文档"
  1805. * @param {boolean} filter (可选)是否过滤 type 不以 "image/" 开头的 blob; 默认为 true
  1806. * @param {boolean} blob (可选)是否返回 blob,默认 false
  1807. */
  1808. img_blobs_to_pdf: async function(blobs, title="文档", filter=true, blob=false) {
  1809. // 格式转换:img blob -> bmp
  1810. let tasks = blobs;
  1811. if (filter) {
  1812. tasks = blobs.filter(
  1813. blob => blob.type.startsWith("image/")
  1814. );
  1815. }
  1816. tasks = await this.gather(
  1817. tasks.map(blob => blob.arrayBuffer())
  1818. );
  1819. tasks = tasks.map(buffer => new Uint8Array(buffer));
  1820. // 导出PDF
  1821. return this.imgs_to_pdf(tasks, title, 0, 0, blob);
  1822. },
  1823.  
  1824. /**
  1825. * 下载可以简单直接请求的图片,合并到 PDF 并导出
  1826. * @param {Iterable<string>} urls 图片链接列表
  1827. * @param {string} title 文档名称
  1828. * @param {number} min_num 如果成功获取的图片数量 < min_num, 则等待 2 秒后重试; 默认 0 不重试
  1829. * @param {boolean} clear 是否在请求完成后清理控制台输出,默认false
  1830. * @param {boolean} blobs 是否返回二进制图片列表,默认 false(即直接导出PDF)
  1831. */
  1832. img_urls_to_pdf: async function(urls, title, min_num=0, clear=false, blobs=false) {
  1833. // 强制转换为迭代器类型
  1834. urls = urls[Symbol.iterator]();
  1835. const first = urls.next().value;
  1836. // 如果不符合同源策略,在打开新标签页
  1837. if (!this.is_same_origin(first)) {
  1838. console.info("URL 不符合同源策略;转为新标签页打开目标网站");
  1839. this.open_in_new_tab((new URL(first)).origin);
  1840. return;
  1841. }
  1842.  
  1843. let tasks, img_blobs, i = 3;
  1844. // 根据请求成功数量判断是否循环
  1845. do {
  1846. i -= 1;
  1847. // 发起请求
  1848. tasks = [this.xhr_get_blob(first)]; // 初始化时加入第一个
  1849. // 然后加入剩余的
  1850. for (const [j, url] of this.enumerate(urls)) {
  1851. tasks.push(this.xhr_get_blob(url));
  1852. this.update_popup(`已经请求 ${j} 张图片`);
  1853. }
  1854. // 接收响应
  1855. img_blobs = (await this.gather(tasks)).filter(
  1856. blob => blob.type.startsWith("image/")
  1857. );
  1858.  
  1859. if (clear) {
  1860. console.clear();
  1861. }
  1862.  
  1863. if (
  1864. min_num
  1865. && img_blobs.length < min_num
  1866. && i
  1867. ) {
  1868. // 下轮行动前冷却
  1869. console.log(`打盹 2 秒`);
  1870. await utils.sleep(2000);
  1871. } else {
  1872. // 结束循环
  1873. break;
  1874. }
  1875. } while (true)
  1876.  
  1877. if (blobs) return img_blobs;
  1878. await this.img_blobs_to_pdf(img_blobs, title, false);
  1879. },
  1880.  
  1881. /**
  1882. * 返回子串个数
  1883. * @param {string} str
  1884. * @param {string} sub
  1885. */
  1886. count_sub_str: function(str, sub) {
  1887. return [...str.matchAll(sub)].length;
  1888. },
  1889.  
  1890. /**
  1891. * 返回按钮区引用
  1892. * @returns
  1893. */
  1894. sec: function() {
  1895. const sec = wk$(".wk-box .btns-sec")[0];
  1896. if (!sec) throw new Error("wk 按钮区找不到");
  1897. return sec;
  1898. },
  1899.  
  1900. _monkey: function() {
  1901. const mky = wk$(".wk-box .wk-fold-btn")[0];
  1902. if (!mky) throw new Error("wk 小猴子找不到");
  1903. return mky;
  1904. },
  1905.  
  1906. /**
  1907. * 折叠按钮区,返回是否转换了状态
  1908. */
  1909. fold_box: function() {
  1910. const sec = this.sec();
  1911. const mky = this._monkey();
  1912. const display = getComputedStyle(sec).display;
  1913. if (display !== "block") return false;
  1914. // 显示 -> 隐藏
  1915. [sec, mky].forEach(
  1916. elem => elem.classList.add("folded")
  1917. );
  1918. return true;
  1919. },
  1920.  
  1921. /**
  1922. * 展开按钮区,返回是否转换了状态
  1923. */
  1924. unfold_box: function() {
  1925. const sec = this.sec();
  1926. const mky = this._monkey();
  1927. const display = getComputedStyle(sec).display;
  1928. if (display === "block") return false;
  1929. // 隐藏 -> 显示
  1930. // 显示 -> 隐藏
  1931. [sec, mky].forEach(
  1932. elem => elem.classList.remove("folded")
  1933. );
  1934. return true;
  1935. },
  1936.  
  1937. /**
  1938. * 运行基于按钮的、显示进度条的函数
  1939. * @param {number} i 按钮序号
  1940. * @param {Function} task 需要等待的耗时函数
  1941. */
  1942. run_with_prog: async function(i, task) {
  1943. const btn = utils.btn(i);
  1944. let new_btn;
  1945.  
  1946. if (!wk$("#wk-popup")[0]) {
  1947. this.add_popup();
  1948. }
  1949.  
  1950. this.fold_box();
  1951. this.toID("wk-popup");
  1952.  
  1953. new_btn = btn.cloneNode(true);
  1954. btn.replaceWith(new_btn);
  1955. this.onclick(
  1956. () => utils.toID("wk-popup"), i, "显示进度"
  1957. );
  1958.  
  1959. try {
  1960. await task();
  1961. } catch(e) {
  1962. console.error(e);
  1963. }
  1964.  
  1965. this.toID("");
  1966. this.unfold_box();
  1967. this.remove_popup();
  1968. new_btn.replaceWith(btn);
  1969. },
  1970.  
  1971. /**
  1972. * 创建5个按钮:展开文档、导出图片、导出PDF、未设定4、未设定5;除第1个外默认均为隐藏
  1973. */
  1974. create_btns: function() {
  1975. // 添加样式
  1976. document.head.insertAdjacentHTML("beforeend", style);
  1977. // 添加按钮区
  1978. document.body.insertAdjacentHTML("beforeend", box);
  1979.  
  1980. // 绑定小猴子按钮回调
  1981. const monkey = wk$(".wk-fold-btn")[0];
  1982. // 隐藏【🙈】,展开【🐵】
  1983. monkey.onclick = () => this.fold_box() || this.unfold_box();
  1984. },
  1985.  
  1986. /**
  1987. * 添加弹窗到 body, 通过 utils.toID("wk-popup") 激发
  1988. */
  1989. add_popup: function() {
  1990. document.body.insertAdjacentHTML("beforeend", popup);
  1991. },
  1992.  
  1993. /**
  1994. * 设置弹窗正文
  1995. * @param {string} text
  1996. */
  1997. update_popup: function(text) {
  1998. const body = wk$(".wk-popup-body")[0];
  1999. if (!body) return;
  2000. body.textContent = text;
  2001. },
  2002.  
  2003. /**
  2004. * 移除弹窗
  2005. */
  2006. remove_popup: function() {
  2007. this.remove(wk$(".wk-popup-container")[0]);
  2008. },
  2009.  
  2010. /**
  2011. * 滚动页面到id位置的元素处
  2012. * @param {string} id
  2013. */
  2014. toID: function(id) {
  2015. location.hash = `#${id}`;
  2016. }
  2017. };
  2018.  
  2019.  
  2020. /**
  2021. * ---------------------------------------------------------------------
  2022. * 绑定使用 this 的函数到 utils,使其均成为绑定方法
  2023. * ---------------------------------------------------------------------
  2024. */
  2025.  
  2026. /**
  2027. * 确保特定外部脚本加载的装饰器
  2028. * @param {string} global_obj_name
  2029. * @param {string} cdn_url
  2030. * @param {Function} func
  2031. * @returns
  2032. */
  2033. function ensure_script_existed(global_obj_name, cdn_url, func) {
  2034. async function inner(...args) {
  2035. if (!window[global_obj_name]) {
  2036. // 根据需要加载依赖
  2037. await utils.load_web_script(cdn_url);
  2038. }
  2039. return func(...args);
  2040. }
  2041. // 存储参数定义
  2042. base.superAssign(inner, func);
  2043. return inner;
  2044. }
  2045.  
  2046.  
  2047. /**
  2048. * 确保引用外部依赖的函数都在调用前加载了依赖
  2049. */
  2050. for (const prop of Object.keys(utils)) {
  2051. // 跳过非函数
  2052. if (
  2053. !(typeof utils[prop] === "function")
  2054. && !`${utils[prop]}`.startsWith("class")
  2055. ) {
  2056. continue;
  2057. }
  2058.  
  2059. // 绑定this到utils
  2060. if (/ this[.[][a-z_]/.test(`${utils[prop]}`)) {
  2061. // 存储参数定义
  2062. const doc = utils.help(utils[prop], false);
  2063. // 绑死this,同时提供 __func__ 来取回原先的函数
  2064. const fn = utils[prop];
  2065. utils[prop] = utils[prop].bind(utils);
  2066. utils[prop].__func__ = fn;
  2067. // 重设参数定义
  2068. utils[prop].__doc__ = doc;
  2069. }
  2070.  
  2071. // 设定 __doc__ 为访问器属性
  2072. const doc_box = [
  2073. utils.help(utils[prop], false)
  2074. ];
  2075. Object.defineProperty(utils[prop], "__doc__", {
  2076. configurable: true,
  2077. enumerable: true,
  2078. get() { return doc_box.join("\n"); },
  2079. set(new_doc) { doc_box.push(new_doc); },
  2080. });
  2081.  
  2082. // 为有外部依赖的函数做包装
  2083. let obj, url;
  2084. const name = prop.toLowerCase();
  2085.  
  2086. if (name.includes("_to_zip")) {
  2087. obj = "JSZip";
  2088. url = "https://cdn.staticfile.org/jszip/3.7.1/jszip.min.js";
  2089.  
  2090. } else if (name.includes("_to_pdf")) {
  2091. obj = "jspdf";
  2092. url = "https://cdn.staticfile.org/jspdf/2.5.1/jspdf.umd.min.js";
  2093.  
  2094. } else {
  2095. continue;
  2096. }
  2097. utils[prop] = ensure_script_existed(obj, url, utils[prop]);
  2098. }
  2099.  
  2100.  
  2101. /**
  2102. * ---------------------------------------------------------------------
  2103. * 为 utils 部分函数绑定更详细的说明
  2104. * ---------------------------------------------------------------------
  2105. */
  2106.  
  2107. utils.b64_to_bytes.__doc__ = `
  2108. /**
  2109. * b64编码字符串转Uint8Array
  2110. * @param {string} sBase64 b64编码的字符串
  2111. * @param {number} nBlockSize 字节数
  2112. * @returns {Uint8Array} arr
  2113. */
  2114. `;
  2115.  
  2116. utils.blobs_to_zip.__doc__ = `
  2117. /**
  2118. * 合并blobs到压缩包,然后下载
  2119. * @param {Iterable<Blob>} blobs
  2120. * @param {string} base_name 文件名通用部分,如 image-1.jpg 中的 image
  2121. * @param {string} ext 扩展名,如 jpg
  2122. * @param {string} zip_name 压缩包名称
  2123. */
  2124. `;
  2125.  
  2126. utils.imgs_to_pdf.__doc__ = `
  2127. /**
  2128. * 合并图像并导出PDF
  2129. * @param {Iterable<HTMLCanvasElement | Uint8Array | HTMLImageElement>} imgs 图像元素列表
  2130. * @param {string} title 文档标题
  2131. * @param {number} width (可选)页面宽度 默认 0
  2132. * @param {number} height (可选)页面高度 默认 0
  2133. * @param {boolean} blob (可选)是否返回 blob 默认 false
  2134. */
  2135. `;
  2136.  
  2137. utils.img_urls_to_pdf.__doc__ = `
  2138. /**
  2139. * 下载可以简单直接请求的图片,合并到 PDF 并导出
  2140. * @param {Iterable<string>} urls 图片链接列表
  2141. * @param {string} title 文档名称
  2142. * @param {number} min_num 如果成功获取的图片数量 < min_num, 则等待 2 秒后重试; 默认 0 不重试
  2143. * @param {boolean} clear 是否在请求完成后清理控制台输出,默认false
  2144. */
  2145. `;
  2146.  
  2147. utils.img_blobs_to_pdf.__doc__ = `
  2148. /**
  2149. * 图片blobs合并并导出为单个PDF
  2150. * @param {Array<Blob>} blobs
  2151. * @param {string} title (可选)文档名称, 不含后缀, 默认为"文档"
  2152. * @param {boolean} filter (可选)是否过滤 type 不以 "image/" 开头的 blob; 默认为 true
  2153. * @param {boolean} blob (可选)是否返回 blob
  2154. */
  2155. `;
  2156.  
  2157.  
  2158. /**
  2159. * ---------------------------------------------------------------------
  2160. * 绑定 utils 成员到 wk$,允许外部轻松调用
  2161. * ---------------------------------------------------------------------
  2162. */
  2163.  
  2164. base.superAssign(wk$, utils);
  2165. console.info("wk: `wk$` 已经挂载到全局");
  2166.  
  2167. /**
  2168. * 展开道客巴巴的文档
  2169. */
  2170. async function readAllDoc88() {
  2171. // 获取“继续阅读”按钮
  2172. let continue_btn = wk$("#continueButton")[0];
  2173. // 如果存在“继续阅读”按钮
  2174. if (continue_btn) {
  2175. // 跳转到文末(等同于展开全文)
  2176. let cur_page = wk$("#pageNumInput")[0];
  2177. // 取得最大页码
  2178. let page_max = cur_page.parentElement.textContent.replace(" / ", "");
  2179. // 跳转到尾页
  2180. utils.to_page(cur_page, page_max, "keypress");
  2181. // 返回顶部
  2182. await utils.sleep(1000);
  2183. utils.to_page(cur_page, "1", "keypress");
  2184. }
  2185. // 文档展开后,显示按钮
  2186. else {
  2187. for (const i of utils.range(1, 6)) {
  2188. utils.toggle_btn(i);
  2189. }
  2190. }
  2191. }
  2192.  
  2193.  
  2194. /**
  2195. * 隐藏选择文字的弹窗
  2196. */
  2197. async function hideSelectPopup() {
  2198. const
  2199. elem = (await wk$$("#left-menu"))[0],
  2200. hide = elem => elem.style.zIndex = -1;
  2201. return utils.until_visible(elem).then(hide);
  2202. }
  2203.  
  2204.  
  2205. /**
  2206. * 初始化任务
  2207. */
  2208. async function initService() {
  2209. // 初始化
  2210. console.log("正在执行初始化任务");
  2211.  
  2212. // 1. 查找复制文字可能的api名称
  2213. const prop = getCopyAPIValue();
  2214. globalThis.doc88JS._apis = Object
  2215. .getOwnPropertyNames(prop)
  2216. .filter(name => {
  2217. if (!name.startsWith("_")) {
  2218. return false;
  2219. }
  2220. if (prop[name] === "") {
  2221. return true;
  2222. }
  2223. });
  2224. // 2. 隐藏选中文字的提示框
  2225. await hideSelectPopup();
  2226. // 3. 隐藏搜索框
  2227. // hideSearchBox();
  2228. // 4. 移除vip复制弹窗
  2229. // hideCopyPopup();
  2230. }
  2231.  
  2232.  
  2233. /**
  2234. * 取得 doc88JS.copy_api 所指向属性的值
  2235. * @returns
  2236. */
  2237. function getCopyAPIValue() {
  2238. let aim = globalThis;
  2239. for (let name of globalThis.doc88JS.copy_api) {
  2240. aim = aim[name];
  2241. }
  2242. return aim;
  2243. }
  2244.  
  2245.  
  2246. /**
  2247. * 返回选中的文字
  2248. * @returns {string}
  2249. */
  2250. function getSelectedText() {
  2251. // 首次复制文字,需要先找出api
  2252. if (globalThis.doc88JS.copy_api.length === 3) {
  2253. // 拼接出路径,得到属性
  2254. let prop = getCopyAPIValue(); // 此时是属性,尚未取得值
  2255.  
  2256. // 查询值
  2257. for (let name of globalThis.doc88JS._apis) {
  2258. let value = prop[name];
  2259. // 值从空字符串变为非空字符串了,确认是目标api名称
  2260. if (typeof value === 'string'
  2261. && value.length > 0
  2262. && !value.match(/\d/) // 开头不能是数字,因为可能是 '1-179-195' 这种值
  2263. ) {
  2264. globalThis.doc88JS.copy_api.push(name);
  2265. break;
  2266. }
  2267. }
  2268. }
  2269. return getCopyAPIValue();
  2270. }
  2271.  
  2272.  
  2273. /**
  2274. * 输出选中的文字到剪贴板和控制台,返回是否复制了文档
  2275. * @returns {boolean} doc_is_copied
  2276. */
  2277. function copySelected() {
  2278. // 尚未选中文字
  2279. if (getComputedStyle(wk$("#left-menu")[0]).display === "none") {
  2280. console.log("尚未选中文字");
  2281. return false;
  2282. }
  2283. // 输出到控制台和剪贴板
  2284. utils.copy_text(getSelectedText());
  2285. return true;
  2286. }
  2287.  
  2288.  
  2289. /**
  2290. * 捕获 ctrl + c 以复制文字
  2291. * @param {KeyboardEvent} e
  2292. * @returns
  2293. */
  2294. function onCtrlC(e) {
  2295. // 判断是否为 ctrl + c
  2296. if (!(e.code === "KeyC" && e.ctrlKey === true)) {
  2297. return;
  2298. }
  2299.  
  2300. // 判断触发间隔
  2301. let now = Date.now();
  2302. // 距离上次小于0.5秒
  2303. if (now - doc88JS.last_copy_time < 500 * 1) {
  2304. doc88JS.last_copy_time = now;
  2305. return;
  2306. }
  2307. // 大于1秒
  2308. // 刷新最近一次触发时间
  2309. doc88JS.last_copy_time = now;
  2310. // 复制文字
  2311. copySelected();
  2312. // if (!copySelected()) return;
  2313. // 停止传播
  2314. e.stopImmediatePropagation();
  2315. e.stopPropagation();
  2316. }
  2317.  
  2318.  
  2319. /**
  2320. * 浏览并加载所有页面
  2321. */
  2322. async function walkThrough$2() {
  2323. // 文档容器
  2324. let container = wk$("#pageContainer")[0];
  2325. container.style.display = "none";
  2326. // 页码
  2327. let page_num = wk$("#pageNumInput")[0];
  2328. // 文末提示
  2329. let tail = wk$("#readEndDiv > p")[0];
  2330. let origin = tail.textContent;
  2331. // 按钮
  2332. wk$('.btns_section > [class*="btn-"]').forEach(
  2333. elem => elem.style.display = "none"
  2334. );
  2335.  
  2336. // 逐页渲染
  2337. let total = parseInt(Config.p_pagecount);
  2338. try {
  2339. for (let i = 1; i <= total; i++) {
  2340. // 前往页码
  2341. GotoPage(i);
  2342. await utils.wait_until(async() => {
  2343. let page = wk$(`#page_${i}`)[0];
  2344. // page无法选中说明有弹窗
  2345. if (!page) {
  2346. // 关闭弹窗,等待,然后递归
  2347. wk$("#ym-window .DOC88Window_close")[0].click();
  2348. await utils.sleep(500);
  2349. walkThrough$2();
  2350. throw new Error("walkThrough 递归完成,终止函数");
  2351. }
  2352. // canvas尚未绘制时width=300
  2353. return page.width !== 300;
  2354. });
  2355. // 凸显页码
  2356. utils.emphasize_text(page_num);
  2357. tail.textContent = `请勿反复点击按钮,耐心等待页面渲染:${i}/${total}`;
  2358. }
  2359. } catch(e) {
  2360. // 捕获退出信号,然后退出
  2361. console.log(e);
  2362. return;
  2363. }
  2364.  
  2365. // 恢复原本显示
  2366. container.style.display = "";
  2367. page_num.style = "";
  2368. tail.textContent = origin;
  2369. // 按钮
  2370. wk$('.btns_section > [class*="btn-"]').forEach(
  2371. elem => elem.style.display = "block"
  2372. );
  2373. wk$(".btns_section > .btn-1")[0].style.display = "none";
  2374. }
  2375.  
  2376.  
  2377. /**
  2378. * 道客巴巴文档下载策略
  2379. */
  2380. async function doc88() {
  2381. // 全局对象
  2382. globalThis.doc88JS = {
  2383. last_copy_time: 0, // 上一次 ctrl + c 的时间戳(毫秒)
  2384. copy_api: ["Core", "Annotation", "api"]
  2385. };
  2386.  
  2387. // 创建脚本启动按钮1、2
  2388. utils.create_btns();
  2389.  
  2390. // 绑定主函数
  2391. let prepare = function() {
  2392. // 获取canvas元素列表
  2393. let node_list = wk$(".inner_page");
  2394. // 获取文档标题
  2395. let title;
  2396. if (wk$(".doctopic h1")[0]) {
  2397. title = wk$(".doctopic h1")[0].title;
  2398. } else {
  2399. title = "文档";
  2400. }
  2401. return [node_list, title];
  2402. };
  2403.  
  2404. // btn_1: 展开文档
  2405. utils.onclick(readAllDoc88, 1);
  2406.  
  2407. // // btn_2: 加载全部页面
  2408. utils.onclick(walkThrough$2, 2, "加载所有页面");
  2409. // btn_3: 导出PDF
  2410. function imgsToPDF() {
  2411. if (confirm("确定每页内容都加载完成了吗?")) {
  2412. utils.run_with_prog(
  2413. 3, () => utils.imgs_to_pdf(...prepare())
  2414. );
  2415. }
  2416. } utils.onclick(imgsToPDF, 3, "导出图片到PDF");
  2417.  
  2418. // btn_4: 导出ZIP
  2419. utils.onclick(() => {
  2420. if (confirm("确定每页内容都加载完成了吗?")) {
  2421. utils.canvases_to_zip(...prepare());
  2422. }
  2423. }, 4, "导出图片到ZIP");
  2424.  
  2425. // btn_5: 复制选中文字
  2426. utils.onclick(btn => {
  2427. if (!copySelected()) {
  2428. btn.textContent = "未选中文字";
  2429. } else {
  2430. btn.textContent = "复制成功!";
  2431. }
  2432. }, 5, "复制选中文字");
  2433.  
  2434. // 为 ctrl + c 添加响应
  2435. window.addEventListener("keydown", onCtrlC, true);
  2436. // 执行一次初始化任务
  2437. window.addEventListener(
  2438. "mousedown", initService, { once: true, capture: true }
  2439. );
  2440. }
  2441.  
  2442. function get_title$1() {
  2443. return document.title.slice(0,-6);
  2444. }
  2445.  
  2446.  
  2447. function save_canvases(type) {
  2448. return () => {
  2449. if (!wk$(".hkswf-content2 canvas").length) {
  2450. alert("当前页面不适用此按钮");
  2451. return;
  2452. }
  2453. if (confirm("页面加载完毕了吗?")) {
  2454. const title = get_title$1();
  2455. const canvases = wk$(".hkswf-content2 canvas");
  2456. let data_to;
  2457.  
  2458. switch (type) {
  2459. case "pdf":
  2460. data_to = utils.imgs_to_pdf;
  2461. break;
  2462.  
  2463. case "zip":
  2464. data_to = utils.canvases_to_zip;
  2465. break;
  2466. default:
  2467. data_to = () => utils.raise(`未知 type: ${type}`);
  2468. break;
  2469. }
  2470. data_to(canvases, title);
  2471. }
  2472. }
  2473. }
  2474.  
  2475.  
  2476. function get_base_url() {
  2477. // https://docimg1.docin.com/docinpic.jsp?file=2179420769&width=1000&sid=bZh4STs-f4NA88IA02INyapgA9Z5X3NN1sGo4WnpquIvk4CyflMk1Oxey1BsO1BG&pageno=2&pcimg=1
  2478. return `https://docimg1.docin.com/docinpic.jsp?` +
  2479. `file=` + location.pathname.match(/p-(\d+)[.]html/)[1] +
  2480. `&width=1000&sid=` + window.readerConfig.flash_param_hzq +
  2481. `&pcimg=1&pageno=`;
  2482. }
  2483.  
  2484.  
  2485. /**
  2486. * 返回总页码
  2487. * @returns {number}
  2488. */
  2489. function get_page_num() {
  2490. return parseInt(
  2491. wk$(".page_num")[0].textContent.slice(1)
  2492. );
  2493. }
  2494.  
  2495. function init_save_imgs() {
  2496. const iframe = document.createElement("iframe");
  2497. iframe.src = "https://docimg1.docin.com/?wk=true";
  2498. iframe.style.display = "none";
  2499. let sock;
  2500.  
  2501. /**
  2502. * @param {MessageEvent} event
  2503. */
  2504. function on_client_msg(event) {
  2505. if (event.data.author !== "wk"
  2506. || event.data.action !== "finish"
  2507. ) return;
  2508. sock.notListen(on_client_msg);
  2509. iframe.remove();
  2510. utils.toggle_btn(1);
  2511. utils.toggle_btn(3);
  2512. }
  2513. /**
  2514. * @param {string} type "pdf" | "zip"
  2515. */
  2516. return (type) => {
  2517. return async function() {
  2518. if (!wk$("[id*=img_] img").length) {
  2519. alert("当前页面不适用此按钮");
  2520. return;
  2521. }
  2522. utils.toggle_btn(1);
  2523. utils.toggle_btn(3);
  2524.  
  2525. document.body.append(iframe);
  2526. await utils.sleep(500);
  2527. sock = new utils.Socket(iframe.contentWindow);
  2528. await sock.connect(false);
  2529. sock.listen(on_client_msg);
  2530. sock.talk({
  2531. author: "wk",
  2532. type,
  2533. title: get_title$1(),
  2534. base_url: get_base_url(),
  2535. max: get_page_num()
  2536. });
  2537. }
  2538. }
  2539. }
  2540.  
  2541.  
  2542. const save_imgs = init_save_imgs();
  2543.  
  2544.  
  2545. async function walk_through() {
  2546. // 隐藏按钮
  2547. utils.toggle_btn(5);
  2548. // 隐藏文档页面
  2549. wk$("#contentcontainer")[0].setAttribute("style", "visibility: hidden;");
  2550.  
  2551. const total = get_page_num();
  2552. const input = wk$("#page_cur")[0];
  2553. for (let i = 1; i <= total; i++) {
  2554. utils.to_page(input, i, "keydown");
  2555. await utils.wait_until(
  2556. () => {
  2557. const page = wk$(`#page_${i}`)[0];
  2558. const contents = wk$.call(page, `.canvas_loaded, img`);
  2559. return contents.length > 0;
  2560. },
  2561. 5000
  2562. );
  2563. }
  2564.  
  2565. // 显示文档页面
  2566. wk$("#contentcontainer")[0].removeAttribute("style");
  2567. }
  2568.  
  2569.  
  2570. function main_page() {
  2571. // 创建脚本启动按钮
  2572. utils.create_btns();
  2573. utils.onclick(
  2574. save_imgs("pdf"), 1, "合并图片为PDF"
  2575. );
  2576. utils.onclick(
  2577. save_canvases("pdf"), 2, "合并画布为PDF"
  2578. );
  2579. utils.toggle_btn(2);
  2580.  
  2581. utils.onclick(
  2582. save_imgs("zip"), 3, "打包图片到ZIP"
  2583. );
  2584. utils.toggle_btn(3);
  2585. utils.onclick(
  2586. save_canvases("zip"), 4, "打包画布到ZIP"
  2587. );
  2588. utils.toggle_btn(4);
  2589.  
  2590. utils.onclick(
  2591. walk_through, 5, "自动浏览页面"
  2592. );
  2593. utils.toggle_btn(5);
  2594. }
  2595.  
  2596.  
  2597.  
  2598. function init_background() {
  2599. const sock = new utils.Socket(window.top);
  2600.  
  2601. /**
  2602. * @param {MessageEvent} event
  2603. */
  2604. async function on_server_msg(event) {
  2605. if (event.data.author !== "wk") return;
  2606. const { title, base_url, max, type } = event.data;
  2607. const urls = Array
  2608. .from(utils.range(1, max + 1))
  2609. .map(i => (base_url + i));
  2610. const imgs = await utils.img_urls_to_pdf(
  2611. urls, title, 0, false, true
  2612. );
  2613. switch (type) {
  2614. case "pdf":
  2615. await utils.img_blobs_to_pdf(imgs, title);
  2616. break;
  2617. case "zip":
  2618. const ext = imgs[0].type ? imgs[0].type.split("/")[1] : "png";
  2619. await utils.blobs_to_zip(
  2620. imgs, "page", ext, title
  2621. );
  2622. break;
  2623.  
  2624. default:
  2625. utils.raise(`未知 type: ${type}`);
  2626. break;
  2627. }
  2628.  
  2629. sock.talk({
  2630. author: "wk",
  2631. action: "finish"
  2632. });
  2633. sock.notListen(on_server_msg);
  2634. }
  2635. return async function() {
  2636. sock.listen(on_server_msg);
  2637. await sock.connect(true);
  2638. }
  2639. }
  2640.  
  2641.  
  2642. const background = init_background();
  2643.  
  2644.  
  2645. /**
  2646. * 豆丁文档下载策略
  2647. */
  2648. function docin() {
  2649. const host = location.hostname;
  2650. switch (host) {
  2651. case "jz.docin.com":
  2652. case "www.docin.com":
  2653. main_page();
  2654. break;
  2655.  
  2656. case "docimg1.docin.com":
  2657. background();
  2658. break;
  2659. default:
  2660. console.log(`未知域名: ${host}`);
  2661. break;
  2662. }
  2663. }
  2664.  
  2665. function jumpToHost() {
  2666. // https://swf.ishare.down.sina.com.cn/1DrH4Qt2cvKd.jpg?ssig=DUf5x%2BXnKU&Expires=1673867307&KID=sina,ishare&range={}-{}
  2667. let url = wk$(".data-detail img, .data-detail embed")[0].src;
  2668. if (!url) {
  2669. alert("找不到图片元素");
  2670. return;
  2671. }
  2672.  
  2673. let url_obj = new URL(url);
  2674. let path = url_obj.pathname.slice(1);
  2675. let query = url_obj.search.slice(1).split("&range")[0];
  2676. let title = document.title.split(" - ")[0];
  2677. let target = `${url_obj.protocol}//${url_obj.host}?path=${path}&fname=${title}&${query}`;
  2678. // https://swf.ishare.down.sina.com.cn/
  2679. globalThis.open(target, "hostage");
  2680. // 然后在跳板页面发起对图片的请求
  2681. }
  2682.  
  2683.  
  2684. /**
  2685. * 爱问文库下载跳转策略
  2686. */
  2687. function ishare() {
  2688. // 创建按钮区
  2689. utils.create_btns();
  2690.  
  2691. // btn_1: 识别文档类型 -> 导出PDF
  2692. utils.onclick(jumpToHost, 1, "到下载页面");
  2693. // btn_2: 不支持爱问办公
  2694. utils.onclick(() => null, 2, "不支持爱问办公");
  2695. // utils.toggleBtnStatus(4);
  2696. }
  2697.  
  2698. /**
  2699. * 返回包含对于数量svg元素的html元素
  2700. * @param {string} data
  2701. * @returns {HTMLDivElement} article
  2702. */
  2703. function _createDiv(data) {
  2704. let num = utils.count_sub_str(data, data.slice(0, 10));
  2705. let article = document.createElement("div");
  2706. article.id = "article";
  2707. article.innerHTML = `
  2708. <style class="wk-settings">
  2709. body {
  2710. margin: 0px;
  2711. width: 100%;
  2712. background-color: rgb(95,99,104);
  2713. }
  2714. #article {
  2715. width: 100%;
  2716. display: flex;
  2717. flex-direction: row;
  2718. justify-content: space-around;
  2719. }
  2720. #root-box {
  2721. display: flex;
  2722. flex-direction: column;
  2723. background-color: white;
  2724. padding: 0 2em;
  2725. }
  2726. .gap {
  2727. height: 50px;
  2728. width: 100%;
  2729. background-color: transparent;
  2730. }
  2731. </style>
  2732. <div id="root-box">
  2733. ${
  2734. `<object class="svg-box"></object>
  2735. <div class="gap"></div>`.repeat(num)
  2736. }
  2737. `;
  2738. // 移除最后一个多出的gap
  2739. Array.from(article.querySelectorAll(".gap")).at(-1).remove();
  2740. return article;
  2741. }
  2742.  
  2743.  
  2744. function setGap(height) {
  2745. let style = wk$(".wk-settings")[0].innerHTML;
  2746. wk$(".wk-settings")[0].innerHTML = style.replace(
  2747. /[.]gap.*?{.*?height:.+?;/s,
  2748. `.gap { height: ${parseInt(height)}px;`
  2749. );
  2750. }
  2751.  
  2752.  
  2753. function setGapGUI() {
  2754. let now = getComputedStyle(wk$(".gap")[0]).height;
  2755. let new_h = prompt(`当前间距:${now}\n请输入新间距:`);
  2756. if (new_h) {
  2757. setGap(new_h);
  2758. }
  2759. }
  2760.  
  2761.  
  2762. function getSVGtext(data) {
  2763. let div = document.createElement("div");
  2764. div.innerHTML = data;
  2765. return div.textContent;
  2766. }
  2767.  
  2768.  
  2769. function toDisplayMode1() {
  2770. let content = globalThis["ishareJS"].content_1;
  2771. if (!content) {
  2772. content = globalThis["ishareJS"].text
  2773. .replace(/\n{2,}/g, "<hr>")
  2774. .replace(/\n/g, "<br>")
  2775. .replace(/\s/g, "&nbsp;")
  2776. .replace(/([a-z])([A-Z])/g, "$1 $2"); // 英文简单分词
  2777.  
  2778. globalThis["ishareJS"].content_1 = content;
  2779. }
  2780.  
  2781. wk$("#root-box")[0].innerHTML = content;
  2782. }
  2783.  
  2784.  
  2785. function toDisplayMode2() {
  2786. let content = globalThis["ishareJS"].content_2;
  2787. if (!content) {
  2788. content = globalThis["ishareJS"].text
  2789. .replace(/\n{2,}/g, "<hr>")
  2790. .replace(/\n/g, "")
  2791. .replace(/\s/g, "&nbsp;")
  2792. .replace(/([a-z])([A-Z])/g, "$1 $2")
  2793. .split("<hr>")
  2794. .map(paragraph => `<p>${paragraph}</p>`)
  2795. .join("");
  2796. globalThis["ishareJS"].content_2 = content;
  2797. wk$(".wk-settings")[0].innerHTML += `
  2798. #root-box > p {
  2799. text-indent: 2em;
  2800. width: 40em;
  2801. word-break: break-word;
  2802. }
  2803. `;
  2804. }
  2805.  
  2806. wk$("#root-box")[0].innerHTML = content;
  2807. }
  2808.  
  2809.  
  2810. function changeDisplayModeWrapper() {
  2811. let flag = true;
  2812.  
  2813. function inner() {
  2814. if (flag) {
  2815. toDisplayMode1();
  2816. } else {
  2817. toDisplayMode2();
  2818. }
  2819. flag = !flag;
  2820. }
  2821. return inner;
  2822. }
  2823.  
  2824.  
  2825. function handleSVGtext() {
  2826. globalThis["ishareJS"].text = getSVGtext(
  2827. globalThis["ishareJS"].data
  2828. );
  2829.  
  2830. let change = changeDisplayModeWrapper();
  2831. utils.onclick(change, 4, "切换显示模式");
  2832.  
  2833. utils.toggle_btn(2);
  2834. utils.toggle_btn(3);
  2835. utils.toggle_btn(4);
  2836. change();
  2837. }
  2838.  
  2839.  
  2840. /**
  2841. * 处理svg的url
  2842. * @param {string} svg_url
  2843. */
  2844. async function handleSVGurl(svg_url) {
  2845. let resp = await fetch(svg_url);
  2846. let data = await resp.text();
  2847. globalThis["ishareJS"].data = data;
  2848.  
  2849. let sep = data.slice(0, 10);
  2850. let svg_texts = data
  2851. .split(sep)
  2852. .slice(1)
  2853. .map(svg_text => sep + svg_text);
  2854.  
  2855. console.log(`共 ${svg_texts.length} 张图片`);
  2856.  
  2857. let article = _createDiv(data);
  2858. let boxes = article.querySelectorAll(".svg-box");
  2859. boxes.forEach((obj, i) => {
  2860. let blob = new Blob([svg_texts[i]], {type: "image/svg+xml"});
  2861. let url = URL.createObjectURL(blob);
  2862. obj.data = url;
  2863. URL.revokeObjectURL(blob);
  2864. });
  2865.  
  2866. let body = wk$("body")[0];
  2867. body.innerHTML = "";
  2868. body.appendChild(article);
  2869.  
  2870. utils.create_btns();
  2871. utils.onclick(utils.print_page, 1, "打印页面到PDF");
  2872. utils.onclick(setGapGUI, 2, "重设页间距");
  2873. utils.onclick(handleSVGtext, 3, "显示空白点我");
  2874.  
  2875. utils.toggle_btn(2);
  2876. utils.toggle_btn(3);
  2877. }
  2878.  
  2879.  
  2880. /**
  2881. * 取得图片下载地址
  2882. * @param {string} fname
  2883. * @param {string} path
  2884. * @returns
  2885. */
  2886. function getImgUrl(fname, path) {
  2887. if (!fname) {
  2888. throw new Error("URL Param `fname` does not exist.");
  2889. }
  2890. return location.href
  2891. .replace(/[?].+?&ssig/, "?ssig")
  2892. .replace("?", path + "?");
  2893. }
  2894.  
  2895.  
  2896. /**
  2897. * 下载整个图片包
  2898. * @param {string} img_url
  2899. * @returns
  2900. */
  2901. async function getData(img_url) {
  2902. let resp = await fetch(img_url);
  2903. // window.data = await resp.blob();
  2904. // throw Error("stop");
  2905. let buffer = await resp.arrayBuffer();
  2906. return new Uint8Array(buffer);
  2907. }
  2908.  
  2909.  
  2910. /**
  2911. * 分切图片包为若干图片
  2912. * @param {Uint8Array} data 多张图片合集数据包
  2913. * @returns {Array<Uint8Array>} 图片列表
  2914. */
  2915. function parseData(data) {
  2916. // 判断图像类型/拿到文件头
  2917. let head = data.slice(0, 8);
  2918. return utils.split_files_by_head(data, head);
  2919. }
  2920.  
  2921.  
  2922. /**
  2923. * 图像Uint8数组列表合并然后导出PDF
  2924. * @param {string} fname
  2925. * @param {Array<Uint8Array>} img_data_list
  2926. */
  2927. async function imgDataArrsToPDF(fname, img_data_list) {
  2928. return utils.imgs_to_pdf(
  2929. img_data_list,
  2930. fname
  2931. );
  2932. }
  2933.  
  2934.  
  2935. /**
  2936. *
  2937. * @param {string} fname 文件名
  2938. * @param {Array<Uint8Array>} img_data_list 数据列表
  2939. */
  2940. async function saveAsZip(fname, img_data_list) {
  2941. await utils.blobs_to_zip(
  2942. img_data_list,
  2943. "page",
  2944. "png",
  2945. fname
  2946. );
  2947. }
  2948.  
  2949.  
  2950. /**
  2951. * 取得图片集合体并切分,如果是 SVG 则对应处理
  2952. * @returns {Array<Uint8Array>} imgs
  2953. */
  2954. async function getImgs() {
  2955. let [fname, path] = [
  2956. window.ishareJS.fname,
  2957. window.ishareJS.path
  2958. ];
  2959.  
  2960. let img_url = getImgUrl(fname, path);
  2961.  
  2962. // 处理svg
  2963. if (path.includes(".svg")) {
  2964. document.title = fname;
  2965. await handleSVGurl(img_url);
  2966. return;
  2967. }
  2968. // 处理常规图像
  2969. let data = await getData(img_url);
  2970. let img_data_list = parseData(data);
  2971. console.log(`共 ${img_data_list.length} 张图片`);
  2972.  
  2973. window.ishareJS.imgs = img_data_list;
  2974.  
  2975. // 下载完成,可以导出
  2976. utils.onclick(exportPDF$3, 2, "下载并导出PDF");
  2977. utils.toggle_btn(1);
  2978. utils.toggle_btn(2);
  2979. }
  2980.  
  2981.  
  2982. async function exportPDF$3() {
  2983. let args = [
  2984. window.ishareJS.fname,
  2985. window.ishareJS.imgs
  2986. ];
  2987.  
  2988. try {
  2989. await imgDataArrsToPDF(...args);
  2990. } catch(e) {
  2991. console.error(e);
  2992. // 因 jsPDF 字符串拼接溢出导致的 Error
  2993. if (`${e}`.includes("RangeError: Invalid string length")) {
  2994. // 提示失败
  2995. alert("图片合并为 PDF 时失败,请尝试下载图片压缩包");
  2996. // 备选方案:导出图片压缩包
  2997. utils.onclick(
  2998. () => saveAsZip(...args),
  2999. 3,
  3000. "导出ZIP"
  3001. );
  3002. utils.toggle_btn(3); // 显示导出ZIP按钮
  3003. utils.toggle_btn(2); // 隐藏导出PDF按钮
  3004. } else {
  3005. throw e;
  3006. }
  3007. }
  3008.  
  3009. }
  3010.  
  3011.  
  3012. function showHints() {
  3013. wk$("h1")[0].textContent = "wk 温馨提示";
  3014. wk$("p")[0].innerHTML = [
  3015. "下载 270 页的 PPT (70 MB) 需要约 30 秒",
  3016. "请耐心等待,无需反复点击按钮",
  3017. "如果很久没反应,请加 QQ 群反馈问题"
  3018. ].join("<br>");
  3019. wk$("hr")[0].nextSibling.textContent = "403 Page Hostaged By Wenku Doc Downloader";
  3020. }
  3021.  
  3022.  
  3023. /**
  3024. * 爱问文库下载策略
  3025. */
  3026. async function ishareData() {
  3027. // 全局对象
  3028. globalThis["ishareJS"] = {
  3029. data: "",
  3030. imgs: [],
  3031. text: "",
  3032. content_1: "",
  3033. content_2: "",
  3034. fname: utils.get_param("fname"),
  3035. path: utils.get_param("path")
  3036. };
  3037.  
  3038. // 显示提示
  3039. showHints();
  3040.  
  3041. // 创建按钮区
  3042. utils.create_btns();
  3043.  
  3044. // btn_1: 识别文档类型,处理SVG或下载数据
  3045. utils.onclick(getImgs, 1, "下载数据");
  3046. }
  3047.  
  3048. /**
  3049. * 提供提示信息
  3050. */
  3051. function showTips$1() {
  3052. const h2 = document.createElement("h2");
  3053. h2.id = "wk-tips";
  3054. document.body.append(h2);
  3055. }
  3056.  
  3057.  
  3058. /**
  3059. * 更新文字到 h2 元素
  3060. * @param {string} text
  3061. */
  3062. function update(text) {
  3063. wk$("#wk-tips")[0].textContent = text;
  3064. }
  3065.  
  3066.  
  3067. /**
  3068. * 被动连接,取出数据,请求并分割图片,导出PDF
  3069. */
  3070. function mainTask() {
  3071. const sock = new utils.Socket(opener);
  3072. sock.listen(async e => {
  3073. if (e.data.wk && e.data.action) {
  3074. update("图片下载中,请耐心等待...");
  3075.  
  3076. const url = e.data.img_url;
  3077. const resp = await fetch(url);
  3078. update("图片下载完成,正在解析...");
  3079.  
  3080. const buffer = await resp.arrayBuffer();
  3081. const whole_data = new Uint8Array(buffer);
  3082. update("图片解析完成,正在合并...");
  3083. await utils.imgs_to_pdf(
  3084. utils.split_files_by_head(whole_data),
  3085. e.data.title
  3086. );
  3087. update("图片合并完成,正在导出 PDF...");
  3088. }
  3089. });
  3090. sock.connect(true);
  3091. }
  3092.  
  3093.  
  3094. /**
  3095. * 爱问文库图片下载策略v2
  3096. * @returns
  3097. */
  3098. function ishareData2() {
  3099. showTips$1();
  3100. if (!(window.opener && window.opener.window)) {
  3101. update("wk: 抱歉,页面出错了");
  3102. return;
  3103. }
  3104. mainTask();
  3105. }
  3106.  
  3107. function getPageNum() {
  3108. // ' / 6 ' -> ' 6 '
  3109. return parseInt(
  3110. wk$("span.counts")[0].textContent.split("/")[1]
  3111. );
  3112. }
  3113.  
  3114.  
  3115. function jumpToHostage() {
  3116. const
  3117. // '/fileroot/2019-9/23/73598bfa-6b91-4cbe-a548-9996f46653a2/73598bfa-6b91-4cbe-a548-9996f46653a21.gif'
  3118. url = new URL(wk$("#pageflash_1 > img")[0].src),
  3119. num = getPageNum(),
  3120. // '七年级上册地理期末试卷精编.doc-得力文库'
  3121. fname = document.title.slice(0, -5),
  3122. path = url.pathname,
  3123. tail = "1.gif";
  3124. if (!path.endsWith(tail)) {
  3125. throw new Error(`url尾部不为【${tail}】!path:【${path}】`);
  3126. }
  3127. const base_path = path.slice(0, -5);
  3128. open(`${url.protocol}//${url.host}/?num=${num}&lmt=${lmt}&fname=${fname}&path=${base_path}`);
  3129. }
  3130.  
  3131.  
  3132. function deliwenku() {
  3133. utils.create_btns();
  3134. utils.onclick(jumpToHostage, 1, "到下载页面");
  3135. }
  3136.  
  3137. function showTips() {
  3138. const body = `
  3139. <style>
  3140. h1 { color: black; }
  3141. #main {
  3142. margin: 1vw 5%;
  3143. border-radius: 10%;
  3144. }
  3145. p { font-size: large; }
  3146. .info {
  3147. color: rgb(230,214,110);
  3148. background: rgb(39,40,34);
  3149. text-align: right;
  3150. font-size: medium;
  3151. padding: 1vw;
  3152. border-radius: 4px;
  3153. }
  3154. </style>
  3155. <div id="main">
  3156. <h1>wk: 跳板页面</h1>
  3157. <p>有时候点一次下载等半天没反应,就再试一次</p>
  3158. <p>如果试了 2 次还不行加 QQ 群反馈吧...</p>
  3159. <p>导出的 PDF 如果页面数量少于应有的,那么意味着免费页数就这么多,我也爱莫能助</p>
  3160. <p>短时间连续使用导出按钮会导致 IP 被封禁</p>
  3161. <hr>
  3162. <div class="info">
  3163. 文档名称:${deliJS.fname}<br>
  3164. 原始文档页数:${deliJS.num}<br>
  3165. 最大免费页数:${deliJS.lmt}<br>
  3166. </div>
  3167. </div>`;
  3168. document.title = utils.get_param("fname"); document.body.innerHTML = body;
  3169. }
  3170.  
  3171.  
  3172. /**
  3173. * url生成器
  3174. * @param {string} base_url
  3175. * @param {number} num
  3176. */
  3177. function* genURLs(base_url, num) {
  3178. for (let i=1; i<=num; i++) {
  3179. yield `${base_url}${i}.gif`;
  3180. }
  3181. }
  3182.  
  3183.  
  3184. function genBaseURL(path) {
  3185. return `${location.protocol}//${location.host}${path}`;
  3186. }
  3187.  
  3188.  
  3189. function parseParamsToDeliJS() {
  3190. const
  3191. base_url = genBaseURL(utils.get_param("path")),
  3192. fname = utils.get_param("fname"),
  3193. num = parseInt(utils.get_param("num"));
  3194.  
  3195. let lmt = parseInt(utils.get_param("lmt"));
  3196. lmt = lmt > 3 ? lmt : 20;
  3197. lmt = lmt > num ? num : lmt;
  3198.  
  3199. window.deliJS = {
  3200. base_url,
  3201. num,
  3202. fname,
  3203. lmt
  3204. };
  3205. }
  3206.  
  3207.  
  3208. async function exportPDF$2() {
  3209. utils.toggle_btn(1);
  3210. await utils.run_with_prog(
  3211. 1, () => utils.img_urls_to_pdf(
  3212. genURLs(deliJS.base_url, deliJS.num),
  3213. deliJS.fname,
  3214. deliJS.lmt,
  3215. true // 请求完成后清理控制台
  3216. )
  3217. );
  3218. utils.toggle_btn(1);
  3219. }
  3220.  
  3221.  
  3222. /**
  3223. * 得力文库跳板页面下载策略
  3224. */
  3225. async function deliFile() {
  3226. // 从URL解析文档参数
  3227. parseParamsToDeliJS();
  3228. // 显示提示
  3229. showTips();
  3230.  
  3231. // 创建按钮区
  3232. utils.create_btns();
  3233. // btn_1: 导出PDF
  3234. utils.onclick(exportPDF$2, 1, "导出PDF");
  3235. }
  3236.  
  3237. function readAll360Doc() {
  3238. // 展开文档
  3239. document.querySelector(".article_showall a").click();
  3240. // 隐藏按钮
  3241. utils.toggle_btn(1);
  3242. // 显示按钮
  3243. utils.toggle_btn(2);
  3244. utils.toggle_btn(3);
  3245. utils.toggle_btn(4);
  3246. }
  3247.  
  3248.  
  3249. function saveText_360Doc() {
  3250. // 捕获图片链接
  3251. let images = wk$("#artContent img");
  3252. let content = [];
  3253.  
  3254. for (let i = 0; i < images.length; i++) {
  3255. let src = images[i].src;
  3256. content.push(`图${i+1},链接:${src}`);
  3257. }
  3258. // 捕获文本
  3259. let text = wk$("#artContent")[0].textContent;
  3260. content.push(text);
  3261.  
  3262. // 保存纯文本文档
  3263. let title = wk$("#titiletext")[0].textContent;
  3264. utils.save(`${title}.txt`, content.join("\n"));
  3265. }
  3266.  
  3267.  
  3268. /**
  3269. * 使文档在页面上居中
  3270. * @param {string} selector 文档容器的css选择器
  3271. * @param {string} default_offset 文档部分向右偏移的百分比(0-59)
  3272. * @returns 偏移值是否合法
  3273. */
  3274. function centre(selector, default_offset) {
  3275. const elem = wk$(selector)[0];
  3276. const offset = prompt("请输入偏移百分位:", default_offset);
  3277. // 如果输入的数字不在 0-59 内,提醒用户重新设置
  3278. if (offset.length === 1 && offset.search(/[0-9]/) !== -1) {
  3279. elem.style.marginLeft = offset + "%";
  3280. return true;
  3281. }
  3282.  
  3283. if (offset.length === 2 && offset.search(/[1-5][0-9]/) !== -1) {
  3284. elem.style.marginLeft = offset + "%";
  3285. return true;
  3286. }
  3287.  
  3288. alert("请输入一个正整数,范围在0至59之间,用来使文档居中");
  3289. return false;
  3290. }
  3291.  
  3292.  
  3293. function printPage360Doc() {
  3294. if (!confirm("确定每页内容都加载完成了吗?")) {
  3295. return;
  3296. }
  3297. // # 清理并打印360doc的文档页
  3298. // ## 移除页面上无关的元素
  3299. let selector = ".fontsize_bgcolor_controler, .atfixednav, .header, .a_right, .article_data, .prev_next, .str_border, .youlike, .new_plbox, .str_border, .ul-similar, #goTop2, #divtort, #divresaveunder, .bottom_controler, .floatqrcode";
  3300. let elem_list = wk$(selector);
  3301. let under_doc_1, under_doc_2;
  3302. try {
  3303. under_doc_1 = wk$("#bgchange p.clearboth")[0].nextElementSibling;
  3304. under_doc_2 = wk$("#bgchange")[0].nextElementSibling.nextElementSibling;
  3305. } catch (e) {}
  3306. // 执行移除
  3307. for (let elem of elem_list) {
  3308. utils.remove(elem);
  3309. }
  3310. utils.remove(under_doc_1);
  3311. utils.remove(under_doc_2);
  3312. // 执行隐藏
  3313. wk$("a[title]")[0].style.display = "none";
  3314.  
  3315. // 使文档居中
  3316. alert("建议使用:\n偏移量: 20\n缩放: 默认\n");
  3317. if (!centre(".a_left", "20")) {
  3318. return; // 如果输入非法,终止函数调用
  3319. }
  3320. // 隐藏按钮,然后打印页面
  3321. utils.print_page();
  3322. }
  3323.  
  3324.  
  3325. /**
  3326. * 阻止监听器生效
  3327. * @param {Event} e
  3328. */
  3329. function stopSpread(e) {
  3330. e.stopImmediatePropagation();
  3331. e.stopPropagation();
  3332. }
  3333.  
  3334.  
  3335. /**
  3336. * 阻止捕获事件
  3337. */
  3338. function stopCapturing() {
  3339. ["click", "mouseup"].forEach(
  3340. type => {
  3341. document.body.addEventListener(type, stopSpread, true);
  3342. document["on" + type] = undefined;
  3343. }
  3344. );
  3345. ["keypress", "keydown"].forEach(
  3346. type => {
  3347. window.addEventListener(type, stopSpread, true);
  3348. window["on" + type] = undefined;
  3349. }
  3350. );
  3351. }
  3352.  
  3353.  
  3354. /**
  3355. * 重置图像链接和最大宽度
  3356. * @param {Document} doc
  3357. */
  3358. function resetImg(doc=document) {
  3359. wk$.call(doc, "img").forEach(
  3360. elem => {
  3361. elem.style.maxWidth = "100%";
  3362. for (let attr of elem.attributes) {
  3363. if (attr.name.endsWith("-src")) {
  3364. elem.setAttribute("src", attr.value);
  3365. break;
  3366. }
  3367. }
  3368. }
  3369. );
  3370. }
  3371.  
  3372.  
  3373. /**
  3374. * 仅保留全屏文档
  3375. */
  3376. function getFullScreen() {
  3377. FullScreenObj.init();
  3378. wk$("#artContent > p:nth-child(3)")[0]?.remove();
  3379. let data = wk$("#artfullscreen__box_scr > table")[0].outerHTML;
  3380. window.doc360JS = { data };
  3381. let html_str = `
  3382. <html><head></head><body style="display: flex; flex-direction: row; justify-content: space-around">
  3383. ${data}
  3384. </body><html>
  3385. `;
  3386. wk$("html")[0].replaceWith(wk$("html")[0].cloneNode());
  3387. wk$("html")[0].innerHTML = html_str;
  3388. resetImg();
  3389. }
  3390.  
  3391.  
  3392. function cleanPage() {
  3393. getFullScreen();
  3394. stopCapturing();
  3395. }
  3396.  
  3397.  
  3398. /**
  3399. * 360doc个人图书馆下载策略
  3400. */
  3401. function doc360() {
  3402. // 创建按钮区
  3403. utils.create_btns();
  3404. // btn_1: 展开文档
  3405. utils.onclick(readAll360Doc, 1);
  3406. // btn_2: 导出纯文本
  3407. utils.onclick(saveText_360Doc, 2, "导出纯文本");
  3408. // btn_3: 打印页面到PDF
  3409. utils.onclick(printPage360Doc, 3, "打印页面到PDF");
  3410. // btn_3: 清理页面
  3411. utils.onclick(cleanPage, 4, "清理页面(推荐)");
  3412. }
  3413.  
  3414. async function getPDF() {
  3415. if (!window.DEFAULT_URL) {
  3416. alert("当前文档无法解析,请加 QQ 群反馈");
  3417. return;
  3418. }
  3419. let title = document.title.split(" - ")[0] + ".pdf";
  3420. let blob = await utils.xhr_get_blob(DEFAULT_URL);
  3421. utils.save(title, blob);
  3422. }
  3423.  
  3424.  
  3425. function mbalib() {
  3426. utils.create_btns();
  3427. utils.onclick(getPDF, 1, "下载PDF");
  3428. }
  3429.  
  3430. /**
  3431. * 判断是否进入预览模式
  3432. * @returns Boolean
  3433. */
  3434. function isInPreview() {
  3435. let p_elem = wk$("#preview_tips")[0];
  3436. if (p_elem && p_elem.style && p_elem.style.display === "none") {
  3437. return true;
  3438. }
  3439. return false;
  3440. }
  3441.  
  3442.  
  3443. /**
  3444. * 确保进入预览模式
  3445. */
  3446. async function ensureInPreview() {
  3447. while (!isInPreview()) {
  3448. // 如果没有进入预览,则先进入
  3449. if (typeof window.preview !== "function") {
  3450. alert("脚本失效,请加 QQ 群反馈");
  3451. throw new Error("preview 全局函数不存在");
  3452. }
  3453.  
  3454. await utils.sleep(500);
  3455. preview();
  3456. }
  3457. }
  3458.  
  3459.  
  3460. /**
  3461. * 前往页码
  3462. * @param {number} page_num
  3463. */
  3464. function toPage(page_num) {
  3465. // 先尝试官方接口,不行再用模拟的
  3466. try {
  3467. Viewer._GotoPage(page_num);
  3468. } catch(e) {
  3469. console.error(e);
  3470. utils.to_page(
  3471. wk$("#pageNumInput")[0],
  3472. page_num,
  3473. "keydown"
  3474. );
  3475. }
  3476. }
  3477.  
  3478.  
  3479. /**
  3480. * 展开全文预览,当展开完成后再次调用时,返回true
  3481. * @returns
  3482. */
  3483. async function walkThrough$1() {
  3484. // 隐藏页面
  3485. wk$("#pageContainer")[0].style.display = "none";
  3486.  
  3487. // 逐页加载
  3488. let lmt = window.dugenJS.lmt;
  3489. for (let i of utils.range(1, lmt + 1)) {
  3490. toPage(i);
  3491. await utils.wait_until(
  3492. () => wk$(`#outer_page_${i}`)[0].style.width.endsWith("px")
  3493. );
  3494. }
  3495.  
  3496. // 恢复显示
  3497. wk$("#pageContainer")[0].style.display = "";
  3498. console.log(`共 ${lmt} 页加载完毕`);
  3499. }
  3500.  
  3501.  
  3502. /**
  3503. * 返回当前未加载页面的页码
  3504. * @returns not_loaded
  3505. */
  3506. function getNotloadedPages() {
  3507. // 已经取得的页码
  3508. let pages = document.querySelectorAll("[id*=pageflash_]");
  3509. let loaded = new Set();
  3510. pages.forEach((page) => {
  3511. let id = page.id.split("_")[1];
  3512. id = parseInt(id);
  3513. loaded.add(id);
  3514. });
  3515. // 未取得的页码
  3516. let not_loaded = [];
  3517. for (let i of utils.range(1, window.dugenJS.lmt + 1)) {
  3518. if (!loaded.has(i)) {
  3519. not_loaded.push(i);
  3520. }
  3521. }
  3522. return not_loaded;
  3523. }
  3524.  
  3525.  
  3526. /**
  3527. * 取得全部文档页面的链接,返回urls;如果有页面未加载,则返回null
  3528. * @returns
  3529. */
  3530. function getImgUrls() {
  3531. let pages = wk$("[id*=pageflash_]");
  3532. // 尚未浏览完全部页面,返回false
  3533. if (pages.length < window.dugenJS.lmt) {
  3534. let hints = [
  3535. "尚未加载完全部页面",
  3536. "以下页面需要浏览并加载:",
  3537. getNotloadedPages().join(",")
  3538. ];
  3539. alert(hints.join("\n"));
  3540. return [false, []];
  3541. }
  3542. // 浏览完全部页面,返回urls
  3543. return [true, pages.map(page => page.querySelector("img").src)];
  3544. }
  3545.  
  3546.  
  3547. function exportImgUrls() {
  3548. let [ok, urls] = getImgUrls();
  3549. if (!ok) {
  3550. return;
  3551. }
  3552. utils.save("urls.csv", urls.join("\n"));
  3553. }
  3554.  
  3555.  
  3556. function exportPDF$1() {
  3557. let [ok, urls] = getImgUrls();
  3558. if (!ok) {
  3559. return;
  3560. }
  3561. let title = document.title.split("-")[0];
  3562. return utils.run_with_prog(
  3563. 3, () => utils.img_urls_to_pdf(urls, title)
  3564. );
  3565. }
  3566.  
  3567.  
  3568. /**
  3569. * dugen文档下载策略
  3570. */
  3571. async function dugen() {
  3572. await ensureInPreview();
  3573. // 全局对象
  3574. window.dugenJS = {
  3575. lmt: window.lmt ? window.lmt : 20
  3576. };
  3577.  
  3578. // 创建按钮区
  3579. utils.create_btns();
  3580.  
  3581. // 绑定监听器
  3582. // 按钮1:展开文档
  3583. utils.onclick(walkThrough$1, 1, "加载可预览页面");
  3584. // 按钮2:导出图片链接
  3585. utils.onclick(exportImgUrls, 2, "导出图片链接");
  3586. utils.toggle_btn(2);
  3587. // 按钮3:导出PDF
  3588. utils.onclick(exportPDF$1, 3, "导出PDF");
  3589. utils.toggle_btn(3);
  3590. }
  3591.  
  3592. // 域名级全局常量
  3593. const img_tasks = [];
  3594.  
  3595.  
  3596. /**
  3597. * 取得文档类型
  3598. * @returns {String} 文档类型str
  3599. */
  3600. function getDocType() {
  3601. const
  3602. // ["icon", "icon-format", "icon-format-doc"]
  3603. elem = wk$(".title .icon.icon-format")[0],
  3604. // "icon-format-doc"
  3605. cls = elem.classList[2];
  3606. return cls.split("-")[2];
  3607. }
  3608.  
  3609.  
  3610. /**
  3611. * 判断文档类型是否为type_list其中之一
  3612. * @returns 是否为type
  3613. */
  3614. function isTypeof(type_list) {
  3615. const type = getDocType();
  3616. if (type_list.includes(type)) {
  3617. return true;
  3618. }
  3619. return false;
  3620. }
  3621.  
  3622.  
  3623. /**
  3624. * 判断文档类型是否为PPT
  3625. * @returns 是否为PPT
  3626. */
  3627. function is_ppt() {
  3628. return isTypeof(["ppt", "pptx"]);
  3629. }
  3630.  
  3631.  
  3632. /**
  3633. * 判断文档类型是否为Excel
  3634. * @returns 是否为Excel
  3635. */
  3636. function is_excel() {
  3637. return isTypeof(["xls", "xlsm", "xlsx"]);
  3638. }
  3639.  
  3640.  
  3641. /**
  3642. * 取得未加载页面的页码
  3643. * @returns {Array} not_loaded 未加载页码列表
  3644. */
  3645. function getNotLoaded() {
  3646. const loaded = wk$("[data-id] img[src]").map(
  3647. img => parseInt(
  3648. img.closest("[data-id]").getAttribute("data-id")
  3649. )
  3650. );
  3651. return Array.from(
  3652. utils.diff(
  3653. utils.range(1, window.book118JS.page_counts + 1),
  3654. loaded
  3655. )
  3656. );
  3657. }
  3658.  
  3659.  
  3660. /**
  3661. * 取得全部文档页的url
  3662. * @returns [<是否全部加载>, <urls列表>, <未加载页码列表>]
  3663. */
  3664. function getUrls() {
  3665. const urls = wk$("[data-id] img[src]").map(
  3666. img => img.src
  3667. );
  3668. // 如果所有页面加载完毕
  3669. if (urls.length === book118JS.page_counts) {
  3670. return [true, urls, []];
  3671. }
  3672. // 否则收集未加载页面的url
  3673. return [false, urls, getNotLoaded()];
  3674. }
  3675.  
  3676.  
  3677. /**
  3678. * 展开全文
  3679. */
  3680. async function walkThrough() {
  3681. // 遍历期间隐藏按钮区
  3682. utils.toggle_box();
  3683.  
  3684. // 取得总页码
  3685. // preview.getPage()
  3686. // {current: 10, actual: 38, preview: 38, remain: 14}
  3687. const { preview: all } = preview.getPage();
  3688. for (let i = 1; i <= all; i++) {
  3689. // 逐页加载
  3690. preview.jump(i);
  3691. await utils.wait_until(
  3692. () => wk$(`[data-id="${i}"] img`)[0].src, 1000
  3693. );
  3694. }
  3695. console.log("遍历完成");
  3696. utils.toggle_box();
  3697. }
  3698.  
  3699.  
  3700. /**
  3701. * btn_2: 导出图片链接
  3702. */
  3703. function wantUrls() {
  3704. let [flag, urls, escaped] = getUrls();
  3705. // 页面都加载完毕,下载urls
  3706. if (!flag) {
  3707. // 没有加载完,提示出未加载好的页码
  3708. const hint = [
  3709. "仍有页面没有加载",
  3710. "请浏览并加载如下页面",
  3711. "是否继续导出图片链接?",
  3712. "[" + escaped.join(",") + "]"
  3713. ].join("\n");
  3714. // 终止导出
  3715. if (!confirm(hint)) {
  3716. return
  3717. }
  3718. }
  3719. utils.save("urls.csv", urls.join("\n"));
  3720. }
  3721.  
  3722.  
  3723. /**
  3724. * 打开PPT预览页面
  3725. */
  3726. async function open_iframe() {
  3727. wk$(".front a")[0].click();
  3728. const iframes = await wk$$("iframe.preview-iframe");
  3729. window.open(iframes[0].src);
  3730. }
  3731.  
  3732.  
  3733. /**
  3734. * 取得最大页码
  3735. * @returns {number} 最大页码
  3736. */
  3737. function getPageCounts$1() {
  3738. return window?.preview?.getPage()?.preview || NaN;
  3739. }
  3740.  
  3741.  
  3742. /**
  3743. * 原创力文档(非PPT或Excel)下载策略
  3744. */
  3745. async function common_doc() {
  3746. await utils.wait_until(
  3747. () => !!wk$(".counts")[0]
  3748. );
  3749.  
  3750. // 创建全局对象
  3751. window.book118JS = {
  3752. doc_type: getDocType(),
  3753. page_counts: getPageCounts$1()
  3754. };
  3755.  
  3756. // 处理非PPT文档
  3757. // 创建按钮组
  3758. utils.create_btns();
  3759. // 绑定监听器到按钮
  3760. // 按钮1:加载全文
  3761. utils.onclick(walkThrough, 1, "加载全文");
  3762. // 按钮2:导出图片链接
  3763. utils.onclick(wantUrls, 2, "导出图片链接");
  3764. utils.toggle_btn(2);
  3765. }
  3766.  
  3767.  
  3768. /**
  3769. * @returns {string}
  3770. */
  3771. function table_to_tsv() {
  3772. // 取得所有行
  3773. const rows = wk$("tr[id]");
  3774.  
  3775. // 遍历行
  3776. return rows.map(row => {
  3777. // 取得所有列
  3778. const cells = wk$.call(row, "td[class*=fi], td.tdrl");
  3779.  
  3780. // 遍历列(单元格)
  3781. return cells.map(cell => {
  3782. // 判断单元格是否存储图片
  3783. const img = cell.querySelector("img");
  3784. if (img) {
  3785. // 如果是图片,保存图片链接
  3786. return img.src;
  3787. }
  3788. // 否则保存单元格文本
  3789. return cell
  3790. .textContent
  3791. .trim()
  3792. .replace(/\n/g, " ")
  3793. .replace(/\t/g, " ");
  3794. }).join("\t");
  3795. }).join("\n").trim();
  3796. }
  3797.  
  3798.  
  3799. /**
  3800. * 下载当前表格内容,保存为csv(utf-8编码)
  3801. */
  3802. function wantEXCEL() {
  3803. const tsv = table_to_tsv();
  3804. const bytes = utils.encode_to_gbk(tsv);
  3805. const fname = "原创力表格.tsv";
  3806. utils.save(fname, bytes);
  3807. }
  3808.  
  3809.  
  3810. /**
  3811. * 在Excel预览页面给出操作提示
  3812. */
  3813. function help$1() {
  3814. const hint = [
  3815. "【导出表格到TSV】只能导出当前 sheet",
  3816. "如果有多张 sheet 请在每个 sheet 上用按钮分别导出 TSV",
  3817. "TSV 文件请用记事本或 Excel 打开",
  3818. "TSV 不能存储图片,所以用图片链接代替",
  3819. "或使用此脚本复制表格到剪贴板:",
  3820. "https://greasyfork.org/zh-CN/scripts/469550",
  3821. ];
  3822. alert(hint.join("\n"));
  3823. }
  3824.  
  3825.  
  3826. /**
  3827. * 原创力文档(EXCEL)下载策略
  3828. */
  3829. function excel() {
  3830. // 创建按钮区
  3831. utils.create_btns();
  3832. // 绑定监听器到按钮
  3833. utils.onclick(wantEXCEL, 1, "导出表格到TSV");
  3834. utils.onclick(help$1, 2, "使用说明");
  3835. // 显示按钮
  3836. utils.toggle_btn(2);
  3837. }
  3838.  
  3839.  
  3840. /**
  3841. * ------------------------------ PPT 策略 ---------------------------------
  3842. */
  3843.  
  3844.  
  3845. /**
  3846. * 返回当前页码
  3847. * @returns {number}
  3848. */
  3849. function cur_page_num() {
  3850. return parseInt(
  3851. wk$("#PageIndex")[0].textContent
  3852. );
  3853. }
  3854.  
  3855.  
  3856. function add_page() {
  3857. const view = wk$("#view")[0];
  3858. view.setAttribute("style", "");
  3859.  
  3860. const i = cur_page_num() - 1;
  3861. const cur_view = wk$(`#view${i}`)[0];
  3862.  
  3863. img_tasks.push(
  3864. html2canvas(cur_view)
  3865. );
  3866. utils.btn(1).textContent = `截图: ${img_tasks.length}`;
  3867. }
  3868.  
  3869.  
  3870. function reset_tasks() {
  3871. img_tasks.splice(0);
  3872. utils.btn(1).textContent = `截图: 0`;
  3873. }
  3874.  
  3875.  
  3876. function canvas_to_blob(canvas) {
  3877. return utils.canvas_to_blob(canvas);
  3878. }
  3879.  
  3880.  
  3881. async function export_imgs_as_pdf() {
  3882. alert("正在合并截图,请耐心等待");
  3883. utils.toggle_btn(3);
  3884.  
  3885. try {
  3886. const imgs = await utils.gather(img_tasks);
  3887. const blobs = await utils.gather(
  3888. imgs.map(canvas_to_blob)
  3889. );
  3890.  
  3891. if (!blobs.length) {
  3892. alert("你尚未截取任何页面!");
  3893. } else {
  3894. await utils.img_blobs_to_pdf(blobs, "原创力幻灯片");
  3895. }
  3896. } catch(err) {
  3897. console.error(err);
  3898. }
  3899. utils.toggle_btn(3);
  3900. }
  3901.  
  3902.  
  3903.  
  3904. function ppt() {
  3905. utils.create_btns();
  3906.  
  3907. const btn1 = utils.btn(1);
  3908. btn1.onclick = add_page;
  3909. btn1.textContent = "截图当前页面";
  3910.  
  3911. utils.onclick(reset_tasks, 2, "清空截图");
  3912. utils.onclick(export_imgs_as_pdf, 3, "合并为PDF");
  3913.  
  3914. utils.toggle_btn(2);
  3915. utils.toggle_btn(3);
  3916. }
  3917.  
  3918.  
  3919. /**
  3920. * 原创力文档下载策略
  3921. */
  3922. function book118() {
  3923. const host = window.location.hostname;
  3924.  
  3925. if (host === 'max.book118.com') {
  3926. if (is_excel()) {
  3927. utils.create_btns();
  3928. utils.onclick(open_iframe, 1, "访问EXCEL");
  3929. } else if (is_ppt()) {
  3930. utils.create_btns();
  3931. utils.onclick(open_iframe, 1, "访问PPT");
  3932. } else {
  3933. common_doc();
  3934. }
  3935. } else if (host === "view-cache.book118.com") {
  3936. if (window.top !== window) return;
  3937. ppt();
  3938. } else if (host.match(/view[0-9]{1,3}.book118.com/)) {
  3939. excel();
  3940. } else {
  3941. console.log(`wk: Unknown host: ${host}`);
  3942. }
  3943. }
  3944.  
  3945. // test url: https://openstd.samr.gov.cn/bzgk/gb/newGbInfo?hcno=E86BBCE32DA8E67F3DA04ED98F2465DB
  3946.  
  3947.  
  3948. /**
  3949. * 绘制0x0的bmp, 作为请求失败时返回的page
  3950. * @returns {Promise<ImageBitmap>} blank_page
  3951. */
  3952. async function blankBMP() {
  3953. let canvas = document.createElement("canvas");
  3954. [canvas.width, canvas.height] = [0, 0];
  3955. return createImageBitmap(canvas);
  3956. }
  3957.  
  3958.  
  3959. /**
  3960. * resp导出bmp
  3961. * @param {string} page_url
  3962. * @param {Promise<Response> | ImageBitmap} pms_or_bmp
  3963. * @returns {Promise<ImageBitmap>} page
  3964. */
  3965. async function respToPage(page_url, pms_or_bmp) {
  3966. let center = globalThis.gb688JS;
  3967. // 此时是bmp
  3968. if (pms_or_bmp instanceof ImageBitmap) {
  3969. return pms_or_bmp;
  3970. }
  3971.  
  3972. // 第一次下载, 且无人处理
  3973. if (!center.pages_status.get(page_url)) {
  3974. // 处理中, 设为占用
  3975. center.pages_status.set(page_url, 1);
  3976.  
  3977. // 处理
  3978. let resp;
  3979. try {
  3980. resp = await pms_or_bmp;
  3981. } catch(err) {
  3982. console.log("下载页面失败");
  3983. console.error(err);
  3984. return blankBMP();
  3985. }
  3986.  
  3987. let page_blob = await resp.blob();
  3988. let page = await createImageBitmap(page_blob);
  3989. center.pages.set(page_url, page);
  3990. // 处理结束, 设为释放
  3991. center.pages_status.set(page_url, 0);
  3992. return page;
  3993. }
  3994.  
  3995. // 有人正在下载且出于处理中
  3996. while (center.pages_status.get(page_url)) {
  3997. await utils.sleep(500);
  3998. }
  3999. return center.pages.get(page_url);
  4000. }
  4001.  
  4002.  
  4003. /**
  4004. * 获得PNG页面
  4005. * @param {string} page_url
  4006. * @returns {Promise<ImageBitmap>} bmp
  4007. */
  4008. async function getPage(page_url) {
  4009. // 如果下载过, 直接返回缓存
  4010. let pages = globalThis.gb688JS.pages;
  4011. if (pages.has(page_url)) {
  4012. return respToPage(page_url, pages.get(page_url));
  4013. }
  4014.  
  4015. // 如果从未下载过, 就下载
  4016. let resp = fetch(page_url, {
  4017. "headers": {
  4018. "accept": "image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8",
  4019. "accept-language": "zh-CN,zh;q=0.9,en;q=0.8",
  4020. "proxy-connection": "keep-alive"
  4021. },
  4022. "referrer": location.href,
  4023. "referrerPolicy": "strict-origin-when-cross-origin",
  4024. "body": null,
  4025. "method": "GET",
  4026. "mode": "cors",
  4027. "credentials": "include"
  4028. });
  4029. pages.set(page_url, resp);
  4030. return respToPage(page_url, resp);
  4031. }
  4032.  
  4033.  
  4034. /**
  4035. * 返回文档页div的裁切和粘贴位置信息: [[cut_x, cut_y, paste_x%, paset_y%],...]
  4036. * @param {HTMLDivElement} page_div 文档页元素
  4037. * @returns {Array<Array<number>>} positions
  4038. */
  4039. function getPostions(page_div) {
  4040. let positions = [];
  4041.  
  4042. Array.from(page_div.children).forEach(span => {
  4043. // 'pdfImg-3-8' -> {left: 30%; top: 80%;}
  4044. let paste_pos = span.className.split("-").slice(1).map(
  4045. v => parseInt(v) / 10
  4046. );
  4047. // '-600px 0px' -> [600, 0]
  4048. let cut_pos = span.style.backgroundPosition.split(" ").map(
  4049. v => Math.abs(parseInt(v))
  4050. );
  4051. positions.push([...cut_pos, ...paste_pos]);
  4052. });
  4053. return positions;
  4054. }
  4055.  
  4056.  
  4057. /**
  4058. * 取得文档页的图像url
  4059. * @param {HTMLDivElement} page_div
  4060. * @returns {string} url
  4061. */
  4062. function getPageURL(page_div) {
  4063. // 拿到目标图像url
  4064. let path = location.pathname.split("/").slice(0, -1).join("/");
  4065. let prefix = location.origin + path + "/";
  4066. let url = page_div.getAttribute("bg");
  4067. if (!url) {
  4068. // 'url("viewGbImg?fileName=VS72l67k0jw5g3j0vErP8DTsnWvk5QsqnNLLxaEtX%2FM%3D")'
  4069. url = page_div.children[0].style.backgroundImage.split('"')[1];
  4070. }
  4071. return prefix + url;
  4072. }
  4073.  
  4074.  
  4075. /**
  4076. * 下载目标图像并拆解重绘, 返回canvas
  4077. * @param {number} i 第 i 页 (从0开始)
  4078. * @param {HTMLDivElement} page_div
  4079. * @returns {Promise<Array>} [页码, Canvas]
  4080. */
  4081. async function getAndDrawPage(i, page_div) {
  4082. // 拿到目标图像
  4083. let url = getPageURL(page_div);
  4084. let page = await getPage(url);
  4085.  
  4086. // 绘制空白A4纸背景
  4087. let [page_w, page_h] = [1190, 1680];
  4088. let bg = document.createElement("canvas");
  4089. bg.width = page_w; // 注意canvas作为取景框的大小
  4090. bg.height = page_h; // 如果不设置等于一个很小的取景框
  4091. let bg_ctx = bg.getContext("2d");
  4092. bg_ctx.fillStyle = "white";
  4093. bg_ctx.fillRect(0, 0, page_w, page_h);
  4094.  
  4095. // 逐个区块剪切取出并粘贴
  4096. // wk$("#viewer .page").forEach(page_div => {
  4097. getPostions(page_div).forEach(pos => {
  4098. bg_ctx.drawImage(
  4099. page, // image source
  4100. pos[0], // source x
  4101. pos[1], // source y
  4102. 120, // source width
  4103. 169, // source height
  4104. pos[2] * page_w, // destination x = left: x%
  4105. pos[3] * page_h, // destination y = top: y%
  4106. 120, // destination width
  4107. 169 // destination height
  4108. );
  4109. });
  4110. // });
  4111. return [i, bg];
  4112. }
  4113.  
  4114.  
  4115. /**
  4116. * 页面批量请求、裁剪重绘, 合成PDF并下载
  4117. */
  4118. async function turnPagesToPDF() {
  4119. // 渲染每页
  4120. const tasks = wk$("#viewer .page").map(
  4121. (page_div, i) => getAndDrawPage(i, page_div)
  4122. );
  4123. // 等待每页渲染完成后,排序
  4124. const results = await utils.gather(tasks);
  4125. results.sort((prev, next) => prev[0] - next[0]);
  4126. // 合并为PDF并导出
  4127. return utils.imgs_to_pdf(
  4128. results.map(item => item[1]),
  4129. // '在线预览|GB 14023-2022'
  4130. document.title.split("|")[1]
  4131. );
  4132. }
  4133.  
  4134.  
  4135. /**
  4136. * 提示预估下载耗时,然后下载
  4137. */
  4138. function hintThenDownload$1() {
  4139. // '/93'
  4140. let page_num = parseInt(wk$("#numPages")[0].textContent.slice(1));
  4141. let estimate = Math.ceil(page_num / 3);
  4142. alert(`页数: ${page_num},预计花费: ${estimate}秒;如遇网络异常可能更久\n请勿反复点击按钮;如果无法导出请 QQ 群反馈`);
  4143. turnPagesToPDF();
  4144. }
  4145.  
  4146.  
  4147. /**
  4148. * gb688文档下载策略
  4149. */
  4150. async function gb688() {
  4151. // 创建全局对象
  4152. globalThis.gb688JS = {
  4153. pages: new Map(), // {url: bmp}
  4154. pages_status: new Map() // {url: 0或1} 0释放, 1占用
  4155. };
  4156.  
  4157. // 创建按钮区
  4158. utils.create_btns();
  4159. // 绑定监听器
  4160. // 按钮1:导出PDF
  4161. utils.onclick(hintThenDownload$1, 1, "导出PDF");
  4162. }
  4163.  
  4164. function getPageCounts() {
  4165. // " / 39"
  4166. const counts_str = wk$(".counts")[0].textContent.split("/")[1];
  4167. const counts = parseInt(counts_str);
  4168. return counts > 20 ? 20 : counts;
  4169. }
  4170.  
  4171.  
  4172. /**
  4173. * 返回图片基础路径
  4174. * @returns {string} base_url
  4175. */
  4176. function getImgBaseURL() {
  4177. return wk$("#dp")[0].value;
  4178. }
  4179.  
  4180.  
  4181. function* genImgURLs$1() {
  4182. let counts = getPageCounts();
  4183. let base_url = getImgBaseURL();
  4184. for (let i = 1; i <= counts; i++) {
  4185. yield base_url + `${i}.gif`;
  4186. }
  4187. }
  4188.  
  4189.  
  4190. /**
  4191. * 下载图片,转为canvas,合并为PDF并下载
  4192. * @returns {Promise<void>}
  4193. */
  4194. function fetchThenExportPDF() {
  4195. // db2092-2014-河北特种设备使用安全管理规范_安全文库网safewk.com
  4196. let title = document.title.split("_")[0];
  4197. return utils.img_urls_to_pdf(genImgURLs$1(), title);
  4198. }
  4199.  
  4200.  
  4201. /**
  4202. * 提示预估下载耗时,然后下载
  4203. */
  4204. function hintThenDownload() {
  4205. let hint = [
  4206. "只能导出可预览的页面(最多20页)",
  4207. "请勿短时间反复点击按钮,导出用时大约不到 10 秒",
  4208. "点完后很久没动静请至 QQ 群反馈"
  4209. ];
  4210. alert(hint.join("\n"));
  4211. return utils.run_with_prog(
  4212. 1, fetchThenExportPDF
  4213. );
  4214. }
  4215.  
  4216.  
  4217. /**
  4218. * safewk文档下载策略
  4219. */
  4220. async function safewk() {
  4221. // 创建按钮区
  4222. utils.create_btns();
  4223. // 绑定监听器
  4224. // 按钮1:导出PDF
  4225. utils.onclick(
  4226. hintThenDownload, 1, "导出PDF"
  4227. );
  4228. }
  4229.  
  4230. /**
  4231. * 跳转到页码
  4232. * @param {string | number} num
  4233. */
  4234. function _to_page(num) {
  4235. if (window.WebPreview
  4236. && WebPreview.Page
  4237. && WebPreview.Page.jump
  4238. ) {
  4239. WebPreview.Page.jump(parseInt(num));
  4240. } else {
  4241. console.error("window.WebPreview.Page.jump doesn't exist");
  4242. }
  4243. }
  4244.  
  4245.  
  4246. /**
  4247. * 跳转页码GUI版
  4248. */
  4249. function to_page() {
  4250. let num = prompt("请输入要跳转的页码")?.trim();
  4251. if (/^[0-9]+$/.test(num)) {
  4252. _to_page(num);
  4253. } else {
  4254. console.log(`输入值 [${num}] 不是合法整数`);
  4255. }
  4256. }
  4257.  
  4258.  
  4259. function capture_urls() {
  4260. if (!confirm(
  4261. "只能导出已经预览页面的链接,是否继续?"
  4262. )) return;
  4263.  
  4264. let imgs = wk$("[data-id] img");
  4265. if (imgs.length === 0) {
  4266. imgs = wk$("img[data-page]");
  4267. }
  4268. console.log(imgs);
  4269.  
  4270. const urls = imgs.map(img => {
  4271. const src = img.dataset.src || img.src;
  4272. if (!src) return;
  4273. return src.startsWith("//") ? "https:" + src : src
  4274. });
  4275. const lacked = [];
  4276. const existed = urls.filter((url, i) => {
  4277. if (url) return true;
  4278. lacked.push(i + 1);
  4279. });
  4280.  
  4281. utils.save_urls(existed);
  4282. alert(
  4283. `已经浏览的页面中有 ${lacked.length} 页图片尚未加载,` +
  4284. `已经从结果中剔除。\n它们的页码是:\n${lacked}`
  4285. );
  4286. }
  4287.  
  4288.  
  4289. function* genImgURLs() {
  4290. const params = window?.previewParams;
  4291. if (!params) throw new Error(
  4292. "接口为空: window.previewParams"
  4293. );
  4294.  
  4295. let i = -4;
  4296. const
  4297. base = "https://openapi.renrendoc.com/preview/getPreview?",
  4298. query = {
  4299. temp_view: 0,
  4300. jsoncallback: "a",
  4301. callback: "b",
  4302. encrypt: params.encrypt,
  4303. doc_id: params.doc_id,
  4304. get _() { return Date.now() },
  4305. get start() { return i += 5; },
  4306. };
  4307. while (true) {
  4308. const keys = Reflect.ownKeys(query);
  4309. yield base + keys.map(
  4310. key => `${key}=${query[key]}`
  4311. ).join("&");
  4312. }
  4313. }
  4314.  
  4315.  
  4316. async function _fetch_preview_urls() {
  4317. let
  4318. is_empty = true,
  4319. switch_counts = 0,
  4320. previews = [];
  4321. for (const [i, url] of utils.enumerate(genImgURLs())) {
  4322. const resp = await fetch(url);
  4323. utils.raise_for_status(resp);
  4324. const raw_data = await resp.text(),
  4325. data = raw_data.slice(2, -1),
  4326. img_urls = JSON
  4327. .parse(data)
  4328. .data
  4329. ?.preview_list
  4330. ?.map(pair => pair.url);
  4331. if (!img_urls) break;
  4332.  
  4333. previews = previews.concat(...img_urls);
  4334. utils.update_popup(`已经请求 ${i + 1} 组图片链接`);
  4335. if (is_empty !== (img_urls.length ? false : true)) {
  4336. is_empty = !is_empty;
  4337. switch_counts++;
  4338. }
  4339. if (switch_counts === 2) break;
  4340.  
  4341. await utils.sleep(1000);
  4342. }
  4343. const
  4344. params = window.previewParams,
  4345. free = params.freepage || 20,
  4346. base = params.pre || wk$(".page img")[0].src.slice(0, -5),
  4347. free_urls = Array.from(
  4348. utils.range(1, free + 1)
  4349. ).map(
  4350. n => `${base}${n}.gif`
  4351. );
  4352.  
  4353. const urls = free_urls.concat(...previews);
  4354. utils.save_urls(urls);
  4355. }
  4356.  
  4357.  
  4358. function fetch_preview_urls() {
  4359. return utils.run_with_prog(
  4360. 3, _fetch_preview_urls
  4361. );
  4362. }
  4363.  
  4364.  
  4365. function help() {
  4366. alert(
  4367. "【捕获】和【请求】图片链接的区别:\n" +
  4368. " - 【捕获】是从当前已经加载的文档页中提取图片链接\n" +
  4369. " - 【请求】是使用官方接口直接下载图片链接\n" +
  4370. " - 【捕获】使用麻烦,但是稳定\n" +
  4371. " - 【请求】使用简单,速度快,但可能失效"
  4372. );
  4373. }
  4374.  
  4375.  
  4376. /**
  4377. * 人人文档下载策略
  4378. */
  4379. async function renrendoc() {
  4380. utils.create_btns();
  4381. utils.onclick(to_page, 1, "跳转到页码");
  4382. utils.onclick(capture_urls, 2, "捕获图片链接");
  4383. utils.onclick(fetch_preview_urls, 3, "请求图片链接");
  4384. utils.onclick(help, 4, "使用说明");
  4385.  
  4386. utils.toggle_btn(2);
  4387. utils.toggle_btn(3);
  4388. utils.toggle_btn(4);
  4389. }
  4390.  
  4391. /**
  4392. * 取得全部图片连接
  4393. * @returns {Array<string>}
  4394. */
  4395. function get_img_urls() {
  4396. const src = wk$("#page1 img")[0]?.src;
  4397.  
  4398. // 适用于图片类型
  4399. if (src) {
  4400. const path = src.split("?")[0].split("/").slice(3, -1).join("/");
  4401. const origin = new URL(location.href).origin;
  4402. const urls = window.htmlConfig.fliphtml5_pages.map(obj => {
  4403. const fname = obj.n[0].split("?")[0].split("/").at(-1);
  4404. return `${origin}/${path}/${fname}`;
  4405. });
  4406. const unique = [...new Set(urls)];
  4407. window.img_urls = unique;
  4408. return unique;
  4409. }
  4410.  
  4411. // 适用于其他类型
  4412. const relative_path = wk$(".side-image img")[0].getAttribute("src").split("?")[0];
  4413. // ../files/large/
  4414. const relative_dir = relative_path.split("/").slice(0, -1).join("/") + "/";
  4415.  
  4416. const base = location.href;
  4417. const urls = window.htmlConfig.fliphtml5_pages.map(obj => {
  4418. // "../files/large/d8b6c26f987104455efb3ec5addca7c9.jpg"
  4419. const path = relative_dir + obj.n[0].split("?")[0];
  4420. const url = new URL(path, base);
  4421. // https://book.yunzhan365.com/mctl/itid/files/large/d8b6c26f987104455efb3ec5addca7c9.jpg
  4422. return url.href.replace("/thumb/", "/content-page/");
  4423. });
  4424.  
  4425. window.img_urls = urls;
  4426. return urls;
  4427. }
  4428.  
  4429.  
  4430. function imgs_to_pdf() {
  4431. const urls = get_img_urls();
  4432. const title = document.title;
  4433. const task = () => utils.img_urls_to_pdf(urls, title);
  4434.  
  4435. utils.run_with_prog(1, task);
  4436. alert(
  4437. "正在下载图片,请稍等,时长取决于图片数量\n" +
  4438. "如果导出的文档只有一页空白页,说明当前文档不适用"
  4439. );
  4440. }
  4441.  
  4442.  
  4443. /**
  4444. * 将数组中的连续数字描述为字符串
  4445. * 例如 [1, 2, 3, 5] => "1 - 3, 5"
  4446. * @param {number[]} nums 整数数组
  4447. * @returns {string} 描述数组的字符串
  4448. */
  4449. function describe_nums(nums) {
  4450. let result = "";
  4451. let start = nums[0];
  4452. let end = nums[0];
  4453. for (let i = 1; i < nums.length; i++) {
  4454. if (nums[i] === end + 1) {
  4455. end = nums[i];
  4456. } else {
  4457. if (start === end) {
  4458. result += start + ", ";
  4459. } else {
  4460. result += start + " - " + end + ", ";
  4461. }
  4462. start = nums[i];
  4463. end = nums[i];
  4464. }
  4465. }
  4466. if (start === end) {
  4467. result += start;
  4468. } else {
  4469. result += start + " - " + end;
  4470. }
  4471. return result;
  4472. }
  4473.  
  4474.  
  4475. /**
  4476. * 取得总页码(作为str)
  4477. * @returns {string}
  4478. */
  4479. function get_total() {
  4480. const total = window?.bookConfig?.totalPageCount;
  4481. if (total) {
  4482. return String(total);
  4483. }
  4484. return wk$("#tfPageIndex input")[0].value.split("/")[1].trim();
  4485. }
  4486.  
  4487.  
  4488. /**
  4489. * 下载稀疏数组的pdf数据,每个元素应该是 [pdf_blob, pwd_str]
  4490. * @param {Array} pdfs_data
  4491. */
  4492. async function data_to_zip(pdfs_data) {
  4493. // 导入jszip
  4494. await utils.blobs_to_zip([], "empty", "dat", "empty", false);
  4495.  
  4496. // 分装截获的数据
  4497. const page_nums = Object.keys(pdfs_data)
  4498. .map(index => parseInt(index) + 1);
  4499. const len = page_nums.length;
  4500. const pwds = new Array(len + 1);
  4501. pwds[0] = "page-num,password";
  4502. // 创建压缩包,归档加密的PDF页面
  4503. const zip = new window.JSZip();
  4504. const total = get_total();
  4505. const digits = total.length;
  4506.  
  4507. // 归档
  4508. for (let i = 0; i < len; i++) {
  4509. // 页码左侧补零
  4510. const page_no = page_nums[i];
  4511. const page_no_str = page_no.toString().padStart(digits, "0");
  4512. // 记录密码
  4513. pwds[i+1] = `${page_no_str},${pdfs_data[page_no - 1][1]}`;
  4514. // 添加pdf内容到压缩包
  4515. const blob = pdfs_data[page_no - 1][0];
  4516. zip.file(`page-${page_no_str}.pdf`, blob, { binary: true });
  4517. }
  4518. console.log("zip:", zip);
  4519.  
  4520. // 添加密码本到压缩包
  4521. const pwds_blob = new Blob([pwds.join("\n")], { type: "text/plain" });
  4522. zip.file(`密码本.txt`, pwds_blob, { binary: true });
  4523. // 下载
  4524. console.info("正在合成压缩包并导出,请耐心等待几分钟......");
  4525. const zip_blob = await zip.generateAsync({ type: "blob" });
  4526. utils.save(`${document.title}.zip`, zip_blob, "application/zip");
  4527. }
  4528.  
  4529.  
  4530. /**
  4531. * 下载多个pdf为一个压缩包,其中包含一个密码本
  4532. * @param {PointerEvent} event
  4533. */
  4534. async function export_zip(event) {
  4535. // 异常判断
  4536. if (!window.pdfs_data) utils.raise(`pdfs_data 不存在!`);
  4537.  
  4538. // 确认是否继续导出PDF
  4539. const page_nums = Object.keys(pdfs_data)
  4540. .map(index => parseInt(index) + 1);
  4541. const donwload = confirm(
  4542. `已经捕获 ${page_nums.length} 个页面,是否导出?\n` +
  4543. `已捕获的页码:${describe_nums(page_nums)}\n` +
  4544. `(如果某页缺失可以先多向后翻几页,然后翻回来,来重新加载它)`
  4545. );
  4546. if (!donwload) return;
  4547. // 隐藏按钮
  4548. const btn = event.target;
  4549. btn.style.display = "none";
  4550.  
  4551. // 下载压缩包
  4552. await data_to_zip(pdfs_data);
  4553.  
  4554. // 显示按钮
  4555. btn.style.display = "block";
  4556. }
  4557.  
  4558.  
  4559. function steal_pdf_when_page_loaded() {
  4560. // 共用变量
  4561. // 存放pdf数据,[[<pdf_blob>, <pwd_str>], ...]
  4562. window.pdfs_data = [];
  4563. // 代表当前页码
  4564. let page_no = NaN;
  4565.  
  4566. // hook PdfLoadingTask.prototype.start
  4567. const _start = PdfLoadingTask.prototype.start;
  4568. wk$._start = _start;
  4569. PdfLoadingTask.prototype.start = function() {
  4570. // 取得页码
  4571. page_no = this.index;
  4572.  
  4573. // 如果不存在此页,则准备捕获此页面
  4574. if (!pdfs_data[page_no - 1]) {
  4575. pdfs_data[page_no - 1] = [];
  4576. }
  4577. return _start.call(this);
  4578. };
  4579.  
  4580. // hook getBlob
  4581. const _get_blob = getBlob;
  4582. wk$._get_blob = _get_blob;
  4583. window.getBlob = async function(param) {
  4584. const result = await _get_blob.call(this, param);
  4585. // 如果当前页面需要捕获,则设置对应项的密码
  4586. if (page_no > 0) {
  4587. const resp = await fetch(result.url);
  4588. const blob = await resp.blob();
  4589.  
  4590. pdfs_data[page_no - 1] = [blob, result.password];
  4591. page_no = NaN;
  4592. }
  4593. return result;
  4594. };
  4595.  
  4596. utils.onclick(export_zip, 1, "导出PDF压缩包");
  4597. }
  4598.  
  4599.  
  4600. /**
  4601. * 请求 url 并将资源转为 [pdf_blob, password_str]
  4602. * @param {string} url
  4603. * @returns {Array}
  4604. */
  4605. async function url_to_item(url) {
  4606. // 取得pdf数据
  4607. const resp = await fetch(url);
  4608. const buffer = await resp.arrayBuffer();
  4609. const bytes = new Uint8Array(buffer);
  4610. const len = bytes.length;
  4611.  
  4612. // 更新进度
  4613. window.downloaded_count++;
  4614. window.downloaded_size += len;
  4615. console.log(
  4616. `已经下载了 ${downloaded_count} 页,\n` +
  4617. `累计下载了 ${(downloaded_size / 1024 / 1024).toFixed(1)} MB`
  4618. );
  4619.  
  4620. // 取出密钥
  4621. const pwd = new Uint8Array(6);
  4622. pwd.set(bytes.subarray(1080, 1083));
  4623. pwd.set(bytes.subarray(-1003, -1000), 3);
  4624. const pwd_str = new TextDecoder().decode(pwd);
  4625.  
  4626. // 解密出数据
  4627. const pdf = bytes.subarray(1083, -1003);
  4628. pdf.subarray(0, 4000).forEach((byte, i) => {
  4629. pdf[i] = 255 - byte;
  4630. });
  4631. return [
  4632. new Blob([pdf, pdf.subarray(4000)], { type: "application/pdf" }),
  4633. pwd_str
  4634. ];
  4635. }
  4636.  
  4637.  
  4638. /**
  4639. * 直接下载并解析原始数据,导出PDF压缩包
  4640. * @param {PointerEvent} event
  4641. */
  4642. async function donwload_zip(event) {
  4643. // 隐藏按钮
  4644. const btn = event.target;
  4645. btn.style.display = "none";
  4646. // 共用进度变量
  4647. window.downloaded_count = 0;
  4648. window.downloaded_size = 0;
  4649.  
  4650. // 取得数据地址
  4651. const urls = get_img_urls()
  4652. .map(url => url.replace("/thumb/", "/content-page/"));
  4653. // 批量下载
  4654. const item_tasks = urls.map(url_to_item);
  4655. const items = await utils.gather(item_tasks);
  4656. // 导出ZIP
  4657. await data_to_zip(items);
  4658.  
  4659. // 显示按钮
  4660. btn.style.display = "block";
  4661. }
  4662.  
  4663.  
  4664. /**
  4665. * 导出图片到PDF
  4666. */
  4667. function judge_file_type() {
  4668. const ext = window
  4669. ?.htmlConfig
  4670. ?.fliphtml5_pages[0]
  4671. ?.n[0]
  4672. ?.split("?")[0]
  4673. ?.split(".").at(-1);
  4674.  
  4675. console.log("ext:", ext);
  4676.  
  4677. if (["zip"].includes(ext)
  4678. && window?.PdfLoadingTask
  4679. && window?.getBlob) {
  4680.  
  4681. utils.onclick(steal_pdf_when_page_loaded, 1, "开始捕获");
  4682. utils.onclick(donwload_zip, 2, "下载PDF压缩包");
  4683. utils.toggle_btn(2);
  4684. }
  4685. else if (wk$("#page1 img")[0]) {
  4686. utils.onclick(imgs_to_pdf, 1, "导出PDF");
  4687. }
  4688. else {
  4689. utils.onclick(() => null, 1, "此文档不适用");
  4690. }
  4691. }
  4692.  
  4693.  
  4694. /**
  4695. * 云展网文档下载策略
  4696. */
  4697. async function yunzhan365() {
  4698. // 根据网址分别处理
  4699. if (location.pathname.startsWith("/basic")) {
  4700. return;
  4701. }
  4702.  
  4703. // 创建脚本启动按钮
  4704. utils.create_btns();
  4705. judge_file_type();
  4706. }
  4707.  
  4708. /**
  4709. * 截取当前对话,返回canvas
  4710. * @param {Function} once 一次性任务
  4711. * @returns {Promise<HTMLCanvasElement>}
  4712. */
  4713. async function captureChat(once) {
  4714. // 执行一次性任务
  4715. await once();
  4716. return html2canvas(window.chat)
  4717. }
  4718.  
  4719.  
  4720. /**
  4721. * 导出图片到文件和剪贴板
  4722. * @param {HTMLCanvasElement} canvas
  4723. */
  4724. async function exportCanvas(canvas) {
  4725. const blob = await utils.canvas_to_blob(canvas);
  4726. utils.save("BingAI对话.png", blob);
  4727. utils.copy(blob);
  4728. }
  4729.  
  4730.  
  4731. /**
  4732. * 截图并导出对话
  4733. */
  4734. async function exportChatAsImage() {
  4735. // 定义一次性任务
  4736. const once = utils.once(() => {
  4737. // 移除video元素
  4738. utils.remove_multi(wk$("video"));
  4739. // 移除对话外元素
  4740. [...document.body.children].slice(2).forEach(
  4741. elem => utils.remove(elem)
  4742. );
  4743. // 移除顶部欢迎
  4744. utils.remove(
  4745. wk$.call(window.chat, "cib-welcome-container")[0]
  4746. );
  4747. });
  4748.  
  4749. return exportCanvas(
  4750. await captureChat(once)
  4751. );
  4752. }
  4753.  
  4754.  
  4755. /**
  4756. * 初始化工作
  4757. */
  4758. async function init() {
  4759. const
  4760. box1 = (await wk$$(".cib-serp-main"))[0],
  4761. box2 = (await wk$$.call(box1.shadowRoot, "#cib-conversation-main"))[0],
  4762. chat = (await wk$$.call(box2.shadowRoot, "#cib-chat-main"))[0],
  4763. outer_bar = (await wk$$.call(box1.shadowRoot, "#cib-action-bar-main"))[0],
  4764. bar = (await wk$$.call(outer_bar.shadowRoot, ".root"))[0],
  4765. ori_btn = (await wk$$.call(bar, ".outside-left-container"))[0];
  4766. window.chat = chat;
  4767. if (!(ori_btn instanceof HTMLElement)) {
  4768. throw Error(`${ori_btn} 的类型不是 HTMLElement!`);
  4769. }
  4770.  
  4771. // 复制按钮
  4772. let button = ori_btn.cloneNode(false);
  4773. window.button = button;
  4774. // 更换按钮图标
  4775. button.innerHTML = `
  4776. <div class="button-compose-wrapper">
  4777. <button class="button-compose" type="button" aria-label="新主题" collapsed="">
  4778. <div class="button-compose-content">
  4779. <div size="32" style="--icon-size:32px;" class="button-compose-icon">
  4780. <svg version="1.0" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32" preserveAspectRatio="xMidYMid meet">
  4781. <metadata>
  4782. Created by potrace 1.15, written by Peter Selinger 2001-2017, edited by Allen Lv 2023
  4783. </metadata>
  4784. <g transform="translate(4.7, 4.7) scale(0.02, 0.02)">
  4785. <g transform="translate(0.000000,1280.000000) scale(0.100000,-0.100000)" fill="#ffffff" stroke="none">
  4786. <path d="M3998 12784 c-62 -33 -58 154 -58 -2439 l0 -2365 -800 0 -800 0 -31
  4787. -28 c-39 -35 -43 -89 -8 -136 66 -90 3316 -4165 3340 -4188 22 -21 39 -28 69
  4788. -28 29 0 47 7 70 28 16 15 779 968 1695 2117 1163 1460 1667 2099 1671 2121 7
  4789. 36 -8 70 -40 96 -19 16 -78 18 -823 18 l-803 0 0 2375 0 2375 -29 32 -29 33
  4790. -1699 2 c-1488 2 -1701 0 -1725 -13z"></path>
  4791. <path d="M1687 4649 l-288 -8 -32 -27 c-17 -15 -332 -286 -699 -602 l-668
  4792. -575 0 -1719 0 -1718 5630 0 5630 0 0 1718 -1 1717 -700 603 -700 603 -252 5
  4793. c-139 3 -267 7 -284 9 l-33 4 0 -800 0 -799 -3645 0 c-2423 0 -3645 -3 -3645
  4794. -10 0 -5 -3 -10 -7 -10 -5 0 -7 364 -5 810 1 445 -1 809 -5 808 -5 0 -137 -5
  4795. -296 -9z"></path></g></g></svg></div>
  4796. <div class="button-compose-text"><font style="vertical-align: inherit;"><font style="vertical-align: inherit;">保存对话</font></font></div></div>
  4797. </button>
  4798. <div class="button-compose-hint"><font style="vertical-align: inherit;"><font style="vertical-align: inherit;">保存对话</font></font></div>
  4799. </div>`;
  4800. // 为点击绑定回调
  4801. button
  4802. .querySelector(".button-compose-content")
  4803. .addEventListener("click", exportChatAsImage);
  4804.  
  4805. // 添加到DOM
  4806. bar.append(button);
  4807. }
  4808.  
  4809.  
  4810. /**
  4811. * BingAI对话辅助工具
  4812. */
  4813. async function bingAIChat() {
  4814. init();
  4815. }
  4816.  
  4817. /**
  4818. * 导出图片链接
  4819. */
  4820. function exportURLs$1() {
  4821. const all = parseInt(
  4822. wk$("[class*=total]")[0]
  4823. );
  4824. const imgs = wk$("[class*=imgContainer] img");
  4825. const got = imgs.length;
  4826.  
  4827. if (got < all) {
  4828. if (!confirm(
  4829. `当前浏览页数:${got},总页数:${all}\n建议浏览剩余页面以导出全部链接\n是否继续导出链接?`
  4830. )) {
  4831. return;
  4832. }
  4833. }
  4834. utils.save_urls(
  4835. imgs.map(img => img.src)
  4836. );
  4837. }
  4838.  
  4839.  
  4840. /**
  4841. * 360文库文档下载策略
  4842. */
  4843. function wenku360() {
  4844. utils.create_btns();
  4845. utils.onclick(
  4846. exportURLs$1, 1, "导出图片链接"
  4847. );
  4848.  
  4849. // utils.onclick(
  4850. // callAgent, 2, "导出PDF"
  4851. // );
  4852. // utils.toggle_btn(2);
  4853. }
  4854.  
  4855. async function getFileInfo() {
  4856. const
  4857. uid = new URL(location.href).searchParams.get("contentUid"),
  4858. resp = await fetch("https://zyjy-resource.webtrn.cn/sdk/api/u/open/getResourceDetail", {
  4859. "headers": {
  4860. "accept": "application/json, text/javascript, */*; q=0.01",
  4861. "content-type": "application/json",
  4862. },
  4863. "referrer": "https://jg.class.com.cn/",
  4864. "body": `{"params":{"contentUid":"${uid}"}}`,
  4865. "method": "POST",
  4866. }),
  4867. data = await resp.json(),
  4868. url = data["data"]["downloadUrl"],
  4869. fname = data["data"]["title"];
  4870.  
  4871. let ext;
  4872. try {
  4873. // validate the URL format
  4874. // and get the file format
  4875. ext = new URL(url).pathname.split(".").at(-1);
  4876. } catch(e) {
  4877. console.log(data);
  4878. throw new Error("API changed, the script is invalid now.");
  4879. }
  4880. return { url, fname, ext };
  4881. }
  4882.  
  4883.  
  4884. /**
  4885. * 保存文件
  4886. * @param {{fname: string, url: string, ext: string}} info
  4887. */
  4888. async function saveFile(info) {
  4889. const
  4890. resp = await fetch(info.url),
  4891. blob = await resp.blob();
  4892. utils.save(info.fname + `.${info.ext}`, blob);
  4893. }
  4894.  
  4895.  
  4896. /**
  4897. * 劫持保存网页,改为保存文件
  4898. * @param {KeyboardEvent} e
  4899. */
  4900. function onCtrlS(e) {
  4901. if (e.code === "KeyS" &&
  4902. e.ctrlKey) {
  4903. console.log("ctrl + s is captured!!");
  4904. getFileInfo().then(info => saveFile(info));
  4905.  
  4906. e.preventDefault();
  4907. e.stopImmediatePropagation();
  4908. e.stopPropagation();
  4909. }
  4910. }
  4911.  
  4912.  
  4913. /**
  4914. * 技工教育网文档策略
  4915. */
  4916. function jg() {
  4917. window.addEventListener(
  4918. "keydown", onCtrlS, true
  4919. );
  4920. }
  4921.  
  4922. async function estimateTimeCost() {
  4923. wk$(".w-page").at(-1).scrollIntoView();
  4924. await utils.sleep(1000);
  4925.  
  4926. let total = wk$("#pageNumber-text")[0].textContent.split("/")[1];
  4927. total = parseInt(total);
  4928. return confirm([
  4929. "注意,一旦开始截图就无法停止,除非刷新页面。",
  4930. "浏览器窗口最小化会导致截图提前结束!",
  4931. "建议将窗口最大化,这将【显著增大清晰度和文件体积】",
  4932. `预计耗时 ${1.1 * total} 秒,是否继续?`,
  4933. ].join("\n"));
  4934. }
  4935.  
  4936.  
  4937. /**
  4938. * 逐页捕获canvas
  4939. * @returns {Promise<Array<Blob>>}
  4940. */
  4941. async function collectAll() {
  4942. const imgs = [];
  4943. let div = wk$(".w-page")[0];
  4944. let i = 0;
  4945. while (true) {
  4946. // 取得 div
  4947. const anchor = Date.now();
  4948. while (!div && (Date.now() - anchor < 1000)) {
  4949. console.log(`retry on page ${i+1}`);
  4950. await utils.sleep(200);
  4951. }
  4952. if (!div) throw new Error(
  4953. `can not fetch <div>: page ${i}`
  4954. );
  4955. // 移动到 div
  4956. div.scrollIntoView({ behavior: "smooth" });
  4957. await utils.sleep(1000);
  4958. // 取得 canvas
  4959. let canvas = wk$.call(div, "canvas")[0];
  4960. let j = 0;
  4961. while (!canvas && j < 100) {
  4962. div = div.nextElementSibling;
  4963. canvas = wk$.call(div, "canvas")[0];
  4964. j++;
  4965. }
  4966. if (!div) throw new Error(
  4967. `can not fetch <div>: page ${i}*`
  4968. );
  4969.  
  4970. // 存储 canvas
  4971. imgs.push(
  4972. await utils.canvas_to_blob(canvas)
  4973. );
  4974. console.log(`canvas stored: ${++i}`);
  4975.  
  4976. // 下一轮循环
  4977. div = div.nextElementSibling;
  4978. if (!div) break;
  4979. }
  4980. console.log("done");
  4981. return imgs;
  4982. }
  4983.  
  4984.  
  4985. /**
  4986. * 放大或缩小文档画面
  4987. * @param {boolean} up
  4988. */
  4989. async function scale(up) {
  4990. let s = "#magnifyBtn";
  4991. if (!up) {
  4992. s = "#shrinkBtn";
  4993. }
  4994. const btn = wk$(s)[0];
  4995. for (let _ of utils.range(10)) {
  4996. btn.click();
  4997. await utils.sleep(500);
  4998. }
  4999. }
  5000.  
  5001.  
  5002. /**
  5003. * 获取全部canvas,显示功能按钮
  5004. * @returns
  5005. */
  5006. async function prepare() {
  5007. if (! await estimateTimeCost()) {
  5008. return;
  5009. }
  5010.  
  5011. // 隐藏按钮
  5012. utils.toggle_btn(1);
  5013. // 放大画面
  5014. await scale(true);
  5015.  
  5016. let imgs;
  5017. try {
  5018. imgs = await collectAll();
  5019. } catch(e) {
  5020. console.error(e);
  5021. } finally {
  5022. // 缩小画面
  5023. scale(false);
  5024. }
  5025. // window.imgs = imgs;
  5026. // 显示功能按钮
  5027. const fname = "技工教育网文档";
  5028. utils.onclick(
  5029. () => utils.img_blobs_to_pdf(imgs, fname),
  5030. 2,
  5031. "导出PDF"
  5032. );
  5033. utils.toggle_btn(2);
  5034.  
  5035. utils.onclick(
  5036. () => utils.blobs_to_zip(imgs, "page", "png", fname),
  5037. 3,
  5038. "导出ZIP"
  5039. );
  5040. utils.toggle_btn(3);
  5041. }
  5042.  
  5043.  
  5044. /**
  5045. * 技工教育文档预览页面策略
  5046. */
  5047. function jgPreview() {
  5048. utils.create_btns();
  5049. utils.onclick(
  5050. prepare, 1, "截图文档"
  5051. );
  5052. }
  5053.  
  5054. /**
  5055. * 取得文档标题
  5056. * @returns {string}
  5057. */
  5058. function getTitle() {
  5059. return document.title.slice(0, -4);
  5060. }
  5061.  
  5062.  
  5063. /**
  5064. * 取得基础URL
  5065. * @returns {string}
  5066. */
  5067. function getBaseURL$1() {
  5068. return wk$("#dp")[0].value;
  5069. }
  5070.  
  5071.  
  5072. /**
  5073. * 获取总页码
  5074. * @returns {number}
  5075. */
  5076. function getTotalPageNum() {
  5077. const num = wk$(".shop3 > li:nth-child(3)")[0]
  5078. .textContent
  5079. .split("/")[1]
  5080. .trim();
  5081. return parseInt(num);
  5082. }
  5083.  
  5084.  
  5085. /**
  5086. * 返回图片链接生成器
  5087. * @param {string} base 基础图片链接地址
  5088. * @param {number} max 最大数量
  5089. * @returns {Generator<string, void, unknown>}
  5090. */
  5091. function* imgURLsMaker(base, max) {
  5092. for (let i of utils.range(1, max + 1)) {
  5093. yield `${base}${i}.gif`;
  5094. }
  5095. }
  5096.  
  5097.  
  5098. /**
  5099. * 取得当前页面全部图片链接(生成器)
  5100. * @returns {Generator<string, void, unknown>}
  5101. */
  5102. function getImgURLs() {
  5103. const
  5104. base = getBaseURL$1(),
  5105. total = getTotalPageNum();
  5106. return imgURLsMaker(base, total)
  5107. }
  5108.  
  5109.  
  5110. function exportPDF() {
  5111. const urls = getImgURLs();
  5112. const title = getTitle();
  5113. return utils.run_with_prog(
  5114. 2, () => utils.img_urls_to_pdf(urls, title)
  5115. );
  5116. }
  5117.  
  5118.  
  5119. function exportURLs() {
  5120. const urls = getImgURLs();
  5121. utils.save_urls(urls);
  5122. }
  5123.  
  5124.  
  5125. /**
  5126. * 文库吧文档下载策略
  5127. */
  5128. function wenkub() {
  5129. utils.create_btns();
  5130. utils.onclick(
  5131. exportURLs, 1, "导出图片链接"
  5132. );
  5133.  
  5134. utils.onclick(
  5135. exportPDF, 2, "导出PDF(测试)"
  5136. );
  5137. utils.toggle_btn(2);
  5138. }
  5139.  
  5140. function* pageURLGen() {
  5141. const
  5142. url = new URL(location.href),
  5143. params = url.searchParams,
  5144. base = url.origin + (window.basePath || "/manuscripts/pdf"),
  5145. type = params.get("type") || "pdf",
  5146. id = params.get("id")
  5147. || new URL(wk$("#pdfContent")[0].src).searchParams.get("id")
  5148. || utils.raise("书本ID未知");
  5149. let i = 0;
  5150. if (window.wk_sklib_url) {
  5151. console.log(`sklib 使用自定义 url: ${window.wk_sklib_url}`);
  5152.  
  5153. while (true) {
  5154. yield [i, window.wk_sklib_url.replace("{id}", id).replace("{index}", `${i++}`)];
  5155. }
  5156. } else {
  5157. while (true) {
  5158. yield [i, `${base}/data/${type}/${id}/${i++}?random=null`];
  5159. }
  5160. }
  5161.  
  5162. }
  5163.  
  5164.  
  5165. /**
  5166. * 下载所有pdf文件数据,返回字节串数组
  5167. * @returns {Promise<Array<Uint8Array>>}
  5168. */
  5169. async function fetch_all_pdfs() {
  5170. // 如果已经下载完成,则直接返回之前的结果
  5171. if (window.download_finished) {
  5172. return window.pdfs;
  5173. }
  5174.  
  5175. // 显示进度的按钮
  5176. const prog_btn = utils.btn(3);
  5177. window.download_finished = false;
  5178.  
  5179. // 存储pdf字节串
  5180. const pdfs = [];
  5181. let
  5182. last_digest = NaN,
  5183. size = NaN;
  5184.  
  5185. // 读取每个PDF的页数
  5186. if (window.loadPdfInfo) {
  5187. try {
  5188. const resp = await loadPdfInfo();
  5189. const info = JSON.parse(resp.data);
  5190. size = parseInt(info.size) || size;
  5191. } catch(e) {
  5192. console.error(e);
  5193. }
  5194. }
  5195.  
  5196. for (const [i, url] of pageURLGen()) {
  5197. // 取得数据
  5198. const b64_data = await fetch(url).then(resp => resp.text());
  5199. // 如果获取完毕,则退出
  5200. if (!b64_data.length) break;
  5201. // 计算摘要
  5202. const digest = utils.crc32(b64_data);
  5203. // 如果摘要重复了,说明到达最后一页,退出
  5204. if (digest === last_digest) break;
  5205. // 否则继续
  5206. last_digest = digest;
  5207. pdfs.push(
  5208. utils.b64_to_bytes(b64_data)
  5209. );
  5210.  
  5211. // 更新进度
  5212. const progress = `已经获取 ${i + 1} 组页面,每组`
  5213. + (size ? ` ${size} 页` : '页数未知');
  5214. console.info(progress);
  5215. prog_btn.textContent = `${i + 1} / ${size} 页`;
  5216. }
  5217.  
  5218. window.pdfs = pdfs;
  5219. window.download_finished = true;
  5220. return pdfs;
  5221. }
  5222.  
  5223.  
  5224. /**
  5225. * @param {Function} async_fn
  5226. * @returns {Function}
  5227. */
  5228. function toggle_dl_btn_wrapper(async_fn) {
  5229. return async function() {
  5230. utils.toggle_btn(1);
  5231. utils.toggle_btn(2);
  5232. await async_fn();
  5233. utils.toggle_btn(1);
  5234. utils.toggle_btn(2);
  5235. }
  5236. }
  5237.  
  5238.  
  5239. async function downloadPDF() {
  5240. alert(
  5241. "如果看不到进度条请使用开发者工具(F12)查看日志\n" +
  5242. "如果文档页数过多可能导致合并PDF失败\n" +
  5243. "此时请使用【下载PDF数据集】按钮"
  5244. );
  5245.  
  5246. const pdfs = await fetch_all_pdfs();
  5247. const combined = await utils.join_pdfs(pdfs);
  5248. utils.save(
  5249. document.title + ".pdf",
  5250. combined,
  5251. "application/pdf"
  5252. );
  5253. }
  5254. downloadPDF = toggle_dl_btn_wrapper(downloadPDF);
  5255.  
  5256.  
  5257. async function download_data_bundle() {
  5258. alert(
  5259. "下载的是 <文档名称>.dat 数据集\n" +
  5260. "等价于若干 PDF 的文件顺序拼接\n" +
  5261. "请使用工具切割并合并为一份 PDF\n" +
  5262. "工具(pdfs-merger)链接在脚本主页"
  5263. );
  5264.  
  5265. const pdfs = await fetch_all_pdfs();
  5266. const blob = new Blob(pdfs, { type: "application/octet-stream" });
  5267. const url = URL.createObjectURL(blob);
  5268. const a = document.createElement("a");
  5269. a.download = document.title + ".dat";
  5270. a.href = url;
  5271. a.click();
  5272.  
  5273. URL.revokeObjectURL(url);
  5274. console.log("pdf数据集", blob);
  5275. }
  5276. download_data_bundle = toggle_dl_btn_wrapper(download_data_bundle);
  5277.  
  5278.  
  5279. /**
  5280. * 中国社会科学文库文档策略
  5281. */
  5282. function sklib() {
  5283. // 创建按钮区
  5284. utils.create_btns();
  5285. // 设置功能
  5286. utils.onclick(downloadPDF, 1, "下载PDF");
  5287. utils.onclick(download_data_bundle, 2, "下载PDF数据集");
  5288. utils.onclick(() => false, 3, "进度条");
  5289. // 显示按钮
  5290. utils.toggle_btn(2);
  5291. utils.toggle_btn(3);
  5292. // 设置按钮样式
  5293. utils.btn(3).style.pointerEvents = "none";
  5294. }
  5295.  
  5296. /**
  5297. * 返回基础图片地址,接上 <页码>.gif 即为完整URL
  5298. * @returns {string}
  5299. */
  5300. function getBaseURL() {
  5301. const
  5302. elem = wk$("#page_1 img")[0],
  5303. src = elem.src;
  5304.  
  5305. if (!src) {
  5306. alert("当前页面不能解析!");
  5307. return;
  5308. }
  5309. if (!src.endsWith("1.gif")) {
  5310. alert("当前文档不能解析!");
  5311. throw new Error("第一页图片不以 1.gif 结尾");
  5312. }
  5313. return src.slice(0, -5);
  5314. }
  5315.  
  5316.  
  5317. function* imgURLGen() {
  5318. const
  5319. base = getBaseURL(),
  5320. max = parseInt(
  5321. // ' / 23 '
  5322. wk$(".counts")[0].textContent.split("/")[1]
  5323. );
  5324.  
  5325. for (const i of utils.range(1, max + 1)) {
  5326. yield `${base}${i}.gif`;
  5327. }
  5328. }
  5329.  
  5330.  
  5331. function getURLs() {
  5332. utils.save_urls(
  5333. imgURLGen()
  5334. );
  5335. }
  5336.  
  5337.  
  5338. function jinchutou() {
  5339. utils.create_btns();
  5340. utils.onclick(
  5341. getURLs, 1, "导出图片链接"
  5342. );
  5343. }
  5344.  
  5345. // http://www.nrsis.org.cn/mnr_kfs/file/read/55806d6159b7d8e19e633f05fa62fefa
  5346.  
  5347.  
  5348. function get_pdfs() {
  5349. // 34
  5350. const size = window?.Page.size;
  5351. if (!size) utils.raise("无法确定总页码");
  5352.  
  5353. // '/mnr_kfs/file/readPage'
  5354. const path = window
  5355. ?.loadPdf
  5356. .toString()
  5357. .match(/url:'(.+?)',/)[1];
  5358. if (!path) utils.raise("无法确定PDF路径");
  5359.  
  5360. const code = location.pathname.split("/").at(-1);
  5361.  
  5362. const tasks = [...utils.range(1, size + 1)].map(
  5363. async i => {
  5364. const resp = await fetch(path + "?wk=true", {
  5365. "headers": {
  5366. "content-type": "application/x-www-form-urlencoded; charset=UTF-8",
  5367. },
  5368. "body": `code=${code}&page=${i}`,
  5369. "method": "POST",
  5370. });
  5371.  
  5372. if (!resp.ok) utils.raise(`第 ${i} 页获取失败!`);
  5373. utils.update_popup(`已经获取第 ${i} 页`);
  5374.  
  5375. const b64_str = await resp.text();
  5376. return utils.b64_to_bytes(b64_str);
  5377. }
  5378. );
  5379. return utils.gather(tasks);
  5380. }
  5381.  
  5382.  
  5383. function get_title() {
  5384. return document.title.slice(0, -5);
  5385. }
  5386.  
  5387.  
  5388. function download_pdf() {
  5389. utils.run_with_prog(1, async () => {
  5390. const pdfs = await get_pdfs();
  5391. debugger;
  5392. const pdf = await utils.join_pdfs(pdfs);
  5393. utils.save(
  5394. get_title(), pdf, "application/pdf"
  5395. );
  5396. });
  5397. }
  5398.  
  5399.  
  5400. function add_style() {
  5401. const style = `
  5402. <style>
  5403. #nprogress .nprogress-spinner-icon.forbidden {
  5404. border-top-color: #b171ff;
  5405. border-left-color: #bf8aff;
  5406. animation: nprogress-spinner 2.4s linear infinite;
  5407. }
  5408. </style>
  5409. `;
  5410. document.body.insertAdjacentHTML(
  5411. "beforeend", style
  5412. );
  5413. }
  5414.  
  5415.  
  5416. function init_forbid_origin_pdf_fetch() {
  5417. console.log("hooked xhr.open");
  5418.  
  5419. // 修改转圈图标
  5420. wk$(".nprogress-spinner-icon")[0]
  5421. .classList.add("forbidden");
  5422.  
  5423. const open = XMLHttpRequest.prototype.open;
  5424.  
  5425. // 重写 XMLHttpRequest.prototype.open 方法
  5426. XMLHttpRequest.prototype.open = function() {
  5427. const args = Array.from(arguments);
  5428. const url = args[1];
  5429.  
  5430. if (!(url.includes("readPage") &&
  5431. !url.includes("wk=true")
  5432. )) return;
  5433. this.send = () => undefined;
  5434. open.apply(this, args);
  5435. };
  5436.  
  5437. return function regain_open() {
  5438. const url = new URL(location.href);
  5439. url.searchParams.set("intercept", "0");
  5440. location.assign(url.toString());
  5441. }
  5442. }
  5443.  
  5444.  
  5445. /**
  5446. * nrsis 文档策略
  5447. */
  5448. function nrsis() {
  5449. utils.create_btns();
  5450. utils.onclick(download_pdf, 1, "下载PDF");
  5451. if (!utils.get_param("intercept")) {
  5452. add_style();
  5453. const regain_open = init_forbid_origin_pdf_fetch();
  5454. utils.onclick(regain_open, 2, "恢复页面加载");
  5455. utils.toggle_btn(2);
  5456. }
  5457. }
  5458.  
  5459. /**
  5460. * 主函数:识别网站,执行对应文档下载策略
  5461. */
  5462. function main(host=null) {
  5463. // 绑定函数到全局
  5464. window.wk_main = main;
  5465.  
  5466. // 显示当前位置
  5467. host = host || location.hostname;
  5468. const params = new URL(location.href).searchParams;
  5469. console.log(`当前 host: ${host}`);
  5470.  
  5471. if (host.includes("docin.com")) {
  5472. docin();
  5473. } else if (host === "swf.ishare.down.sina.com.cn") {
  5474. if (params.get("wk") === "true") {
  5475. ishareData2();
  5476. } else {
  5477. ishareData();
  5478. }
  5479. } else if (host.includes("ishare.iask")) {
  5480. ishare();
  5481. } else if (host === "www.deliwenku.com") {
  5482. deliwenku();
  5483. } else if (host.includes("file") && host.includes("deliwenku.com")) {
  5484. deliFile();
  5485. } else if (host === "www.doc88.com") {
  5486. doc88();
  5487. } else if (host === "www.360doc.com") {
  5488. doc360();
  5489. } else if (host === "doc.mbalib.com") {
  5490. mbalib();
  5491. } else if (host === "www.dugen.com") {
  5492. dugen();
  5493. } else if (host === "c.gb688.cn") {
  5494. gb688();
  5495. } else if (host === "www.safewk.com") {
  5496. safewk();
  5497. } else if (host.includes("book118.com")) {
  5498. book118();
  5499. } else if (host === "www.renrendoc.com") {
  5500. renrendoc();
  5501. } else if (host.includes("yunzhan365.com")) {
  5502. yunzhan365();
  5503. } else if (host === "www.bing.com") {
  5504. bingAIChat();
  5505. } else if (host === "wenku.so.com") {
  5506. wenku360();
  5507. } else if (host === "jg.class.com.cn") {
  5508. jg();
  5509. } else if (host === "preview.imm.aliyuncs.com") {
  5510. jgPreview();
  5511. } else if (host === "www.wenkub.com") {
  5512. wenkub();
  5513. } else if (/www([.-])sklib\1cn/.test(host)) {
  5514. sklib();
  5515. } else if (host === "www.jinchutou.com") {
  5516. jinchutou();
  5517. } else if (host === "www.nrsis.org.cn") {
  5518. nrsis();
  5519. } else {
  5520. console.log("匹配到了无效网页");
  5521. }
  5522. }
  5523.  
  5524.  
  5525. setTimeout(main, 1000);
  5526.  
  5527. })();