Backend folders organize and finshed send audio

backend/src/services/wbotMessageListener.js

@@ -0,0 +1,138 @@
+const Contact = require("../models/Contact");
+const Message = require("../models/Message");
+
+const path = require("path");
+const fs = require("fs");
+
+const { getIO } = require("../libs/socket");
+const { getWbot, init } = require("../libs/wbot");
+
+const wbotMessageListener = () => {
+	const io = getIO();
+	const wbot = getWbot();
+
+	wbot.on("message", async msg => {
+		console.log(msg);
+		let newMessage;
+		// console.log(msg);
+		if (msg.from === "status@broadcast") {
+			return;
+		}
+		try {
+			const msgContact = await msg.getContact();
+			const profilePicUrl = await msgContact.getProfilePicUrl();
+			try {
+				let contact = await Contact.findOne({
+					where: { number: msgContact.number },
+				});
+
+				if (contact) {
+					await contact.update({ profilePicUrl: profilePicUrl });
+				}
+
+				if (!contact) {
+					try {
+						contact = await Contact.create({
+							name: msgContact.pushname || msgContact.number.toString(),
+							number: msgContact.number,
+							profilePicUrl: profilePicUrl,
+						});
+					} catch (err) {
+						console.log(err);
+					}
+				}
+
+				if (msg.hasQuotedMsg) {
+					const quotedMessage = await msg.getQuotedMessage();
+					console.log("quoted", quotedMessage);
+				}
+
+				if (msg.hasMedia) {
+					const media = await msg.downloadMedia();
+
+					if (media) {
+						if (!media.filename) {
+							let ext = media.mimetype.split("/")[1].split(";")[0];
+							media.filename = `${new Date().getTime()}.${ext}`;
+						}
+
+						fs.writeFile(
+							path.join(__dirname, "..", "public", media.filename),
+							media.data,
+							"base64",
+							err => {
+								console.log(err);
+							}
+						);
+
+						newMessage = await contact.createMessage({
+							id: msg.id.id,
+							messageBody: msg.body || media.filename,
+							mediaUrl: media.filename,
+							mediaType: media.mimetype.split("/")[0],
+						});
+						await contact.update({ lastMessage: msg.body || media.filename });
+					}
+				} else {
+					newMessage = await contact.createMessage({
+						id: msg.id.id,
+						messageBody: msg.body,
+					});
+					await contact.update({ lastMessage: msg.body });
+				}
+
+				io.to(contact.id)
+					.to("notification")
+					.emit("appMessage", {
+						action: "create",
+						message: {
+							...newMessage.dataValues,
+							mediaUrl: `${
+								newMessage.mediaUrl
+									? `http://${process.env.HOST}:${process.env.PORT}/public/${newMessage.mediaUrl}`
+									: ""
+							}`,
+						},
+						contact: {
+							...contact.dataValues,
+							unreadMessages: 1,
+							lastMessage: newMessage.messageBody,
+						},
+					});
+
+				let chat = await msg.getChat();
+				chat.sendSeen();
+			} catch (err) {
+				console.log(err);
+			}
+		} catch (err) {
+			console.log(err);
+		}
+	});
+
+	wbot.on("message_ack", async (msg, ack) => {
+		try {
+			const messageToUpdate = await Message.findOne({
+				where: { id: msg.id.id },
+			});
+			if (!messageToUpdate) {
+				// will throw an error is msg wasn't sent from app
+				const error = new Error(
+					"Erro ao alterar o ack da mensagem no banco de dados"
+				);
+				error.statusCode = 501;
+				throw error;
+			}
+			await messageToUpdate.update({ ack: ack });
+
+			io.to(messageToUpdate.contactId).emit("appMessage", {
+				action: "update",
+				message: messageToUpdate,
+			});
+		} catch (err) {
+			console.log(err);
+		}
+	});
+};
+
+module.exports = wbotMessageListener;

backend/src/services/wbotMonitor.js

@@ -0,0 +1,50 @@
+const Whatsapp = require("../models/Whatsapp");
+const wbotMessageListener = require("./wbotMessageListener");
+
+const { getIO } = require("../libs/socket");
+const { getWbot, init } = require("../libs/wbot");
+
+const wbotMonitor = () => {
+	const io = getIO();
+	const wbot = getWbot();
+
+	try {
+		wbot.on("change_state", newState => {
+			console.log("monitor", newState);
+		});
+
+		wbot.on("change_battery", async batteryInfo => {
+			// Battery percentage for attached device has changed
+			const { battery, plugged } = batteryInfo;
+			try {
+				await Whatsapp.update({ battery, plugged }, { where: { id: 1 } });
+			} catch (err) {
+				console.log(err);
+			}
+
+			console.log(`Battery: ${battery}% - Charging? ${plugged}`); //todo> save batery state to db
+		});
+
+		wbot.on("disconnected", reason => {
+			console.log("disconnected", reason); //todo> save connection status to DB
+			setTimeout(
+				() =>
+					init()
+						.then(res => {
+							wbotMessageListener();
+							wbotMonitor();
+						})
+						.catch(err => console.log(err)),
+				2000
+			);
+		});
+
+		// setInterval(() => {
+		// 	wbot.resetState();
+		// }, 20000);
+	} catch (err) {
+		console.log(err);
+	}
+};
+
+module.exports = wbotMonitor;