Update index.js

index.js

@@ -181,6 +181,137 @@ app.get('/get/message', (req, res) => {
 });
 
 
+const db_chatHistory = {};
+
+function generateUID(length) {
+  const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+  const charactersLength = characters.length;
+  let uid = '';
+  for (let i = 0; i < length; i++) {
+    uid += characters.charAt(Math.floor(Math.random() * charactersLength));
+  }
+  return uid;
+}
+
+async function sendMessageV2(data) {
+  try {
+    const response = await axios.post('https://chat.chatgptdemo.net/chat_api_stream', {
+      data
+    }, {
+      headers: {
+        'Content-Type': 'application/json',
+        'User-Agent': 'Mozilla/5.0 (iPhone14,3; U; CPU iPhone OS 15_0 like Mac OS X) AppleWebKit/602.1.50 (KHTML, like Gecko) Version/10.0 Mobile/19A346 Safari/602.1',
+        'X-Forwarded-For': generateRandomIP()
+      }
+    });
+    return response.data;
+  } catch (error) {
+    console.error(error);
+  }
+}
+
+async function cleanInactiveUsersV2() {
+  const currentTime = DateTime.local();
+
+  Object.keys(db_chatHistory).forEach((userId) => {
+    const user = db_chatHistory[userId];
+
+    if (user && user.lastChat && user.lastChat.plus(Duration.fromObject({ minutes: 20 })) < currentTime) {
+      // User's last message was more than 10 minutes ago, remove the user from db_chatHistory
+      delete db_chatHistory[userId];
+    }
+  });
+
+  // Schedule the next execution after 1 minute
+  setTimeout(cleanInactiveUsersV2, 180000);
+}
+
+// Start the continuous execution of cleanInactiveUsers
+cleanInactiveUsersV2();
+
+app.get('/v2/send', async (req, res) => {
+  try {
+    const { id, message } = req.query;
+    if (!id) return res.status(400).json({
+      success: false,
+      response: "input param id & input id user"
+    });
+    if (!message) return res.status(400).json({
+      success: false,
+      response: "input param message & input a message from user"
+    });
+    if (!db_chatHistory[id]) {
+      db_chatHistory[id] = {
+        lastChat: DateTime.local(),
+        data: {
+          botId: 'default',
+          customId: null,
+          session: 'N/A',
+          chatId: generateUID(11),
+          contextId: 58,
+          messages: [{
+            id: generateUID(10),
+            role: 'assistant',
+            content: 'Hi, How can I help you?',
+            who: 'AI: ',
+            timestamp: Date.now() - (4 * 60 * 1000)
+          }],
+          newMessage: '',
+          newFileId: null,
+          stream: false
+        }
+      };
+    }
+    db_chatHistory[id].lastChat = DateTime.local();
+    db_chatHistory[id].data.newMessage = message;
+    const botResponse = await sendMessageV2(db_chatHistory[id].data);
+    //setelah terkirim simpan pesan user
+    db_chatHistory[id].data.messages.push({
+      id: generateUID(10),
+      role: 'user',
+      content: message,
+      who: 'User: ',
+      timestamp: Date.now()
+    });
+    db_chatHistory[id].data.newMessage = ""; //set ke semula sesudah post
+    //save pesan bot juga
+    db_chatHistory[id].data.messages.push({
+      id: generateUID(10),
+      role: 'assistant',
+      content: botResponse.reply,
+      who: 'AI: ',
+      timestamp: Date.now()
+    });
+    db_chatHistory[id].lastChat = DateTime.local();
+
+    res.status(200).json({
+      success: true,
+      response: botResponse.reply
+    });
+  } catch (e) {
+    res.status(400).json({
+      success: false,
+      response: format(e)
+    });
+  }
+});
+
+app.get('/v2/listuser', (req, res) => {
+  const userList = Object.keys(db_chatHistory);
+  res.status(200).json({ userList });
+});
+
+app.get('/v2/get/message', (req, res) => {
+  const id = req.query.id;
+  if (!id) return res.status(400).json({ success: false, response: "input id" });
+  if (!db_chatHistory[id]) {
+    return res.status(404).json({ error: 'User not found' });
+  }
+  const userMessages = db_chatHistory[id].data.messages;
+  res.status(200).json({ messages: userMessages });
+});
+
+
 app.listen(port, () => {
 	console.log(`Server is running on port ${port}`);
 });