wa_strapi
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

262 rivejä
7.7 KiB

  1. const qrcode = require("qrcode-terminal");
  2. const fs = require("fs");
  3. // const path = require("path");
  4. const { Client, Chat, Message } = require("whatsapp-web.js");
  5. // const SESSION_FOLDER = path.join(__dirname, "../data/");
  6. // const SESSION_FILE_PATH = path.join(__dirname, "../data/session.json");
  7. // const sessions = require("../api/session_data");
  8. const { parseMultipartData, sanitizeEntity } = require("strapi-utils");
  9. var client;
  10. //Login Session when there is Session File
  11. function callbacks(client, callback, type = "create") {
  12. // console.log("starting", type);
  13. let sessionData;
  14. client.on("auth_failure", (session) => {
  15. // console.log("auth_failure");
  16. return callback({
  17. STATUS: 0,
  18. TYPE: "AUTH_FAILURE",
  19. DATA: "SESSION HAS BEEN ENDED , PLEASE CREATE A NEW ONE !",
  20. });
  21. });
  22. client.on("disconnected", async function (session) {
  23. console.log("disconnected from ", type, session);
  24. if (session == "UNPAIRED" || session == "CONFLICT") {
  25. try {
  26. if (type == "load" && client && client.info) {
  27. console.log(getID);
  28. getID = await strapi
  29. .query("session-data")
  30. .findOne({ phonenumber: client.info.wid.user });
  31. console.log(getID);
  32. if (getID) {
  33. let id = getID["id"];
  34. await strapi.api["session-data"].services[
  35. "session-data"
  36. ].delete({ id });
  37. }
  38. await client.destroy();
  39. }
  40. } catch (e) {}
  41. }
  42. });
  43. client.on("ready", async () => {
  44. // console.log("ready");
  45. var clientInfo = client.info;
  46. clientInfo["sessionData"] = sessionData;
  47. if (type == "create") await client.destroy();
  48. return callback({
  49. STATUS: 1,
  50. TYPE: "READY",
  51. DATA: type == "create" ? clientInfo : client,
  52. });
  53. });
  54. client.on("message", async (msg) => {
  55. // console.log("MESSAGE RECEIVED", msg.body);
  56. });
  57. client.on("qr", (qr) => {
  58. return callback({ STATUS: 1, TYPE: "QR_AUTH", DATA: qr });
  59. });
  60. client.on("change_state", (state) => {
  61. // console.log(`state => ${state}`);
  62. });
  63. client.on("authenticated", (session) => {
  64. console.log("authenticated");
  65. try {
  66. sessionData = session;
  67. // fs.writeFileSync(SESSION_FILE_PATH, JSON.stringify(session));
  68. } catch (e) {
  69. return callback({
  70. STATUS: 0,
  71. TYPE: "AUTHENTICATED_ERROR",
  72. DATA: e.message,
  73. });
  74. }
  75. });
  76. }
  77. var loadClient = async function (clientSessionData, callback) {
  78. try {
  79. // if(!client){
  80. // console.log(client.info);
  81. // return callback({"STATUS" : 1 , "TYPE" : "READY" , "DATA" : client.info})
  82. // }
  83. // let sessionPath = path.join(SESSION_FOLDER, `${phoneNumber}.json`);
  84. // if (fs.existsSync(sessionPath)) {
  85. // sessionData = require(sessionPath);
  86. client = new Client({
  87. puppeteer: {
  88. args: ["--no-sandbox", "--disable-setuid-sandbox"],
  89. },
  90. session: clientSessionData,
  91. restartOnAuthFail: true,
  92. });
  93. callbacks(client, callback, "load");
  94. // console.log("initialize");
  95. client.initialize();
  96. } catch (e) {
  97. // console.log(e);
  98. return callback({
  99. STATUS: 0,
  100. TYPE: "ERROR",
  101. DATA: e.message,
  102. });
  103. }
  104. };
  105. function asyncLoadClient(i) {
  106. return new Promise((resolve, reject) => {
  107. try {
  108. loadClient(i["session_data"], async (data) => {
  109. return resolve(data);
  110. });
  111. } catch (e) {
  112. return reject(e.message);
  113. }
  114. });
  115. }
  116. var createNewClient = async function (callback) {
  117. try {
  118. client = new Client({
  119. puppeteer: {
  120. args: ["--no-sandbox", "--disable-setuid-sandbox"],
  121. },
  122. });
  123. callbacks(client, callback);
  124. console.log("initialize");
  125. client.initialize();
  126. //var clientInfo = await getClientInfo();
  127. //return clientInfo;
  128. } catch (e) {
  129. return callback({ STATUS: 0, TYPE: "READY", DATA: e.message });
  130. }
  131. };
  132. var sendMessage = async function (client, phone_number, textMessage) {
  133. try {
  134. if (!client.info) {
  135. throw { message: "Client Is Not Ready !" };
  136. }
  137. var phoneNumber = reformatMobilePhone(phone_number);
  138. // var state = await client.isRegisteredUser(phoneNumber);
  139. // console.log(state);
  140. await client.sendMessage(phoneNumber, textMessage);
  141. return { STATUS: 1, DATA: "Messages Succesfully send !" };
  142. } catch (e) {
  143. console.log(e);
  144. return { STATUS: 0, DATA: e.message };
  145. }
  146. };
  147. var getChatbyPhoneNumber = async function (phone_number) {
  148. try {
  149. if (!client.info) {
  150. throw { message: "Client Is Not Ready !" };
  151. }
  152. let phoneNumber = reformatMobilePhone(phone_number);
  153. let chats = await client.getChatById(phoneNumber);
  154. let messageList = await chats.fetchMessages();
  155. return { STATUS: 1, DATA: messageList };
  156. } catch (e) {
  157. return { STATUS: 0, DATA: e.message };
  158. }
  159. };
  160. var getAllChats = async function () {
  161. try {
  162. if (!client.info) {
  163. throw { message: "Client Is Not Ready !" };
  164. }
  165. let chatList = await client.getChats();
  166. return { STATUS: 1, DATA: chatList };
  167. } catch (e) {
  168. return { STATUS: 0, DATA: e.message };
  169. }
  170. };
  171. var getClientInfo = function () {
  172. try {
  173. var a;
  174. while (!a) {
  175. a = client.info;
  176. }
  177. return { STATUS: 1, DATA: a };
  178. } catch (e) {
  179. return { STATUS: 0, DATA: e.message };
  180. }
  181. };
  182. var getClientState = async function () {
  183. try {
  184. if (!client.info) {
  185. throw { message: "Client Is Not Ready !" };
  186. }
  187. let state = await client.getState();
  188. return { STATUS: 1, DATA: state };
  189. } catch (e) {
  190. return { STATUS: 0, DATA: e.message };
  191. }
  192. };
  193. var destroySession = async function (phoneNumber) {
  194. try {
  195. if (client && client.info) {
  196. await client.logout();
  197. await client.destroy();
  198. clearFileSession(phoneNumber);
  199. return { STATUS: 1, DATA: "Session cleared" };
  200. } else throw { message: "Client Is Not Ready !" };
  201. } catch (e) {
  202. return { STATUS: 0, DATA: e.message };
  203. }
  204. };
  205. function reformatMobilePhone(phoneNumber) {
  206. var newPhoneNumber;
  207. if (phoneNumber[0].toString() == "0") {
  208. newPhoneNumber = "62" + phoneNumber.substring(1) + "@c.us";
  209. } else if (phoneNumber[0] == "+") {
  210. newPhoneNumber = phoneNumber.substring(1) + "@c.us";
  211. } else {
  212. newPhoneNumber = phoneNumber + "@c.us";
  213. }
  214. return newPhoneNumber;
  215. }
  216. function clearFileSession(phoneNumber) {
  217. try {
  218. let sessionPath = path.join(SESSION_FOLDER, `${phoneNumber}.json`);
  219. if (fs.existsSync(sessionPath)) {
  220. console.log(`clearing session ${phoneNumber}`);
  221. fs.unlinkSync(sessionPath);
  222. }
  223. } catch (e) {
  224. throw e;
  225. }
  226. }
  227. module.exports = {
  228. sendMessage: sendMessage,
  229. createNewClient: createNewClient,
  230. loadClient: loadClient,
  231. getClientInfo: getClientInfo,
  232. destroySession: destroySession,
  233. getClientState: getClientState,
  234. getClientState: getClientState,
  235. getChatbyPhoneNumber: getChatbyPhoneNumber,
  236. asyncLoadClient: asyncLoadClient,
  237. getAllChats: getAllChats,
  238. };