move db functions

api/models/Conversation.js

@@ -1,5 +1,4 @@
 const mongoose = require('mongoose');
-const crypto = require('crypto');
 const { getMessages, deleteMessages } = require('./Message');
 
 const convoSchema = mongoose.Schema(
@@ -65,13 +64,14 @@ const getConvo = async (user, conversationId) => {
 };
 
 module.exports = {
+  Conversation,
   saveConvo: async (user, { conversationId, newConversationId, title, ...convo }) => {
     try {
       const messages = await getMessages({ conversationId });
       const update = { ...convo, messages };
       if (title) {
         update.title = title;
-        update.user = user
+        update.user = user;
       }
       if (newConversationId) {
         update.conversationId = newConversationId;
@@ -97,9 +97,13 @@ module.exports = {
   },
   updateConvo: async (user, { conversationId, ...update }) => {
     try {
-      return await Conversation.findOneAndUpdate({ conversationId: conversationId, user }, update, {
-        new: true
-      }).exec();
+      return await Conversation.findOneAndUpdate(
+        { conversationId: conversationId, user },
+        update,
+        {
+          new: true
+        }
+      ).exec();
     } catch (error) {
       console.log(error);
       return { message: 'Error updating conversation' };
@@ -132,63 +136,8 @@ module.exports = {
     }
   },
   deleteConvos: async (user, filter) => {
-    let deleteCount = await Conversation.deleteMany({...filter, user}).exec();
+    let deleteCount = await Conversation.deleteMany({ ...filter, user }).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: '[Migrate] No conversations to migrate' };
-
-      for (let convo of conversations) {
-        const messages = await getMessages({
-          conversationId: convo.conversationId,
-          messageId: { $exists: false }
-        });
-
-        let model;
-        let oldId;
-        const promises = [];
-        messages.forEach((message, i) => {
-          const msgObj = message.toObject();
-          const newId = msgObj.id;
-          if (i === 0) {
-            message.parentMessageId = '00000000-0000-0000-0000-000000000000';
-          } else {
-            message.parentMessageId = oldId;
-          }
-
-          oldId = newId;
-          message.messageId = newId;
-          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 },
-          { new: true }
-        ).exec();
-      }
-
-      try {
-        await mongoose.connection.db.collection('messages').dropIndex('id_1');
-      } catch (error) {
-        console.log("[Migrate] Index doesn't exist or already dropped");
-      }
-    } catch (error) {
-      console.log(error);
-      return { message: '[Migrate] Error migrating conversations' };
-    }
   }
 };

api/models/dbConnect.js

@@ -1,44 +0,0 @@
-require('dotenv').config();
-const mongoose = require('mongoose');
-const MONGO_URI = process.env.MONGO_URI;
-
-if (!MONGO_URI) {
-  throw new Error('Please define the MONGO_URI environment variable inside .env.local');
-}
-
-/**
- * Global is used here to maintain a cached connection across hot reloads
- * in development. This prevents connections growing exponentially
- * during API Route usage.
- */
-let cached = global.mongoose;
-
-if (!cached) {
-  cached = global.mongoose = { conn: null, promise: null };
-}
-
-async function dbConnect() {
-  if (cached.conn) {
-    return cached.conn;
-  }
-
-  if (!cached.promise) {
-    const opts = {
-      useNewUrlParser: true,
-      useUnifiedTopology: true,
-      bufferCommands: false
-      // bufferMaxEntries: 0,
-      // useFindAndModify: true,
-      // useCreateIndex: true
-    };
-
-    mongoose.set('strictQuery', true);
-    cached.promise = mongoose.connect(MONGO_URI, opts).then((mongoose) => {
-      return mongoose;
-    });
-  }
-  cached.conn = await cached.promise;
-  return cached.conn;
-}
-
-module.exports = dbConnect;

api/models/index.js

@@ -1,15 +1,15 @@
-const { saveMessage, deleteMessagesSince, deleteMessages } = require('./Message');
+const { getMessages, saveMessage, deleteMessagesSince, deleteMessages } = require('./Message');
 const { getCustomGpts, updateCustomGpt, updateByLabel, deleteCustomGpts } = require('./CustomGpt');
-const { getConvoTitle, getConvo, saveConvo, migrateDb } = require('./Conversation');
+const { getConvoTitle, getConvo, saveConvo } = require('./Conversation');
 
 module.exports = {
+  getMessages,
   saveMessage,
   deleteMessagesSince,
   deleteMessages,
   getConvoTitle,
   getConvo,
   saveConvo,
-  migrateDb,
   getCustomGpts,
   updateCustomGpt,
   updateByLabel,