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-03 提交的版本,查看 最新版本

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