From ee36ebc6a6090629a743abc83bfec24497aad92b Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Tue, 29 Oct 2024 11:43:39 +0530
Subject: [PATCH] [mob] Add migration for filedata

---
 mobile/lib/db/ml/db.dart        |  2 ++
 mobile/lib/db/ml/db_fields.dart | 13 +++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/mobile/lib/db/ml/db.dart b/mobile/lib/db/ml/db.dart
index 89c4860506..77cc2ef532 100644
--- a/mobile/lib/db/ml/db.dart
+++ b/mobile/lib/db/ml/db.dart
@@ -46,6 +46,7 @@ class MLDataDB {
     createNotPersonFeedbackTable,
     fcClusterIDIndex,
     createClipEmbeddingsTable,
+    createFileDataTable,
   ];
 
   // only have a single app-wide reference to the database
@@ -237,6 +238,7 @@ class MLDataDB {
     await db.execute(deleteClusterSummaryTable);
     await db.execute(deleteNotPersonFeedbackTable);
     await db.execute(deleteClipEmbeddingsTable);
+    await db.execute(deleteFileDataTable);
   }
 
   Future<Iterable<Uint8List>> getFaceEmbeddingsForCluster(
diff --git a/mobile/lib/db/ml/db_fields.dart b/mobile/lib/db/ml/db_fields.dart
index 7a3a9d7631..4910a35a66 100644
--- a/mobile/lib/db/ml/db_fields.dart
+++ b/mobile/lib/db/ml/db_fields.dart
@@ -106,3 +106,16 @@ CREATE TABLE IF NOT EXISTS $clipTable (
 ''';
 
 const deleteClipEmbeddingsTable = 'DELETE FROM $clipTable';
+
+const fileDataTable = 'filedata';
+const createFileDataTable = '''
+CREATE TABLE IF NOT EXISTS $fileDataTable ( 
+  $fileIDColumn INTEGER NOT NULL,
+  type TEXT NOT NULL,
+  size INTEGER NOT NULL,
+  updated_at INTEGER NOT NULL,
+  PRIMARY KEY ($fileIDColumn, type)
+  );
+''';
+
+const deleteFileDataTable = 'DELETE FROM $fileDataTable';