Add support to get book info from book info page

This commit is contained in:
2026-02-16 11:44:32 +08:00
parent ab46dcc2c1
commit 0cf5eccaf1
8 changed files with 204 additions and 1 deletions

103
src/qdbook.ts Normal file
View File

@@ -0,0 +1,103 @@
import type { BookGData, QdBookTag } from "./qdtypes";
import type { SendMessage, Message } from "./types";
import { QdBookTagType } from "./qdtypes";
let g_data: BookGData | undefined;
function get_book_name() {
const bookName = document.getElementById('bookName') as HTMLHeadingElement | null;
if (!bookName) {
throw new Error('Failed to find book name element');
}
return bookName.innerText.trim();
}
function get_book_tags() {
const tags: QdBookTag[] = [];
const attribute = document.querySelector('p.book-attribute');
if (attribute) {
for (const children of attribute.children) {
if (children instanceof HTMLSpanElement) {
if (children.classList.contains("dot")) {
continue;
}
tags.push({
type: QdBookTagType.System,
name: children.innerText.trim(),
});
} else if (children instanceof HTMLAnchorElement) {
tags.push({
type: QdBookTagType.Category,
name: children.innerText.trim(),
url: children.href,
});
} else {
throw new Error(`Unknown tag element: ${children.outerHTML}`);
}
}
}
const allLabel = document.querySelector('p.all-label');
if (allLabel) {
for (const label of allLabel.children) {
if (label instanceof HTMLAnchorElement) {
tags.push({
type: QdBookTagType.User,
name: label.innerText.trim(),
url: label.href,
});
} else {
throw new Error(`Unknown user tag element: ${label.outerHTML}`);
}
}
}
return tags;
}
window.addEventListener('message', (event) => {
const data = event.data;
if (data && data['@type'] === 'g_data') {
g_data = data.g_data;
console.log('Received g_data:', g_data);
}
});
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
const m = message as SendMessage;
try {
if (m.type === 'GetQdBookInfo') {
if (!g_data) {
const msg: Message = {
ok: false,
code: 404,
msg: '没有找到g_data,可能是页面数据还没有加载完成,请稍后再试或者尝试刷新页面',
for: m.type,
};
sendResponse(msg);
return;
}
const bookName = get_book_name();
const msg: Message = {
ok: true,
code: 0,
body: {
type: 'QdBookInfo',
bookInfo: g_data,
bookName,
id: g_data.pageJson.bookId,
tags: get_book_tags(),
},
for: m.type,
};
sendResponse(msg);
}
} catch (e) {
console.error(e);
const msg: Message = {
ok: false,
code: 500,
msg: e instanceof Error ? e.message : 'Unknown error',
for: m.type,
};
sendResponse(msg);
}
});