init

lib/models/strength.ts

@@ -29,6 +29,7 @@ export type StrengthWorkoutInput = z.infer<typeof strengthWorkoutSchema>;
 
 export type StrengthWorkout = StrengthWorkoutInput & {
   _id: ObjectId;
+  userId: string;
   createdAt: Date;
 };
 
@@ -37,30 +38,34 @@ const COLLECTION = "strength_workouts";
 async function getCollection() {
   const db = await getDb();
   const collection = db.collection<StrengthWorkout>(COLLECTION);
-  await collection.createIndex({ sourceKey: 1 }, { unique: true });
+  await collection.createIndex({ userId: 1, sourceKey: 1 }, { unique: true });
   return collection;
 }
 
 export async function upsertStrengthWorkout(
+  userId: string,
   workout: StrengthWorkoutInput
 ): Promise<void> {
   const collection = await getCollection();
   await collection.updateOne(
-    { sourceKey: workout.sourceKey },
+    { userId, sourceKey: workout.sourceKey },
     {
-      $set: workout,
+      $set: { ...workout, userId },
       $setOnInsert: { createdAt: new Date() },
     },
     { upsert: true }
   );
 }
 
-export async function listStrengthWorkouts(): Promise<StrengthWorkout[]> {
+export async function listStrengthWorkouts(userId: string): Promise<StrengthWorkout[]> {
   const collection = await getCollection();
-  return collection.find().sort({ date: -1 }).toArray();
+  return collection.find({ userId }).sort({ date: -1 }).toArray();
 }
 
-export async function getStrengthWorkout(id: string): Promise<StrengthWorkout | null> {
+export async function getStrengthWorkout(
+  userId: string,
+  id: string
+): Promise<StrengthWorkout | null> {
   const collection = await getCollection();
-  return collection.findOne({ _id: new ObjectId(id) });
+  return collection.findOne({ _id: new ObjectId(id), userId });
 }