Add group chat functions, slight refactoring

This commit is contained in:
Pedro Lopez 2019-02-20 02:42:19 -04:00
commit 1b7376885d

View file

@ -1,105 +0,0 @@
'use strict';
const EventEmitter = require('events');
const puppeteer = require('puppeteer');
const Util = require('../util/Util');
const { WhatsWebURL, UserAgent, DefaultOptions, Events } = require('../util/Constants');
const { ExposeStore, LoadExtraProps } = require('../util/Injected');
const Chat = require('../models/Chat');
const Message = require('../models/Message')
/**
* Starting point for interacting with the WhatsApp Web API
* @extends {EventEmitter}
*/
class Client extends EventEmitter {
constructor(options = {}) {
super();
this.options = Util.mergeDefault(DefaultOptions, options);
this.pupBrowser = null;
this.pupPage = null;
}
async initialize() {
const browser = await puppeteer.launch(this.options.puppeteer);
const page = await browser.newPage();
page.setUserAgent(UserAgent);
await page.goto(WhatsWebURL);
await page.evaluate(ExposeStore);
// Wait for QR Code
await page.waitForSelector('._1jjYO');
const qr = await page.$eval('._2EZ_m', node => node.getAttribute('data-ref'));
this.emit(Events.QR_RECEIVED, qr);
// Wait for Auth
await page.waitForSelector('._2Uo0Z', {timeout: 0});
this.emit(Events.AUTHENTICATED);
// Check Store Injection
await page.waitForFunction('window.Store != undefined');
// Load extra serialized props
const models = [Chat, Message];
for (let model of models) {
await page.evaluate(LoadExtraProps, model.WAppModel, model.extraFields);
}
// Register events
await page.exposeFunction('onAddMessageEvent', msg => {
if (msg.id.fromMe || !msg.isNewMsg) return;
this.emit(Events.MESSAGE_CREATE, new Message(this, msg));
});
await page.exposeFunction('onConnectionChangedEvent', (conn, connected) => {
if (!connected) {
this.emit(Events.DISCONNECTED);
this.destroy();
}
})
await page.evaluate(() => {
Store.Msg.on('add', onAddMessageEvent);
Store.Conn.on('change:connected', onConnectionChangedEvent);
})
this.pupBrowser = browser;
this.pupPage = page;
this.emit(Events.READY);
}
async destroy() {
await this.pupBrowser.close();
}
async sendMessage(chatId, message) {
await this.pupPage.evaluate((chatId, message) => {
Store.Chat.get(chatId).sendMessage(message);
}, chatId, message)
}
async getChats() {
let chats = await this.pupPage.evaluate(() => {
return Store.Chat.serialize()
});
return chats.map(chatData => new Chat(this, chatData));
}
async getChatById(chatId) {
let chat = await this.pupPage.evaluate(chatId => {
return Store.Chat.get(chatId).serialize();
}, chatId);
return new Chat(this, chat);
}
}
module.exports = Client;