From 0bff6b40c051c99d0e9bbffebf603ae214a4a330 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daphn=C3=A9=20Popin?= <PopDaph@users.noreply.github.com>
Date: Fri, 13 Dec 2024 17:16:35 +0100
Subject: [PATCH] Tracker models: add fields to track data processed by Notif
 workflow (#9370)

---
 front/lib/models/doc_tracker.ts       | 11 +++++++++++
 front/migrations/db/migration_130.sql |  2 ++
 2 files changed, 13 insertions(+)
 create mode 100644 front/migrations/db/migration_130.sql

diff --git a/front/lib/models/doc_tracker.ts b/front/lib/models/doc_tracker.ts
index 1ec0a776276c..15985a4891c2 100644
--- a/front/lib/models/doc_tracker.ts
+++ b/front/lib/models/doc_tracker.ts
@@ -26,6 +26,7 @@ export class TrackerConfigurationModel extends SoftDeletableModel<TrackerConfigu
   declare prompt: string | null;
 
   declare frequency: string | null;
+  declare lastNotifiedAt: Date | null;
 
   declare recipients: string[] | null;
 
@@ -87,6 +88,10 @@ TrackerConfigurationModel.init(
       type: DataTypes.STRING,
       allowNull: true,
     },
+    lastNotifiedAt: {
+      type: DataTypes.DATE,
+      allowNull: true,
+    },
     recipients: {
       type: DataTypes.ARRAY(DataTypes.STRING),
       allowNull: true,
@@ -215,6 +220,8 @@ export class TrackerGenerationModel extends SoftDeletableModel<TrackerGeneration
   declare dataSourceId: ForeignKey<DataSourceModel["id"]>;
   declare documentId: string;
 
+  declare consumedAt: Date | null;
+
   declare trackerConfiguration: NonAttribute<TrackerConfigurationModel>;
 }
 
@@ -245,6 +252,10 @@ TrackerGenerationModel.init(
       type: DataTypes.STRING,
       allowNull: false,
     },
+    consumedAt: {
+      type: DataTypes.DATE,
+      allowNull: true,
+    },
   },
   {
     modelName: "tracker_generation",
diff --git a/front/migrations/db/migration_130.sql b/front/migrations/db/migration_130.sql
new file mode 100644
index 000000000000..f88a4d201950
--- /dev/null
+++ b/front/migrations/db/migration_130.sql
@@ -0,0 +1,2 @@
+ALTER TABLE "public"."tracker_configurations" ADD COLUMN "lastNotifiedAt" TIMESTAMP WITH TIME ZONE;
+ALTER TABLE "public"."tracker_generations" ADD COLUMN "consumedAt" TIMESTAMP WITH TIME ZONE;