82 lines
2.5 KiB
JavaScript
82 lines
2.5 KiB
JavaScript
const crypto = require("crypto");
|
||
|
||
/**
|
||
* 获取微信读书的书籍 ID
|
||
* 感谢 @MCBBC 及 ChatGPT
|
||
* @param {string} bookId - 书籍 ID
|
||
* @returns {string} - 唯一的书籍 ID
|
||
*/
|
||
const getWereadID = (bookId) => {
|
||
try {
|
||
// 使用 MD5 哈希算法创建哈希对象
|
||
const hash = crypto.createHash("md5");
|
||
hash.update(bookId);
|
||
const str = hash.digest("hex");
|
||
|
||
// 取哈希结果的前三个字符作为初始值
|
||
let strSub = str.substring(0, 3);
|
||
|
||
// 判断书籍 ID 的类型并进行转换
|
||
let fa;
|
||
if (/^\d*$/.test(bookId)) {
|
||
// 如果书籍 ID 只包含数字，则将其拆分成长度为 9 的子字符串，并转换为十六进制表示
|
||
const chunks = [];
|
||
for (let i = 0; i < bookId.length; i += 9) {
|
||
const chunk = bookId.substring(i, i + 9);
|
||
chunks.push(parseInt(chunk).toString(16));
|
||
}
|
||
fa = ["3", chunks];
|
||
} else {
|
||
// 如果书籍 ID 包含其他字符，则将每个字符的 Unicode 编码转换为十六进制表示
|
||
let hexStr = "";
|
||
for (let i = 0; i < bookId.length; i++) {
|
||
hexStr += bookId.charCodeAt(i).toString(16);
|
||
}
|
||
fa = ["4", [hexStr]];
|
||
}
|
||
|
||
// 将类型添加到初始值中
|
||
strSub += fa[0];
|
||
|
||
// 将数字2和哈希结果的后两个字符添加到初始值中
|
||
strSub += "2" + str.substring(str.length - 2);
|
||
|
||
// 处理转换后的子字符串数组
|
||
for (let i = 0; i < fa[1].length; i++) {
|
||
const sub = fa[1][i];
|
||
const subLength = sub.length.toString(16);
|
||
|
||
// 如果长度只有一位数，则在前面添加0
|
||
const subLengthPadded = subLength.length === 1 ? "0" + subLength : subLength;
|
||
|
||
// 将长度和子字符串添加到初始值中
|
||
strSub += subLengthPadded + sub;
|
||
|
||
// 如果不是最后一个子字符串，则添加分隔符 'g'
|
||
if (i < fa[1].length - 1) {
|
||
strSub += "g";
|
||
}
|
||
}
|
||
|
||
// 如果初始值长度不足 20，从哈希结果中取足够的字符补齐
|
||
if (strSub.length < 20) {
|
||
strSub += str.substring(0, 20 - strSub.length);
|
||
}
|
||
|
||
// 使用 MD5 哈希算法创建新的哈希对象
|
||
const finalHash = crypto.createHash("md5");
|
||
finalHash.update(strSub);
|
||
const finalStr = finalHash.digest("hex");
|
||
|
||
// 取最终哈希结果的前三个字符并添加到初始值的末尾
|
||
strSub += finalStr.substring(0, 3);
|
||
|
||
return strSub;
|
||
} catch (error) {
|
||
console.error("处理微信读书 ID 时出现错误：" + error);
|
||
return null;
|
||
}
|
||
};
|
||
|
||
module.exports = getWereadID;
|