cheveguerra/whaticket-community
1
0
Fork 0
mirror of https://github.com/cheveguerra/whaticket-community.git synced 2026-04-18 03:39:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

improvement: better names on wbot methods

Browse Source
This commit is contained in:
canove
2020-09-05 15:43:52 -03:00
parent b8ff993e6f
commit 40e2e5e8a6
8 changed files with 160 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/src/app.js
View File

@@ -7,7 +7,7 @@ const cors = require("cors");
const multer = require("multer");
const Sentry = require("@sentry/node");
const wBot = require("./libs/wbot");
const { initWbot } = require("./libs/wbot");
const wbotMessageListener = require("./services/wbotMessageListener");
const wbotMonitor = require("./services/wbotMonitor");
const Whatsapp = require("./models/Whatsapp");
@@ -61,8 +61,7 @@ const startWhatsAppSessions = async () => {
if (whatsapps.length > 0) {
whatsapps.forEach(dbSession => {
wBot
.init(dbSession)
initWbot(dbSession)
.then(() => {
wbotMessageListener(dbSession);
wbotMonitor(dbSession);

2
backend/src/controllers/ImportPhoneContactsController.js
View File

@@ -1,6 +1,6 @@
const Contact = require("../models/Contact");
const { getIO } = require("../libs/socket");
const { getWbot, init } = require("../libs/wbot");
const { getWbot, initWbot } = require("../libs/wbot");
exports.store = async (req, res, next) => {
const io = getIO();

56
backend/src/controllers/WhatsAppSessionController.js
View File

@@ -1,6 +1,8 @@
const Whatsapp = require("../models/Whatsapp");
const { getIO } = require("../libs/socket");
const { getWbot } = require("../libs/wbot");
const { getWbot, initWbot, removeWbot } = require("../libs/wbot");
const wbotMessageListener = require("../services/wbotMessageListener");
const wbotMonitor = require("../services/wbotMonitor");
exports.index = async (req, res) => {
const dbSession = await Whatsapp.findAll();
@@ -8,6 +10,29 @@ exports.index = async (req, res) => {
return res.status(200).json(dbSession);
};
exports.store = async (req, res) => {
const io = getIO();
const dbSession = await Whatsapp.create(req.body);
if (!dbSession) {
return res.status(400).json({ error: "Cannot create whatsapp session." });
}
initWbot(dbSession)
.then(() => {
wbotMessageListener(dbSession);
wbotMonitor(dbSession);
})
.catch(err => console.log(err));
io.emit("session", {
action: "update",
session: dbSession,
});
return res.status(200).json({ message: "Session created sucessfully." });
};
exports.show = async (req, res) => {
const { sessionId } = req.params;
const dbSession = await Whatsapp.findByPk(sessionId);
@@ -20,6 +45,7 @@ exports.show = async (req, res) => {
};
exports.update = async (req, res) => {
const io = getIO();
const { sessionId } = req.params;
const dbSession = await Whatsapp.findByPk(sessionId);
@@ -29,15 +55,37 @@ exports.update = async (req, res) => {
}
const wbot = getWbot(dbSession.id);
const io = getIO();
wbot.logout();
await dbSession.update(req.body);
wbot.logout();
io.emit("session", {
action: "update",
session: dbSession,
});
return res.status(200).json({ message: "session disconnected" });
return res.status(200).json({ message: "Session updated" });
};
exports.delete = async (req, res) => {
const io = getIO();
const { sessionId } = req.params;
const dbSession = await Whatsapp.findByPk(sessionId);
if (!dbSession) {
return res.status(404).json({ message: "Session not found" });
}
const wbot = getWbot(dbSession.id);
await dbSession.destroy();
removeWbot(dbSession.id);
io.emit("session", {
action: "delete",
sessionId: dbSession.id,
});
return res.status(200).json({ message: "Session deleted." });
};

167
backend/src/libs/wbot.js
View File

@@ -6,90 +6,83 @@ const { getIO } = require("../libs/socket");
let sessions = [];
module.exports = {
init: async dbSession => {
const io = getIO();
const sessionName = dbSession.name;
let sessionCfg;
initWbot: async dbSession => {
try {
const io = getIO();
const sessionName = dbSession.name;
let sessionCfg;
if (dbSession && dbSession.session) {
sessionCfg = JSON.parse(dbSession.session);
if (dbSession && dbSession.session) {
sessionCfg = JSON.parse(dbSession.session);
}
const sessionIndex = sessions.findIndex(s => s.id === dbSession.id);
if (sessionIndex !== -1) {
sessions[sessionIndex].destroy();
sessions.splice(sessionIndex, 1);
}
const wbot = new Client({
session: sessionCfg,
restartOnAuthFail: true,
});
wbot.initialize();
wbot.on("qr", async qr => {
console.log("Session:", sessionName);
qrCode.generate(qr, { small: true });
await dbSession.update({ id: 1, qrcode: qr, status: "qrcode" });
io.emit("session", {
action: "update",
session: dbSession,
});
});
wbot.on("authenticated", async session => {
console.log("Session:", sessionName, "AUTHENTICATED");
await dbSession.update({
session: JSON.stringify(session),
status: "authenticated",
});
io.emit("session", {
action: "update",
session: dbSession,
});
});
wbot.on("auth_failure", async msg => {
console.error("Session:", sessionName, "AUTHENTICATION FAILURE", msg);
await dbSession.update({ session: "" });
});
wbot.on("ready", async () => {
console.log("Session:", sessionName, "READY");
await dbSession.update({
status: "CONNECTED",
qrcode: "",
});
io.emit("session", {
action: "update",
session: dbSession,
});
wbot.sendPresenceAvailable();
});
wbot.id = dbSession.id;
sessions.push(wbot);
} catch (err) {
console.log(err);
}
const sessionIndex = sessions.findIndex(s => s.id === dbSession.id);
if (sessionIndex !== -1) {
sessions[sessionIndex].destroy();
sessions.splice(sessionIndex, 1);
}
const wbot = new Client({
session: sessionCfg,
restartOnAuthFail: true,
});
wbot.initialize();
wbot.on("qr", async qr => {
console.log("Session:", sessionName);
qrCode.generate(qr, { small: true });
await dbSession.update({ id: 1, qrcode: qr, status: "qrcode" });
io.emit("session", {
action: "update",
session: dbSession,
});
});
wbot.on("authenticated", async session => {
console.log("Session:", sessionName, "AUTHENTICATED");
await dbSession.update({
session: JSON.stringify(session),
status: "authenticated",
});
io.emit("session", {
action: "update",
session: dbSession,
});
});
wbot.on("auth_failure", async msg => {
console.error("Session:", sessionName, "AUTHENTICATION FAILURE", msg);
await dbSession.update({ session: "" });
});
wbot.on("ready", async () => {
console.log("Session:", sessionName, "READY");
await dbSession.update({
status: "CONNECTED",
qrcode: "",
});
io.emit("session", {
action: "update",
session: dbSession,
});
// const chats = await wbot.getChats(); // pega as mensagens nao lidas (recebidas quando o bot estava offline)
// let unreadMessages; // todo > salvar isso no DB pra mostrar no frontend
// for (let chat of chats) {
// if (chat.unreadCount > 0) {
// unreadMessages = await chat.fetchMessages({
// limit: chat.unreadCount,
// });
// }
// }
// console.log(unreadMessages);
wbot.sendPresenceAvailable();
});
wbot.id = dbSession.id;
sessions.push(wbot);
return null;
},
@@ -103,4 +96,16 @@ module.exports = {
}
return sessions[sessionIndex];
},
removeWbot: sessionId => {
try {
const sessionIndex = sessions.findIndex(s => s.id === sessionId);
if (sessionIndex !== -1) {
sessions[sessionIndex].destroy();
sessions.splice(sessionIndex, 1);
}
} catch (err) {
console.log(err);
}
},
};

8
backend/src/router/routes/whatsapp.js
View File

@@ -7,6 +7,8 @@ const routes = express.Router();
routes.get("/whatsapp/session/", isAuth, WhatsAppSessionController.index);
routes.post("/whatsapp/session", isAuth, WhatsAppSessionController.store);
routes.get(
"/whatsapp/session/:sessionId",
isAuth,
@@ -19,4 +21,10 @@ routes.put(
WhatsAppSessionController.update
);
routes.delete(
"/whatsapp/session/:sessionId",
isAuth,
WhatsAppSessionController.delete
);
module.exports = routes;

2
backend/src/services/wbotMessageListener.js
View File

@@ -9,7 +9,7 @@ const Ticket = require("../models/Ticket");
const Message = require("../models/Message");
const { getIO } = require("../libs/socket");
const { getWbot, init } = require("../libs/wbot");
const { getWbot, initWbot } = require("../libs/wbot");
const verifyContact = async (msgContact, profilePicUrl) => {
let contact = await Contact.findOne({

4
backend/src/services/wbotMonitor.js
View File

@@ -3,7 +3,7 @@ const Sentry = require("@sentry/node");
const wbotMessageListener = require("./wbotMessageListener");
const { getIO } = require("../libs/socket");
const { getWbot, init } = require("../libs/wbot");
const { getWbot, initWbot } = require("../libs/wbot");
const wbotMonitor = dbSession => {
const io = getIO();
@@ -61,7 +61,7 @@ const wbotMonitor = dbSession => {
setTimeout(
() =>
init(dbSession)
initWbot(dbSession)
.then(() => {
wbotMessageListener(dbSession);
wbotMonitor(dbSession);