@wppconnect-team/wppconnect
Version:
WPPConnect is an open source project developed by the JavaScript community with the aim of exporting functions from WhatsApp Web to the node, which can be used to support the creation of any interaction, such as customer service, media sending, intelligen
108 lines (107 loc) • 5.84 kB
JavaScript
;
/*
* This file is part of WPPConnect.
*
* WPPConnect is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* WPPConnect is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with WPPConnect. If not, see <https://www.gnu.org/licenses/>.
*/
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (g && (g = 0, op[0] && (_ = 0)), _) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.scrapeImg = scrapeImg;
function scrapeImg(page) {
return __awaiter(this, void 0, void 0, function () {
var click, result;
return __generator(this, function (_a) {
switch (_a.label) {
case 0: return [4 /*yield*/, page
.evaluate(function () {
var selectorImg = document.querySelector('canvas');
var selectorUrl = selectorImg.closest('[data-ref]');
//const buttonReload = selectorUrl.querySelector('[role="button"]') as HTMLButtonElement;
var buttonReload = selectorUrl.querySelector('button');
if (buttonReload != null) {
buttonReload.click();
return true;
}
return false;
})
.catch(function () { return false; })];
case 1:
click = _a.sent();
if (!click) return [3 /*break*/, 3];
return [4 /*yield*/, page.waitForFunction(function () {
var selectorImg = document.querySelector('canvas');
var selectorUrl = selectorImg.closest('[data-ref]');
return selectorUrl.getAttribute('data-ref');
})];
case 2:
_a.sent();
_a.label = 3;
case 3: return [4 /*yield*/, page
.evaluate(function () {
var selectorImg = document.querySelector('canvas');
var selectorUrl = selectorImg.closest('[data-ref]');
if (selectorImg != null && selectorUrl != null) {
var data = {
base64Image: selectorImg.toDataURL(),
urlCode: selectorUrl.getAttribute('data-ref'),
};
return data;
}
else {
return undefined;
}
})
.catch(function () { return undefined; })];
case 4:
result = _a.sent();
return [2 /*return*/, result];
}
});
});
}