t-comm
Version:
专业、稳定、纯粹的工具库
216 lines (212 loc) • 6.94 kB
JavaScript
import _toConsumableArray from '@babel/runtime/helpers/toConsumableArray';
import _regeneratorRuntime from '@babel/runtime/regenerator';
import { _ as __awaiter } from './tslib.es6-848120af.js';
import { getOs } from './nodejs/os.mjs';
// eslint-disable-next-line @typescript-eslint/naming-convention
var DEVICE_TYPE;
(function (DEVICE_TYPE) {
DEVICE_TYPE["PC"] = "PC";
DEVICE_TYPE["MOBILE_HOR"] = "MOBILE_HORPC";
DEVICE_TYPE["MOBILE_VERT"] = "MOBILE_VERT";
})(DEVICE_TYPE || (DEVICE_TYPE = {}));
function initBrowser(_ref) {
var puppeteer = _ref.puppeteer,
_ref$args = _ref.args,
args = _ref$args === void 0 ? [] : _ref$args,
_ref$headless = _ref.headless,
headless = _ref$headless === void 0 ? true : _ref$headless,
_ref$devtools = _ref.devtools,
devtools = _ref$devtools === void 0 ? true : _ref$devtools;
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
var defaultArgs, type, isLinux, launchArgs, browser;
return _regeneratorRuntime.wrap(function (_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
defaultArgs = ['--disable-setuid-sandbox'];
type = getOs().type();
isLinux = type == 'Linux';
if (isLinux) {
defaultArgs.push('--no-sandbox');
}
launchArgs = {
headless: headless,
devtools: devtools,
args: [].concat(defaultArgs, _toConsumableArray(args))
};
_context.next = 1;
return puppeteer.launch(launchArgs);
case 1:
browser = _context.sent;
return _context.abrupt("return", browser);
case 2:
case "end":
return _context.stop();
}
}, _callee);
}));
}
function getNewPage(browser, device) {
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee2() {
var page;
return _regeneratorRuntime.wrap(function (_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
_context2.next = 1;
return browser.newPage();
case 1:
page = _context2.sent;
if (!(device === DEVICE_TYPE.MOBILE_VERT)) {
_context2.next = 3;
break;
}
_context2.next = 2;
return page.setViewport({
width: 414,
height: 736,
isMobile: true
});
case 2:
_context2.next = 4;
break;
case 3:
if (!(device === DEVICE_TYPE.MOBILE_HOR)) {
_context2.next = 4;
break;
}
_context2.next = 4;
return page.setViewport({
width: 736,
height: 414,
isMobile: true
});
case 4:
return _context2.abrupt("return", page);
case 5:
case "end":
return _context2.stop();
}
}, _callee2);
}));
}
var openedPageList = new Set();
function openOrFindPage(browser, href, device) {
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee3() {
var page, target;
return _regeneratorRuntime.wrap(function (_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
if (!openedPageList.has(href)) {
_context3.next = 5;
break;
}
_context3.prev = 1;
_context3.next = 2;
return browser.waitForTarget(function (t) {
return t.url().includes(href);
}, {
timeout: 2000
});
case 2:
target = _context3.sent;
_context3.next = 3;
return target.page();
case 3:
page = _context3.sent;
return _context3.abrupt("return", page);
case 4:
_context3.prev = 4;
_context3["catch"](1);
case 5:
_context3.next = 6;
return getNewPage(browser, device);
case 6:
page = _context3.sent;
_context3.next = 7;
return page["goto"](href);
case 7:
openedPageList.add(href);
return _context3.abrupt("return", page);
case 8:
case "end":
return _context3.stop();
}
}, _callee3, null, [[1, 4]]);
}));
}
function setUserAgent(useragent, page) {
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee4() {
return _regeneratorRuntime.wrap(function (_context4) {
while (1) switch (_context4.prev = _context4.next) {
case 0:
_context4.next = 1;
return page.setUserAgent(useragent);
case 1:
_context4.next = 2;
return page.evaluate(function () {
return location.reload();
});
case 2:
_context4.next = 3;
return page.waitForNavigation();
case 3:
case "end":
return _context4.stop();
}
}, _callee4);
}));
}
function setSessionStorage(key, value, page) {
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee5() {
return _regeneratorRuntime.wrap(function (_context5) {
while (1) switch (_context5.prev = _context5.next) {
case 0:
_context5.next = 1;
return page.evaluate(function (_ref2) {
var key = _ref2.key,
value = _ref2.value;
sessionStorage.setItem(key, value);
}, {
key: key,
value: value
});
case 1:
case "end":
return _context5.stop();
}
}, _callee5);
}));
}
function setRoute(page) {
var route = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : '/';
return __awaiter(this, void 0, void 0, /*#__PURE__*/_regeneratorRuntime.mark(function _callee6() {
return _regeneratorRuntime.wrap(function (_context6) {
while (1) switch (_context6.prev = _context6.next) {
case 0:
if (page) {
_context6.next = 1;
break;
}
return _context6.abrupt("return");
case 1:
_context6.next = 2;
return page.evaluate(function (route) {
return window.app.$router.push(route);
}, route);
case 2:
case "end":
return _context6.stop();
}
}, _callee6);
}));
}
var page = /*#__PURE__*/Object.freeze({
__proto__: null,
get DEVICE_TYPE () { return DEVICE_TYPE; },
initBrowser: initBrowser,
getNewPage: getNewPage,
openOrFindPage: openOrFindPage,
setUserAgent: setUserAgent,
setSessionStorage: setSessionStorage,
setRoute: setRoute
});
export { DEVICE_TYPE as D, setSessionStorage as a, setRoute as b, getNewPage as g, initBrowser as i, openOrFindPage as o, page as p, setUserAgent as s };