upd

server/src/models/Token.ts

@@ -11,6 +11,10 @@ const tokenSchema = new Schema(
       type: String,
       required: true,
     },
+    refreshToken: {
+      type: String,
+      required: true,
+    },
   },
   {
     timestamps: true,

server/src/models/User.ts

@@ -11,6 +11,11 @@ const userSchema = new Schema(
       type: String,
       required: true,
     },
+    companyId: {
+      type: Schema.Types.ObjectId,
+      ref: "Company",
+      required: true,
+    },
     role: {
       type: String,
       required: true,
@@ -19,14 +24,6 @@ const userSchema = new Schema(
       type: String,
       required: true,
     },
-    avatar: {
-      type: String,
-    },
-    companyId: {
-      type: Schema.Types.ObjectId,
-      ref: "Company",
-      required: true,
-    },
     buildIds: {
       type: [Schema.Types.ObjectId],
       ref: "Build",
@@ -39,6 +36,12 @@ const userSchema = new Schema(
   }
 );
 
+userSchema.virtual("tokens", {
+  ref: "Token",
+  foreignField: "userId",
+  localField: "_id",
+});
+
 const User = model("User", userSchema);
 
 export default User;