jebbs/data-extracter-extesion
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

code optimize

Browse Source
This commit is contained in:
jebbs
2018-09-27 11:57:21 +08:00
parent c3b765ca8e
commit fb2bb7b59e
2 changed files with 76 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
scripts/content.js
View File

@ -1,19 +1,22 @@
chrome.runtime.onMessage.addListener(
function (request, sender, sendResponse) {
if (!request.from) return;
let [ext, act] = request.from.split(":");
if (ext.toLowerCase() !== 'dataextracter') return;
// console.log(request);
switch (request.from) {
case "DataExtracter:Extract":
switch (act.toLowerCase()) {
case "extract":
let data = extractTabData(request.itemsSelector, request.fieldSelectors);
if (sendResponse) sendResponse(data);
break;
case "DataExtracter:GotoUrl":
case "gotourl":
window.location.replace(request.url);
if (sendResponse) sendResponse(request.url);
break;
case "DataExtracter:ReportIn":
case "reportin":
if (sendResponse) sendResponse(request.from);
break;
case "DataExtracter:QueryUrl":
case "queryurl":
if (sendResponse) sendResponse(window.location.href);
break;
default:
@ -22,19 +25,7 @@
}
);
function extractTabData(itemsSelector, fieldSelectors) {
return $(itemsSelector).toArray().map(
item => fieldSelectors.map(
selector => {
let [cls, attr] = selector.split('@').slice(0, 2);
return $(item).find(cls).toArray().map(find => attr ? find[attr] : find.textContent.trim()).join('\n')
}
)
);
}
function extract(...args) {
let sig = `
const sig = `
# DataExtracter Help
----------------------------
@ -66,7 +57,24 @@ extract(".list-item", ["a.title", "p.content"], "http://sample.com/?pn=\${page}"
### Extract link text and target (use 'selector@attribute')
extract('.list-item', ['a.title', 'a.title@href'])
### Collect links from page(s) & Extract data of each link (only available in console of extension background page)
extract('body',["a.title", "p.content"], await getData('.list-item', ['.item a@href'],["http://sample.com/abc"]))
`.trim();
function extractTabData(itemsSelector, fieldSelectors) {
return $(itemsSelector).toArray().map(
item => fieldSelectors.map(
selector => {
let [cls, attr] = selector.split('@').slice(0, 2);
return $(item).find(cls).toArray().map(find => attr ? find[attr] : find.textContent.trim()).join('\n')
}
)
);
}
function extract(...args) {
if (!testArgs(...args)) {
console.log(sig);
return;
@ -90,12 +98,41 @@ extract('.list-item', ['a.title', 'a.title@href'])
}
function testArgs(...args) {
if (args.length < 2) return false;
if (args.length == 2)
return (args[0] && args[1] && (typeof args[0] == "string") && (args[1] instanceof Array))
let urls = [];
if (args.length > 2) return (typeof args[2] == "string") && (
(args[3] instanceof Array) ||
(!isNaN(args[3]) && !isNaN(args[4]) && !isNaN(args[5]))
)
switch (args.length) {
case 0, 1:
return false;
case 2:
return args[0] && args[1] &&
(typeof args[0] == "string") &&
(args[1] instanceof Array) &&
testArrayVals(args[1], v => typeof v == "string");
case 3:
return args[0] && args[1] &&
typeof args[0] == "string" &&
args[1] instanceof Array &&
testArrayVals(args[1], v => typeof v == "string") &&
args[2] instanceof Array &&
testArrayVals(args[2], v => typeof v == "string");
case 4:
return args[0] && args[1] &&
typeof args[0] == "string" &&
args[1] instanceof Array &&
testArrayVals(args[1], v => typeof v == "string") &&
typeof args[2] == "string" &&
args[3] instanceof Array &&
testArrayVals(args[3], v => typeof v == "number");
case 5:
return args[0] && args[1] &&
typeof args[0] == "string" &&
args[1] instanceof Array &&
testArrayVals(args[1], v => typeof v == "string") &&
typeof args[2] == "string" &&
!isNaN(args[3]) && !isNaN(args[4]) && !isNaN(args[5]);
default:
return false;
}
function testArrayVals(arr, tester) {
return arr.reduce((p, c) => p && tester(c), true);
}
}