From eadab903409b40dba5b9227543120828125f6651 Mon Sep 17 00:00:00 2001
From: Evert Prants <evert@lunasqu.ee>
Date: Sat, 10 Sep 2022 09:29:03 +0300
Subject: [PATCH] audit changes

---
 src/modules/objects/audit/audit.enum.ts    |  2 ++
 src/modules/objects/audit/audit.service.ts | 11 +++++++----
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/modules/objects/audit/audit.enum.ts b/src/modules/objects/audit/audit.enum.ts
index bcf8db7..db3af0a 100644
--- a/src/modules/objects/audit/audit.enum.ts
+++ b/src/modules/objects/audit/audit.enum.ts
@@ -7,4 +7,6 @@ export enum AuditAction {
   EMAIL_CHANGE = 'email_change',
   MALICIOUS_REQUEST = 'malicious_request',
   THROTTLE = 'throttle',
+  DEACTIVATION_REQUEST = 'deactivation_request',
+  DATA_DOWNLOAD_REQUEST = 'data_download_request',
 }
diff --git a/src/modules/objects/audit/audit.service.ts b/src/modules/objects/audit/audit.service.ts
index 9fa3bb8..6c8f864 100644
--- a/src/modules/objects/audit/audit.service.ts
+++ b/src/modules/objects/audit/audit.service.ts
@@ -17,6 +17,12 @@ import { AuditSearchClause, UserLoginEntry } from './audit.interfaces';
 
 const PLUCK_LOCATION = ['country', 'city', 'timezone', 'll'];
 const PLUCK_USER_AGENT = ['browser', 'version', 'os', 'platform'];
+const AUTOFLAG = [
+  AuditAction.MALICIOUS_REQUEST,
+  AuditAction.THROTTLE,
+  AuditAction.DEACTIVATION_REQUEST,
+  AuditAction.DATA_DOWNLOAD_REQUEST,
+];
 
 @Injectable()
 export class AuditService {
@@ -40,10 +46,7 @@ export class AuditService {
     audit.actor_ua = ua;
     audit.actor = user;
 
-    if (
-      action === AuditAction.MALICIOUS_REQUEST ||
-      action === AuditAction.THROTTLE
-    ) {
+    if (AUTOFLAG.includes(action)) {
       audit.flagged = true;
       // TODO: email administrator
     }