fix: migrate old schema to new

api/models/Conversation.js

@@ -1,44 +1,48 @@
 const mongoose = require('mongoose');
+const crypto = require('crypto');
 const { getMessages, deleteMessages } = require('./Message');
 
-const convoSchema = mongoose.Schema({
-  conversationId: {
-    type: String,
-    unique: true,
-    required: true
+const convoSchema = mongoose.Schema(
+  {
+    conversationId: {
+      type: String,
+      unique: true,
+      required: true
+    },
+    parentMessageId: {
+      type: String,
+      required: true
+    },
+    title: {
+      type: String,
+      default: 'New Chat'
+    },
+    jailbreakConversationId: {
+      type: String
+    },
+    conversationSignature: {
+      type: String
+    },
+    clientId: {
+      type: String
+    },
+    invocationId: {
+      type: String
+    },
+    chatGptLabel: {
+      type: String
+    },
+    promptPrefix: {
+      type: String
+    },
+    model: {
+      type: String
+    },
+    suggestions: [{ type: String }],
+    messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }]
   },
-  parentMessageId: {
-    type: String,
-    required: true
-  },
-  title: {
-    type: String,
-    default: 'New Chat'
-  },
-  jailbreakConversationId: {
-    type: String
-  },
-  conversationSignature: {
-    type: String
-  },
-  clientId: {
-    type: String
-  },
-  invocationId: {
-    type: String
-  },
-  chatGptLabel: {
-    type: String
-  },
-  promptPrefix: {
-    type: String
-  },
-  model: {
-    type: String
-  },
-  suggestions: [{ type: String }],
-  messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }],
-}, { timestamps: true });
+  { timestamps: true }
+);
 
 const Conversation =
   mongoose.models.Conversation || mongoose.model('Conversation', convoSchema);
@@ -114,5 +118,62 @@ module.exports = {
     let deleteCount = await Conversation.deleteMany(filter).exec();
     deleteCount.messages = await deleteMessages(filter);
     return deleteCount;
+  },
+  migrateDb: async () => {
+    try {
+      const conversations = await Conversation.find({ model: null }).exec();
+
+      if (!conversations || conversations.length === 0)
+        return { message: 'No conversations to migrate' };
+
+      for (let convo of conversations) {
+        const messages = await getMessages({
+          conversationId: convo.conversationId,
+          messageId: { $exists: false }
+        });
+
+        const promises = [];
+        let model;
+        let oldId;
+        messages.forEach((message, i) => {
+          const msgObj = message.toObject();
+          const newId = msgObj.id;
+          if (i === 0) {
+            message.parentMessageId = '00000000-0000-0000-0000-000000000000';
+            oldId = newId;
+          } else {
+            message.parentMessageId = oldId;
+            oldId = newId;
+          }
+          
+          message.messageId = newId;
+          message.createdAt = message.created;
+          if (message.sender.toLowerCase() !== 'user' && !model) {
+            model = message.sender.toLowerCase();
+          }
+
+          if (message.sender.toLowerCase() === 'user') {
+            message.isCreatedByUser = true;
+          }
+          promises.push(message.save());
+        });
+        await Promise.all(promises);
+
+        await Conversation.findOneAndUpdate(
+          { conversationId: convo.conversationId },
+          { model, createdAt: convo.created },
+          { new: true }
+        ).exec();
+      }
+
+      try {
+        await mongoose.connection.db.collection('messages').dropIndex('id_1');
+      } catch (error) {
+        console.log("Index doesn't exist or already dropped");
+      }
+    } catch (error) {
+      console.log(error);
+      return { message: 'Error migrating conversations' };
+    }
   }
 };