add querying support

ActivityWatch · Mar 26, 2024 · b3f5b37 · b3f5b37
1 parent 12d6e53
commit b3f5b37
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/firebase/data.ts b/src/firebase/data.ts
@@ -50,10 +50,14 @@ export async function addScreenTimeData(userId: number, data: ScreenTimeData) {
   }
 }
 
-export async function getScreenTimeData(userId: string): Promise<ScreenTimeData[] | null> {
-  const colPath = `screentime/${userId}/${userId}`
-  const colRef = collection(db, colPath)
-  const snapshot = await getDocs(colRef)
+export async function getScreenTimeData(userId: string, since: Date | null = null, _public: Boolean = true): Promise<ScreenTimeData[] | null> {
+  const q = query(
+    collection(db, 'screentime/' + userId + '/' + userId),
+    // where('date', '>=', since || new Date('1900-1-1')),
+    where('public', '==', _public)
+  )
+
+  const snapshot = await getDocs(q)
   if (snapshot.empty) {
     return null
   }