mirror of
https://github.com/wassname/chatGPTBox.git
synced 2026-07-14 11:15:11 +08:00
64 lines
1.7 KiB
JavaScript
64 lines
1.7 KiB
JavaScript
import { getPossibleElementByQuerySelector } from './get-possible-element-by-query-selector.mjs'
|
|
|
|
function getArea(e) {
|
|
const rect = e.getBoundingClientRect()
|
|
return rect.width * rect.height
|
|
}
|
|
|
|
const adapters = {
|
|
'scholar.google': ['#gs_res_ccl_mid'],
|
|
google: ['#search'],
|
|
csdn: ['#content_views'],
|
|
bing: ['#b_results'],
|
|
wikipedia: ['#mw-content-text'],
|
|
}
|
|
|
|
function findLargestElement(e) {
|
|
let maxArea = 0
|
|
let largestElement = null
|
|
const limitedArea = 0.8 * getArea(e)
|
|
|
|
function traverseDOM(node) {
|
|
if (node.nodeType === Node.ELEMENT_NODE) {
|
|
const area = getArea(node)
|
|
|
|
if (area > maxArea && area < limitedArea) {
|
|
maxArea = area
|
|
largestElement = node
|
|
}
|
|
|
|
Array.from(node.children).forEach(traverseDOM)
|
|
}
|
|
}
|
|
|
|
traverseDOM(e)
|
|
return largestElement
|
|
}
|
|
|
|
export function getCoreContentText() {
|
|
for (const [siteName, selectors] of Object.entries(adapters)) {
|
|
if (location.hostname.includes(siteName)) {
|
|
const element = getPossibleElementByQuerySelector(selectors)
|
|
if (element) return element.innerText || element.textContent
|
|
break
|
|
}
|
|
}
|
|
|
|
const largestElement = findLargestElement(document.body)
|
|
const secondLargestElement = findLargestElement(largestElement)
|
|
console.log(largestElement)
|
|
console.log(secondLargestElement)
|
|
|
|
if (!largestElement) return
|
|
|
|
let ret
|
|
if (secondLargestElement && getArea(secondLargestElement) > 0.5 * getArea(largestElement)) {
|
|
ret = secondLargestElement.innerText || secondLargestElement.textContent
|
|
console.log('use second')
|
|
} else {
|
|
ret = largestElement.innerText || largestElement.textContent
|
|
console.log('use first')
|
|
}
|
|
return ret.trim().replaceAll(' ', '').replaceAll('\n\n', '').replaceAll(',,', '')
|
|
}
|